googlecloud 0.0.4 → 0.0.6

data/packages/gsutil/boto/tests/unit/ec2/test_instance.py

@@ -0,0 +1,243 @@
+#!/usr/bin/env python
+
+from tests.unit import unittest
+from tests.unit import AWSMockServiceTestCase
+
+import mock
+
+from boto.ec2.connection import EC2Connection
+
+DESCRIBE_INSTANCE_VPC = r"""<?xml version="1.0" encoding="UTF-8"?>
+<DescribeInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2012-10-01/">
+    <requestId>c6132c74-b524-4884-87f5-0f4bde4a9760</requestId>
+    <reservationSet>
+        <item>
+            <reservationId>r-72ef4a0a</reservationId>
+            <ownerId>184906166255</ownerId>
+            <groupSet/>
+            <instancesSet>
+                <item>
+                    <instanceId>i-instance</instanceId>
+                    <imageId>ami-1624987f</imageId>
+                    <instanceState>
+                        <code>16</code>
+                        <name>running</name>
+                    </instanceState>
+                    <privateDnsName/>
+                    <dnsName/>
+                    <reason/>
+                    <keyName>mykeypair</keyName>
+                    <amiLaunchIndex>0</amiLaunchIndex>
+                    <productCodes/>
+                    <instanceType>m1.small</instanceType>
+                    <launchTime>2012-12-14T23:48:37.000Z</launchTime>
+                    <placement>
+                        <availabilityZone>us-east-1d</availabilityZone>
+                        <groupName/>
+                        <tenancy>default</tenancy>
+                    </placement>
+                    <kernelId>aki-88aa75e1</kernelId>
+                    <monitoring>
+                        <state>disabled</state>
+                    </monitoring>
+                    <subnetId>subnet-0dc60667</subnetId>
+                    <vpcId>vpc-id</vpcId>
+                    <privateIpAddress>10.0.0.67</privateIpAddress>
+                    <sourceDestCheck>true</sourceDestCheck>
+                    <groupSet>
+                        <item>
+                            <groupId>sg-id</groupId>
+                            <groupName>WebServerSG</groupName>
+                        </item>
+                    </groupSet>
+                    <architecture>x86_64</architecture>
+                    <rootDeviceType>ebs</rootDeviceType>
+                    <rootDeviceName>/dev/sda1</rootDeviceName>
+                    <blockDeviceMapping>
+                        <item>
+                            <deviceName>/dev/sda1</deviceName>
+                            <ebs>
+                                <volumeId>vol-id</volumeId>
+                                <status>attached</status>
+                                <attachTime>2012-12-14T23:48:43.000Z</attachTime>
+                                <deleteOnTermination>true</deleteOnTermination>
+                            </ebs>
+                        </item>
+                    </blockDeviceMapping>
+                    <virtualizationType>paravirtual</virtualizationType>
+                    <clientToken>foo</clientToken>
+                    <tagSet>
+                        <item>
+                            <key>Name</key>
+                            <value/>
+                        </item>
+                    </tagSet>
+                    <hypervisor>xen</hypervisor>
+                    <networkInterfaceSet>
+                        <item>
+                            <networkInterfaceId>eni-id</networkInterfaceId>
+                            <subnetId>subnet-id</subnetId>
+                            <vpcId>vpc-id</vpcId>
+                            <description>Primary network interface</description>
+                            <ownerId>ownerid</ownerId>
+                            <status>in-use</status>
+                            <privateIpAddress>10.0.0.67</privateIpAddress>
+                            <sourceDestCheck>true</sourceDestCheck>
+                            <groupSet>
+                                <item>
+                                    <groupId>sg-id</groupId>
+                                    <groupName>WebServerSG</groupName>
+                                </item>
+                            </groupSet>
+                            <attachment>
+                                <attachmentId>eni-attach-id</attachmentId>
+                                <deviceIndex>0</deviceIndex>
+                                <status>attached</status>
+                                <attachTime>2012-12-14T23:48:37.000Z</attachTime>
+                                <deleteOnTermination>true</deleteOnTermination>
+                            </attachment>
+                            <privateIpAddressesSet>
+                                <item>
+                                    <privateIpAddress>10.0.0.67</privateIpAddress>
+                                    <primary>true</primary>
+                                </item>
+                                <item>
+                                    <privateIpAddress>10.0.0.54</privateIpAddress>
+                                    <primary>false</primary>
+                                </item>
+                                <item>
+                                    <privateIpAddress>10.0.0.55</privateIpAddress>
+                                    <primary>false</primary>
+                                </item>
+                            </privateIpAddressesSet>
+                        </item>
+                    </networkInterfaceSet>
+                    <ebsOptimized>false</ebsOptimized>
+                </item>
+            </instancesSet>
+        </item>
+    </reservationSet>
+</DescribeInstancesResponse>
+"""
+
+RUN_INSTANCE_RESPONSE = r"""
+<RunInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2012-06-01/">
+    <requestId>ad4b83c2-f606-4c39-90c6-5dcc5be823e1</requestId>
+    <reservationId>r-c5cef7a7</reservationId>
+    <ownerId>ownerid</ownerId>
+    <groupSet>
+        <item>
+            <groupId>sg-id</groupId>
+            <groupName>SSH</groupName>
+        </item>
+    </groupSet>
+    <instancesSet>
+        <item>
+            <instanceId>i-ff0f1299</instanceId>
+            <imageId>ami-ed65ba84</imageId>
+            <instanceState>
+                <code>0</code>
+                <name>pending</name>
+            </instanceState>
+            <privateDnsName/>
+            <dnsName/>
+            <reason/>
+            <keyName>awskeypair</keyName>
+            <amiLaunchIndex>0</amiLaunchIndex>
+            <productCodes/>
+            <instanceType>t1.micro</instanceType>
+            <launchTime>2012-05-30T19:21:18.000Z</launchTime>
+            <placement>
+                <availabilityZone>us-east-1a</availabilityZone>
+                <groupName/>
+                <tenancy>default</tenancy>
+            </placement>
+            <kernelId>aki-b6aa75df</kernelId>
+            <monitoring>
+                <state>disabled</state>
+            </monitoring>
+            <groupSet>
+                <item>
+                    <groupId>sg-99a710f1</groupId>
+                    <groupName>SSH</groupName>
+                </item>
+            </groupSet>
+            <stateReason>
+                <code>pending</code>
+                <message>pending</message>
+            </stateReason>
+            <architecture>i386</architecture>
+            <rootDeviceType>ebs</rootDeviceType>
+            <rootDeviceName>/dev/sda1</rootDeviceName>
+            <blockDeviceMapping/>
+            <virtualizationType>paravirtual</virtualizationType>
+            <clientToken/>
+            <hypervisor>xen</hypervisor>
+            <networkInterfaceSet/>
+            <iamInstanceProfile>
+                <arn>arn:aws:iam::ownerid:instance-profile/myinstanceprofile</arn>
+                <id>iamid</id>
+            </iamInstanceProfile>
+        </item>
+    </instancesSet>
+</RunInstancesResponse>
+"""
+
+
+class TestRunInstanceResponseParsing(unittest.TestCase):
+    def testIAMInstanceProfileParsedCorrectly(self):
+        ec2 = EC2Connection(aws_access_key_id='aws_access_key_id',
+                            aws_secret_access_key='aws_secret_access_key')
+        mock_response = mock.Mock()
+        mock_response.read.return_value = RUN_INSTANCE_RESPONSE
+        mock_response.status = 200
+        ec2.make_request = mock.Mock(return_value=mock_response)
+        reservation = ec2.run_instances(image_id='ami-12345')
+        self.assertEqual(len(reservation.instances), 1)
+        instance = reservation.instances[0]
+        self.assertEqual(instance.image_id, 'ami-ed65ba84')
+        # iamInstanceProfile has an ID element, so we want to make sure
+        # that this does not map to instance.id (which should be the
+        # id of the ec2 instance).
+        self.assertEqual(instance.id, 'i-ff0f1299')
+        self.assertDictEqual(
+            instance.instance_profile,
+            {'arn': ('arn:aws:iam::ownerid:'
+                     'instance-profile/myinstanceprofile'),
+             'id': 'iamid'})
+
+
+class TestDescribeInstances(AWSMockServiceTestCase):
+    connection_class = EC2Connection
+
+    def default_body(self):
+        return DESCRIBE_INSTANCE_VPC
+
+    def test_multiple_private_ip_addresses(self):
+        self.set_http_response(status_code=200)
+
+        api_response = self.service_connection.get_all_instances()
+        self.assertEqual(len(api_response), 1)
+
+        instances = api_response[0].instances
+        self.assertEqual(len(instances), 1)
+
+        instance = instances[0]
+        self.assertEqual(len(instance.interfaces), 1)
+
+        interface = instance.interfaces[0]
+        self.assertEqual(len(interface.private_ip_addresses), 3)
+
+        addresses = interface.private_ip_addresses
+        self.assertEqual(addresses[0].private_ip_address, '10.0.0.67')
+        self.assertTrue(addresses[0].primary)
+
+        self.assertEqual(addresses[1].private_ip_address, '10.0.0.54')
+        self.assertFalse(addresses[1].primary)
+
+        self.assertEqual(addresses[2].private_ip_address, '10.0.0.55')
+        self.assertFalse(addresses[2].primary)
+
+
+if __name__ == '__main__':
+    unittest.main()

