from decimal import Decimal as D

from django.core.exceptions import ValidationError
from django.test import TestCase

from oscar.apps.dashboard.offers import forms
from oscar.apps.offer.custom import create_benefit, create_condition
from oscar.apps.offer.models import Benefit, Range
from oscar.test.factories import create_product
from tests._site.model_tests_app.models import CustomBenefitModel, CustomConditionModel


class TestBenefitForm(TestCase):
    def setUp(self):
        self.range = Range.objects.create(
            name="All products", includes_all_products=True
        )
        self.prod = create_product()

    def test_init_without_custom_benefit(self):
        """
        If no custom benefit exists, the type field should be required.
        """
        form = forms.BenefitForm()
        self.assertTrue(form.fields["type"].required)

    def test_init_with_custom_benefit(self):
        """
        If a custom benefit exists, the type field should not be required.
        """
        create_benefit(CustomBenefitModel)
        form = forms.BenefitForm()
        self.assertFalse(form.fields["type"].required)
        self.assertEqual(form.fields["custom_benefit"].initial, None)

    def test_init_with_custom_benefit_with_instance(self):
        """
        If a custom benefit exists and the kwargs instance is passed to init, the initial value for the custom_benefit
        should be the instance.
        """
        benefit = create_benefit(CustomBenefitModel)
        form = forms.BenefitForm(instance=benefit)
        self.assertFalse(form.fields["type"].required)
        self.assertEqual(form.fields["custom_benefit"].initial, benefit.id)

    def test_is_valid_no_data(self):
        """
        If not data is supplied, is_valid should evaluate to false
        """
        form = forms.BenefitForm()
        self.assertFalse(form.is_valid())

    def test_clean_no_value_data(self):
        """
        If data is supplied without any values, the form should evaluate to not valid +
        and the clean method should throw a ValidationError
        """
        form = forms.BenefitForm(
            data={"range": "", "type": "", "value": "", "custom_benefit": ""}
        )
        self.assertFalse(form.is_valid())
        self.assertRaises(ValidationError, form.clean)

    def test_clean_new_incentive(self):
        """
        If a range, type and value is supplied, the clean method should return the cleaned data without errors.
        """
        form = forms.BenefitForm(
            data={
                "range": self.range.id,
                "type": Benefit.FIXED,
                "value": 5,
                "custom_benefit": "",
            }
        )

        self.assertTrue(form.is_valid())
        self.assertEqual(
            {
                "range": self.range,
                "type": Benefit.FIXED,
                "value": D("5"),
                "custom_benefit": "",
                "max_affected_items": None,
            },
            form.clean(),
        )

    def test_clean_new_incentive_only_range(self):
        """
        If only a range is supplied, the clean method should throw a ValidationError.
        """
        form = forms.BenefitForm(
            data={"range": self.range.id, "type": "", "value": "", "custom_benefit": ""}
        )
        self.assertFalse(form.is_valid())
        self.assertRaises(ValidationError, form.clean)

    def test_clean_validation_with_custom_benefit(self):
        """
        If a custom benefit is selected, the form should be valid.
        """
        benefit = create_benefit(CustomBenefitModel)

        form = forms.BenefitForm(
            data={"range": "", "type": "", "value": "", "custom_benefit": benefit.id}
        )

        self.assertTrue(form.is_valid())
        self.assertEqual(
            {
                "range": None,
                "type": "",
                "value": None,
                "custom_benefit": str(benefit.id),
                "max_affected_items": None,
            },
            form.clean(),
        )

    def test_clean_only_range_custom_exists(self):
        """
        If a custom benefit exists, the type field is not required. Still, the clean method should throw a
        ValidationError, if only the range is supplied.
        """
        create_benefit(CustomBenefitModel)

        form = forms.BenefitForm(
            data={"range": self.range, "type": "", "value": "", "custom_benefit": ""}
        )

        self.assertFalse(form.is_valid())
        self.assertRaises(ValidationError, form.clean)

    def test_clean_validation_custom_exists(self):
        """
        If a custom benefit exists, and the data for range, type and value is supplied, the form should validate.
        Clean should return the cleaned data.
        """

        create_benefit(CustomBenefitModel)

        form = forms.BenefitForm(
            data={
                "range": self.range.id,
                "type": Benefit.FIXED,
                "value": 5,
                "custom_benefit": "",
            }
        )

        self.assertTrue(form.is_valid())
        self.assertEqual(
            {
                "range": self.range,
                "type": Benefit.FIXED,
                "value": D("5"),
                "custom_benefit": "",
                "max_affected_items": None,
            },
            form.clean(),
        )


class TestConditionForm(TestCase):
    def setUp(self):
        self.range = Range.objects.create(
            name="All products", includes_all_products=True
        )

    def test_clean_all_data(self):
        """
        If a custom condition exists, and the data for range, type, value is supplied,
        the form should be valid.
        """
        create_condition(CustomConditionModel)
        form = forms.ConditionForm(
            data={
                "range": self.range.id,
                "type": "Count",
                "value": 1,
                "custom_condition": "",
            }
        )
        self.assertTrue(form.is_valid())

    def test_clean_no_value(self):
        """
        If a custom condition exists, and the data for one of range, type, value is supplied,
        the form should be invalid.
        """
        create_condition(CustomConditionModel)
        test_data = [("range", self.range), ("type", "Count"), ("value", 1)]
        for field, value in test_data:
            data = {"range": "", "type": "", "value": "", "custom_condition": ""}
            data[field] = value
            form = forms.ConditionForm(data=data)
            self.assertFalse(form.is_valid())

    def test_clean_custom_condition(self):
        """
        If a custom condition is selected, the form should be valid.
        """
        custom_condition = create_condition(CustomConditionModel)
        form = forms.ConditionForm(
            data={
                "range": "",
                "type": "",
                "value": "",
                "custom_condition": custom_condition.id,
            }
        )
        self.assertTrue(form.is_valid())
        self.assertEqual(
            {
                "range": None,
                "type": "",
                "value": None,
                "custom_condition": str(custom_condition.id),
            },
            form.clean(),
        )

    def test_clean_custom_condition_with_range_type_and_value(self):
        """
        If a custom condition is selected, but a range, type and value is selected as well,
        it should throw a ValidationError as you may only have a custom condition.
        """
        custom_condition = create_condition(CustomConditionModel)
        form = forms.ConditionForm(
            data={
                "range": self.range.id,
                "type": "Count",
                "value": "5",
                "custom_condition": custom_condition.id,
            }
        )
        self.assertFalse(form.is_valid())
        self.assertRaises(ValidationError, form.clean)