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)