def add_user():
Utility.load_evironment()
connect(Utility.environment["mongo_db"],
host=Utility.environment["mongo_url"])
user = AccountProcessor.account_setup(
{
"email": "*****@*****.**",
"first_name": "Demo",
"last_name": "User",
"password": "******",
"account": "integration",
"bot": "integration"
},
user="******")
processor = MongoProcessor()
processor.save_from_path("tests/testing_data/all", user['bot'],
"testAdmin")
AccountProcessor.account_setup(
{
"email": "*****@*****.**",
"first_name": "Demo",
"last_name": "User",
"password": "******",
"account": "integration2",
"bot": "integration2"
},
user="******")
def test_account_setup_user_info(self):
account = {
"account": "Test_Account",
"bot": "Test",
"first_name": "Test_First",
"last_name": "Test_Last",
"password": "******"
}
with pytest.raises(ValidationError):
AccountProcessor.account_setup(account_setup=account,
user="******")
def test_account_setup_missing_bot_name(self):
account = {
"account": "TestAccount",
"email": "*****@*****.**",
"first_name": "Test_First",
"last_name": "Test_Last",
"password": "******"
}
with pytest.raises(ValidationError):
AccountProcessor.account_setup(account_setup=account,
user="******")
def test_account_setup_empty_values(self):
account = {}
with pytest.raises(AssertionError):
loop = asyncio.new_event_loop()
loop.run_until_complete(
AccountProcessor.account_setup(account_setup=account,
user="******"))
def test_account_setup_user_info(self):
account = {
"account": "Test_Account",
"bot": "Test",
"first_name": "Test_First",
"last_name": "Test_Last",
"password": SecretStr("Welcome@1"),
}
with pytest.raises(AssertionError):
loop = asyncio.new_event_loop()
loop.run_until_complete(
AccountProcessor.account_setup(account_setup=account,
user="******"))
def test_account_setup_missing_bot_name(self):
account = {
"account": "TestAccount",
"email": "*****@*****.**",
"first_name": "Test_First",
"last_name": "Test_Last",
"password": "******",
}
with pytest.raises(AssertionError):
loop = asyncio.new_event_loop()
loop.run_until_complete(
AccountProcessor.account_setup(account_setup=account,
user="******"))
def test_account_setup(self):
account = {
"account": "Test_Account",
"bot": "Test",
"email": "*****@*****.**",
"first_name": "Test_First",
"last_name": "Test_Last",
"password": "******"
}
actual = AccountProcessor.account_setup(account_setup=account,
user="******")
assert actual['role'] == "admin"
assert actual['_id']
assert actual['account']
assert actual['bot']
def test_account_setup(self):
account = {
"account": "Test_Account",
"bot": "Test",
"email": "*****@*****.**",
"first_name": "Test_First",
"last_name": "Test_Last",
"password": SecretStr("Welcome@1"),
}
actual = AccountProcessor.account_setup(account_setup=account,
user="******")
assert actual["role"] == "admin"
assert actual["_id"]
assert actual["account"]
assert actual["bot"]
def test_account_setup(self):
account = {
"account": "Test_Account",
"bot": "Test",
"email": "*****@*****.**",
"first_name": "Test_First",
"last_name": "Test_Last",
"password": SecretStr("Welcome@1"),
}
loop = asyncio.new_event_loop()
actual = loop.run_until_complete(
AccountProcessor.account_setup(account_setup=account,
user="******"))
assert actual["role"] == "admin"
assert actual["_id"]
assert actual["account"]
assert actual["bot"]
async def register_account(register_account: RegisterAccount):
"""new account registeration api"""
AccountProcessor.account_setup(register_account.dict(), "sysadmin")
return {"message": "Account Registered!"}
def test_account_setup_empty_values(self):
account = {}
with pytest.raises(ValidationError):
AccountProcessor.account_setup(account_setup=account,
user="******")
