def test_return_accounts_as_json_when_account_data_is_valid(self):
account_data = {"account_name1": {"id": 42, "email": "*****@*****.**"}}
result = ac.get_converted_aws_accounts(account_data)
json_result = result["accounts.json"]
decoded_result = json.loads(json_result)
self.assertEqual(decoded_result, account_data)
def test_return_accounts_as_yaml_when_account_data_is_valid(self):
account_data = {"account_name1": {"id": 42, "email": "*****@*****.**"}}
result = ac.get_converted_aws_accounts(account_data)
yaml_result = result["accounts.yaml"]
decoded_result = yaml.safe_load(yaml_result)
self.assertEqual(decoded_result, account_data)
def test_return_accounts_as_json_when_account_data_is_valid(self):
account_data = {
"account_name1": {
"id": 42,
"email": "*****@*****.**"
}
}
result = ac.get_converted_aws_accounts(account_data)
json_result = result["accounts.json"]
decoded_result = json.loads(json_result)
self.assertEqual(decoded_result, account_data)
def test_return_accounts_as_yaml_when_account_data_is_valid(self):
account_data = {
"account_name1": {
"id": 42,
"email": "*****@*****.**"
}
}
result = ac.get_converted_aws_accounts(account_data)
yaml_result = result["accounts.yaml"]
decoded_result = yaml.safe_load(yaml_result)
self.assertEqual(decoded_result, account_data)
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)
