diff --git a/glance/tests/unit/async_/flows/test_api_image_import.py b/glance/tests/unit/async_/flows/test_api_image_import.py index b0fd098e73..5d88640333 100644 --- a/glance/tests/unit/async_/flows/test_api_image_import.py +++ b/glance/tests/unit/async_/flows/test_api_image_import.py @@ -569,6 +569,7 @@ class TestImportCopyImageTask(test_utils.BaseTestCase): admin_repo=admin_repo, image_id=IMAGE_ID1, import_req=fake_req, + context=self.context, backend=['cheap']) # Assert that we saved the image with the admin repo instead of the diff --git a/glance/tests/unit/async_/test_async.py b/glance/tests/unit/async_/test_async.py index 31e27173a4..1f97f94b26 100644 --- a/glance/tests/unit/async_/test_async.py +++ b/glance/tests/unit/async_/test_async.py @@ -93,7 +93,8 @@ class TestImportTaskFlow(test_utils.BaseTestCase): 'task_repo': mock.MagicMock(), 'image_repo': mock.MagicMock(), 'image_id': mock.MagicMock(), - 'import_req': import_req or mock.MagicMock() + 'import_req': import_req or mock.MagicMock(), + 'context': mock.MagicMock(), } inputs['image_repo'].get.return_value = mock.MagicMock( extra_properties={'os_glance_import_task': mock.sentinel.task_id})