Mock method only if necessary

pagarme · Jun 16, 2017 · c08349a · c08349a
1 parent 1fcbe3f
commit c08349a
Showing 1 changed file with 25 additions and 17 deletions.
diff --git a/tests/unit/ClientTest.php b/tests/unit/ClientTest.php
@@ -37,8 +37,10 @@ public function setup()
      */
     public function mustSendRequest()
     {
-        $this->guzzleClientMock->method('createRequest')
-            ->willReturn($this->getMock('GuzzleHttp\Message\RequestInterface'));
+        if ($this->isUsingLegacyGuzzle()) {
+            $this->guzzleClientMock->method('createRequest')
+                ->willReturn($this->getMock('GuzzleHttp\Message\RequestInterface'));
+        }
 
         $responseMock = $this->getResponseMock();
 
@@ -58,18 +60,20 @@ public function mustSendRequest()
      */
     public function mustSendRequestWithProperContent()
     {
-        $this->guzzleClientMock->method('createRequest')
-            ->with(
-                RequestInterface::HTTP_POST,
-                self::REQUEST_PATH,
-                [
-                    'json' => [
-                        'content'        => self::CONTENT,
-                        'api_key'        => self::API_KEY
+        if ($this->isUsingLegacyGuzzle()) {
+            $this->guzzleClientMock->method('createRequest')
+                ->with(
+                    RequestInterface::HTTP_POST,
+                    self::REQUEST_PATH,
+                    [
+                        'json' => [
+                            'content'        => self::CONTENT,
+                            'api_key'        => self::API_KEY
+                        ]
                     ]
-                ]
-            )
-            ->willReturn($this->getMock('GuzzleHttp\Message\RequestInterface'));
+                )
+                ->willReturn($this->getMock('GuzzleHttp\Message\RequestInterface'));
+        }
 
         $responseMock = $this->getResponseMock();
 
@@ -94,8 +98,10 @@ public function mustReturnClientExeptionWhenGetRequestException()
             $this->getGuzzleRequestInterfaceName()
         );
 
-        $this->guzzleClientMock->method('createRequest')
-            ->willReturn($guzzleRequestMock);
+        if ($this->isUsingLegacyGuzzle()) {
+            $this->guzzleClientMock->method('createRequest')
+                ->willReturn($guzzleRequestMock);
+        }
 
         $this->guzzleClientMock->method('send')
             ->will(
@@ -127,8 +133,10 @@ public function mustSetDefaultTimeout()
             )->disableOriginalConstructor()
             ->getMock();
 
-        $this->guzzleClientMock->method('createRequest')
-            ->willReturn($guzzleRequestMock);
+        if ($this->isUsingLegacyGuzzle()) {
+            $this->guzzleClientMock->method('createRequest')
+                ->willReturn($guzzleRequestMock);
+        }
 
         $responseMock = $this->getResponseMock();