_context =& $context; } public function expectContext($key, $value) { $this->_expected_context[$key] = $value; } public function expectContextAt($step, $key, $value) { $this->_expected_context_at[$step][$key] = $value; } public function send($v1, $v2) { // test for context $context = SimpleTest::getContext(); $test = $context->getTest(); // compat if (empty($this->_mock)) { $mock = $this; } else { $mock = $this->_mock; } foreach ($this->_expected_context as $key => $value) { $test->assertEqual($value, $this->_context->get($key)); } $step = $mock->getCallCount('send'); if (isset($this->_expected_context_at[$step])) { foreach ($this->_expected_context_at[$step] as $key => $value) { $test->assertEqual($value, $this->_context->get($key)); } } // boilerplate mock code, does not have return value or references $args = func_get_args(); $mock->_invoke('send', $args); } }