re2o/cotisations/test_views.py

from django.test import TestCase
from django.urls import reverse
from django.contrib.auth.models import Permission

import datetime
from dateutil.relativedelta import relativedelta
from django.utils import timezone

from users.models import Adherent
from .models import Vente, Facture, Cotisation, Paiement, Article

class NewFactureTests(TestCase):
    def tearDown(self):
        self.user.facture_set.all().delete()
        self.user.delete()
        self.paiement.delete()
        self.article_one_day.delete()
        self.article_one_month.delete()
        self.article_one_month_and_one_week.delete()

    def setUp(self):
        self.user = Adherent.objects.create(
            pseudo="testUser",
            email="test@example.org",
        )
        self.user.set_password('plopiplop')
        self.user.user_permissions.set(
            [
                Permission.objects.get_by_natural_key("add_facture", "cotisations", "Facture"),
                Permission.objects.get_by_natural_key("use_every_payment", "cotisations", "Paiement"),
            ]
        )
        self.user.save()

        self.paiement = Paiement.objects.create(
            moyen="test payment",

        )
        self.article_one_day = Article.objects.create(
            name="One day",
            prix=0,
            duration=0,
            duration_days=1,
            type_cotisation='All',
            available_for_everyone=True
        )
        self.article_one_month = Article.objects.create(
            name="One day",
            prix=0,
            duration=1,
            duration_days=0,
            type_cotisation='All',
            available_for_everyone=True
        )
        self.article_one_month_and_one_week = Article.objects.create(
            name="One day",
            prix=0,
            duration=1,
            duration_days=7,
            type_cotisation='All',
            available_for_everyone=True
        )
        self.client.login(
            username="testUser",
            password="plopiplop"
        )

    def test_invoice_with_one_day(self):
        data = {
            "Facture-paiement": self.paiement.pk,
            "form-TOTAL_FORMS": 1,
            "form-INITIAL_FORMS": 0,
            "form-MIN_NUM_FORMS": 0,
            "form-MAX_NUM_FORMS": 1000,
            "form-0-article": 1,
            "form-0-quantity": 1,
        }
        date = timezone.now()
        response = self.client.post(reverse('cotisations:new-facture', kwargs={'userid':self.user.pk}), data)
        self.assertEqual(
            response.status_code,
            302
        )
        self.assertEqual(
            response.url,
            "/users/profil/%d"%self.user.pk
        )
        self.assertAlmostEqual(
            self.user.end_connexion() - date,
            datetime.timedelta(days=1),
            delta=datetime.timedelta(seconds=1)
        )

    def test_invoice_with_one_month(self):
        data = {
            "Facture-paiement": self.paiement.pk,
            "form-TOTAL_FORMS": 1,
            "form-INITIAL_FORMS": 0,
            "form-MIN_NUM_FORMS": 0,
            "form-MAX_NUM_FORMS": 1000,
            "form-0-article": 2,
            "form-0-quantity": 1,
        }
        date = timezone.now()
        response = self.client.post(reverse('cotisations:new-facture', kwargs={'userid':self.user.pk}), data)
        self.assertEqual(
            response.status_code,
            302
        )
        self.assertEqual(
            response.url,
            "/users/profil/%d"%self.user.pk
        )
        delta = relativedelta(self.user.end_connexion(), date)
        delta.microseconds=0
        self.assertEqual(
            delta,
            relativedelta(months=1),
        )

    def test_invoice_with_one_month_and_one_week(self):
        data = {
            "Facture-paiement": self.paiement.pk,
            "form-TOTAL_FORMS": 2,
            "form-INITIAL_FORMS": 0,
            "form-MIN_NUM_FORMS": 0,
            "form-MAX_NUM_FORMS": 1000,
            "form-0-article": 1,
            "form-0-quantity": 7,
            "form-1-article": 2,
            "form-1-quantity": 1,
        }
        date = timezone.now()
        response = self.client.post(reverse('cotisations:new-facture', kwargs={'userid':self.user.pk}), data)
        self.assertEqual(
            response.status_code,
            302
        )
        self.assertEqual(
            response.url,
            "/users/profil/%d"%self.user.pk
        )
        invoice = self.user.facture_set.first()
        delta = relativedelta(self.user.end_connexion(), date)
        delta.microseconds=0
        self.assertEqual(
            delta,
            relativedelta(months=1, days=7),
        )


    def test_several_articles_creates_several_purchases(self):
        data = {
            "Facture-paiement": self.paiement.pk,
            "form-TOTAL_FORMS": 2,
            "form-INITIAL_FORMS": 0,
            "form-MIN_NUM_FORMS": 0,
            "form-MAX_NUM_FORMS": 1000,
            "form-0-article": 2,
            "form-0-quantity": 1,
            "form-1-article": 2,
            "form-1-quantity": 1,
        }
        response = self.client.post(reverse('cotisations:new-facture', kwargs={'userid':self.user.pk}), data)
        f = self.user.facture_set.first()
        self.assertEqual(f.vente_set.count(), 2)