#    Copyright (C) 2017  Alban Gruin
#
#    celcatsanitizer is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as published
#    by the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    celcatsanitizer is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with celcatsanitizer.  If not, see <http://www.gnu.org/licenses/>.

from django.test import TestCase
from .models import Course, Group, Timetable, Year
from .utils import tz_now


class CourseTestCase(TestCase):
    def setUp(self):
        dt = tz_now()

        self.year = Year(name="L2", slug="l2")
        self.year.save()

        self.timetable = Timetable(year=self.year, name="Test timetable 2", url="http://example.org/", slug="test-timetable2")
        self.timetable.save()

        cma = Group.objects.create(celcat_name="L1 info s2 CMA", timetable=self.timetable)
        tda2 = Group.objects.create(celcat_name="L1 info s2 TDA2", timetable=self.timetable)
        self.tpa21 = Group.objects.create(celcat_name="L1 info s2 TPA21", timetable=self.timetable)

        cmb = Group.objects.create(celcat_name="L1 info s2 CMB", timetable=self.timetable)
        tdb2 = Group.objects.create(celcat_name="L1 info s2 TDB2", timetable=self.timetable)
        self.tpb21 = Group.objects.create(celcat_name="L1 info s2 TPB21", timetable=self.timetable)

        for group in (cma, tda2, self.tpa21, cmb, tdb2, self.tpb21,):
            course = Course.objects.create(name="{0} course".format(group.name), type="cours", timetable=self.timetable, begin=dt, end=dt)
            course.groups.add(group)

    def test_get_courses_for_group(self):
        tpa21_courses = Course.objects.get_courses_for_group(self.tpa21)
        tpb21_courses = Course.objects.get_courses_for_group(self.tpb21)

        tpa21_course_names = ["L1 info s2 CMA course", "L1 info s2 TDA2 course", "L1 info s2 TPA21 course"]
        tpb21_course_names = ["L1 info s2 CMB course", "L1 info s2 TDB2 course", "L1 info s2 TPB21 course"]

        for courses, names in ((tpa21_courses, tpa21_course_names,), (tpb21_courses, tpb21_course_names,),):
            for course in courses:
                self.assertIn(course.name, names)
                names.remove(course.name)


