def process_kickstart(self, data):
"""Process the kickstart data."""
configuration = LiveImageConfigurationData()
configuration.url = data.liveimg.url or ""
configuration.proxy = data.liveimg.proxy or ""
configuration.checksum = data.liveimg.checksum or ""
configuration.ssl_verification_enabled = not data.liveimg.noverifyssl
self.set_configuration(configuration)
def test_invalid_image(self):
"""Test an invalid image."""
configuration = LiveImageConfigurationData()
configuration.url = "file:///my/invalid/path"
task = SetUpLocalImageSourceTask(configuration)
with pytest.raises(SourceSetupError) as cm:
task.run()
assert str(cm.value) == "File /my/invalid/path does not exist."
def invalid_image_test(self):
"""Test an invalid image."""
configuration = LiveImageConfigurationData()
configuration.url = "file:///my/invalid/path"
task = SetUpLocalImageSourceTask(configuration)
with self.assertRaises(SourceSetupError) as cm:
task.run()
self.assertEqual(str(cm.exception),
"File /my/invalid/path does not exist.")
def empty_image_test(self):
"""Test an empty image."""
with tempfile.NamedTemporaryFile("w") as f:
# Run the task.
configuration = LiveImageConfigurationData()
configuration.url = "file://" + f.name
task = SetUpLocalImageSourceTask(configuration)
result = task.run()
# Check the result.
self.assertIsInstance(result, SetupImageResult)
self.assertEqual(result, SetupImageResult(None))
def test_empty_image(self, os_stat):
"""Test an empty image."""
os_stat.return_value = Mock(st_blocks=0)
with tempfile.NamedTemporaryFile("w") as f:
# Run the task.
configuration = LiveImageConfigurationData()
configuration.url = "file://" + f.name
task = SetUpLocalImageSourceTask(configuration)
result = task.run()
# Check the result.
assert isinstance(result, SetupImageResult)
assert result == SetupImageResult(None)
def test_failed_request(self, session_getter):
"""Test a request that fails to be send."""
# Prepare the session.
session = session_getter.return_value.__enter__.return_value
session.head.side_effect = RequestException("Fake!")
# Run the task.
configuration = LiveImageConfigurationData()
configuration.url = "http://my/fake/path"
task = SetUpRemoteImageSourceTask(configuration)
with pytest.raises(SourceSetupError) as cm:
task.run()
# Check the exception.
assert str(cm.value) == "Error while handling a request: Fake!"
def invalid_response_test(self, session_getter):
"""Test an invalid response."""
# Prepare the session.
session = session_getter.return_value.__enter__.return_value
response = session.head.return_value
response.status_code = 303
# Run the task.
configuration = LiveImageConfigurationData()
configuration.url = "http://my/fake/path"
task = SetUpRemoteImageSourceTask(configuration)
with self.assertRaises(SourceSetupError) as cm:
task.run()
# Check the exception.
self.assertEqual(str(cm.exception), "The request has failed: 303")
def fake_image_test(self):
"""Test a fake image."""
with tempfile.NamedTemporaryFile("w") as f:
# Create a fake image.
f.write("MY FAKE IMAGE")
f.flush()
# Run the task.
configuration = LiveImageConfigurationData()
configuration.url = "file://" + f.name
task = SetUpLocalImageSourceTask(configuration)
result = task.run()
# Check the result.
self.assertIsInstance(result, SetupImageResult)
self.assertGreater(result.required_space, 0)
def fake_request_test(self, session_getter):
"""Test a fake request."""
# Prepare the session.
session = session_getter.return_value.__enter__.return_value
response = session.head.return_value
response.status_code = 200
response.headers = {'content-length': 1000}
# Run the task.
configuration = LiveImageConfigurationData()
configuration.url = "http://my/fake/path"
task = SetUpRemoteImageSourceTask(configuration)
result = task.run()
# Check the result.
self.assertIsInstance(result, SetupImageResult)
self.assertEqual(result, SetupImageResult(4000))
def missing_size_test(self, session_getter):
"""Test a request with a missing size."""
# Prepare the session.
session = session_getter.return_value.__enter__.return_value
response = session.head.return_value
response.status_code = 200
response.headers = {}
# Run the task.
configuration = LiveImageConfigurationData()
configuration.url = "http://my/fake/path"
task = SetUpRemoteImageSourceTask(configuration)
result = task.run()
# Check the result.
self.assertIsInstance(result, SetupImageResult)
self.assertEqual(result, SetupImageResult(None))
def fake_image_test(self, os_stat):
"""Test a fake image."""
os_stat.return_value = Mock(st_blocks=2)
with tempfile.NamedTemporaryFile("w") as f:
# Create a fake image.
f.write("MY FAKE IMAGE")
f.flush()
# Run the task.
configuration = LiveImageConfigurationData()
configuration.url = "file://" + f.name
task = SetUpLocalImageSourceTask(configuration)
result = task.run()
# Check the result.
self.assertIsInstance(result, SetupImageResult)
self.assertEqual(result, SetupImageResult(3072))
