Add expected arguments param to createCallback()

Author: trowski

src/AsyncTestCase.php

@@ -191,10 +191,15 @@ final protected function setTimeout(float $seconds): void
     /**
      * @param int           $invocationCount Number of times the callback must be invoked or the test will fail.
      * @param callable|null $returnCallback Callable providing a return value for the callback.
+     * @param array         $expectArgs Arguments expected to be passed to the callback.
      *
      * @return \Closure
      */
-    final protected function createCallback(int $invocationCount, ?callable $returnCallback = null): \Closure
+    final protected function createCallback(
+        int $invocationCount,
+        ?callable $returnCallback = null,
+        array $expectArgs = [],
+    ): \Closure
     {
         $mock = $this->createMock(CallbackStub::class);
         $invocationMocker = $mock->expects(self::exactly($invocationCount))
@@ -204,6 +209,10 @@ final protected function createCallback(int $invocationCount, ?callable $returnC
             $invocationMocker->willReturnCallback($returnCallback);
         }
 
+        if ($expectArgs) {
+            $invocationMocker->with(...$expectArgs);
+        }
+
         return \Closure::fromCallable($mock);
     }
 }

test/AsyncTestCaseTest.php

@@ -8,6 +8,7 @@
 use Amp\PHPUnit\TestException;
 use Amp\PHPUnit\UnhandledException;
 use PHPUnit\Framework\AssertionFailedError;
+use PHPUnit\Framework\ExpectationFailedException;
 use Revolt\EventLoop;
 use function Amp\async;
 use function Amp\delay;
@@ -175,10 +176,7 @@ public function testSetMinimumRunTime(): void
 
     public function testCreateCallback(): void
     {
-        $mock = $this->createCallback(1, function (int $value): int {
-            return $value + 1;
-        });
-
+        $mock = $this->createCallback(1, fn (int $value) => $value + 1, [1]);
         self::assertSame(2, $mock(1));
     }