def test_get_capabilities(self, driver_manager): driver_manager.return_value = FakeStorageDriver() storage_id = '12345' capabilities = API().get_capabilities(context, storage_id) self.assertTrue('resource_metrics' in capabilities) driver_manager.assert_called_once()
def test_list_masking_views(self, driver_manager, mock_fake): driver_manager.return_value = FakeStorageDriver() mock_fake.return_value = [] api = API() storage_id = '12345' api.list_masking_views(context, storage_id) driver_manager.assert_called_once() mock_fake.assert_called_once()
def test_collect_perf_metrics(self, driver_manager): driver_manager.return_value = FakeStorageDriver() storage_id = '12345' capabilities = API().get_capabilities(context, storage_id) metrics = API().collect_perf_metrics(context, storage_id, capabilities['resource_metrics'], 1622808000000, 1622808000001) self.assertTrue('resource_metrics' in capabilities) self.assertTrue(True, isinstance(metrics[0], constants.metric_struct)) self.assertEqual(driver_manager.call_count, 2)
def test_get_storage(self, driver_manager, mock_fake): driver_manager.return_value = FakeStorageDriver() storage = copy.deepcopy(STORAGE) storage['id'] = '12345' mock_fake.return_value = storage api = API() storage_id = '12345' api.get_storage(context, storage_id) driver_manager.assert_called_once() mock_fake.assert_called()
def test_parse_alert(self, mock_access_info, driver_manager, mock_fake): mock_access_info.return_value = ACCESS_INFO driver_manager.return_value = FakeStorageDriver() mock_fake.return_value = [] api = API() storage_id = '12345' api.parse_alert(context, storage_id, 'alert') mock_access_info.assert_called_once() driver_manager.assert_called_once() mock_fake.assert_called_once()
def test_remove_storage(self, mock_storage, mock_access_info, mock_storage_create, mock_get_storage, mock_dm): storage = copy.deepcopy(STORAGE) storage['id'] = '12345' mock_storage.return_value = None mock_access_info.return_value = ACCESS_INFO mock_storage_create.return_value = storage api = API() api.discover_storage(context, ACCESS_INFO) mock_get_storage.return_value = None mock_dm.return_value = FakeStorageDriver() storage_id = '12345' # Verify that driver instance not added to factory driver = api.driver_manager.driver_factory.get(storage_id, None) self.assertIsNone(driver) api.remove_storage(context, storage_id) driver = api.driver_manager.driver_factory.get(storage_id, None) self.assertIsNone(driver)