import pytest
from django.core.files.uploadedfile import SimpleUploadedFile
from django.db import IntegrityError
from django.test import TestCase
from django.utils.datastructures import MultiValueDict

from alphabetlearning.resources.factories import ResourceModelFactory
from alphabetlearning.resources.forms import ResourceCreateForm
from alphabetlearning.resources.models import ResourceCategory
from alphabetlearning.resources.models import ResourceType


class ResourceCreateFormTest(TestCase):
    def setUp(self):
        self.resource_type = ResourceType.objects.create(name="Test Resource Type")
        self.resource_category = ResourceCategory.objects.create(name="Test Resource Category")
        self.form_data = {
            "name": "Test Resource",
            "description": "Test Description",
            "card_description": "Test Card Description",
            "resource_type": self.resource_type.id,
            "age_range": "Reception (4-5yrs)",
            "curriculum": "English",
            "feature_slot": 1,
            "main_resource_category": self.resource_category.id,
        }
        self.pdf_files = [
            # use the correct PDF file header - this should pass validation
            SimpleUploadedFile(
                f"file{i}.pdf",
                b"\x25\x50\x44\x46\x2d",
                content_type="application/pdf",
            )
            for i in range(11)
        ]
        self.thumbnail_files = [
            # use the correct JPG file header - this should pass validation
            SimpleUploadedFile(
                f"thumbnail{i}.jpg",
                b"\xff\xd8\xff\xdb",
                content_type="image/jpeg",
            )
            # 5 is the max number currently set in ALLOWED_THUMBNAILS
            for i in range(5)
        ]
        # This needs to be a MultiValueDict for the test to mimic production code
        # see clean_pdf_files - self.files.getlist()
        self.form_files = MultiValueDict(
            {"pdf_files": self.pdf_files, "thumbnail_files": self.thumbnail_files},
        )

    def test_form_valid(self):
        form = ResourceCreateForm(data=self.form_data, files=self.form_files)
        assert form.is_valid()

    @pytest.mark.django_db()
    def test_featured_slots_must_be_unique(self):
        r1 = ResourceModelFactory(feature_slot=1)
        with pytest.raises(IntegrityError):
            ResourceModelFactory(feature_slot=1)

    @pytest.mark.django_db()
    def test_featured_slots_allowable(self):
        form_data = {
            "name": "Test Resource",
            "description": "Test Description",
            "resource_type": self.resource_type.id,
            "age_range": "5-7",
            "curriculum": "English",
            "feature_slot": 4,
            "main_resource_category": self.resource_category.id,
        }
        form = ResourceCreateForm(data=form_data, files=self.form_files)
        assert not form.is_valid()