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]
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