edan_f0/tests/integration/voucher/test_forms.py

import datetime

import pytest
from django.utils import timezone
from django.utils.datastructures import MultiValueDict

from oscar.apps.dashboard.vouchers import forms
from oscar.core.loading import get_model
from oscar.test.factories.offer import (
    BenefitFactory,
    ConditionalOfferFactory,
    ConditionFactory,
    RangeFactory,
)
from oscar.test.factories.voucher import VoucherSetFactory

ConditionalOffer = get_model("offer", "ConditionalOffer")
Voucher = get_model("voucher", "Voucher")


@pytest.mark.django_db
def test_voucher_set_form_create():
    a_range = RangeFactory(includes_all_products=True)
    offer = ConditionalOfferFactory(
        offer_type=ConditionalOffer.VOUCHER,
        benefit=BenefitFactory(range=a_range),
        condition=ConditionFactory(range=a_range, value=1),
    )
    data = MultiValueDict(
        {
            "name": ["10% Discount"],
            "code_length": ["10"],
            "count": ["10"],
            "description": ["This is a 10% discount for mailing X"],
            "start_datetime": ["2014-10-01"],
            "end_datetime": ["2018-10-01"],
            "usage": [Voucher.MULTI_USE],
            "offers": [offer.pk],
        }
    )
    form = forms.VoucherSetForm(data)
    assert form.is_valid(), form.errors
    voucher_set = form.save()
    assert voucher_set.vouchers.count() == 10


@pytest.mark.django_db
def test_voucher_set_form_update_with_unchanged_count():
    tzinfo = timezone.get_current_timezone()
    voucher_set = VoucherSetFactory(
        name="Dummy name",
        count=5,
        code_length=12,
        description="Dummy description",
        start_datetime=datetime.datetime(2021, 2, 1, tzinfo=tzinfo),
        end_datetime=datetime.datetime(2021, 2, 28, tzinfo=tzinfo),
    )
    voucher = voucher_set.vouchers.first()
    assert voucher.usage == Voucher.MULTI_USE
    new_offers = [
        ConditionalOfferFactory(offer_type=ConditionalOffer.VOUCHER),
        ConditionalOfferFactory(offer_type=ConditionalOffer.VOUCHER),
    ]
    data = {
        "name": "New name",
        "code_length": 10,
        "description": "New description",
        "start_datetime": datetime.datetime(2021, 3, 1, tzinfo=tzinfo),
        "end_datetime": datetime.datetime(2021, 3, 31, tzinfo=tzinfo),
        "count": voucher_set.count,
        "usage": Voucher.SINGLE_USE,
        "offers": new_offers,
    }
    form = forms.VoucherSetForm(data, instance=voucher_set)
    assert form.is_valid(), form.errors
    voucher_set = form.save()
    assert voucher_set.vouchers.count() == 5
    for i, v in enumerate(voucher_set.vouchers.order_by("date_created")):
        assert v.name == "New name - %d" % (i + 1)
        assert len(v.code) == 14  # The code is not modified
        assert v.start_datetime == datetime.datetime(2021, 3, 1, tzinfo=tzinfo)
        assert v.end_datetime == datetime.datetime(2021, 3, 31, tzinfo=tzinfo)
        assert v.usage == Voucher.SINGLE_USE
        assert list(v.offers.all()) == new_offers


@pytest.mark.django_db
def test_voucher_set_form_update_with_changed_count():
    tzinfo = timezone.get_current_timezone()
    voucher_set = VoucherSetFactory(
        name="Dummy name",
        count=5,
        code_length=12,
        description="Dummy description",
        start_datetime=datetime.datetime(2021, 2, 1, tzinfo=tzinfo),
        end_datetime=datetime.datetime(2021, 2, 28, tzinfo=tzinfo),
    )
    voucher = voucher_set.vouchers.first()
    assert voucher.usage == Voucher.MULTI_USE
    new_offers = [
        ConditionalOfferFactory(offer_type=ConditionalOffer.VOUCHER),
        ConditionalOfferFactory(offer_type=ConditionalOffer.VOUCHER),
    ]
    data = {
        "name": "New name",
        "code_length": 10,
        "description": "New description",
        "start_datetime": datetime.datetime(2021, 3, 1, tzinfo=tzinfo),
        "end_datetime": datetime.datetime(2021, 3, 31, tzinfo=tzinfo),
        "count": 10,
        "usage": Voucher.SINGLE_USE,
        "offers": new_offers,
    }
    form = forms.VoucherSetForm(data, instance=voucher_set)
    assert form.is_valid(), form.errors
    voucher_set = form.save()
    voucher_set.refresh_from_db()
    assert voucher_set.count == 10  # "count" is updated
    assert voucher_set.vouchers.count() == 10
    for i, v in enumerate(voucher_set.vouchers.order_by("date_created")):
        assert v.name == "New name - %d" % (i + 1)
        if i < 5:
            # Original vouchers
            assert len(v.code) == 14  # The code is not modified
        else:
            # New vouchers
            assert len(v.code) == 12
        assert v.start_datetime == datetime.datetime(2021, 3, 1, tzinfo=tzinfo)
        assert v.end_datetime == datetime.datetime(2021, 3, 31, tzinfo=tzinfo)
        assert v.usage == Voucher.SINGLE_USE
        assert list(v.offers.all()) == new_offers