| OLD | NEW |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import urllib | 5 import urllib |
| 6 | 6 |
| 7 from testing_utils import testing | 7 from testing_utils import testing |
| 8 | 8 |
| 9 from common import retry_http_client | 9 from libs.http import retry_http_client |
| 10 | 10 |
| 11 | 11 |
| 12 class DummyHttpClient(retry_http_client.RetryHttpClient): | 12 class DummyHttpClient(retry_http_client.RetryHttpClient): |
| 13 def __init__(self, simulated_failures, failure_status): | 13 def __init__(self, simulated_failures, failure_status): |
| 14 super(DummyHttpClient, self).__init__() | 14 super(DummyHttpClient, self).__init__() |
| 15 self.requests = [] | 15 self.requests = [] |
| 16 self.request_count = 0 | 16 self.request_count = 0 |
| 17 self.simulated_failures = simulated_failures | 17 self.simulated_failures = simulated_failures |
| 18 self.failure_status = failure_status | 18 self.failure_status = failure_status |
| 19 | 19 |
| (...skipping 160 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 180 self.assertEquals(expected_status_code, status_code) | 180 self.assertEquals(expected_status_code, status_code) |
| 181 self.assertEquals('failure - PUT', content) | 181 self.assertEquals('failure - PUT', content) |
| 182 | 182 |
| 183 def testRetryForPut(self): | 183 def testRetryForPut(self): |
| 184 dummy_http_client = DummyHttpClient(5, 503) | 184 dummy_http_client = DummyHttpClient(5, 503) |
| 185 status_code, content = dummy_http_client.Put('http://test', {'data': 0}, | 185 status_code, content = dummy_http_client.Put('http://test', {'data': 0}, |
| 186 max_retries=3) | 186 max_retries=3) |
| 187 self.assertEquals(3, dummy_http_client.request_count) | 187 self.assertEquals(3, dummy_http_client.request_count) |
| 188 self.assertEquals(503, status_code) | 188 self.assertEquals(503, status_code) |
| 189 self.assertEquals('failure - PUT', content) | 189 self.assertEquals('failure - PUT', content) |
| OLD | NEW |