Method test_retry

tests/test_http.py:998–1032

(self)

        self.assertEqual({"foo": "bar"}, response)

    def test_retry(self):
        num_retries = 6
        resp_seq = [({"status": "500"}, "")] * (num_retries - 4)
        resp_seq.append(({"status": "403"}, RATE_LIMIT_EXCEEDED_RESPONSE))
        resp_seq.append(
            ({"status": "403"}, USER_RATE_LIMIT_EXCEEDED_RESPONSE_NO_STATUS)
        )
        resp_seq.append(
            ({"status": "403"}, USER_RATE_LIMIT_EXCEEDED_RESPONSE_WITH_STATUS)
        )
        resp_seq.append(({"status": "429"}, ""))
        resp_seq.append(({"status": "200"}, "{}"))

        http = HttpMockSequence(resp_seq)
        model = JsonModel()
        uri = "https://www.googleapis.com/someapi/v1/collection/?foo=bar"
        method = "POST"
        request = HttpRequest(
            http,
            model.response,
            uri,
            method=method,
            body="{}",
            headers={"content-type": "application/json"},
        )

        sleeptimes = []
        request._sleep = lambda x: sleeptimes.append(x)
        request._rand = lambda: 10

        request.execute(num_retries=num_retries)

        self.assertEqual(num_retries, len(sleeptimes))
        for retry_num in range(num_retries):
            self.assertEqual(10 * 2 ** (retry_num + 1), sleeptimes[retry_num])

    def test_no_retry_succeeds(self):
        num_retries = 5