Allow passing request type id as the first parameter to MockClientBuilder::shouldHandle()

Author: rybakit

src/Client/MockClientBuilder.php

@@ -53,11 +53,15 @@ public static function buildDefault() : Client
     }
 
     /**
-     * @param Request|Constraint $request
+     * @param Request|Constraint|int $request
      * @param Response ...$responses
      */
     public function shouldHandle($request, ...$responses) : self
     {
+        if (is_int($request)) {
+            $request = new IsRequestType($request);
+        }
+
         $this->requests->attach($request, $responses);
 
         return $this;

tests/Expectation/ExpressionContext/RequestCountContextTest.php

@@ -19,7 +19,6 @@
 use Tarantool\Client\RequestTypes;
 use Tarantool\PhpUnit\Client\ClientMocking;
 use Tarantool\PhpUnit\Client\DummyFactory;
-use Tarantool\PhpUnit\Client\IsRequestType;
 use Tarantool\PhpUnit\Expectation\ExpressionContext\RequestCountContext;
 use Tarantool\PhpUnit\Expectation\ExpressionContext\RequestCounter;
 
@@ -31,7 +30,7 @@ public function testGetValuesReturnsCorrectValues() : void
     {
         $mockClient = $this->getMockClientBuilder()
             ->shouldHandle(
-                new IsRequestType(RequestTypes::EVALUATE),
+                RequestTypes::EVALUATE,
                 DummyFactory::createResponseFromData([2]),
                 DummyFactory::createResponseFromData([3])
             )->build();