def _pick_attach_data(self, attach_type):
"""
Helper function to create desired volume attach data.
:param NamedConstant attach_type: Type of attach data to create.
:returns: Volume attachment set that conforms to requested attach type.
:rtype: AttachmentSet
"""
if attach_type == self.A.MISSING_ATTACH_DATA:
return None
elif attach_type == self.A.MISSING_INSTANCE_ID:
attach_data = AttachmentSet()
attach_data.device = u'/dev/sdf'
attach_data.instance_id = ''
return attach_data
elif attach_type == self.A.MISSING_DEVICE:
attach_data = AttachmentSet()
attach_data.device = ''
attach_data.instance_id = u'i-xyz'
return attach_data
elif attach_type == self.A.ATTACH_SUCCESS:
attach_data = AttachmentSet()
attach_data.device = u'/dev/sdf'
attach_data.instance_id = u'i-xyz'
return attach_data
elif attach_type == self.A.DETACH_SUCCESS:
return None
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"
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)
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 _pick_attach_data(self, attach_type):
"""
Helper function to create desired volume attach data.
:param NamedConstant attach_type: Type of attach data to create.
:returns: Volume attachment set that conforms to requested attach type.
:rtype: AttachmentSet
"""
if attach_type == self.A.MISSING_ATTACH_DATA:
return None
elif attach_type == self.A.MISSING_INSTANCE_ID:
attach_data = AttachmentSet()
attach_data.device = u'/dev/sdf'
attach_data.instance_id = ''
return attach_data
elif attach_type == self.A.MISSING_DEVICE:
attach_data = AttachmentSet()
attach_data.device = ''
attach_data.instance_id = u'i-xyz'
return attach_data
elif attach_type == self.A.ATTACH_SUCCESS:
attach_data = AttachmentSet()
attach_data.device = u'/dev/sdf'
attach_data.instance_id = u'i-xyz'
return attach_data
elif attach_type == self.A.DETACH_SUCCESS:
return None
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)
