Sith/subscription/tests/test_dates.py

#
# Copyright 2023 © AE UTBM
# ae@utbm.fr / ae.info@utbm.fr
#
# This file is part of the website of the UTBM Student Association (AE UTBM),
# https://ae.utbm.fr.
#
# You can find the source code of the website at https://github.com/ae-utbm/sith
#
# LICENSED UNDER THE GNU GENERAL PUBLIC LICENSE VERSION 3 (GPLv3)
# SEE : https://raw.githubusercontent.com/ae-utbm/sith/master/LICENSE
# OR WITHIN THE LOCAL FILE "LICENSE"
#
#
"""Tests focused on the computing of subscription end, start and duration"""

from datetime import date

import freezegun
import pytest
from django.conf import settings
from django.test import TestCase

from core.models import User
from subscription.models import Subscription


@pytest.mark.parametrize(
    ("today", "duration", "expected_start"),
    [
        (date(2020, 9, 18), 1, date(2020, 9, 18)),
        (date(2020, 9, 18), 2, date(2020, 9, 18)),
        (date(2020, 5, 17), 3, date(2020, 2, 15)),
        (date(2021, 1, 18), 4, date(2020, 8, 15)),
        (date(2020, 9, 18), 4, date(2020, 8, 15)),
    ],
)
def test_subscription_compute_start_from_today(today, duration, expected_start):
    with freezegun.freeze_time(today):
        assert Subscription.compute_start(duration=duration) == expected_start


@pytest.mark.parametrize(
    ("start_date", "duration", "expected_start"),
    [
        (date(2020, 5, 17), 1, date(2020, 5, 17)),
        (date(2020, 5, 17), 2, date(2020, 5, 17)),
        (date(2020, 5, 17), 3, date(2020, 2, 15)),
        (date(2020, 1, 11), 3, date(2019, 8, 15)),
    ],
)
def test_subscription_compute_start_explicit(start_date, duration, expected_start):
    assert Subscription.compute_start(start_date, duration=duration) == expected_start


@pytest.mark.parametrize(
    ("today", "duration", "expected_end"),
    [
        (date(2020, 9, 18), 1, date(2021, 3, 18)),
        (date(2020, 9, 18), 2, date(2021, 9, 18)),
        (date(2020, 9, 18), 3, date(2022, 2, 15)),
        (date(2020, 5, 17), 4, date(2022, 2, 15)),
        (date(2020, 9, 18), 0.33, date(2020, 11, 18)),
        (date(2020, 9, 18), 0.67, date(2021, 1, 19)),
        (date(2020, 9, 18), 0.5, date(2020, 12, 18)),
    ],
)
def test_subscription_compute_end_from_today(today, duration, expected_end):
    with freezegun.freeze_time(today):
        assert Subscription.compute_end(duration=duration) == expected_end


@pytest.mark.parametrize(
    ("start_date", "duration", "expected_end"),
    [
        (date(2020, 9, 18), 3, date(2022, 3, 18)),
        (date(2020, 9, 18), 4, date(2022, 9, 18)),
    ],
)
def test_subscription_compute_end(start_date, duration, expected_end):
    assert Subscription.compute_end(duration, start_date) == expected_end


