def serialize_account(account): ''' Serializes data for a single account ''' return { 'id': account.id, 'user': { 'id': account.user.id, 'username': account.user.username, }, 'name': account.name, 'balance': account.balance, 'address': account.address, 'phoneNumber': account.phone_number, 'transactions': serialize_transactions(Transaction.get_by_account(account)) }
def test_an_account_can_have_many_transactions(self): """ Tests that can get all transactions for account """ user = User.objects.create_user('john', '*****@*****.**', 'johnpassword') account = Account.objects.create(user=user, balance=1000, address="New York", phone_number="9176910399", name="John's private account", creator=user) transaction1 = Transaction.objects.create( transaction_type="withdrawal", amount=30, account_from=account, account_to=None, creator=user) transaction2 = Transaction.objects.create(transaction_type="deposit", amount=30, account_from=None, account_to=account, creator=user) self.assertEqual(Transaction.get_by_account(account).count(), 2)