Unit test for context.go