Added unit test for context