Index: third_party/gsutil/boto/tests/unit/ec2/test_address.py |
diff --git a/third_party/gsutil/boto/tests/unit/ec2/test_address.py b/third_party/gsutil/boto/tests/unit/ec2/test_address.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f26619794e16a9a3d2d9029c51959202f04ecf1a |
--- /dev/null |
+++ b/third_party/gsutil/boto/tests/unit/ec2/test_address.py |
@@ -0,0 +1,39 @@ |
+import mock |
+import 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("192.168.1.1") |
+ |
+ def test_associate_calls_connection_associate_address_with_correct_args(self): |
+ self.address.associate(1) |
+ self.address.connection.associate_address.assert_called_with(1, "192.168.1.1") |
+ |
+ def test_disassociate_calls_connection_disassociate_address_with_correct_args(self): |
+ self.address.disassociate() |
+ self.address.connection.disassociate_address.assert_called_with("192.168.1.1") |
+ |
+if __name__ == "__main__": |
+ unittest.main() |