Added unit test to context.go