def make_boot_resources_create_action(self):
self.stdio = self.useFixture(CaptureStandardIO())
action_bases = (BootResourcesCreateAction,)
action_ns = {
"action": {"method": "POST"},
"handler": {"uri": b"/MAAS/api/2.0/boot-resources/", "params": []},
"profile": {"credentials": make_api_credentials()},
}
action_class = type("create", action_bases, action_ns)
action = action_class(Mock())
return action
def make_sshkeys_import_action(self):
self.stdio = self.useFixture(CaptureStandardIO())
action_bases = (SSHKeysImportAction,)
action_ns = {
"action": {"method": "POST"},
"handler": {"uri": b"/MAAS/api/2.0/sshkeys/", "params": []},
"profile": {"credentials": make_api_credentials()},
}
action_class = type("import", action_bases, action_ns)
action = action_class(Mock())
return action
def test_sends_boot_images_to_server(self):
self.useFixture(ConfigFixture({'tftp': {'root': self.make_dir()}}))
self.set_maas_url()
auth.record_api_credentials(':'.join(make_api_credentials()))
image = make_boot_image_params()
self.patch(tftppath, 'list_boot_images', Mock(return_value=[image]))
self.patch(boot_images, "get_cluster_uuid")
self.patch(MAASClient, 'post')
report_boot_images.delay()
args, kwargs = MAASClient.post.call_args
self.assertItemsEqual([image], json.loads(kwargs['images']))
def test_sends_boot_images_to_server(self):
self.useFixture(ConfigFixture({'tftp': {'root': self.make_dir()}}))
self.set_maas_url()
auth.record_api_credentials(':'.join(make_api_credentials()))
image = make_boot_image_params()
self.patch(tftppath, 'list_boot_images', Mock(return_value=[image]))
self.patch(boot_images, "get_cluster_uuid")
self.patch(MAASClient, 'post')
report_boot_images.delay()
args, kwargs = MAASClient.post.call_args
self.assertItemsEqual([image], json.loads(kwargs['images']))
def setUp(self):
# Initialise the knowledge cache.
super(TestPeriodicTask, self).setUp()
uuid = factory.getRandomUUID()
maas_url = 'http://%s.example.com/%s/' % (
factory.make_name('host'),
factory.getRandomString(),
)
api_credentials = make_api_credentials()
cache.cache.set(NODEGROUP_UUID_CACHE_KEY, uuid)
os.environ["MAAS_URL"] = maas_url
cache.cache.set('api_credentials', ':'.join(api_credentials))
self.knowledge = dict(nodegroup_uuid=uuid,
api_credentials=api_credentials,
maas_url=maas_url)
def make_boot_resources_create_action(self):
self.stdio = self.useFixture(CaptureStandardIO())
action_bases = (BootResourcesCreateAction, )
action_ns = {
"action": {
'method': 'POST'
},
"handler": {
'uri': b'/MAAS/api/2.0/boot-resources/',
'params': []
},
"profile": {
'credentials': make_api_credentials()
}
}
action_class = type("create", action_bases, action_ns)
action = action_class(Mock())
return action
def test_updates_api_credentials(self):
credentials = make_api_credentials()
refresh_secrets(
api_credentials=convert_tuple_to_string(credentials))
self.assertEqual(credentials, auth.get_recorded_api_credentials())
def set_api_credentials(self):
record_api_credentials(':'.join(make_api_credentials()))
def set_api_credentials(self):
record_api_credentials(':'.join(make_api_credentials()))
def test_updates_api_credentials(self):
credentials = make_api_credentials()
refresh_secrets(api_credentials=convert_tuple_to_string(credentials))
self.assertEqual(credentials, auth.get_recorded_api_credentials())
def test_get_recorded_api_credentials_returns_credentials_as_tuple(self):
creds = make_api_credentials()
auth.record_api_credentials(convert_tuple_to_string(creds))
self.assertEqual(creds, auth.get_recorded_api_credentials())
def test_record_api_credentials_records_credentials_string(self):
creds_string = convert_tuple_to_string(make_api_credentials())
auth.record_api_credentials(creds_string)
self.assertEqual(creds_string,
cache.cache.get(auth.API_CREDENTIALS_CACHE_KEY))
def test_get_recorded_api_credentials_returns_credentials_as_tuple(self):
creds = make_api_credentials()
auth.record_api_credentials(convert_tuple_to_string(creds))
self.assertEqual(creds, auth.get_recorded_api_credentials())
def test_record_api_credentials_records_credentials_string(self):
creds_string = convert_tuple_to_string(make_api_credentials())
auth.record_api_credentials(creds_string)
self.assertEqual(
creds_string, cache.cache.get(auth.API_CREDENTIALS_CACHE_KEY))