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

third_party/gsutil/third_party/boto/tests/unit/ec2/test_address.py

Index: third_party/gsutil/third_party/boto/tests/unit/ec2/test_address.py
diff --git a/third_party/gsutil/third_party/boto/tests/unit/ec2/test_address.py b/third_party/gsutil/third_party/boto/tests/unit/ec2/test_address.py
new file mode 100644
index 0000000000000000000000000000000000000000..519918e752dd65b88356a29d53d270249613501d
--- /dev/null
+++ b/third_party/gsutil/third_party/boto/tests/unit/ec2/test_address.py
@@ -0,0 +1,145 @@
+from tests.compat import mock, unittest
+
+from boto.ec2.address import Address
+
+
+class AddressTest(unittest.TestCase):
+    def setUp(self):
+        self.address = Address()
+        self.address.connection = mock.Mock()
+        self.address.public_ip = "192.168.1.1"
+
+    def check_that_attribute_has_been_set(self, name, value, attribute):
+        self.address.endElement(name, value, None)
+        self.assertEqual(getattr(self.address, attribute), value)
+
+    def test_endElement_sets_correct_attributes_with_values(self):
+        for arguments in [("publicIp", "192.168.1.1", "public_ip"),
+                          ("instanceId", 1, "instance_id"),
+                          ("domain", "some domain", "domain"),
+                          ("allocationId", 1, "allocation_id"),
+                          ("associationId", 1, "association_id"),
+                          ("somethingRandom", "somethingRandom", "somethingRandom")]:
+            self.check_that_attribute_has_been_set(arguments[0], arguments[1], arguments[2])
+
+    def test_release_calls_connection_release_address_with_correct_args(self):
+        self.address.release()
+        self.address.connection.release_address.assert_called_with(
+            public_ip="192.168.1.1",
+            dry_run=False
+        )
+
+    def test_associate_calls_connection_associate_address_with_correct_args(self):
+        self.address.associate(1)
+        self.address.connection.associate_address.assert_called_with(
+            instance_id=1,
+            public_ip="192.168.1.1",
+            allow_reassociation=False,
+            network_interface_id=None,
+            private_ip_address=None,
+            dry_run=False
+        )
+
+    def test_disassociate_calls_connection_disassociate_address_with_correct_args(self):
+        self.address.disassociate()
+        self.address.connection.disassociate_address.assert_called_with(
+            public_ip="192.168.1.1",
+            dry_run=False
+        )
+
+
+class AddressWithAllocationTest(unittest.TestCase):
+    def setUp(self):
+        self.address = Address()
+        self.address.connection = mock.Mock()
+        self.address.public_ip = "192.168.1.1"
+        self.address.allocation_id = "aid1"
+
+    def check_that_attribute_has_been_set(self, name, value, attribute):
+        self.address.endElement(name, value, None)
+        self.assertEqual(getattr(self.address, attribute), value)
+
+    def test_endElement_sets_correct_attributes_with_values(self):
+        for arguments in [("publicIp", "192.168.1.1", "public_ip"),
+                          ("instanceId", 1, "instance_id"),
+                          ("domain", "some domain", "domain"),
+                          ("allocationId", 1, "allocation_id"),
+                          ("associationId", 1, "association_id"),
+                          ("somethingRandom", "somethingRandom", "somethingRandom")]:
+            self.check_that_attribute_has_been_set(arguments[0], arguments[1], arguments[2])
+
+    def test_release_calls_connection_release_address_with_correct_args(self):
+        self.address.release()
+        self.address.connection.release_address.assert_called_with(
+            allocation_id="aid1",
+            dry_run=False
+        )
+
+    def test_associate_calls_connection_associate_address_with_correct_args(self):
+        self.address.associate(1)
+        self.address.connection.associate_address.assert_called_with(
+            instance_id=1,
+            public_ip="192.168.1.1",
+            allocation_id="aid1",
+            network_interface_id=None,
+            private_ip_address=None,
+            allow_reassociation=False,
+            dry_run=False
+        )
+
+    def test_disassociate_calls_connection_disassociate_address_with_correct_args(self):
+        self.address.disassociate()
+        self.address.connection.disassociate_address.assert_called_with(
+            public_ip="192.168.1.1",
+            dry_run=False
+        )
+
+class AddressWithNetworkInterfaceTest(unittest.TestCase):
+    def setUp(self):
+        self.address = Address()
+        self.address.connection = mock.Mock()
+        self.address.public_ip = "192.168.1.1"
+        self.address.allocation_id = "aid1"
+
+    def check_that_attribute_has_been_set(self, name, value, attribute):
+        self.address.endElement(name, value, None)
+        self.assertEqual(getattr(self.address, attribute), value)
+
+    def test_endElement_sets_correct_attributes_with_values(self):
+        for arguments in [("publicIp", "192.168.1.1", "public_ip"),
+                          ("instanceId", 1, "instance_id"),
+                          ("domain", "some domain", "domain"),
+                          ("allocationId", 1, "allocation_id"),
+                          ("associationId", 1, "association_id"),
+                          ("somethingRandom", "somethingRandom", "somethingRandom")]:
+            self.check_that_attribute_has_been_set(arguments[0], arguments[1], arguments[2])
+
+
+    def test_release_calls_connection_release_address_with_correct_args(self):
+        self.address.release()
+        self.address.connection.release_address.assert_called_with(
+            allocation_id="aid1",
+            dry_run=False
+        )
+
+    def test_associate_calls_connection_associate_address_with_correct_args(self):
+        self.address.associate(network_interface_id=1)
+        self.address.connection.associate_address.assert_called_with(
+            instance_id=None,
+            public_ip="192.168.1.1",
+            network_interface_id=1,
+            private_ip_address=None,
+            allocation_id="aid1",
+            allow_reassociation=False,
+            dry_run=False
+        )
+
+    def test_disassociate_calls_connection_disassociate_address_with_correct_args(self):
+        self.address.disassociate()
+        self.address.connection.disassociate_address.assert_called_with(
+            public_ip="192.168.1.1",
+            dry_run=False
+        )
+
+if __name__ == "__main__":
+    unittest.main()