OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # | 2 # |
3 # Copyright 2014 Google Inc. All Rights Reserved. | 3 # Copyright 2014 Google Inc. All Rights Reserved. |
4 # | 4 # |
5 # Licensed under the Apache License, Version 2.0 (the "License"); | 5 # Licensed under the Apache License, Version 2.0 (the "License"); |
6 # you may not use this file except in compliance with the License. | 6 # you may not use this file except in compliance with the License. |
7 # You may obtain a copy of the License at | 7 # You may obtain a copy of the License at |
8 # | 8 # |
9 # http://www.apache.org/licenses/LICENSE-2.0 | 9 # http://www.apache.org/licenses/LICENSE-2.0 |
10 # | 10 # |
(...skipping 617 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
628 HttpMockWithSSLErrors(3, {'status': '200'}, '{"foo": "bar"}'), | 628 HttpMockWithSSLErrors(3, {'status': '200'}, '{"foo": "bar"}'), |
629 model.response, | 629 model.response, |
630 u'https://www.example.com/json_api_endpoint') | 630 u'https://www.example.com/json_api_endpoint') |
631 request._sleep = lambda _x: 0 # do nothing | 631 request._sleep = lambda _x: 0 # do nothing |
632 request._rand = lambda: 10 | 632 request._rand = lambda: 10 |
633 response = request.execute(num_retries=3) | 633 response = request.execute(num_retries=3) |
634 self.assertEqual({u'foo': u'bar'}, response) | 634 self.assertEqual({u'foo': u'bar'}, response) |
635 | 635 |
636 def test_retry_ssl_errors_resumable(self): | 636 def test_retry_ssl_errors_resumable(self): |
637 with open(datafile('small.png'), 'rb') as small_png_file: | 637 with open(datafile('small.png'), 'rb') as small_png_file: |
638 small_png_file_descriptor = BytesIO(small_png_file.read()) | 638 small_png_fd = BytesIO(small_png_file.read()) |
639 upload = MediaIoBaseUpload(fd=small_png_file, mimetype='image/png', | 639 upload = MediaIoBaseUpload(fd=small_png_fd, mimetype='image/png', |
640 chunksize=500, resumable=True) | 640 chunksize=500, resumable=True) |
641 model = JsonModel() | 641 model = JsonModel() |
642 | 642 |
643 request = HttpRequest( | 643 request = HttpRequest( |
644 HttpMockWithSSLErrors( | 644 HttpMockWithSSLErrors( |
645 3, {'status': '200', 'location': 'location'}, '{"foo": "bar"}'), | 645 3, {'status': '200', 'location': 'location'}, '{"foo": "bar"}'), |
646 model.response, | 646 model.response, |
647 u'https://www.example.com/file_upload', | 647 u'https://www.example.com/file_upload', |
648 method='POST', | 648 method='POST', |
649 resumable=upload) | 649 resumable=upload) |
650 request._sleep = lambda _x: 0 # do nothing | 650 request._sleep = lambda _x: 0 # do nothing |
651 request._rand = lambda: 10 | 651 request._rand = lambda: 10 |
652 response = request.execute(num_retries=3) | 652 response = request.execute(num_retries=3) |
653 self.assertEqual({u'foo': u'bar'}, response) | 653 self.assertEqual({u'foo': u'bar'}, response) |
654 | 654 |
655 def test_retry(self): | 655 def test_retry(self): |
656 num_retries = 5 | 656 num_retries = 5 |
657 resp_seq = [({'status': '500'}, '')] * num_retries | 657 resp_seq = [({'status': '500'}, '')] * num_retries |
658 resp_seq.append(({'status': '200'}, '{}')) | 658 resp_seq.append(({'status': '200'}, '{}')) |
659 | 659 |
660 http = HttpMockSequence(resp_seq) | 660 http = HttpMockSequence(resp_seq) |
661 model = JsonModel() | 661 model = JsonModel() |
662 uri = u'https://www.googleapis.com/someapi/v1/collection/?foo=bar' | 662 uri = u'https://www.googleapis.com/someapi/v1/collection/?foo=bar' |
663 method = u'POST' | 663 method = u'POST' |
(...skipping 420 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1084 model.response, | 1084 model.response, |
1085 'https://www.googleapis.com/someapi/v1/collection/?foo=bar', | 1085 'https://www.googleapis.com/someapi/v1/collection/?foo=bar', |
1086 method='GET', | 1086 method='GET', |
1087 headers={}) | 1087 headers={}) |
1088 self.assertRaises(HttpError, request.execute) | 1088 self.assertRaises(HttpError, request.execute) |
1089 | 1089 |
1090 | 1090 |
1091 if __name__ == '__main__': | 1091 if __name__ == '__main__': |
1092 logging.getLogger().setLevel(logging.ERROR) | 1092 logging.getLogger().setLevel(logging.ERROR) |
1093 unittest.main() | 1093 unittest.main() |
OLD | NEW |