Fix tests for #658

w7corp · Apr 21, 2017 · 7ad8b97 · 7ad8b97
1 parent 0c66f95
commit 7ad8b97
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/Payment/PaymentAPITest.php b/tests/Payment/PaymentAPITest.php
@@ -45,7 +45,10 @@ public function getAPI()
                 'notify_url' => 'merchant_default_notify_url',
             ]);
 
-        $api = \Mockery::mock('EasyWeChat\Payment\API[getHttp]', [$merchant])->shouldAllowMockingProtectedMethods();
+        $api = \Mockery::mock('EasyWeChat\Payment\API[getHttp]', [$merchant])
+                 ->shouldAllowMockingProtectedMethods();
+
+        $api->shouldReceive('wrapApi')->passthru();
         $api->shouldReceive('getHttp')->andReturn($http);
 
         return $api;
@@ -204,7 +207,9 @@ public function testDownloadBill()
                 'notify_url' => 'merchant_default_notify_url',
             ]);
 
-        $api = \Mockery::mock('EasyWeChat\Payment\API[getHttp]', [$merchant]);
+        $api = \Mockery::mock('EasyWeChat\Payment\API[getHttp]', [$merchant])->shouldAllowMockingProtectedMethods();
+
+        $api->shouldReceive('wrapApi')->passthru();
         $api->shouldReceive('getHttp')->andReturn($http);
 
         $response = $api->downloadBill('20150901');