def test_update_aws_account(self):
from security_monkey.account_manager import account_registry
for name, account_manager in account_registry.items():
manager.add_command("add_account_%s" % name.lower(), AddAccount(account_manager()))
# Create the account:
from security_monkey.account_manager import account_registry
for name, am in account_registry.items():
if name == "AWS":
break
account_manager = am()
account_manager.create(account_manager.account_type, "test", True, False, "Tests", "99999999999",
custom_fields=dict(canonical_id="bcaf1ffd86f41161ca5fb16fd081034f", s3_id=None))
# Create a second account:
account_manager.create(account_manager.account_type, "test2", True, False, "Tests", "99999999990",
custom_fields=dict(canonical_id="bcaf1ffd86f41161ca5fb16fd081asdf", s3_id=None))
# Get the ID of the first account:
id = Account.query.filter(Account.name == "test").one().id
# Try to rename the account:
account_manager.update(id, account_manager.account_type, "lololol", True, False, "Tests", "99999999999",
custom_fields=dict(canonical_id="bcaf1ffd86f41161ca5fb16fd081034f", s3_id=None))
assert not Account.query.filter(Account.name == "test").first()
assert Account.query.filter(Account.name == "lololol").first().id == id
# Try to update it to an existing name:
with self.assertRaises(AccountNameExists):
account_manager.update(id, account_manager.account_type, "test2", True, False, "Tests", "99999999999",
custom_fields=dict(canonical_id="bcaf1ffd86f41161ca5fb16fd081034f", s3_id=None))
def main():
from security_monkey.account_manager import account_registry
for name, account_manager in account_registry.items():
manager.add_command("add_account_%s" % name.lower(), AddAccount(account_manager()))
manager.add_command("run_api_server", APIServer())
manager.run()
def test_create_aws_account(self):
from security_monkey.account_manager import account_registry
for name, account_manager in list(account_registry.items()):
manager.add_command("add_account_%s" % name.lower(),
AddAccount(account_manager()))
manager.handle("manage.py", [
"add_account_aws", "-n", "test", "--active", "--id", "99999999999",
"--canonical_id", "bcaf1ffd86f41161ca5fb16fd081034f",
"--role_name", "SecurityMonkey"
])
account = Account.query.filter(Account.name == "test").first()
assert account
assert account.identifier == "99999999999"
assert account.active
assert len(account.custom_fields) == 4
# Get the canonical ID field:
c_id = AccountTypeCustomValues.query.filter(
AccountTypeCustomValues.name == "canonical_id",
AccountTypeCustomValues.account_id == account.id).first()
assert c_id
assert c_id.value == "bcaf1ffd86f41161ca5fb16fd081034f"
# Already exists:
assert manager.handle("manage.py", [
"add_account_aws", "-n", "test", "--active", "--id", "99999999999",
"--canonical_id", "bcaf1ffd86f41161ca5fb16fd081034f",
"--role_name", "SecurityMonkey"
]) == -1
def test_update_account_with_canonical(self):
from security_monkey.account_manager import account_registry
for name, account_manager in list(account_registry.items()):
manager.add_command("add_account_%s" % name.lower(),
AddAccount(account_manager()))
# Update:
manager.handle("manage.py", [
"add_account_aws", "-n", "account0", "--active", "--id",
"012345678910", "--canonical_id",
"bcaf1ffd86f41161ca5fb16fd081034f", "--s3_name", "test",
"--role_name", "SecurityMonkey", "--update-existing"
])
account = Account.query.filter(Account.name == "account0").first()
assert account
assert account.identifier == "012345678910"
assert account.active
assert len(account.custom_fields) == 4
# Get the canonical ID field:
c_id = AccountTypeCustomValues.query.filter(
AccountTypeCustomValues.name == "canonical_id",
AccountTypeCustomValues.account_id == account.id).first()
assert c_id
assert c_id.value == "bcaf1ffd86f41161ca5fb16fd081034f"
def test_create_account_with_canonical(self):
from security_monkey.account_manager import account_registry
for name, account_manager in account_registry.items():
manager.add_command("add_account_%s" % name.lower(), AddAccount(account_manager()))
manager.handle("manage.py", ["add_account_aws", "-n", "test", "--active", "--id", "99999999999",
"--canonical_id", "bcaf1ffd86f41161ca5fb16fd081034f", "--s3_name", "test",
"--role_name", "SecurityMonkey"])
account = Account.query.filter(Account.name == "test").first()
assert account
assert account.identifier == "99999999999"
assert account.active
assert len(account.custom_fields) == 3
# Get the canonical ID field:
c_id = AccountTypeCustomValues.query.filter(AccountTypeCustomValues.name == "canonical_id",
AccountTypeCustomValues.account_id == account.id).first()
assert c_id
assert c_id.value == "bcaf1ffd86f41161ca5fb16fd081034f"
assert manager.handle("manage.py", ["add_account_aws", "-n", "test", "--active", "--id", "99999999999",
"--canonical_id", "bcaf1ffd86f41161ca5fb16fd081034f", "--s3_name", "test",
"--role_name", "SecurityMonkey"]) == -1
def main():
from security_monkey.account_manager import account_registry
for name, account_manager in list(account_registry.items()):
manager.add_command("add_account_%s" % name.lower(), AddAccount(account_manager()))
manager.add_command("run_api_server", APIServer())
manager.run()
def test_update_aws_account(self):
from security_monkey.account_manager import account_registry
for name, account_manager in list(account_registry.items()):
manager.add_command("add_account_%s" % name.lower(),
AddAccount(account_manager()))
# Create the account:
from security_monkey.account_manager import account_registry
for name, am in list(account_registry.items()):
if name == "AWS":
break
account_manager = am()
account_manager.create(
account_manager.account_type,
"test",
True,
False,
"Tests",
"99999999999",
custom_fields=dict(canonical_id="bcaf1ffd86f41161ca5fb16fd081034f",
s3_id=None))
# Create a second account:
account_manager.create(
account_manager.account_type,
"test2",
True,
False,
"Tests",
"99999999990",
custom_fields=dict(canonical_id="bcaf1ffd86f41161ca5fb16fd081asdf",
s3_id=None))
# Get the ID of the first account:
id = Account.query.filter(Account.name == "test").one().id
# Try to rename the account:
account_manager.update(
id,
account_manager.account_type,
"lololol",
True,
False,
"Tests",
"99999999999",
custom_fields=dict(canonical_id="bcaf1ffd86f41161ca5fb16fd081034f",
s3_id=None))
assert not Account.query.filter(Account.name == "test").first()
assert Account.query.filter(Account.name == "lololol").first().id == id
# Try to update it to an existing name:
with self.assertRaises(AccountNameExists):
account_manager.update(
id,
account_manager.account_type,
"test2",
True,
False,
"Tests",
"99999999999",
custom_fields=dict(
canonical_id="bcaf1ffd86f41161ca5fb16fd081034f",
s3_id=None))
