use HTTP mock

stripe · Sep 10, 2015 · 829c559 · 829c559
1 parent 23f6d70
commit 829c559
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 10 deletions.
diff --git a/tests/RateLimitErrorTest.php b/tests/RateLimitErrorTest.php
@@ -4,17 +4,18 @@
 
 class RateLimitErrorTest extends TestCase
 {
+    private function rateLimitErrorResponse()
+    {
+        return array(
+            'error' => array(),
+        );
+    }
+
     public function testRateLimit()
     {
         try {
-            throw new Error\RateLimit(
-                "rate limited",
-                "some_param",
-                429,
-                "{'foo':'bar'}",
-                array('foo' => 'bar')
-            );
-            $this->fail("Did not raise error");
+            $this->mockRequest('GET', '/v1/accounts/acct_DEF', array(), $this->rateLimitErrorResponse(), 429);
+            Account::retrieve('acct_DEF');
         } catch (Error\RateLimit $e) {
             $this->assertSame(429, $e->getHttpStatus());
         }

diff --git a/tests/TestCase.php b/tests/TestCase.php
@@ -29,13 +29,13 @@ protected function setUp()
         $this->call = 0;
     }
 
-    protected function mockRequest($method, $path, $params = array(), $return = array('id' => 'myId'))
+    protected function mockRequest($method, $path, $params = array(), $return = array('id' => 'myId'), $rcode = 200)
     {
         $mock = $this->setUpMockRequest();
         $mock->expects($this->at($this->call++))
              ->method('request')
              ->with(strtolower($method), 'https://api.stripe.com' . $path, $this->anything(), $params, false)
-             ->willReturn(array(json_encode($return), 200, array()));
+             ->willReturn(array(json_encode($return), $rcode, array()));
     }
 
     private function setUpMockRequest()