PHPUnit willReturnMap 方法 - 回调参数

Question

我的用例比较复杂，但为了简单起见:

class MockObject {
    public function test($param1, callable $callback = null) {
        return is_null($callback) ? $param1 : $callback($param1);
    }
}

我想使用 returnMap 为我的其他测试模拟这个类。

$map = [
    ['a', null, 'a'],
    ['b', $WHAT_SHOULD_BE_HERE?, 'b']
];

$mock = $this->getMock('MockObject');
$mock->expects($this->atLeastOnce())->method('test')->willReturnMap($map);

$this->assertEquals('a', $mock->test('a')); // Works 
$this->assertEquals('b', $mock->test('b', function($value){return $value})); // Doesn't work 

Answer 1

只需定义一个 $callback 并在映射和断言中使用它:

$callback = function($value) { return $value; };

$map = [
    ['a', null, 'a'],
    ['b', $callback, 'b']
];

$mock = $this->getMock('MockObject');
$mock->expects($this->atLeastOnce())->method('test')->will($this->returnValueMap($map));

$this->assertEquals('a', $mock->test('a')); // Works 
$this->assertEquals('b', $mock->test('b', $callback)); // Should work

您可以通过使用计算返回值的函数来更好地控制所有情况:

$mock = $this->getMock('MockObject');
$mock->expects($this->atLeastOnce())
     ->method('test')
     ->will($this->returnCallback(function($param1, $param2) {
          // return value depends on parameters
     }));