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)
