| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 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 textwrap | 5 import textwrap |
| 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 infra_api_clients.codereview.rietveld import Rietveld |
| 10 from common.rietveld import Rietveld | 10 from libs.http import retry_http_client |
| 11 | 11 |
| 12 | 12 |
| 13 class DummyHttpClient(retry_http_client.RetryHttpClient): | 13 class DummyHttpClient(retry_http_client.RetryHttpClient): |
| 14 | 14 |
| 15 def __init__(self): | 15 def __init__(self): |
| 16 super(DummyHttpClient, self).__init__() | 16 super(DummyHttpClient, self).__init__() |
| 17 self.responses = {} | 17 self.responses = {} |
| 18 self.requests = [] | 18 self.requests = [] |
| 19 | 19 |
| 20 def SetResponse(self, url, result): | 20 def SetResponse(self, url, result): |
| (...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 101 self.assertEqual(1, len(self.http_client.requests)) | 101 self.assertEqual(1, len(self.http_client.requests)) |
| 102 | 102 |
| 103 def testPostMessageFailOnPublish(self): | 103 def testPostMessageFailOnPublish(self): |
| 104 rietveld_url = 'https://test' | 104 rietveld_url = 'https://test' |
| 105 issue_url = '%s/123' % rietveld_url | 105 issue_url = '%s/123' % rietveld_url |
| 106 message_publish_url = '%s/publish' % issue_url | 106 message_publish_url = '%s/publish' % issue_url |
| 107 self.http_client.SetResponse('%s/xsrf_token' % rietveld_url, (302, 'login')) | 107 self.http_client.SetResponse('%s/xsrf_token' % rietveld_url, (302, 'login')) |
| 108 self.http_client.SetResponse(message_publish_url, (429, 'Error')) | 108 self.http_client.SetResponse(message_publish_url, (429, 'Error')) |
| 109 self.assertFalse(self.rietveld.PostMessage(issue_url, 'message')) | 109 self.assertFalse(self.rietveld.PostMessage(issue_url, 'message')) |
| 110 self.assertEqual(1, len(self.http_client.requests)) | 110 self.assertEqual(1, len(self.http_client.requests)) |
| OLD | NEW |