예제 #1
0
def test_set_bank_detail(user: User, session: Session,
                         application: Application):
    assert not user.bank_detail
    account_name = "Test Account Name"
    account_number = "Test Account Number"
    application.set_bank_detail(session, user, account_name, account_number)
    xuser = session.query(User).get(user.id)
    assert xuser.bank_detail
    assert xuser.bank_detail.account_name == account_name
    assert xuser.bank_detail.account_number == account_number
    assert xuser.bank_detail.user_id == user.id
예제 #2
0
def set_user_bank_details(
        data: BankDetailSchema,
        user: User = Depends(dependencies.get_current_user),
        app: Application = Depends(dependencies.get_application),
        session: Session = Depends(dependencies.get_database_session),
):
    try:
        return app.set_bank_detail(session, user, data.account_name,
                                   data.account_number)
    except ApplicationError as e:
        if "not found" in str(e).lower():
            raise HTTPException(404, str(e))
        raise HTTPException(400, str(e))