class TestRackspaceCtrl(unittest.TestCase):
def setUp(self):
""" Set up the objects used by most of the tests. """
self.ctrl = RackspaceCtrl("valid_user", "valid_api_key", "http://foo.bar:8888")
self.ctrl.post_fn = stub_rackspace_identity_post
self.driver_cls = MockLibCloudDriver
def test_authenticate_valid_user(self):
""" Test authentication with a valid user and api key. """
auth_result = self.ctrl.authenticate()
self.assertEqual(auth_result, True)
self.assertIsNotNone(self.ctrl.token)
def test_authenticate_empty_user(self):
""" Ensure authentication with empty string as username fails. """
ctrl = RackspaceCtrl("", "valid_api_key", "http://foo.bar:8888")
ctrl.post_fn = stub_rackspace_identity_post
auth_result = ctrl.authenticate()
self.assertEqual(auth_result, False)
self.assertIsNone(ctrl.token)
def test_authenticate_empty_apikey(self):
""" Ensure authentication with empty string as api_key fails. """
ctrl = RackspaceCtrl("valid_user", "", "http://foo.bar:8888")
ctrl.post_fn = stub_rackspace_identity_post
auth_result = ctrl.authenticate()
self.assertEqual(auth_result, False)
self.assertIsNone(ctrl.token)
def test_authenticate_invalid_user(self):
""" Ensure authentication with invalid user credentials fails. """
ctrl = RackspaceCtrl("invalid_user", "invalid_api_key", "http://foo.bar:8888")
ctrl.post_fn = stub_rackspace_identity_post
auth_result = ctrl.authenticate()
self.assertEqual(auth_result, False)
self.assertIsNone(ctrl.token)
def test_list_regions(self):
""" Ensure that list_regions returns the correct result. """
self.ctrl.authenticate()
regions = self.ctrl.list_regions()
expected_regions = [{"IAD": "iad"}, {"DFW": "dfw"}, {"SYD": "syd"}, {"ORD": "ord"}]
self.assertCountEqual(regions, expected_regions)
def test_set_region(self):
""" Ensure that set_region sets 'region' and 'driver'. """
self.ctrl.authenticate()
result = self.ctrl.set_region("iad")
self.assertEqual(result, True)
self.assertEqual(self.ctrl.region, "iad")
self.assertIsNotNone(self.ctrl.driver)
def test_set_invalid_region(self):
""" Ensure that calling 'set_region' with an invalid param fails. """
self.ctrl.authenticate()
result = self.ctrl.set_region("invalid")
self.assertEqual(result, False)
self.assertIsNone(self.ctrl.region)
self.assertIsNone(self.ctrl.driver)
def test_token_parsed(self):
""" Ensure that the token is set. """
ctrl = RackspaceCtrl("valid_user", "valid_api_key", "http://foo.bar:8888")
ctrl.post_fn = stub_rackspace_identity_post
ctrl.authenticate()
self.assertEqual("abcdefgh0123456789", ctrl.token)
def test_upload_file(self):
self.ctrl.storage_driver = mock.MagicMock()
mock_container = mock.MagicMock()
mock_container.list_objects = mock.MagicMock(return_value=[])
self.ctrl.storage_driver.create_container = mock.MagicMock(return_value=mock_container)
test_data = b"abcdef"
test_data_hash = hashlib.md5(test_data).hexdigest()
test_file = tempfile.NamedTemporaryFile()
with test_file.file as f:
f.write(test_data)
return_value = self.ctrl.upload_file(test_file.name, "test_folder/test.txt")
upload_file_args = self.ctrl.storage_driver.upload_object_via_stream.call_args_list[0]
upload_hash_args = self.ctrl.storage_driver.upload_object_via_stream.call_args_list[1]
self.assertEqual(upload_file_args[0][2], "test_folder/test.txt")
self.assertEqual(upload_hash_args[0][2], "test_folder/test.txt.md5")
self.assertEqual(upload_file_args[0][0].name, test_file.name)
self.assertEqual(upload_hash_args[0][0].read(), test_data_hash)
self.assertEqual(return_value, True)
def test_upload_file__exists(self):
test_data = b"abcdef"
test_data_hash = hashlib.md5(test_data).hexdigest()
test_file = tempfile.NamedTemporaryFile()
with test_file.file as f:
f.write(test_data)
self.ctrl.storage_driver = mock.MagicMock()
mock_container = mock.MagicMock()
mock_container.list_objects = mock.MagicMock(
return_value=[MockStorageObject("test_folder/test.txt"), MockStorageObject("test_folder/test.txt.md5")]
)
self.ctrl.storage_driver.create_container = mock.MagicMock(return_value=mock_container)
mock_container.get_object = mock.MagicMock(return_value=MockStorageObject("", bytes(test_data_hash, "utf8")))
return_value = self.ctrl.upload_file(test_file.name, "test_folder/test.txt")
self.assertFalse(self.ctrl.storage_driver.upload_object_via_stream.called)
self.assertEqual(return_value, False)
def test_download_file__exists__same_hash(self):
test_data = b"abcdefghi"
test_data_hash = hashlib.md5(test_data).hexdigest()
test_file = tempfile.NamedTemporaryFile()
with test_file.file as f:
f.write(test_data)
self.ctrl.storage_driver = mock.MagicMock()
mock_container = mock.MagicMock()
self.ctrl.storage_driver.get_container = mock.MagicMock(return_value=mock_container)
file_object = MockStorageObject("test_file.txt")
file_object.download = mock.MagicMock()
file_hash_object = MockStorageObject("test_file.txt", bytes(test_data_hash, "utf8"))
mock_container.get_object = lambda name: {"test_file.txt": file_object, "test_file.txt.md5": file_hash_object}[
name
]
self.ctrl.download_file("test_file.txt", test_file.name)
self.assertFalse(file_object.download.called)
def test_download_file__exists__different_hash(self):
test_data = b"abcdefghij"
test_data_hash = "some_garbage_hash"
test_file = tempfile.NamedTemporaryFile()
with test_file.file as f:
f.write(test_data)
self.ctrl.storage_driver = mock.MagicMock()
mock_container = mock.MagicMock()
self.ctrl.storage_driver.get_container = mock.MagicMock(return_value=mock_container)
file_object = MockStorageObject("test_file.txt")
file_object.download = mock.MagicMock()
file_hash_object = MockStorageObject("test_file.txt", bytes(test_data_hash, "utf8"))
mock_container.get_object = lambda name: {"test_file.txt": file_object, "test_file.txt.md5": file_hash_object}[
name
]
self.ctrl.download_file("test_file.txt", test_file.name)
file_object.download.assert_called_once_with(test_file.name)
def test_find_storage_image_names(self):
self.ctrl.storage_driver = mock.MagicMock()
mock_container = mock.MagicMock()
self.ctrl.storage_driver.get_container = mock.MagicMock(return_value=mock_container)
mock_container.list_objects = mock.MagicMock(
return_value=[
MockStorageObject("images/IOS/test_image_1.image"),
MockStorageObject("images/IOS/test_image_1.image.md5"),
MockStorageObject("images/IOS/test_image_2.img"),
MockStorageObject("images/IOS/test_image_2.img.md5"),
MockStorageObject("images/IOS/test_image_3.img"),
MockStorageObject("projects/project1.zip"),
MockStorageObject("test_image_1.image"),
]
)
image_names = self.ctrl.find_storage_image_names(["test_image_1.image", "test_image_2.img"])
self.assertDictEqual(
image_names,
{"test_image_1.image": "images/IOS/test_image_1.image", "test_image_2.img": "images/IOS/test_image_2.img"},
)
def test_find_storage_image_names__missing(self):
self.ctrl.storage_driver = mock.MagicMock()
mock_container = mock.MagicMock()
self.ctrl.storage_driver.get_container = mock.MagicMock(return_value=mock_container)
mock_container.list_objects = mock.MagicMock(
return_value=[
MockStorageObject("images/IOS/test_image_1.image"),
MockStorageObject("images/IOS/test_image_1.image.md5"),
MockStorageObject("images/IOS/test_image_2.img.md5"),
MockStorageObject("projects/project1.zip"),
MockStorageObject("test_image_1.image"),
]
)
self.assertRaises(Exception, self.ctrl.find_storage_image_names, ["test_image_1.image", "test_image_2.img"])
