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 |