def test_is_called_every_interval(self):
clock = Clock()
service = ImageDownloadService(sentinel.service, sentinel.tftp_root,
clock)
# Avoid actual downloads:
self.patch_download(service, None)
maas_meta_last_modified = self.patch(tftppath,
"maas_meta_last_modified")
maas_meta_last_modified.return_value = None
service.startService()
# The first call is issued at startup.
self.assertEqual(1, len(get_mock_calls(maas_meta_last_modified)))
# Wind clock forward one second less than the desired interval.
clock.advance(service.check_interval - 1)
# No more periodic calls made.
self.assertEqual(1, len(get_mock_calls(maas_meta_last_modified)))
# Wind clock forward one second, past the interval.
clock.advance(1)
# Now there were two calls.
self.assertEqual(2, len(get_mock_calls(maas_meta_last_modified)))
# Forward another interval, should be three calls.
clock.advance(service.check_interval)
self.assertEqual(3, len(get_mock_calls(maas_meta_last_modified)))
def test_download_is_initiated_in_new_thread(self):
clock = Clock()
maas_meta_last_modified = self.patch(
tftppath, 'maas_meta_last_modified')
one_week = timedelta(minutes=15).total_seconds()
maas_meta_last_modified.return_value = clock.seconds() - one_week
http_proxy = factory.make_simple_http_url()
https_proxy = factory.make_simple_http_url()
rpc_client = Mock()
client_call = Mock()
client_call.side_effect = [
defer.succeed(dict(sources=sentinel.sources)),
defer.succeed(dict(
http=urlparse(http_proxy),
https=urlparse(https_proxy))),
]
rpc_client.getClientNow.return_value = defer.succeed(client_call)
rpc_client.maas_url = factory.make_simple_http_url()
# We could patch out 'import_boot_images' instead here but I
# don't do that for 2 reasons:
# 1. It requires spinning the reactor again before being able to
# test the result.
# 2. It means there's no thread to clean up after the test.
deferToThread = self.patch(boot_images, 'deferToThread')
deferToThread.return_value = defer.succeed(None)
service = ImageDownloadService(
rpc_client, sentinel.tftp_root, clock)
service.startService()
self.assertThat(
deferToThread, MockCalledOnceWith(
_run_import, sentinel.sources, rpc_client.maas_url,
http_proxy=http_proxy, https_proxy=https_proxy))
def test_initiates_download_if_no_meta_file(self):
clock = Clock()
service = ImageDownloadService(sentinel.service, sentinel.tftp_root,
clock)
_start_download = self.patch_download(service, None)
self.patch(tftppath, "maas_meta_last_modified").return_value = None
service.startService()
self.assertThat(_start_download, MockCalledOnceWith())
def test_no_download_if_no_rpc_connections(self):
rpc_client = Mock()
failure = NoConnectionsAvailable()
rpc_client.getClientNow.return_value = defer.fail(failure)
deferToThread = self.patch(boot_images, "deferToThread")
service = ImageDownloadService(rpc_client, self.make_dir(), Clock())
service.startService()
self.assertThat(deferToThread, MockNotCalled())
def test_initiates_download_if_15_minutes_has_passed(self):
clock = Clock()
service = ImageDownloadService(sentinel.service, sentinel.tftp_root,
clock)
_start_download = self.patch_download(service, None)
one_week_ago = clock.seconds() - timedelta(minutes=15).total_seconds()
self.patch(tftppath,
"maas_meta_last_modified").return_value = one_week_ago
service.startService()
self.assertThat(_start_download, MockCalledOnceWith())
def test_no_download_if_15_minutes_has_not_passed(self):
clock = Clock()
service = ImageDownloadService(sentinel.service, sentinel.tftp_root,
clock)
_start_download = self.patch_download(service, None)
one_week = timedelta(minutes=15).total_seconds()
self.patch(tftppath,
"maas_meta_last_modified").return_value = clock.seconds()
clock.advance(one_week - 1)
service.startService()
self.assertThat(_start_download, MockNotCalled())