data/packages/gsutil/boto/tests/unit/ec2/test_networkinterface.py

@@ -0,0 +1,140 @@
+#!/usr/bin/env python
+# Copyright (c) 2012 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 boto.ec2.networkinterface import NetworkInterfaceCollection
+from boto.ec2.networkinterface import NetworkInterfaceSpecification
+from boto.ec2.networkinterface import PrivateIPAddress
+
+
+class TestNetworkInterfaceCollection(unittest.TestCase):
+    maxDiff = None
+
+    def setUp(self):
+        self.private_ip_address1 = PrivateIPAddress(
+            private_ip_address='10.0.0.10', primary=False)
+        self.private_ip_address2 = PrivateIPAddress(
+            private_ip_address='10.0.0.11', primary=False)
+        self.network_interfaces_spec1 = NetworkInterfaceSpecification(
+            device_index=1, subnet_id='subnet_id',
+            description='description1',
+            private_ip_address='10.0.0.54', delete_on_termination=False,
+            private_ip_addresses=[self.private_ip_address1,
+                                  self.private_ip_address2])
+
+        self.private_ip_address3 = PrivateIPAddress(
+            private_ip_address='10.0.1.10', primary=False)
+        self.private_ip_address4 = PrivateIPAddress(
+            private_ip_address='10.0.1.11', primary=False)
+        self.network_interfaces_spec2 = NetworkInterfaceSpecification(
+            device_index=2, subnet_id='subnet_id2',
+            description='description2',
+            groups=['group_id1', 'group_id2'],
+            private_ip_address='10.0.1.54', delete_on_termination=False,
+            private_ip_addresses=[self.private_ip_address3,
+                                  self.private_ip_address4])
+
+    def test_param_serialization(self):
+        collection = NetworkInterfaceCollection(self.network_interfaces_spec1,
+                                                self.network_interfaces_spec2)
+        params = {}
+        collection.build_list_params(params)
+        self.assertDictEqual(params, {
+            'NetworkInterface.1.DeviceIndex': '1',
+            'NetworkInterface.1.DeleteOnTermination': 'false',
+            'NetworkInterface.1.Description': 'description1',
+            'NetworkInterface.1.PrivateIpAddress': '10.0.0.54',
+            'NetworkInterface.1.SubnetId': 'subnet_id',
+            'NetworkInterface.1.PrivateIpAddresses.1.Primary': 'false',
+            'NetworkInterface.1.PrivateIpAddresses.1.PrivateIpAddress':
+                '10.0.0.10',
+            'NetworkInterface.1.PrivateIpAddresses.2.Primary': 'false',
+            'NetworkInterface.1.PrivateIpAddresses.2.PrivateIpAddress':
+                '10.0.0.11',
+            'NetworkInterface.2.DeviceIndex': '2',
+            'NetworkInterface.2.Description': 'description2',
+            'NetworkInterface.2.DeleteOnTermination': 'false',
+            'NetworkInterface.2.PrivateIpAddress': '10.0.1.54',
+            'NetworkInterface.2.SubnetId': 'subnet_id2',
+            'NetworkInterface.2.SecurityGroupId.1': 'group_id1',
+            'NetworkInterface.2.SecurityGroupId.2': 'group_id2',
+            'NetworkInterface.2.PrivateIpAddresses.1.Primary': 'false',
+            'NetworkInterface.2.PrivateIpAddresses.1.PrivateIpAddress':
+                '10.0.1.10',
+            'NetworkInterface.2.PrivateIpAddresses.2.Primary': 'false',
+            'NetworkInterface.2.PrivateIpAddresses.2.PrivateIpAddress':
+                '10.0.1.11',
+        })
+
+    def test_add_prefix_to_serialization(self):
+        return
+        collection = NetworkInterfaceCollection(self.network_interfaces_spec1,
+                                                self.network_interfaces_spec2)
+        params = {}
+        collection.build_list_params(params, prefix='LaunchSpecification.')
+        # We already tested the actual serialization previously, so
+        # we're just checking a few keys to make sure we get the proper
+        # prefix.
+        self.assertDictEqual(params, {
+            'LaunchSpecification.NetworkInterface.1.DeviceIndex': '1',
+            'LaunchSpecification.NetworkInterface.1.DeleteOnTermination':
+                'false',
+            'LaunchSpecification.NetworkInterface.1.Description':
+                'description1',
+            'LaunchSpecification.NetworkInterface.1.PrivateIpAddress':
+                '10.0.0.54',
+            'LaunchSpecification.NetworkInterface.1.SubnetId': 'subnet_id',
+            'LaunchSpecification.NetworkInterface.1.PrivateIpAddresses.1.Primary':
+                'false',
+            'LaunchSpecification.NetworkInterface.1.PrivateIpAddresses.1.PrivateIpAddress':
+                '10.0.0.10',
+            'LaunchSpecification.NetworkInterface.1.PrivateIpAddresses.2.Primary': 'false',
+            'LaunchSpecification.NetworkInterface.1.PrivateIpAddresses.2.PrivateIpAddress':
+                '10.0.0.11',
+            'LaunchSpecification.NetworkInterface.2.DeviceIndex': '2',
+            'LaunchSpecification.NetworkInterface.2.Description':
+                'description2',
+            'LaunchSpecification.NetworkInterface.2.DeleteOnTermination':
+                'false',
+            'LaunchSpecification.NetworkInterface.2.PrivateIpAddress':
+                '10.0.1.54',
+            'LaunchSpecification.NetworkInterface.2.SubnetId': 'subnet_id2',
+            'LaunchSpecification.NetworkInterface.2.SecurityGroupId.1':
+                'group_id1',
+            'LaunchSpecification.NetworkInterface.2.SecurityGroupId.2':
+                'group_id2',
+            'LaunchSpecification.NetworkInterface.2.PrivateIpAddresses.1.Primary':
+                'false',
+            'LaunchSpecification.NetworkInterface.2.PrivateIpAddresses.1.PrivateIpAddress':
+                '10.0.1.10',
+            'LaunchSpecification.NetworkInterface.2.PrivateIpAddresses.2.Primary':
+                'false',
+            'LaunchSpecification.NetworkInterface.2.PrivateIpAddresses.2.PrivateIpAddress':
+                '10.0.1.11',
+        })
+
+
+if __name__ == '__main__':
+    unittest.main()

