Add unit test for @context