def test_update_pci_device(self, mock_write, mock_read):
mock_read.side_effect = etcd.EtcdKeyNotFound
values = fake_values
pcidevice = models.PciDevice(values)
updated = self.dbapi.update_pci_device(values['compute_node_uuid'],
values['address'], values)
self.assertEqual(pcidevice.as_dict(), updated.as_dict())
def translate_etcd_result(etcd_result, model_type):
"""Translate etcd unicode result to etcd models."""
try:
data = json.loads(etcd_result.value)
ret = None
if model_type == 'container':
ret = models.Container(data)
elif model_type == 'zun_service':
ret = models.ZunService(data)
elif model_type == 'image':
ret = models.Image(data)
elif model_type == 'resource_class':
ret = models.ResourceClass(data)
elif model_type == 'compute_node':
ret = models.ComputeNode(data)
elif model_type == 'capsule':
ret = models.Capsule(data)
elif model_type == 'pcidevice':
ret = models.PciDevice(data)
elif model_type == 'volume_mapping':
ret = models.VolumeMapping(data)
elif model_type == 'container_action':
ret = models.ContainerAction(data)
elif model_type == 'container_action_event':
ret = models.ContainerActionEvent(data)
else:
raise exception.InvalidParameterValue(
_('The model_type value: %s is invalid.'), model_type)
return ret
except (ValueError, TypeError) as e:
LOG.error("Error occurred while translating etcd result: %s",
six.text_type(e))
raise
def test_get_pci_device_by_addr(self, mock_list):
filters = {
'compute_node_uuid': 'ef5ef3492b57c8b9bc0556a',
'address': 'fake_addr'
}
mock_list.return_value = [models.PciDevice(fake_values)]
self.dbapi.get_pci_device_by_addr('ef5ef3492b57c8b9bc0556a',
'fake_addr')
mock_list.assert_called_with(filters=filters)
def _create_pci_device(self, pci_device_data):
if not pci_device_data.get('uuid'):
pci_device_data['uuid'] = uuidutils.generate_uuid()
pci_device = models.PciDevice(pci_device_data)
try:
pci_device.save()
except Exception:
raise
return pci_device
def test_get_pci_device_by_id(self, mock_list):
filters = {'id': '1'}
mock_list.return_value = [models.PciDevice(fake_values)]
self.dbapi.get_pci_device_by_id('1')
mock_list.assert_called_with(filters=filters)
def test_get_all_pci_device_by_container_uuid(self, mock_list):
filters = {'container_uuid': 'Id64c317ff78e95af2fc'}
mock_list.return_value = [models.PciDevice(fake_values)]
self.dbapi.get_all_pci_device_by_container_uuid('Id64c317ff78e95af2fc')
mock_list.assert_called_with(filters=filters)
