diff --git a/oslo_vmware/image_transfer.py b/oslo_vmware/image_transfer.py index 6ff2acbe..264fb19d 100644 --- a/oslo_vmware/image_transfer.py +++ b/oslo_vmware/image_transfer.py @@ -338,10 +338,11 @@ def upload_image(context, timeout_secs, image_service, image_id, owner_id, 'owner_id': owner_id}} updater = loopingcall.FixedIntervalLoopingCall(read_handle.update_progress) + store_id = kwargs.get('store_id') try: updater.start(interval=NFC_LEASE_UPDATE_PERIOD) image_service.update(context, image_id, image_metadata, - data=read_handle) + data=read_handle, store_id=store_id) finally: updater.stop() read_handle.close() diff --git a/oslo_vmware/tests/test_image_transfer.py b/oslo_vmware/tests/test_image_transfer.py index 08ac0e74..0bb61b72 100644 --- a/oslo_vmware/tests/test_image_transfer.py +++ b/oslo_vmware/tests/test_image_transfer.py @@ -372,6 +372,7 @@ class ImageTransferUtilityTest(base.TestCase): image_name = 'fake_image' image_version = 1 + store_id = 'fake_store' fake_VmdkReadHandle = mock.Mock() fake_rw_handles_VmdkReadHandle.return_value = fake_VmdkReadHandle @@ -389,7 +390,8 @@ class ImageTransferUtilityTest(base.TestCase): vmdk_size=image_size, is_public=is_public, image_name=image_name, - image_version=image_version) + image_version=image_version, + store_id=store_id) fake_rw_handles_VmdkReadHandle.assert_called_once_with(session, host, @@ -408,4 +410,5 @@ class ImageTransferUtilityTest(base.TestCase): image_service.update.assert_called_once_with(context, image_id, image_metadata, - data=fake_VmdkReadHandle) + data=fake_VmdkReadHandle, + store_id=store_id)