| OLD | NEW |
| 1 import mock | 1 import mock |
| 2 from tests.unit import unittest | 2 from tests.unit import unittest |
| 3 | 3 |
| 4 from boto.ec2.snapshot import Snapshot | 4 from boto.ec2.snapshot import Snapshot |
| 5 from boto.ec2.tag import Tag, TagSet | 5 from boto.ec2.tag import Tag, TagSet |
| 6 from boto.ec2.volume import Volume, AttachmentSet, VolumeAttribute | 6 from boto.ec2.volume import Volume, AttachmentSet, VolumeAttribute |
| 7 | 7 |
| 8 | 8 |
| 9 class VolumeTests(unittest.TestCase): | 9 class VolumeTests(unittest.TestCase): |
| 10 def setUp(self): | 10 def setUp(self): |
| (...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 71 retval = volume.startElement("attachmentSet", None, None) | 71 retval = volume.startElement("attachmentSet", None, None) |
| 72 self.assertEqual(retval, volume.attach_data) | 72 self.assertEqual(retval, volume.attach_data) |
| 73 | 73 |
| 74 @mock.patch("boto.ec2.volume.TaggedEC2Object.startElement") | 74 @mock.patch("boto.ec2.volume.TaggedEC2Object.startElement") |
| 75 def test_startElement_else_returns_None(self, startElement): | 75 def test_startElement_else_returns_None(self, startElement): |
| 76 startElement.return_value = None | 76 startElement.return_value = None |
| 77 volume = Volume() | 77 volume = Volume() |
| 78 retval = volume.startElement("not tagSet or attachmentSet", None, None) | 78 retval = volume.startElement("not tagSet or attachmentSet", None, None) |
| 79 self.assertEqual(retval, None) | 79 self.assertEqual(retval, None) |
| 80 | 80 |
| 81 def check_that_attribute_has_been_set(self, name, value, attribute): | 81 def check_that_attribute_has_been_set(self, name, value, attribute, obj_valu
e=None): |
| 82 volume = Volume() | 82 volume = Volume() |
| 83 volume.endElement(name, value, None) | 83 volume.endElement(name, value, None) |
| 84 self.assertEqual(getattr(volume, attribute), value) | 84 expected_value = obj_value if obj_value is not None else value |
| 85 self.assertEqual(getattr(volume, attribute), expected_value) |
| 85 | 86 |
| 86 def test_endElement_sets_correct_attributes_with_values(self): | 87 def test_endElement_sets_correct_attributes_with_values(self): |
| 87 for arguments in [("volumeId", "some value", "id"), | 88 for arguments in [("volumeId", "some value", "id"), |
| 88 ("createTime", "some time", "create_time"), | 89 ("createTime", "some time", "create_time"), |
| 89 ("status", "some status", "status"), | 90 ("status", "some status", "status"), |
| 90 ("size", 5, "size"), | 91 ("size", 5, "size"), |
| 91 ("snapshotId", 1, "snapshot_id"), | 92 ("snapshotId", 1, "snapshot_id"), |
| 92 ("availabilityZone", "some zone", "zone"), | 93 ("availabilityZone", "some zone", "zone"), |
| 93 ("someName", "some value", "someName")]: | 94 ("someName", "some value", "someName"), |
| 94 self.check_that_attribute_has_been_set(arguments[0], arguments[1], a
rguments[2]) | 95 ("encrypted", "true", "encrypted", True)]: |
| 96 self.check_that_attribute_has_been_set(*arguments) |
| 95 | 97 |
| 96 def test_endElement_with_name_status_and_empty_string_value_doesnt_set_statu
s(self): | 98 def test_endElement_with_name_status_and_empty_string_value_doesnt_set_statu
s(self): |
| 97 volume = Volume() | 99 volume = Volume() |
| 98 volume.endElement("status", "", None) | 100 volume.endElement("status", "", None) |
| 99 self.assertNotEqual(volume.status, "") | 101 self.assertNotEqual(volume.status, "") |
| 100 | 102 |
| 101 def test_update_with_result_set_greater_than_0_updates_dict(self): | 103 def test_update_with_result_set_greater_than_0_updates_dict(self): |
| 102 self.volume_two.connection.get_all_volumes.return_value = [self.volume_o
ne] | 104 self.volume_two.connection.get_all_volumes.return_value = [self.volume_o
ne] |
| 103 self.volume_two.update() | 105 self.volume_two.update() |
| 104 | 106 |
| (...skipping 152 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 257 self.volume_attribute.endElement("volumeId", "some_value", None) | 259 self.volume_attribute.endElement("volumeId", "some_value", None) |
| 258 self.assertEqual(self.volume_attribute.id, "some_value") | 260 self.assertEqual(self.volume_attribute.id, "some_value") |
| 259 | 261 |
| 260 def test_endElement_with_other_name_sets_other_name_attribute(self): | 262 def test_endElement_with_other_name_sets_other_name_attribute(self): |
| 261 self.volume_attribute.endElement("someName", "some value", None) | 263 self.volume_attribute.endElement("someName", "some value", None) |
| 262 self.assertEqual(self.volume_attribute.someName, "some value") | 264 self.assertEqual(self.volume_attribute.someName, "some value") |
| 263 | 265 |
| 264 | 266 |
| 265 if __name__ == "__main__": | 267 if __name__ == "__main__": |
| 266 unittest.main() | 268 unittest.main() |
| OLD | NEW |