在Amp项目中实现异步测试的上下文跟踪

背景介绍

在现代PHP开发中，异步编程变得越来越重要。Amp作为一个流行的PHP异步编程框架，提供了强大的协程和Future支持。在编写异步集成测试时，我们经常需要处理多个并发测试用例的执行，同时还需要跟踪每个测试的状态和断言数量。

问题场景

假设我们有以下测试用例：

class MyTest {
    public function test1() {
        $result = $this->request('/delay/3'); // 使用Amp HTTP客户端
        $this->assertTrue($result['foo']);
    }

    public function test2() {
        $result = $this->request('/delay/1'); // 使用Amp HTTP客户端
        $this->assertTrue($result['foo']);
    }
}

传统同步执行方式需要4秒(3+1)，而利用Amp的异步特性，我们可以让这两个测试并发执行，总时间只需3秒。然而，在并发执行时，如何正确跟踪每个测试的断言数量成为一个挑战。

解决方案探索

初始实现

最初的实现使用了一个测试池(Pool)来管理并发测试：

while (!$pool->isEmpty()) {
    $test = $pool->getTestToRun();
    $futures[$test->getIdentifier()] = async(function () use ($test, &$futures) {
        $this->run($test);
        unset($futures[$test->getIdentifier()]);
    });
}

上下文跟踪问题

当测试中遇到I/O操作时，Amp会暂停当前协程并切换到其他协程。这导致在跟踪断言数量时，上下文可能会混淆。例如，test1暂停后执行test2时，test2的断言可能会被错误地计入test1。

FiberLocal解决方案

Amp基于协程(Fiber)实现异步操作。我们可以利用FiberLocal来为每个协程维护独立的上下文数据。FiberLocal类似于线程本地存储，但针对协程进行了优化。

实现方式如下：

$fiberLocal = new FiberLocal(function() {
    return ['assertionCount' => 0];
});

// 在测试中
$fiberLocal->get()['assertionCount']++;

这种方法确保了每个测试协程都有自己的断言计数器，不会互相干扰。

高级考量

嵌套协程处理

如果测试中创建了嵌套协程，FiberLocal的默认行为可能无法满足需求，因为子协程不会自动继承父协程的上下文。这时可以手动传递上下文：

$parentContext = $fiberLocal->get();
async(function() use ($fiberLocal, $parentContext) {
    $fiberLocal->set($parentContext);
    // 子协程逻辑
});

性能影响

虽然FiberLocal提供了便利，但频繁的上下文切换和存储访问可能带来性能开销。在性能敏感的场景中，应该尽量减少上下文数据的体积和访问频率。

最佳实践

1. 为每个测试协程创建独立的上下文存储
2. 在断言方法中自动更新上下文计数器
3. 处理嵌套协程时显式传递上下文
4. 在测试结束时收集和验证上下文数据

总结

通过利用Amp的FiberLocal特性，我们可以有效地解决异步测试中的上下文跟踪问题。这种方法不仅适用于断言计数，还可以扩展到其他需要在协程间隔离的状态管理场景。理解并合理应用这些模式，可以显著提升异步测试的可靠性和可维护性。