def test_delete_with_invalid_id(self): self.mock_object( db, 'storage_get', mock.Mock(side_effect=exception.StorageNotFound('fake_id'))) req = fakes.HTTPRequest.blank('/storages/fake_id') self.assertRaises(exception.StorageNotFound, self.controller.delete, req, 'fake_id')
def test_create(self): self.mock_object( self.controller.driver_api, 'discover_storage', mock.Mock( return_value={ "id": "12c2d52f-01bc-41f5-b73f-7abf6f38a2a6", 'name': 'fake_driver', 'description': 'it is a fake driver.', 'vendor': 'fake_vendor', 'model': 'fake_model', 'status': 'normal', 'serial_number': '2102453JPN12KA000011', 'firmware_version': '1.0.0', 'location': 'HK', 'total_capacity': 1024 * 1024, 'used_capacity': 3126, 'free_capacity': 1045449, "sync_status": constants.SyncStatus.SYNCED, 'raw_capacity': 1610612736000, 'subscribed_capacity': 219902325555200 })) self.mock_object(db, 'access_info_get_all', fakes.fake_access_info_get_all) self.mock_object( db, 'storage_get', mock.Mock(side_effect=exception.StorageNotFound('fake_id'))) self.mock_object(self.controller, 'sync', fakes.fake_sync) body = { 'model': 'fake_driver', 'vendor': 'fake_storage', 'rest': { 'username': '******', 'password': '******', 'host': '10.0.0.76', 'port': 1234 }, 'extra_attributes': { 'array_id': '0001234567891' } } req = fakes.HTTPRequest.blank('/storages') res_dict = self.controller.create(req, body=body) expctd_dict = { "id": "12c2d52f-01bc-41f5-b73f-7abf6f38a2a6", 'name': 'fake_driver', 'description': 'it is a fake driver.', 'vendor': 'fake_vendor', 'model': 'fake_model', 'status': 'normal', 'serial_number': '2102453JPN12KA000011', 'firmware_version': '1.0.0', 'location': 'HK', 'total_capacity': 1024 * 1024, 'used_capacity': 3126, 'free_capacity': 1045449, "sync_status": "SYNCED", 'raw_capacity': 1610612736000, 'subscribed_capacity': 219902325555200 } self.assertDictEqual(expctd_dict, res_dict)
def _storage_get(context, storage_id, session=None): result = (_storage_get_query( context, session=session).filter_by(id=storage_id).first()) if not result: raise exception.StorageNotFound(storage_id) return result
def test_create_with_performance_monitoring(self): self.mock_object( self.controller.driver_api, 'discover_storage', mock.Mock( return_value={ "id": "12c2d52f-01bc-41f5-b73f-7abf6f38a2a6", 'name': 'fake_driver', 'description': 'it is a fake driver.', 'vendor': 'fake_vendor', 'model': 'fake_model', 'status': 'normal', 'serial_number': '2102453JPN12KA000011', 'firmware_version': '1.0.0', 'location': 'HK', 'total_capacity': 1024 * 1024, 'used_capacity': 3126, 'free_capacity': 1045449, "sync_status": constants.SyncStatus.SYNCED, 'raw_capacity': 1610612736000, 'subscribed_capacity': 219902325555200 })) self.mock_object(db, 'access_info_get_all', fakes.fake_access_info_get_all) self.mock_object( db, 'storage_get', mock.Mock(side_effect=exception.StorageNotFound('fake_id'))) self.mock_object(self.controller, 'sync', fakes.fake_sync) body = { 'model': 'fake_driver', 'vendor': 'fake_storage', 'rest': { 'username': '******', 'password': '******', 'host': '10.0.0.76', 'port': 1234 }, 'extra_attributes': { 'array_id': '0001234567891' } } req = fakes.HTTPRequest.blank('/storages') resource_metrics = { "storage": { "throughput": { "unit": "MB/s", "description": "Represents how much data is " "successfully transferred in MB/s" }, } } self.mock_object( self.controller.driver_api, 'get_capabilities', mock.Mock(return_value={ 'is_historic': False, 'resource_metrics': resource_metrics })) self.mock_object( self.controller.driver_api, 'get_capabilities', mock.Mock(return_value={ 'is_historic': False, 'resource_metrics': resource_metrics })) def test_task_create(context, values): self.assertEqual(values['resource_metrics'], resource_metrics) db.task_create = test_task_create res_dict = self.controller.create(req, body=body) expctd_dict = { "id": "12c2d52f-01bc-41f5-b73f-7abf6f38a2a6", 'name': 'fake_driver', 'description': 'it is a fake driver.', 'vendor': 'fake_vendor', 'model': 'fake_model', 'status': 'normal', 'serial_number': '2102453JPN12KA000011', 'firmware_version': '1.0.0', 'location': 'HK', 'total_capacity': 1024 * 1024, 'used_capacity': 3126, 'free_capacity': 1045449, "sync_status": "SYNCED", 'raw_capacity': 1610612736000, 'subscribed_capacity': 219902325555200 } self.assertDictEqual(expctd_dict, res_dict)
def fake_storage_get_exception(ctx, storage_id): raise exception.StorageNotFound('abcd-1234-5678')
def fake_storage_get_exception(ctx, storage_id): raise exception.StorageNotFound(storage_id)