class GroupTestCase(TestCase):
    def setUp(self):
        self.year = Year(name="L1", slug="l1")
        self.year.save()

        self.timetable = Timetable(year=self.year, name="Test timetable", url="http://example.com/", slug="test-timetable")
        self.timetable.save()

        Group.objects.create(celcat_name="L1 info s2 CMA", timetable=self.timetable)
        Group.objects.create(celcat_name="L1 info s2 TDA2", timetable=self.timetable)
        Group.objects.create(celcat_name="L1 info s2 TPA21", timetable=self.timetable)

        Group.objects.create(celcat_name="L1 info s2 CMB", timetable=self.timetable)
        Group.objects.create(celcat_name="L1 info s2 TDB2", timetable=self.timetable)
        Group.objects.create(celcat_name="L1 info s2 TPB21", timetable=self.timetable)

        Group.objects.create(celcat_name="L1 info (toutes sections et semestres confondus)", timetable=self.timetable)

    def test_corresponds(self):
        cma = Group.objects.get(celcat_name="L1 info s2 CMA", timetable=self.timetable)
        tda2 = Group.objects.get(celcat_name="L1 info s2 TDA2", timetable=self.timetable)
        tpa21 = Group.objects.get(celcat_name="L1 info s2 TPA21", timetable=self.timetable)

        cmb = Group.objects.get(celcat_name="L1 info s2 CMB", timetable=self.timetable)
        tdb2 = Group.objects.get(celcat_name="L1 info s2 TDB2", timetable=self.timetable)
        tpb21 = Group.objects.get(celcat_name="L1 info s2 TPB21", timetable=self.timetable)

        general = Group.objects.get(celcat_name="L1 info (toutes sections et semestres confondus)", timetable=self.timetable)

        self.assertFalse(cma.corresponds_to(*tda2.group_info))
        self.assertFalse(cma.corresponds_to(*tpa21.group_info))
        self.assertFalse(tda2.corresponds_to(*tpa21.group_info))

        self.assertFalse(cmb.corresponds_to(*tdb2.group_info))
        self.assertFalse(cmb.corresponds_to(*tpb21.group_info))
        self.assertFalse(tdb2.corresponds_to(*tpb21.group_info))

        self.assertFalse(cmb.corresponds_to(*tda2.group_info))
        self.assertFalse(cmb.corresponds_to(*tpa21.group_info))
        self.assertFalse(tdb2.corresponds_to(*tpa21.group_info))

        self.assertTrue(tda2.corresponds_to(*cma.group_info))
        self.assertTrue(tpa21.corresponds_to(*cma.group_info))
        self.assertTrue(tpa21.corresponds_to(*tda2.group_info))

        self.assertTrue(tdb2.corresponds_to(*cmb.group_info))
        self.assertTrue(tpb21.corresponds_to(*cmb.group_info))
        self.assertTrue(tpb21.corresponds_to(*tdb2.group_info))

        self.assertFalse(tda2.corresponds_to(*cmb.group_info))
        self.assertFalse(tpa21.corresponds_to(*cmb.group_info))
        self.assertFalse(tpa21.corresponds_to(*tdb2.group_info))

        self.assertFalse(general.corresponds_to(*cma.group_info))
        self.assertFalse(general.corresponds_to(*cmb.group_info))
        self.assertFalse(general.corresponds_to(*tda2.group_info))
        self.assertFalse(general.corresponds_to(*tdb2.group_info))
        self.assertFalse(general.corresponds_to(*tpa21.group_info))
        self.assertFalse(general.corresponds_to(*tpb21.group_info))

        self.assertTrue(cma.corresponds_to(*general.group_info))
        self.assertTrue(cmb.corresponds_to(*general.group_info))
        self.assertTrue(tda2.corresponds_to(*general.group_info))
        self.assertTrue(tdb2.corresponds_to(*general.group_info))
        self.assertTrue(tpa21.corresponds_to(*general.group_info))
        self.assertTrue(tpb21.corresponds_to(*general.group_info))

    def test_get(self):
        cma = Group.objects.get(name="L1 info s2 CMA", timetable=self.timetable)
        tda2 = Group.objects.get(name="L1 info s2 TDA2", timetable=self.timetable)
        tpa21 = Group.objects.get(name="L1 info s2 TPA21", timetable=self.timetable)

        cmb = Group.objects.get(name="L1 info s2 CMB", timetable=self.timetable)
        tdb2 = Group.objects.get(name="L1 info s2 TDB2", timetable=self.timetable)
        tpb21 = Group.objects.get(name="L1 info s2 TPB21", timetable=self.timetable)

        general = Group.objects.get(celcat_name="L1 info (toutes sections et semestres confondus)", timetable=self.timetable)

        self.assertEqual(cma.celcat_name, "L1 info s2 CMA")
        self.assertEqual(tda2.celcat_name, "L1 info s2 TDA2")
        self.assertEqual(tpa21.celcat_name, "L1 info s2 TPA21")

        self.assertEqual(cmb.celcat_name, "L1 info s2 CMB")
        self.assertEqual(tdb2.celcat_name, "L1 info s2 TDB2")
        self.assertEqual(tpb21.celcat_name, "L1 info s2 TPB21")

        self.assertEqual(general.celcat_name, "L1 info (toutes sections et semestres confondus)")

    def test_parse(self):
        cma = Group.objects.get(celcat_name="L1 info s2 CMA", timetable=self.timetable)
        tda2 = Group.objects.get(celcat_name="L1 info s2 TDA2", timetable=self.timetable)
        tpa21 = Group.objects.get(celcat_name="L1 info s2 TPA21", timetable=self.timetable)

        cmb = Group.objects.get(celcat_name="L1 info s2 CMB", timetable=self.timetable)
        tdb2 = Group.objects.get(celcat_name="L1 info s2 TDB2", timetable=self.timetable)
        tpb21 = Group.objects.get(celcat_name="L1 info s2 TPB21", timetable=self.timetable)

        general = Group.objects.get(celcat_name="L1 info (toutes sections et semestres confondus)", timetable=self.timetable)

        self.assertEqual(cma.group_info, ("L1 info", 2, "A"))
        self.assertEqual(tda2.group_info, ("L1 info", 2, "A2"))
        self.assertEqual(tpa21.group_info, ("L1 info", 2, "A21"))

        self.assertEqual(cmb.group_info, ("L1 info", 2, "B"))
        self.assertEqual(tdb2.group_info, ("L1 info", 2, "B2"))
        self.assertEqual(tpb21.group_info, ("L1 info", 2, "B21"))

        self.assertEqual(general.group_info, ("L1 info", None, ""))