add test for context