from django.conf import settings
from django.core.management import call_command
from django.test import TestCase
from django.urls import reverse
from model_bakery import baker
from model_bakery.recipe import Recipe

from core.baker_recipes import subscriber_user
from core.models import Group, User
from pedagogy.models import UE


class TestUESearch(TestCase):
    """Test UE guide rights for view and API."""

    @classmethod
    def setUpTestData(cls):
        cls.root = baker.make(User, is_superuser=True)
        cls.url = reverse("api:fetch_ues")
        ue_recipe = Recipe(UE, author=cls.root)
        ues = [
            ue_recipe.prepare(
                code="AP4A",
                credit_type="CS",
                semester="AUTUMN",
                department="GI",
                manager="francky",
                title=(
                    "Programmation Orientée Objet: "
                    "Concepts fondamentaux et mise en pratique avec le langage C++"
                ),
            ),
            ue_recipe.prepare(
                code="MT01",
                credit_type="CS",
                semester="AUTUMN",
                department="TC",
                manager="ben",
                title="Intégration1. Algèbre linéaire - Fonctions de deux variables",
            ),
            ue_recipe.prepare(
                code="PHYS11", credit_type="CS", semester="AUTUMN", department="TC"
            ),
            ue_recipe.prepare(
                code="TNEV",
                credit_type="TM",
                semester="SPRING",
                department="TC",
                manager="moss",
                title="tnetennba",
            ),
            ue_recipe.prepare(
                code="MT10", credit_type="TM", semester="AUTUMN", department="IMSI"
            ),
            ue_recipe.prepare(
                code="DA50",
                credit_type="TM",
                semester="AUTUMN_AND_SPRING",
                department="GI",
                manager="francky",
            ),
        ]
        UE.objects.bulk_create(ues)
        call_command("update_index")

    def test_permissions(self):
        # Test with anonymous user
        response = self.client.get(self.url)
        assert response.status_code == 401

        # Test with not subscribed user
        self.client.force_login(baker.make(User))
        response = self.client.get(self.url)
        assert response.status_code == 403

        for user in (
            self.root,
            subscriber_user.make(),
            baker.make(
                User,
                groups=[Group.objects.get(pk=settings.SITH_GROUP_PEDAGOGY_ADMIN_ID)],
            ),
        ):
            # users that have right
            with self.subTest():
                self.client.force_login(user)
                response = self.client.get(self.url)
                assert response.status_code == 200

    def test_format(self):
        """Test that the return data format is correct"""
        self.client.force_login(self.root)
        res = self.client.get(self.url, query_params={"search": "PA00"})
        ue = UE.objects.get(code="PA00")
        assert res.status_code == 200
        assert res.json() == {
            "count": 1,
            "next": None,
            "previous": None,
            "results": [
                {
                    "id": ue.id,
                    "title": ue.title,
                    "code": ue.code,
                    "credit_type": ue.credit_type,
                    "semester": ue.semester,
                    "department": ue.department,
                    "detail_url": reverse(
                        "pedagogy:ue_detail", kwargs={"ue_id": ue.id}
                    ),
                    "edit_url": reverse("pedagogy:ue_update", kwargs={"ue_id": ue.id}),
                    "delete_url": reverse(
                        "pedagogy:ue_delete", kwargs={"ue_id": ue.id}
                    ),
                }
            ],
        }

    def test_search_by_text(self):
        self.client.force_login(self.root)
        for query, expected in (
            # UE code search case-insensitive
            ("m", {"MT01", "MT10"}),
            ("M", {"MT01", "MT10"}),
            ("mt", {"MT01", "MT10"}),
            ("MT", {"MT01", "MT10"}),
            ("algèbre", {"MT01"}),  # Title search case-insensitive
            # Manager search
            ("moss", {"TNEV"}),
            ("francky", {"DA50", "AP4A"}),
        ):
            res = self.client.get(self.url, query_params={"search": query})
            assert res.status_code == 200
            assert {ue["code"] for ue in res.json()["results"]} == expected

    def test_search_by_credit_type(self):
        self.client.force_login(self.root)
        res = self.client.get(self.url, query_params={"credit_type": "CS"})
        assert res.status_code == 200
        codes = [ue["code"] for ue in res.json()["results"]]
        assert codes == ["AP4A", "MT01", "PHYS11"]
        res = self.client.get(self.url, query_params={"credit_type": ["CS", "OM"]})
        assert res.status_code == 200
        codes = {ue["code"] for ue in res.json()["results"]}
        assert codes == {"AP4A", "MT01", "PHYS11", "PA00"}

    def test_search_by_semester(self):
        self.client.force_login(self.root)
        res = self.client.get(self.url, query_params={"semester": "SPRING"})
        assert res.status_code == 200
        codes = {ue["code"] for ue in res.json()["results"]}
        assert codes == {"DA50", "TNEV", "PA00"}

    def test_search_multiple_filters(self):
        self.client.force_login(self.root)
        res = self.client.get(
            self.url,
            query_params={
                "semester": "AUTUMN",
                "credit_type": "CS",
                "department": "TC",
            },
        )
        assert res.status_code == 200
        codes = {ue["code"] for ue in res.json()["results"]}
        assert codes == {"MT01", "PHYS11"}

    def test_search_fails(self):
        self.client.force_login(self.root)
        res = self.client.get(
            self.url, query_params={"search": "DA", "credit_type": "CS"}
        )
        assert res.status_code == 200
        assert res.json()["results"] == []

    def test_search_closed_status(self):
        # Simple heuristic : take a UE, and mark everything before it as closed
        closed_ue = UE.objects.order_by("id")[2]
        UE.objects.filter(id__lte=closed_ue.id).update(semester="CLOSED")

        self.client.force_login(self.root)

        res = self.client.get(self.url, query_params={"is_open": False})
        assert res.status_code == 200
        ids = {ue["id"] for ue in res.json()["results"]}
        assert ids == set(
            UE.objects.filter(semester="CLOSED").values_list("id", flat=True)
        )

        res = self.client.get(self.url, query_params={"is_open": True})
        assert res.status_code == 200
        ids = {ue["id"] for ue in res.json()["results"]}
        assert ids == set(
            UE.objects.exclude(semester="CLOSED").values_list("id", flat=True)
        )