Added unit test for working context