class TestSubscriptionIntegration(TestCase):
    @classmethod
    def setUpTestData(cls):
        cls.user = User.objects.get(username="public")

    def test_duration_one_month(self):
        s = Subscription(
            member=self.user,
            subscription_type=list(settings.SITH_SUBSCRIPTIONS.keys())[3],
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2017, 8, 29)
        s.subscription_end = s.compute_end(duration=0.166, start=s.subscription_start)
        s.save()
        assert s.subscription_end == date(2017, 9, 29)

    def test_duration_two_months(self):
        s = Subscription(
            member=self.user,
            subscription_type=list(settings.SITH_SUBSCRIPTIONS.keys())[3],
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2017, 8, 29)
        s.subscription_end = s.compute_end(duration=0.333, start=s.subscription_start)
        s.save()
        assert s.subscription_end == date(2017, 10, 29)

    def test_duration_one_day(self):
        s = Subscription(
            member=self.user,
            subscription_type=list(settings.SITH_SUBSCRIPTIONS.keys())[3],
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2017, 8, 29)
        s.subscription_end = s.compute_end(
            duration=settings.SITH_SUBSCRIPTIONS["un-jour"]["duration"],
            start=s.subscription_start,
        )
        s.save()
        assert s.subscription_end == date(2017, 8, 30)

    def test_duration_three_months(self):
        s = Subscription(
            member=self.user,
            subscription_type=list(settings.SITH_SUBSCRIPTIONS.keys())[3],
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2017, 8, 29)
        s.subscription_end = s.compute_end(duration=0.5, start=s.subscription_start)
        s.save()
        assert s.subscription_end == date(2017, 11, 29)

    def test_duration_four_months(self):
        s = Subscription(
            member=self.user,
            subscription_type=list(settings.SITH_SUBSCRIPTIONS.keys())[3],
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2017, 8, 29)
        s.subscription_end = s.compute_end(duration=0.67, start=s.subscription_start)
        s.save()
        assert s.subscription_end == date(2017, 12, 30)

    def test_duration_six_weeks(self):
        s = Subscription(
            member=self.user,
            subscription_type=list(settings.SITH_SUBSCRIPTIONS.keys())[3],
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2018, 9, 1)
        s.subscription_end = s.compute_end(duration=0.23, start=s.subscription_start)
        s.save()
        assert s.subscription_end == date(2018, 10, 13)

    def test_dates_sliding_with_subscribed_user(self):
        user = self.user
        s = Subscription(
            member=user,
            subscription_type="deux-semestres",
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2015, 8, 29)
        s.subscription_end = s.compute_end(
            duration=settings.SITH_SUBSCRIPTIONS[s.subscription_type]["duration"],
            start=s.subscription_start,
        )
        s.save()
        assert s.subscription_end == date(2016, 8, 29)
        with freezegun.freeze_time("2016-08-25"):
            d = Subscription.compute_end(
                duration=settings.SITH_SUBSCRIPTIONS["deux-semestres"]["duration"],
                user=user,
            )
            assert d == date(2017, 8, 29)

    def test_dates_renewal_sliding_during_two_free_monthes(self):
        user = self.user
        s = Subscription(
            member=user,
            subscription_type="deux-mois-essai",
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2015, 8, 29)
        s.subscription_end = s.compute_end(
            duration=settings.SITH_SUBSCRIPTIONS[s.subscription_type]["duration"],
            start=s.subscription_start,
        )
        s.save()
        assert s.subscription_end == date(2015, 10, 29)
        with freezegun.freeze_time("2015-09-25"):
            d = Subscription.compute_end(
                duration=settings.SITH_SUBSCRIPTIONS["deux-semestres"]["duration"],
                user=user,
            )
            assert d == date(2016, 10, 29)

    def test_dates_renewal_sliding_after_two_free_monthes(self):
        user = self.user
        s = Subscription(
            member=user,
            subscription_type="deux-mois-essai",
            payment_method=settings.SITH_SUBSCRIPTION_PAYMENT_METHOD[0],
        )
        s.subscription_start = date(2015, 8, 29)
        s.subscription_end = s.compute_end(
            duration=settings.SITH_SUBSCRIPTIONS[s.subscription_type]["duration"],
            start=s.subscription_start,
        )
        s.save()
        assert s.subscription_end == date(2015, 10, 29)
        with freezegun.freeze_time("2015-11-05"):
            d = Subscription.compute_end(
                duration=settings.SITH_SUBSCRIPTIONS["deux-semestres"]["duration"],
                user=user,
            )
            assert d == date(2016, 11, 5)