def make_boot_image(self, params, purpose, subarch=None, subarches=None):
image = make_image(params, purpose)
if subarch is not None:
image["subarchitecture"] = subarch
if subarches is not None:
image["supported_subarches"] = subarches
return image
def test_list_boot_images_finds_boot_image(self):
params = make_boot_image_storage_params()
self.make_image_dir(params, self.tftproot)
purposes = ['install', 'commissioning', 'xinstall']
make_osystem(self, params['osystem'], purposes)
self.assertItemsEqual(
[make_image(params, purpose) for purpose in purposes],
list_boot_images(self.tftproot))
def test_list_boot_images_finds_boot_image(self):
params = make_boot_image_storage_params()
self.make_image_dir(params, self.tftproot)
purposes = ["install", "commissioning", "xinstall"]
make_osystem(self, params["osystem"], purposes)
self.assertItemsEqual(
[make_image(params, purpose) for purpose in purposes],
list_boot_images(self.tftproot),
)
def test_list_boot_images_enumerates_boot_images(self):
purposes = ['install', 'commissioning', 'xinstall']
params = [make_boot_image_storage_params() for counter in range(3)]
for param in params:
self.make_image_dir(param, self.tftproot)
make_osystem(self, param['osystem'], purposes)
self.assertItemsEqual([
make_image(param, purpose) for param in params
for purpose in purposes
], list_boot_images(self.tftproot))
def test_returns_boot_images_for_one_cluster(self):
factory.make_RackController()
self.useFixture(RunningClusterRPCFixture())
purposes = ['install', 'commissioning', 'xinstall']
params = [make_boot_image_storage_params() for _ in range(3)]
for param in params:
make_image_dir(param, self.tftp_root)
test_tftppath.make_osystem(self, param['osystem'], purposes)
self.assertItemsEqual([
make_image(param, purpose) for param in params
for purpose in purposes
], self.get())
def test_list_boot_images_merges_maas_meta_data(self):
params = make_boot_image_storage_params()
self.make_image_dir(params, self.tftproot)
# The required metadata is called "subarches" in maas.meta
metadata = dict(subarches=factory.make_name("subarches"))
self.make_meta_file(params, metadata, self.tftproot)
purposes = ['install', 'commissioning', 'xinstall']
make_osystem(self, params['osystem'], purposes)
# The API requires "supported_subarches".
expected_metadata = dict(supported_subarches=metadata["subarches"])
self.assertItemsEqual([
make_image(params, purpose, expected_metadata)
for purpose in purposes
], list_boot_images(self.tftproot))
def test_returns_boot_images(self):
rack_controller = factory.make_RackController()
self.useFixture(RunningClusterRPCFixture())
purposes = ["install", "commissioning", "xinstall"]
params = [make_boot_image_storage_params() for _ in range(3)]
for param in params:
make_image_dir(param, self.tftp_root)
test_tftppath.make_osystem(self, param["osystem"], purposes)
self.assertItemsEqual(
[
make_image(param, purpose) for param in params
for purpose in purposes
],
get_boot_images(rack_controller),
)
def make_rpc_boot_images(self, param):
purposes = ["install", "commissioning", "xinstall"]
return [make_image(param, purpose) for purpose in purposes]
def make_rpc_boot_images(self, param):
purposes = ['install', 'commissioning', 'xinstall']
return [make_image(param, purpose) for purpose in purposes]
