PHPUnit - 如何模拟 PDO 准备好的语句

Question

我正在尝试使用 PHPUnit 对映射器类进行单元测试。 我可以轻松地模拟将注入映射器类中的 PDO 实例，但我不知道如何模拟PreparedStatement 类，因为它是由 PDO 类生成的。

在我的例子中，我扩展了 PDO 类，所以我有这个：

public function __construct($dsn, $user, $pass, $driverOptions)
{

    //...

    parent::__construct($dsn, $user, $pass, $driverOptions);
    $this->setAttribute(PDO::ATTR_STATEMENT_CLASS,
        array('Core_Db_Driver_PDOStatement', array($this)));
}

要点是 Core_Db_Driver_PDOStatement 没有注入到 PDO 类的构造函数中，它是静态实例化的。即使我这样做：

public function __construct($dsn, $user, $pass, $driverOptions, $stmtClass = 'Core_Db_Driver_PDOStatement')
{

    //...

    parent::__construct($dsn, $user, $pass, $driverOptions);
    $this->setAttribute(PDO::ATTR_STATEMENT_CLASS,
        array($stmtClass, array($this)));
}

...它仍然是一个静态实例，因为我无法传递我自己的准备好的语句类的模拟实例。

有什么想法吗？

编辑： 解决方案，改编自 anwser：

/**
 * @codeCoverageIgnore
 */
private function getDbStub($result)
{
    $STMTstub = $this->getMock('PDOStatement');
    $STMTstub->expects($this->any())
            ->method('fetchAll')
            ->will($this->returnValue($result));


    $PDOstub = $this->getMock('mockPDO');
    $PDOstub->expects($this->any())
            ->method('prepare')
            ->will($this->returnValue($STMTstub));

    return $PDOstub;
}

public function testGetFooById()
{
    $arrResult = array( ... );
    $PDOstub = $this->getDbStub($arrResult);
}

Answer 1

如果您可以模拟 PDO 类，只需模拟出 pdo 类及其所有依赖项即可。不需要关心语句类或 pdo 类的构造函数，因为您通过模拟定义了输入和输出。

因此，您需要一个返回模拟对象的模拟对象。

它可能看起来有点令人困惑，但由于您应该只测试正在测试的类的功能，而没有其他任何内容，因此您几乎可以摆脱数据库连接的所有其他部分。

在这个例子中，您想要弄清楚的是：

• prepare 被调用了吗？
• fetchAll 是否在准备返回时被调用？
• 调用的结果是否返回？

如果是这样：一切都好。

<?php
class myClass {
     public function __construct(ThePDOObject $pdo) {
         $this->db = $pdo;
     }

     public function doStuff() {
         $x = $this->db->prepare("...");
         return $x->fetchAll();
     }
}

class myClassTest extends PHPUnit_Framework_TestCase {

    public function testDoStuff() {

        $fetchAllMock = $this
           ->getMockBuilder("stdClass" /* or whatever has a fetchAll */)
           ->setMethods(array("fetchAll"))
           ->getMock();
        $fetchAllMock
           ->expects($this->once())->method("fetchAll")
           ->will($this->returnValue("hello!"));

        $mock = $this
           ->getMockBuilder("ThePDOObject")
           ->disableOriginalConstructor()
           ->setMethods(array("prepare"))
           ->getMock();
        $mock
           ->expects($this->once())
           ->method("prepare")
           ->with("...")
           ->will($this->returnValue($fetchAllMock));

        $x = new myClass($mock);
        $this->assertSame("hello!", $x->doStuff());


    }

}