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)