Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1)

Unified Diff: third_party/gsutil/third_party/boto/tests/integration/sns/test_connection.py

Issue 1377933002: [catapult] - Copy Telemetry's gsutilz over to third_party. (Closed) Base URL: https://github.com/catapult-project/catapult.git@master
Patch Set: Rename to gsutil. Created 5 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: third_party/gsutil/third_party/boto/tests/integration/sns/test_connection.py
diff --git a/third_party/gsutil/third_party/boto/tests/integration/sns/test_connection.py b/third_party/gsutil/third_party/boto/tests/integration/sns/test_connection.py
new file mode 100644
index 0000000000000000000000000000000000000000..6a359b1b7f803eea24bf1edeaacb26753553245b
--- /dev/null
+++ b/third_party/gsutil/third_party/boto/tests/integration/sns/test_connection.py
@@ -0,0 +1,68 @@
+# Copyright (c) 2013 Amazon.com, Inc. or its affiliates.
+# All rights reserved.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a
+# copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish, dis-
+# tribute, sublicense, and/or sell copies of the Software, and to permit
+# persons to whom the Software is furnished to do so, subject to the fol-
+# lowing conditions:
+#
+# The above copyright notice and this permission notice shall be included
+# in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
+# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
+# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+# IN THE SOFTWARE.
+from tests.compat import mock, unittest
+from boto.compat import http_client
+from boto.sns import connect_to_region
+
+
+class StubResponse(object):
+ status = 403
+ reason = 'nopenopenope'
+
+ def getheader(self, val):
+ return b''
+
+ def getheaders(self):
+ return b''
+
+ def read(self):
+ return b''
+
+
+class TestSNSConnection(unittest.TestCase):
+
+ sns = True
+
+ def setUp(self):
+ self.connection = connect_to_region('us-west-2')
+
+ def test_list_platform_applications(self):
+ response = self.connection.list_platform_applications()
+
+ def test_forced_host(self):
+ # This test asserts that the ``Host`` header is correctly set.
+ # On Python 2.5(.6), not having this in place would cause any SigV4
+ # calls to fail, due to a signature mismatch (the port would be present
+ # when it shouldn't be).
+ https = http_client.HTTPConnection
+ mpo = mock.patch.object
+
+ with mpo(https, 'request') as mock_request:
+ with mpo(https, 'getresponse', return_value=StubResponse()):
+ with self.assertRaises(self.connection.ResponseError):
+ self.connection.list_platform_applications()
+
+ # Now, assert that the ``Host`` was there & correct.
+ call = mock_request.call_args_list[0]
+ headers = call[0][3]
+ self.assertTrue('Host' in headers)
+ self.assertEqual(headers['Host'], 'sns.us-west-2.amazonaws.com')

Powered by Google App Engine
This is Rietveld 408576698