如何重置PHPUnit模拟对象的期望

时间:2014-09-19 14:05:30

标签: php unit-testing mocking phpunit

我在TestCase中进行了大量测试。我想设置一个模拟对象,在大多数测试中返回相同的值,但在一些测试中我想自定义该值。

我的想法是创建一个set_up()方法(我无法在自动调用的setUp()中设置期望值),并在每次测试开始时手动调用它。在这个方法中,我会设置默认的返回值,然后在需要自定义返回值的少数测试中,我会第二次调用expect,并希望覆盖默认的返回值。这不起作用,返回值不会被覆盖。

这是一个简化的例子:

<?php

class SomeClass {
    function someMethod() {
    }
}

class SomeTest extends PHPUnit_Framework_TestCase {
    private $mock;

    function set_up() {
        $this->mock = $this->getMockBuilder('SomeClass')
            ->disableOriginalConstructor() // This is necessary in actual program
            ->getMock();
        $this->mock->expects($this->any())
            ->method('someMethod')
            ->will($this->returnValue(1));
    }

    function test() {
        $this->set_up();

        $this->mock->expects($this->any())
            ->method('someMethod')
            ->will($this->returnValue(2));

        $this->assertEquals(2, $this->mock->someMethod());
    }
}

这似乎可以通过阅读How to reset a Mock Object with PHPUnit来实现。

PHPUnit mock with multiple expects() calls没有回答我的问题。

我正在使用phpUnit 4.2

1 个答案:

答案 0 :(得分:5)

您可以将参数传递给set_up方法,以便它可以根据需要配置模拟:

function set_up($someMethodReturnValue = 1) {
    $mock = $this->getMockBuilder('SomeClass')
        ->disableOriginalConstructor() // This is necessary in actual program
        ->getMock();
    $mock->expects($this->any())
        ->method('someMethod')
        ->will($this->returnValue($someMethodReturnValue));
    return $mock;
}


function test() {
    $mock = $this->set_up(2);
    $this->assertEquals(2, $this->mock->someMethod());
}

您可以进一步增强set_up()方法。如果有很多选项,最终你可以创建一个Mock创建类。