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_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 |