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