Ejemplo n.º 1
0
    def test_find_all_by_user_returns_all_expected_wallets(self, database):
        user = User.create(first_name=self.first_name,
                           last_name=self.last_name,
                           email=self.email)

        user2 = User.create(first_name=self.first_name,
                            last_name=self.last_name,
                            email='*****@*****.**')

        wallet = Wallet.create(user, self.currency)

        assert Wallet.query.count() == 1
        assert Wallet.find_all_by_user(user2) == []
        assert Wallet.find_all_by_user(user) == [wallet]
Ejemplo n.º 2
0
def wallets():
    user_authentication_details = flask_login.current_user
    user = User.find_by_id(user_authentication_details.user_id)

    wallets = [wallet.as_dict() for wallet in Wallet.find_all_by_user(user)]

    return jsonify(wallets), HTTPStatus.OK