def test_01(self):
"""
Test Stat model manager's method.
"""
e = Emulator()
e.init()
e.create_stripe_transfers(90)
TransferUser.objects.process_users()
assert TransferUser.objects.count() > 0
assert TransferDebt.objects.count() > 0
print(Stat.objects.get_json(e.user.id))
def test_close_account01(self):
"""
Test user's method.
User closes Donkies account.
All required steps listed in "user.close_account" method.
"""
e = Emulator()
e.init()
# Emulate transfers to Stripe
e.create_stripe_transfers(90)
count = TransferStripe.objects.filter(paid=True).count()
assert count > 0
count = TransferStripe.objects.filter(
is_processed_to_user=True).count()
assert count == 0
count = TransferUser.objects.all().count()
assert count == 0
# Close user's account
e.user.close_account()
# All user's items, accounts and transactions should
# be not active
assert Item.objects.filter(is_active=True).count() == 0
assert Account.objects.filter(is_active=True).count() == 0
assert Transaction.objects.filter(is_active=True).count() == 0
# All transfers from TransferStripe should be processed.
count = TransferStripe.objects.filter(
is_processed_to_user=False).count()
assert count == 0
count = TransferStripe.objects.filter(
is_processed_to_user=True).count()
assert count > 0
# TransferUser model should be filled
count = TransferUser.objects.all().count()
assert count > 0
# TransferDebt model should be filled
count = TransferDebt.objects.all().count()
assert count > 0
e.user.refresh_from_db()
e.user.is_closed_account is True
def test_get_transfer_debt(self, client):
e = Emulator()
e.init()
e.create_stripe_transfers(90)
e.run_process_users()
client = self.get_auth_client(e.user)
count = TransferDebt.objects.count()
assert count > 0
url = '/v1/transfers_debt'
response = client.get(url)
assert response.status_code == 200
rd = response.json()
assert len(rd) == count
def test_close_account02(self):
"""
Test API endpoint.
User closes Donkies account.
"""
e = Emulator()
e.init()
e.create_stripe_transfers(90)
client = self.get_auth_client(e.user)
url = '/v1/user_close_account'
dic = {}
data = json.dumps(dic)
response = client.post(url, data, content_type='application/json')
assert response.status_code == 204
e.user.refresh_from_db()
assert e.user.is_closed_account is True