PHPUnit Method Mock不同参数调用的处理策略：如何有效应对？

2024-03-02 02:47:45

PHPUnit中处理不同参数调用的Method Mock

简介

在PHPUnit测试框架中，Method Mock是一种强大的工具，它允许我们模拟类的特定方法的行为，从而在单元测试中控制其输出。当需要处理具有不同参数的不同方法调用时，我们可以采用多种策略。

策略一：使用预期方法

最直接的方法是为每个不同的参数组合创建单独的预期。预期方法允许我们指定对方法调用的特定期望，包括参数和返回值。例如：

// 模拟一个名为DB的类的Query方法
$dbMock = $this->getMockBuilder(DB::class)
    ->setMethods(['Query'])
    ->getMock();

// 创建对Query方法的预期，指定参数和返回值
$dbMock->expects($this->once())
    ->method('Query')
    ->with('SELECT * FROM users')
    ->willReturn($users);

$dbMock->expects($this->once())
    ->method('Query')
    ->with('SELECT * FROM products')
    ->willReturn($products);

策略二：使用预期参数匹配器

如果需要处理大量的参数组合，可以使用预期参数匹配器来简化代码。匹配器允许我们使用回调函数或正则表达式来匹配参数值。例如：

// 使用回调函数匹配器
$dbMock = $this->getMockBuilder(DB::class)
    ->setMethods(['Query'])
    ->getMock();

$dbMock->expects($this->any())
    ->method('Query')
    ->with($this->callback(function (string $query) {
        return strpos($query, 'SELECT * FROM') !== false;
    }))
    ->willReturn($resultSet);

策略三：使用ConsecutiveCalls方法

如果需要模拟对同一方法的连续调用，可以使用consecutiveCalls()方法。此方法允许我们按顺序定义方法的返回值，从而模拟连续的调用行为。例如：

// 模拟对Query方法的连续调用
$dbMock = $this->getMockBuilder(DB::class)
    ->setMethods(['Query'])
    ->getMock();

$dbMock->method('Query')
    ->will($this->consecutiveCalls(
        [$users],
        [$products]
    ));