[catapult] - Copy Telemetry's gsutilz over to third_party.

third_party/gsutil/third_party/boto/tests/unit/ses/test_identity.py

Index: third_party/gsutil/third_party/boto/tests/unit/ses/test_identity.py
diff --git a/third_party/gsutil/third_party/boto/tests/unit/ses/test_identity.py b/third_party/gsutil/third_party/boto/tests/unit/ses/test_identity.py
new file mode 100755
index 0000000000000000000000000000000000000000..1187514abb2098e06ac5726798fd2c510b1005af
--- /dev/null
+++ b/third_party/gsutil/third_party/boto/tests/unit/ses/test_identity.py
@@ -0,0 +1,193 @@
+#!/usr/bin/env python
+# 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.unit import unittest
+from tests.unit import AWSMockServiceTestCase
+
+from boto.jsonresponse import ListElement
+from boto.ses.connection import SESConnection
+
+
+class TestSESIdentity(AWSMockServiceTestCase):
+    connection_class = SESConnection
+
+    def setUp(self):
+        super(TestSESIdentity, self).setUp()
+
+    def default_body(self):
+        return b"""<GetIdentityDkimAttributesResponse \
+xmlns="http://ses.amazonaws.com/doc/2010-12-01/">
+  <GetIdentityDkimAttributesResult>
+    <DkimAttributes>
+      <entry>
+        <key>test@amazon.com</key>
+      <value>
+        <DkimEnabled>true</DkimEnabled>
+        <DkimVerificationStatus>Success</DkimVerificationStatus>
+        <DkimTokens>
+          <member>vvjuipp74whm76gqoni7qmwwn4w4qusjiainivf6f</member>
+          <member>3frqe7jn4obpuxjpwpolz6ipb3k5nvt2nhjpik2oy</member>
+          <member>wrqplteh7oodxnad7hsl4mixg2uavzneazxv5sxi2</member>
+        </DkimTokens>
+      </value>
+    </entry>
+    <entry>
+      <key>secondtest@amazon.com</key>
+        <value>
+          <DkimEnabled>false</DkimEnabled>
+          <DkimVerificationStatus>NotStarted</DkimVerificationStatus>
+        </value>
+      </entry>
+    </DkimAttributes>
+  </GetIdentityDkimAttributesResult>
+  <ResponseMetadata>
+    <RequestId>bb5a105d-c468-11e1-82eb-dff885ccc06a</RequestId>
+  </ResponseMetadata>
+</GetIdentityDkimAttributesResponse>"""
+
+    def test_ses_get_identity_dkim_list(self):
+        self.set_http_response(status_code=200)
+
+        response = self.service_connection\
+                       .get_identity_dkim_attributes(['test@amazon.com', 'secondtest@amazon.com'])
+
+        response = response['GetIdentityDkimAttributesResponse']
+        result = response['GetIdentityDkimAttributesResult']
+
+        first_entry = result['DkimAttributes'][0]
+        entry_key = first_entry['key']
+        attributes = first_entry['value']
+        tokens = attributes['DkimTokens']
+
+        self.assertEqual(entry_key, 'test@amazon.com')
+        self.assertEqual(ListElement, type(tokens))
+        self.assertEqual(3, len(tokens))
+        self.assertEqual('vvjuipp74whm76gqoni7qmwwn4w4qusjiainivf6f',
+                         tokens[0])
+        self.assertEqual('3frqe7jn4obpuxjpwpolz6ipb3k5nvt2nhjpik2oy',
+                         tokens[1])
+        self.assertEqual('wrqplteh7oodxnad7hsl4mixg2uavzneazxv5sxi2',
+                         tokens[2])
+
+        second_entry = result['DkimAttributes'][1]
+        entry_key = second_entry['key']
+        attributes = second_entry['value']
+        dkim_enabled = attributes['DkimEnabled']
+        dkim_verification_status = attributes['DkimVerificationStatus']
+
+        self.assertEqual(entry_key, 'secondtest@amazon.com')
+        self.assertEqual(dkim_enabled, 'false')
+        self.assertEqual(dkim_verification_status, 'NotStarted')
+
+
+class TestSESSetIdentityNotificationTopic(AWSMockServiceTestCase):
+    connection_class = SESConnection
+
+    def setUp(self):
+        super(TestSESSetIdentityNotificationTopic, self).setUp()
+
+    def default_body(self):
+        return b"""<SetIdentityNotificationTopicResponse \
+        xmlns="http://ses.amazonaws.com/doc/2010-12-01/">
+           <SetIdentityNotificationTopicResult/>
+           <ResponseMetadata>
+             <RequestId>299f4af4-b72a-11e1-901f-1fbd90e8104f</RequestId>
+           </ResponseMetadata>
+         </SetIdentityNotificationTopicResponse>"""
+
+    def test_ses_set_identity_notification_topic_bounce(self):
+        self.set_http_response(status_code=200)
+
+        response = self.service_connection\
+                       .set_identity_notification_topic(
+                        identity='user@example.com',
+                        notification_type='Bounce',
+                        sns_topic='arn:aws:sns:us-east-1:123456789012:example')
+
+        response = response['SetIdentityNotificationTopicResponse']
+        result = response['SetIdentityNotificationTopicResult']
+
+        self.assertEqual(2, len(response))
+        self.assertEqual(0, len(result))
+
+    def test_ses_set_identity_notification_topic_complaint(self):
+        self.set_http_response(status_code=200)
+
+        response = self.service_connection\
+                       .set_identity_notification_topic(
+                        identity='user@example.com',
+                        notification_type='Complaint',
+                        sns_topic='arn:aws:sns:us-east-1:123456789012:example')
+
+        response = response['SetIdentityNotificationTopicResponse']
+        result = response['SetIdentityNotificationTopicResult']
+
+        self.assertEqual(2, len(response))
+        self.assertEqual(0, len(result))
+
+
+class TestSESSetIdentityFeedbackForwardingEnabled(AWSMockServiceTestCase):
+    connection_class = SESConnection
+
+    def setUp(self):
+        super(TestSESSetIdentityFeedbackForwardingEnabled, self).setUp()
+
+    def default_body(self):
+        return b"""<SetIdentityFeedbackForwardingEnabledResponse \
+        xmlns="http://ses.amazonaws.com/doc/2010-12-01/">
+          <SetIdentityFeedbackForwardingEnabledResult/>
+          <ResponseMetadata>
+            <RequestId>299f4af4-b72a-11e1-901f-1fbd90e8104f</RequestId>
+          </ResponseMetadata>
+        </SetIdentityFeedbackForwardingEnabledResponse>"""
+
+    def test_ses_set_identity_feedback_forwarding_enabled_true(self):
+        self.set_http_response(status_code=200)
+
+        response = self.service_connection\
+                       .set_identity_feedback_forwarding_enabled(
+                            identity='user@example.com',
+                            forwarding_enabled=True)
+
+        response = response['SetIdentityFeedbackForwardingEnabledResponse']
+        result = response['SetIdentityFeedbackForwardingEnabledResult']
+
+        self.assertEqual(2, len(response))
+        self.assertEqual(0, len(result))
+
+    def test_ses_set_identity_notification_topic_enabled_false(self):
+        self.set_http_response(status_code=200)
+
+        response = self.service_connection\
+                       .set_identity_feedback_forwarding_enabled(
+                            identity='user@example.com',
+                            forwarding_enabled=False)
+
+        response = response['SetIdentityFeedbackForwardingEnabledResponse']
+        result = response['SetIdentityFeedbackForwardingEnabledResult']
+
+        self.assertEqual(2, len(response))
+        self.assertEqual(0, len(result))
+
+
+if __name__ == '__main__':
+    unittest.main()