8
0
Fork 0
mirror of https://gitlab2.federez.net/re2o/re2o synced 2024-12-04 18:42:25 +00:00
re2o/users/test_models.py

45 lines
1.3 KiB
Python
Raw Permalink Normal View History

2019-11-03 23:25:29 +00:00
import datetime
2021-02-10 10:06:09 +00:00
from django.test import TestCase
2019-11-03 23:25:29 +00:00
from django.utils import timezone
2021-02-10 10:06:09 +00:00
from cotisations.models import Facture, Paiement, Vente
2019-11-03 23:25:29 +00:00
from users.models import User
2019-11-03 23:25:29 +00:00
class UserModelTests(TestCase):
def setUp(self):
self.user = User.objects.create(pseudo="testUser")
2019-11-03 23:25:29 +00:00
def tearDown(self):
self.user.facture_set.all().delete()
self.user.delete()
def test_multiple_cotisations_are_taken_into_account(self):
paiement = Paiement.objects.create(moyen="test payment")
invoice = Facture.objects.create(user=self.user, paiement=paiement, valid=True)
2019-11-03 23:25:29 +00:00
date = timezone.now()
purchase1 = Vente.objects.create(
facture=invoice,
number=1,
name="Test purchase",
duration=0,
duration_days=1,
type_cotisation="All",
prix=0,
)
purchase2 = Vente.objects.create(
facture=invoice,
number=1,
name="Test purchase",
duration=0,
duration_days=1,
type_cotisation="All",
prix=0,
)
self.assertAlmostEqual(
self.user.end_connexion() - date,
datetime.timedelta(days=2),
delta=datetime.timedelta(seconds=1),
2019-11-03 23:25:29 +00:00
)