def test_update_last_image_sync_not_performed(self):
get_maas_id = self.patch(boot_images, "get_maas_id")
get_maas_id.return_value = factory.make_string()
getRegionClient = self.patch(boot_images, "getRegionClient")
_run_import = self.patch_autospec(boot_images, '_run_import')
_run_import.return_value = False
yield boot_images._import_boot_images(sentinel.sources)
self.assertThat(_run_import,
MockCalledOnceWith(sentinel.sources, None, None))
self.assertThat(getRegionClient, MockNotCalled())
self.assertThat(get_maas_id, MockNotCalled())
def test_update_last_image_sync_always_updated(self):
get_maas_id = self.patch(boot_images, "get_maas_id")
get_maas_id.return_value = factory.make_string()
getRegionClient = self.patch(boot_images, "getRegionClient")
_run_import = self.patch_autospec(boot_images, '_run_import')
_run_import.return_value = False
yield boot_images._import_boot_images(sentinel.sources)
self.assertThat(_run_import,
MockCalledOnceWith(sentinel.sources, None, None))
self.assertThat(getRegionClient, MockCalledOnceWith())
self.assertThat(get_maas_id, MockCalledOnceWith())
client = getRegionClient.return_value
self.assertThat(
client,
MockCalledOnceWith(UpdateLastImageSync, system_id=get_maas_id()))
def test_update_last_image_sync(self):
get_maas_id = self.patch(boot_images, "get_maas_id")
get_maas_id.return_value = factory.make_string()
getRegionClient = self.patch(boot_images, "getRegionClient")
_run_import = self.patch_autospec(boot_images, "_run_import")
_run_import.return_value = True
maas_url = factory.make_simple_http_url()
yield boot_images._import_boot_images(sentinel.sources, maas_url)
self.assertThat(
_run_import,
MockCalledOnceWith(sentinel.sources, maas_url, None, None),
)
self.assertThat(getRegionClient, MockCalledOnceWith())
self.assertThat(get_maas_id, MockCalledOnceWith())
client = getRegionClient.return_value
self.assertThat(
client,
MockCalledOnceWith(UpdateLastImageSync, system_id=get_maas_id()),
)