def test_loaded_data_is_checked(self, mock_check_account):
directory = tempfile.mkdtemp()
filename = "account.yaml"
content = "account_one:\n id: 1\n email: [email protected]\n owner: me"
try:
with open(os.path.join(directory, filename), "w") as test_file:
test_file.write(content)
ai.read_directory(directory)
mock_check_account.assert_called_once_with(
{'account_one': {
'id': '1', 'email': '*****@*****.**', 'owner': 'me'}})
finally:
shutil.rmtree(directory)
def test_account_id_is_converted_to_string(self):
directory = tempfile.mkdtemp()
filename = "account.yaml"
content = {
"account_name": {
"id": 42,
"email": "*****@*****.**",
"owner": "me"
},
"another": {
"id": "43",
"email": "*****@*****.**",
"owner": "minime"
}
}
try:
with open(os.path.join(directory, filename), "w") as test_file:
test_file.write(yaml.dump(content))
result = ai.read_directory(directory)
expected_content = content
# Must have been converted from int 42 to string "42".
expected_content['account_name']['id'] = '42'
self.assertEqual(expected_content, result)
finally:
shutil.rmtree(directory)
def test_loaded_data_is_checked(self, mock_check_account):
directory = tempfile.mkdtemp()
filename = "account.yaml"
content = "account_one:\n id: 1\n email: [email protected]\n owner: me"
try:
with open(os.path.join(directory, filename), "w") as test_file:
test_file.write(content)
ai.read_directory(directory)
mock_check_account.assert_called_once_with({
'account_one': {
'id': '1',
'email': '*****@*****.**',
'owner': 'me'
}
})
finally:
shutil.rmtree(directory)
def test_loaded_data_is_returned_data(self):
directory = tempfile.mkdtemp()
filename = "account.yaml"
content = {"account_name": {"id": "42", "email": "*****@*****.**", "owner": "me"}}
try:
with open(os.path.join(directory, filename), "w") as test_file:
test_file.write(yaml.dump(content))
result = ai.read_directory(directory)
self.assertEqual(content, result)
finally:
shutil.rmtree(directory)
def upload(data_directory, destination_bucket_name, allowed_ips=None):
allowed_ips = allowed_ips or []
try:
account_data = read_directory(data_directory)
except Exception as e:
raise Exception("Failed to read data directory '{0}': {1} ".format(data_directory, e))
data_to_upload = get_converted_aws_accounts(account_data)
our_account_ids = [account["id"] for account in account_data.values()]
uploader = S3Uploader(destination_bucket_name, allowed_ips=allowed_ips, allowed_aws_account_ids=our_account_ids)
uploader.setup_infrastructure()
uploader.upload_to_S3(data_to_upload)
def upload(data_directory, destination_bucket_name, allowed_ips=None):
allowed_ips = allowed_ips or []
try:
account_data = read_directory(data_directory)
except Exception as e:
raise Exception("Failed to read data directory '{0}': {1} ".format(
data_directory, e))
data_to_upload = get_converted_aws_accounts(account_data)
our_account_ids = [account['id'] for account in account_data.values()]
uploader = S3Uploader(destination_bucket_name,
allowed_ips=allowed_ips,
allowed_aws_account_ids=our_account_ids)
uploader.setup_infrastructure()
uploader.upload_to_S3(data_to_upload)
def test_account_id_is_converted_to_string(self):
directory = tempfile.mkdtemp()
filename = "account.yaml"
content = {
"account_name": {"id": 42, "email": "*****@*****.**", "owner": "me"},
"another": {"id": "43", "email": "*****@*****.**", "owner": "minime"}}
try:
with open(os.path.join(directory, filename), "w") as test_file:
test_file.write(yaml.dump(content))
result = ai.read_directory(directory)
expected_content = content
# Must have been converted from int 42 to string "42".
expected_content['account_name']['id'] = '42'
self.assertEqual(expected_content, result)
finally:
shutil.rmtree(directory)
def test_loaded_data_is_returned_data(self):
directory = tempfile.mkdtemp()
filename = "account.yaml"
content = {
"account_name": {
"id": "42",
"email": "*****@*****.**",
"owner": "me"
}
}
try:
with open(os.path.join(directory, filename), "w") as test_file:
test_file.write(yaml.dump(content))
result = ai.read_directory(directory)
self.assertEqual(content, result)
finally:
shutil.rmtree(directory)