data/packages/gsutil/boto/tests/unit/ec2/test_volume.py

@@ -0,0 +1,248 @@
+import mock
+from tests.unit import unittest
+
+from boto.ec2.snapshot import Snapshot
+from boto.ec2.tag import Tag, TagSet
+from boto.ec2.volume import Volume, AttachmentSet, VolumeAttribute
+
+
+class VolumeTests(unittest.TestCase):
+    def setUp(self):
+        self.attach_data = AttachmentSet()
+        self.attach_data.id = 1
+        self.attach_data.instance_id = 2
+        self.attach_data.status = "some status"
+        self.attach_data.attach_time = 5
+        self.attach_data.device = "/dev/null"
+
+        self.volume_one = Volume()
+        self.volume_one.id = 1
+        self.volume_one.create_time = 5
+        self.volume_one.status = "one_status"
+        self.volume_one.size = "one_size"
+        self.volume_one.snapshot_id = 1
+        self.volume_one.attach_data = self.attach_data
+        self.volume_one.zone = "one_zone"
+
+        self.volume_two = Volume()
+        self.volume_two.connection = mock.Mock()
+        self.volume_two.id = 1
+        self.volume_two.create_time = 6
+        self.volume_two.status = "two_status"
+        self.volume_two.size = "two_size"
+        self.volume_two.snapshot_id = 2
+        self.volume_two.attach_data = None
+        self.volume_two.zone = "two_zone"
+
+    @mock.patch("boto.ec2.volume.TaggedEC2Object.startElement")
+    def test_startElement_calls_TaggedEC2Object_startElement_with_correct_args(self, startElement):
+        volume = Volume()
+        volume.startElement("some name", "some attrs", None)
+        startElement.assert_called_with(volume, "some name", "some attrs", None)
+
+    @mock.patch("boto.ec2.volume.TaggedEC2Object.startElement")
+    def test_startElement_retval_not_None_returns_correct_thing(self, startElement):
+        tag_set = mock.Mock(TagSet)
+        startElement.return_value = tag_set
+        volume = Volume()
+        retval = volume.startElement(None, None, None)
+        self.assertEqual(retval, tag_set)
+
+    @mock.patch("boto.ec2.volume.TaggedEC2Object.startElement")
+    @mock.patch("boto.resultset.ResultSet")
+    def test_startElement_with_name_tagSet_calls_ResultSet(self, ResultSet, startElement):
+        startElement.return_value = None
+        result_set = mock.Mock(ResultSet([("item", Tag)]))
+        volume = Volume()
+        volume.tags = result_set
+        retval = volume.startElement("tagSet", None, None)
+        self.assertEqual(retval, volume.tags)
+
+    @mock.patch("boto.ec2.volume.TaggedEC2Object.startElement")
+    def test_startElement_with_name_attachmentSet_returns_AttachmentSet(self, startElement):
+        startElement.return_value = None
+        attach_data = AttachmentSet()
+        volume = Volume()
+        volume.attach_data = attach_data
+        retval = volume.startElement("attachmentSet", None, None)
+        self.assertEqual(retval, volume.attach_data)
+
+    @mock.patch("boto.ec2.volume.TaggedEC2Object.startElement")
+    def test_startElement_else_returns_None(self, startElement):
+        startElement.return_value = None
+        volume = Volume()
+        retval = volume.startElement("not tagSet or attachmentSet", None, None)
+        self.assertEqual(retval, None)
+
+    def check_that_attribute_has_been_set(self, name, value, attribute):
+        volume = Volume()
+        volume.endElement(name, value, None)
+        self.assertEqual(getattr(volume, attribute), value)
+
+    def test_endElement_sets_correct_attributes_with_values(self):
+        for arguments in [("volumeId", "some value", "id"),
+                          ("createTime", "some time", "create_time"),
+                          ("status", "some status", "status"),
+                          ("size", 5, "size"),
+                          ("snapshotId", 1, "snapshot_id"),
+                          ("availabilityZone", "some zone", "zone"),
+                          ("someName", "some value", "someName")]:
+            self.check_that_attribute_has_been_set(arguments[0], arguments[1], arguments[2])
+
+    def test_endElement_with_name_status_and_empty_string_value_doesnt_set_status(self):
+        volume = Volume()
+        volume.endElement("status", "", None)
+        self.assertNotEqual(volume.status, "")
+
+    def test_update_with_result_set_greater_than_0_updates_dict(self):
+        self.volume_two.connection.get_all_volumes.return_value = [self.volume_one]
+        self.volume_two.update()
+
+        assert all([self.volume_two.create_time == 5,
+                    self.volume_two.status == "one_status",
+                    self.volume_two.size == "one_size",
+                    self.volume_two.snapshot_id == 1,
+                    self.volume_two.attach_data == self.attach_data,
+                    self.volume_two.zone == "one_zone"])
+
+    def test_update_with_validate_true_raises_value_error(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.connection.get_all_volumes.return_value = []
+        with self.assertRaisesRegexp(ValueError, "^1 is not a valid Volume ID$"):
+            self.volume_one.update(True)
+
+    def test_update_returns_status(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.connection.get_all_volumes.return_value = [self.volume_two]
+        retval = self.volume_one.update()
+        self.assertEqual(retval, "two_status")
+
+    def test_delete_calls_delete_volume(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.delete()
+        self.volume_one.connection.delete_volume.assert_called_with(1)
+
+    def test_attach_calls_attach_volume(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.attach("instance_id", "/dev/null")
+        self.volume_one.connection.attach_volume.assert_called_with(1, "instance_id", "/dev/null")
+
+    def test_detach_calls_detach_volume(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.detach()
+        self.volume_one.connection.detach_volume.assert_called_with(
+                1, 2, "/dev/null", False)
+
+    def test_detach_with_no_attach_data(self):
+        self.volume_two.connection = mock.Mock()
+        self.volume_two.detach()
+        self.volume_two.connection.detach_volume.assert_called_with(
+                1, None, None, False)
+
+    def test_detach_with_force_calls_detach_volume_with_force(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.detach(True)
+        self.volume_one.connection.detach_volume.assert_called_with(
+                1, 2, "/dev/null", True)
+
+
+    def test_create_snapshot_calls_connection_create_snapshot(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.create_snapshot()
+        self.volume_one.connection.create_snapshot.assert_called_with(
+                1, None)
+
+    def test_create_snapshot_with_description(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.create_snapshot("some description")
+        self.volume_one.connection.create_snapshot.assert_called_with(
+                1, "some description")
+
+    def test_volume_state_returns_status(self):
+        retval = self.volume_one.volume_state()
+        self.assertEqual(retval, "one_status")
+
+    def test_attachment_state_returns_state(self):
+        retval = self.volume_one.attachment_state()
+        self.assertEqual(retval, "some status")
+
+    def test_attachment_state_no_attach_data_returns_None(self):
+        retval = self.volume_two.attachment_state()
+        self.assertEqual(retval, None)
+
+    def test_snapshots_returns_snapshots(self):
+        snapshot_one = Snapshot()
+        snapshot_one.volume_id = 1
+        snapshot_two = Snapshot()
+        snapshot_two.volume_id = 2
+
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.connection.get_all_snapshots.return_value = [snapshot_one, snapshot_two]
+        retval = self.volume_one.snapshots()
+        self.assertEqual(retval, [snapshot_one])
+
+    def test_snapshots__with_owner_and_restorable_by(self):
+        self.volume_one.connection = mock.Mock()
+        self.volume_one.connection.get_all_snapshots.return_value = []
+        self.volume_one.snapshots("owner", "restorable_by")
+        self.volume_one.connection.get_all_snapshots.assert_called_with(
+                owner="owner", restorable_by="restorable_by")
+
+class AttachmentSetTests(unittest.TestCase):
+    def check_that_attribute_has_been_set(self, name, value, attribute):
+        attachment_set = AttachmentSet()
+        attachment_set.endElement(name, value, None)
+        self.assertEqual(getattr(attachment_set, attribute), value)
+
+    def test_endElement_with_name_volumeId_sets_id(self):
+        return self.check_that_attribute_has_been_set("volumeId", "some value", "id")
+
+    def test_endElement_with_name_instanceId_sets_instance_id(self):
+        return self.check_that_attribute_has_been_set("instanceId", 1, "instance_id")
+
+    def test_endElement_with_name_status_sets_status(self):
+        return self.check_that_attribute_has_been_set("status", "some value", "status")
+
+    def test_endElement_with_name_attachTime_sets_attach_time(self):
+        return self.check_that_attribute_has_been_set("attachTime", 5, "attach_time")
+
+    def test_endElement_with_name_device_sets_device(self):
+        return self.check_that_attribute_has_been_set("device", "/dev/null", "device")
+
+    def test_endElement_with_other_name_sets_other_name_attribute(self):
+        return self.check_that_attribute_has_been_set("someName", "some value", "someName")
+
+class VolumeAttributeTests(unittest.TestCase):
+    def setUp(self):
+        self.volume_attribute = VolumeAttribute()
+        self.volume_attribute._key_name = "key_name"
+        self.volume_attribute.attrs = {"key_name": False}
+
+    def test_startElement_with_name_autoEnableIO_sets_key_name(self):
+        self.volume_attribute.startElement("autoEnableIO", None, None)
+        self.assertEqual(self.volume_attribute._key_name, "autoEnableIO")
+
+    def test_startElement_without_name_autoEnableIO_returns_None(self):
+        retval = self.volume_attribute.startElement("some name", None, None)
+        self.assertEqual(retval, None)
+
+    def test_endElement_with_name_value_and_value_true_sets_attrs_key_name_True(self):
+        self.volume_attribute.endElement("value", "true", None)
+        self.assertEqual(self.volume_attribute.attrs['key_name'], True)
+
+    def test_endElement_with_name_value_and_value_false_sets_attrs_key_name_False(self):
+        self.volume_attribute._key_name = "other_key_name"
+        self.volume_attribute.endElement("value", "false", None)
+        self.assertEqual(self.volume_attribute.attrs['other_key_name'], False)
+
+    def test_endElement_with_name_volumeId_sets_id(self):
+        self.volume_attribute.endElement("volumeId", "some_value", None)
+        self.assertEqual(self.volume_attribute.id, "some_value")
+
+    def test_endElement_with_other_name_sets_other_name_attribute(self):
+        self.volume_attribute.endElement("someName", "some value", None)
+        self.assertEqual(self.volume_attribute.someName, "some value")
+
+
+if __name__ == "__main__":
+    unittest.main()