Sith/pedagogy/tests/test_api.py

import json

from django.conf import settings
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 RealGroup, User
from pedagogy.models import UV


class UVSearchTest(TestCase):
    """Test UV guide rights for view and API."""

    @classmethod
    def setUpTestData(cls):
        cls.root = User.objects.get(username="root")
        cls.url = reverse("api:fetch_uvs")
        uv_recipe = Recipe(UV, author=cls.root)
        uvs = [
            uv_recipe.prepare(
                code="AP4A", credit_type="CS", semester="AUTUMN", department="GI"
            ),
            uv_recipe.prepare(
                code="MT01", credit_type="CS", semester="AUTUMN", department="TC"
            ),
            uv_recipe.prepare(
                code="PHYS11", credit_type="CS", semester="AUTUMN", department="TC"
            ),
            uv_recipe.prepare(
                code="TNEV", credit_type="TM", semester="SPRING", department="TC"
            ),
            uv_recipe.prepare(
                code="MT10", credit_type="TM", semester="AUTUMN", department="IMSI"
            ),
            uv_recipe.prepare(
                code="DA50",
                credit_type="TM",
                semester="AUTUMN_AND_SPRING",
                department="GI",
            ),
        ]
        UV.objects.bulk_create(uvs)

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

        # 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=[
                    RealGroup.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 + "?search=PA00")
        uv = UV.objects.get(code="PA00")
        assert res.status_code == 200
        assert json.loads(res.content) == {
            "count": 1,
            "next": None,
            "previous": None,
            "results": [
                {
                    "id": uv.id,
                    "title": uv.title,
                    "code": uv.code,
                    "credit_type": uv.credit_type,
                    "semester": uv.semester,
                    "department": uv.department,
                }
            ],
        }

    def test_search_by_code(self):
        self.client.force_login(self.root)
        res = self.client.get(self.url + "?search=MT")
        assert res.status_code == 200
        assert {uv["code"] for uv in json.loads(res.content)["results"]} == {
            "MT01",
            "MT10",
        }

    def test_search_by_credit_type(self):
        self.client.force_login(self.root)
        res = self.client.get(self.url + "?credit_type=CS")
        assert res.status_code == 200
        codes = [uv["code"] for uv in json.loads(res.content)["results"]]
        assert codes == ["AP4A", "MT01", "PHYS11"]
        res = self.client.get(self.url + "?credit_type=CS&credit_type=OM")
        assert res.status_code == 200
        codes = {uv["code"] for uv in json.loads(res.content)["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 + "?semester=SPRING")
        assert res.status_code == 200
        codes = {uv["code"] for uv in json.loads(res.content)["results"]}
        assert codes == {"DA50", "TNEV", "PA00"}

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

    def test_search_fails(self):
        self.client.force_login(self.root)
        res = self.client.get(self.url + "?credit_type=CS&search=DA")
        assert res.status_code == 200
        assert json.loads(res.content)["results"] == []

    def test_search_pa00_fail(self):
        self.client.force_login(self.root)
        # Search with UV code
        response = self.client.get(reverse("pedagogy:guide"), {"search": "IFC"})
        self.assertNotContains(response, text="PA00")

        # Search with first letter of UV code
        response = self.client.get(reverse("pedagogy:guide"), {"search": "I"})
        self.assertNotContains(response, text="PA00")

        # Search with UV manager
        response = self.client.get(reverse("pedagogy:guide"), {"search": "GILLES"})
        self.assertNotContains(response, text="PA00")

        # Search with department
        response = self.client.get(reverse("pedagogy:guide"), {"department": "TC"})
        self.assertNotContains(response, text="PA00")

        # Search with semester
        response = self.client.get(reverse("pedagogy:guide"), {"semester": "CLOSED"})
        self.assertNotContains(response, text="PA00")

        # Search with language
        response = self.client.get(reverse("pedagogy:guide"), {"language": "EN"})
        self.assertNotContains(response, text="PA00")

        # Search with credit type
        response = self.client.get(reverse("pedagogy:guide"), {"credit_type": "TM"})
        self.assertNotContains(response, text="PA00")