def test__prepare_image(self):
"""
test prepare image
"""
self.img._prepare_image(self.target_name)
self.assertEqual(
((["si_prepareclient",
"--server " + imager._remove_domain(self.image_server_name),
"--yes"],), {}),
self.mock_executor_class.call_args_list[0])
def test__push_overrides(self):
"""
push overrides on target
"""
self.img._push_overrides(self.target_name)
self.assertEqual(
((["si_pushoverrides", imager._remove_domain(self.target_name)],), {}),
self.mock_executor_class.call_args_list[0])
self.assertEqual(
("ssh_run", (self.image_server_name, "root"), {"check": True}),
self.mock_executor.method_calls[0])
def test__image_incremental_update(self):
"""
incremental image target
"""
self.img._image_incremental_update(self.target_name)
self.assertEqual(
((["si_updateclient", "--server " + imager._remove_domain(self.image_server_name), "--yes"],), {}),
self.mock_executor_class.call_args_list[0])
self.assertEqual(
("ssh_run", (self.target_name, "root"), {"check": True}),
self.mock_executor.method_calls[0])
def test__get_image(self):
"""
test get image
"""
image_name = "image_name"
ip_assignment = "REPLICANT"
self.img._get_image(self.target_name, image_name, ip_assignment)
self.assertEqual(
((["si_getimage",
"--golden-client " + imager._remove_domain(self.target_name),
"--image " + image_name,
"--quiet",
"--update-script YES",
"--ip-assignment REPLICANT",
"--post-install reboot"],), {}),
self.mock_executor_class.call_args_list[0])
self.assertEqual(
("ssh_run", (self.image_server_name, "root"), {"check": True}),
self.mock_executor.method_calls[0])
def test_remove_domain_no_dots(self):
self.assertEqual("omega", imager._remove_domain("omega"))
def test_remove_domain_empty(self):
self.assertEqual("", imager._remove_domain(""))
def test_remove_domain(self):
self.assertEqual("omega", imager._remove_domain("omega.ucd.ie"))
