Adding tests for context passing