def test_query_new_user(database, cleanup):
users = UserManager(database)
test_role = create_role(*user_role)
user1 = {
"email": "*****@*****.**",
"hashed_password": "******",
"salt": "diwriqjroqds",
"private_key": "rweqoasnfa",
"role": test_role.id,
}
user2 = {
"email": "*****@*****.**",
"hashed_password": "******",
"salt": "dgfgsgwrwers",
"private_key": "AHEIJASDAEW",
"role": test_role.id,
}
db_user1 = users.register(**user1)
db_user2 = users.register(**user2)
# Test every database field, except role id
for key, value in list(user1.items())[:-1]:
query_result_1 = users.query(**{key: value})
assert len(query_result_1) == 1
query_result_1 = query_result_1[0]
assert query_result_1.email == db_user1.email
assert query_result_1.hashed_password == db_user1.hashed_password
assert query_result_1.salt == db_user1.salt
assert query_result_1.private_key == db_user1.private_key
assert query_result_1.role == db_user1.role
for key, value in list(user2.items())[:-1]:
query_result_2 = users.query(**{key: value})
assert len(query_result_2) == 1
query_result_2 = query_result_2[0]
assert query_result_2.email == db_user2.email
assert query_result_2.hashed_password == db_user2.hashed_password
assert query_result_2.salt == db_user2.salt
assert query_result_2.private_key == db_user2.private_key
assert query_result_2.role == db_user2.role
query_result_3 = users.query(role=test_role.id)
assert len(query_result_3) == 2
def test_register_new_user(database, cleanup):
users = UserManager(database)
test_role = create_role(*user_role)
user = users.register(
email="*****@*****.**",
hashed_password="******",
salt="aifhaufa",
private_key="afihauhfao",
role=test_role.id,
)
assert user.email == "*****@*****.**"
assert user.hashed_password == "aifhufhaf"
assert user.salt == "aifhaufa"
assert user.private_key == "afihauhfao"
assert user.role == test_role.id
def test_set_email(database, cleanup):
users = UserManager(database)
test_role = create_role(*user_role)
user1 = {
"email": "*****@*****.**",
"hashed_password": "******",
"salt": "diwriqjroqds",
"private_key": "rweqoasnfa",
"role": test_role.id,
}
db_user1 = users.register(**user1)
assert users.query(id=db_user1.id)[0].email == "*****@*****.**"
users.set(user_id=db_user1.id, email="*****@*****.**")
assert users.query(id=db_user1.id)[0].email == "*****@*****.**"
def test_set_role(database, cleanup):
users = UserManager(database)
user_role_obj = create_role(*user_role)
admin_role_obj = create_role(*admin_role)
user1 = {
"email": "*****@*****.**",
"hashed_password": "******",
"salt": "diwriqjroqds",
"private_key": "rweqoasnfa",
"role": user_role_obj.id,
}
db_user1 = users.register(**user1)
assert users.query(id=db_user1.id)[0].role == user_role_obj.id
users.set(user_id=db_user1.id, role=admin_role_obj.id)
assert users.query(id=db_user1.id)[0].role == admin_role_obj.id
