def test_repository_can_get_users(session):
created_users = add_random_users(session)
repo = repository.SqlAlchemyRepository(session)
users = repo.get_users()
assert created_users == users
async def update_user_account_balance(account: Account):
session = get_session()
repo = repository.SqlAlchemyRepository(session)
account = repo.update_account_for_user(user_id=account.user_id,
account_id=account.account_id,
new_balance=account.balance)
return jsonable_encoder({'account_id': account.id})
def test_no_update_non_existing_account_for_user(session):
session.add(model.User())
random_balance = round(random.uniform(0, 2000), 2)
repo = repository.SqlAlchemyRepository(session)
account = repo.update_account_for_user(1, 2, random_balance)
assert not account
def test_get_accounts_from_user(session):
created_users = add_random_users(session)
random_user = random.choice(created_users)
repo = repository.SqlAlchemyRepository(session)
accounts = repo.get_accounts_from_user(random_user.id)
assert random_user.get_accounts() == accounts
async def get_accounts_for_user(user_id: int):
session = get_session()
repo = repository.SqlAlchemyRepository(session)
accounts = repo.get_accounts_from_user(user_id)
if not accounts:
raise HTTPException(status_code=404,
detail="No accounts not found for this user")
return jsonable_encoder(accounts)
def test_user_can_add_account(session):
created_users = add_random_users(session)
random_user = random.choice(created_users)
repo = repository.SqlAlchemyRepository(session)
number_user_accounts = repo.add_account_for_user(user_id=random_user.id)
user_accounts = set(
session.query(model.Account).filter(
model.Account.user_id == random_user.id).all())
assert number_user_accounts == len(user_accounts)
def test_update_account_for_user(session):
created_users = add_random_users(session)
random_user = random.choice(created_users)
random_account = random.choice(list(random_user.get_accounts()))
random_balance = round(random.uniform(0, 2000), 2)
repo = repository.SqlAlchemyRepository(session)
account = repo.update_account_for_user(random_user.id, random_account.id,
random_balance)
assert account.balance == random_balance
def test_repository_can_add_user(session):
repo = repository.SqlAlchemyRepository(session)
user = repo.add_user()
users = session.query(model.User).all()
assert user in users
async def create_account_for_user(user: User):
session = get_session()
repo = repository.SqlAlchemyRepository(session)
account_id = repo.add_account_for_user(user_id=user.user_id)
return jsonable_encoder({'account_id': account_id})
async def create_new_user():
session = get_session()
repo = repository.SqlAlchemyRepository(session)
user = repo.add_user()
return jsonable_encoder({'user_id': user.id})
