Add tests for context helpers