PhpUnit測試存根方法多重返回

Question

我正在用symfony2進行一些PHPUnit測試。我遇到了一個特定測試的問題。

我正在測試一個響應表單，當然我的一個類的響應是真實的一個錯誤。我有一個模擬我的數據庫，我有一個從我的databaseRepository的方法之一存根。

問題是，在一個測試中，我做了一個存根到一個有效數組的方法，第二個測試我只想查詢是無效的null。

我的分貝模擬：

//Setting up mock of database repository class 
    $this->db = $this->getMockBuilder('DatabaseRepository') 
     ->disableOriginalConstructor() 
     ->getMock(); 

    $this->db->expects($this->any()) 
     ->method('getRecord') 
     ->will($this->returnValue(self::$registrationRecord)); 

    $this->db->expects($this->any()) 
     ->method('getRecord') 
     ->willReturn(null); 

所以我想有兩個不同的預期，但這obviousley不行.....是有可能有一個存根方法有兩種不同的回報.. ？

測試1：

<?php 


class UnsubscribeRegistrationTemplateTest extends \PHPUnit_Framework_TestCase 
{ 

    /** 
    * @var UnsubscribeRegistrationTemplate 
    */ 
    protected $object; 

    /** 
    * @var ValidationClass 
    */ 
    public $validate; 

    /** 
    * @var DatabaseRepository 
    */ 
    public $db; 

    //Database Record Mock 
    public static $registrationRecord = array 
    (
     'rowid' => '96', 
     'unsubscription' => 'N', 
     'updated' => 'BB' 
    ); 

    /** 
    * 
    */ 
    protected function setUp() 
    { 
     //Setting up mock of validation class 
     $this->validate = $this->getMockBuilder('ValidationClass') 
      ->disableOriginalConstructor() 
      ->getMock(); 

     $this->validate->expects($this->any()) 
      ->method('validateInput') 
      ->willReturn(true); 

     //Setting up mock of database repository class 
     $this->db = $this->getMockBuilder('DatabaseRepository') 
      ->disableOriginalConstructor() 
      ->getMock(); 

     $this->db->expects($this->any()) 
      ->method('getRegistrationRecord') 
      ->will($this->returnValue(self::$registrationRecord)); 

     $this->db->expects($this->any()) 
      ->method('getRegistrationRecord') 
      ->will($this->returnValue(null)); 

     $this->db->expects($this->any()) 
      ->method('setPreRegistrationEnquiryUnsubscriptionEnabled') 
      ->willReturn(true); 

     $this->object = $this->createUnsubscribeRegistrationTemplateInstance(); 
    } 

    /** 
    * @return UnsubscribeRegistrationTemplate 
    * 
    */ 
    public function createUnsubscribeRegistrationTemplateInstance() 
    { 
     //initialize Unsubscribe Registration Template 
     return new UnsubscribeRegistrationTemplate 
     (
      $this->validate, 
      $this->db 
     ); 
    } 

    /** 
    * @param array $mapping 
    * @return Request 
    */ 
    public function createRequest(array $mapping) 
    { 
     $request = new Request(); 

     foreach ($mapping as $k =>$v) 
     { 
      $request->query->set($k, $v); 
     } 

     return $request; 
    } 

    /** 
    * 
    */ 
    public function testUnsubscribeRegistrationTemplateValidResponse() 
    { 
     $request = $this->createRequest(array(
      'registration_id' => '96', 
      'source_channel' => 'BB' 
     )); 

     $response = new Response(
      true, 
      'Unsubscription successful' 
     ); 

     $this->assertEquals($response, $this->object->create($request)); 
    } 

    /** 
    * 
    */ 
    public function testUnsubscribeRegistrationTemplateEmptyResponse() 
    { 
     $request = $this->createRequest(array(
      'registration_id' => '96', 
      'source_channel' => 'BB' 
     )); 

     $response = new Response(
      false, 
      'Registration Record Not Found.' 
     ); 

     $this->assertEquals($response, $this->object->create($request)); 
    } 

    /** 
    * 
    */ 
    public function testIsAlreadyRegisteredValidResponse() 
    { 
     //Testing record is already unsubscribed 
     $registrationRecord = array(
      'unsubscription_enabled' => 'Y' 
     ); 

     $this->assertTrue($this->object->isAlreadyUnsubscribed($registrationRecord)); 
    } 

    /** 
    * 
    */ 
    public function testIsAlreadyRegisteredInValidResponse() 
    { 
     //Testing record not unsubscribed 
     $registrationRecord = array(
      'unsubscription_enabled' => 'N' 
     ); 
     $this->assertFalse($this->object->isAlreadyUnsubscribed($registrationRecord)); 
    } 

    /** 
    * 
    */ 
    protected function tearDown() 
    { 
     unset($this->object); 
    } 

} 

Answer 1

您可以在許多方面做到這一點。
以下兩種方式可能適合您的需求。

1 - 移動getRecord（）預計測試

/** 
* @test 
*/ 
public function ifTrue() 
{ 
    $this->db->expects($this->once()) 
    ->method('getRecord') 
    ->will($this->returnValue(self::$registrationRecord)); 

    $request = $this->createRequest(array(
     'id' => '10', 
     'code' => 'BB' 
    )); 

    $response = new Response(
     true, 
     'successful' 
    ); 

    $this->assertEquals($response, $this->object->create($request)); 
} 

/** 
* @test 
*/ 
public function ifFalse() 
{ 
    $this->db->expects($this->once()) 
    ->method('getRecord') 
    ->willReturn(null); 

    $request = $this->createRequest(array(
     'id' => '10', 
     'code' => 'BB' 
    )); 

    $response = new Response(
     false, 
     'Record Not Found.' 
    ); 

    $this->assertEquals($response, $this->object->create($request)); 
} 

正如你可以看到，有很多重複的，所以我們使用dataprovider。

2 - 使用@dataProvider

protected function getDataForTest() 
{ 
    return array(
     array(self::$registrationRecord, true, 'successful'), 
     array(null, false, 'Record Not Found.') 
    ); 
} 

/** 
* @dataProvider getDataForTest 
* @test 
*/ 
public function ifTrue($getRecordValue, $bool, $message) 
{ 
    $this->db->expects($this->once()) 
    ->method('getRecord') 
    ->will($this->returnValue($getRecordValue); 

    $request = $this->createRequest(array(
     'id' => '10', 
     'code' => 'BB' 
    )); 

    $response = new Response(
     $bool, 
     $message 
    ); 

    $this->assertEquals($response, $this->object->create($request)); 
} 

使用@dataProvider只要你想測試所有情況下，你可以使用盡可能多的價值。

Answer 2

我想你應該使用PHPUnit at()方法來檢查特定索引處的方法調用。所以你必須用正確的索引來調用expects值參數。

所以，你可以使用下面的代碼：

//Setting up mock of database repository class 
$this->db = $this->getMockBuilder('DatabaseRepository') 
    ->disableOriginalConstructor() 
    ->getMock(); 

$this->db->expects($this->at(0)) // Mock the first call 
    ->method('getRecord') 
    ->will($this->returnValue(self::$registrationRecord)); 

$this->db->expects($this->at(1)) // Mock the second call 
    ->method('getRecord') 
    ->willReturn(null); 

http://www.andrejfarkas.com/2012/07/phpunit-at-method-to-check-method-invocation-at-certain-index/

希望這有助於