aboutsummaryrefslogtreecommitdiff
path: root/tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests.py')
-rw-r--r--tests.py12
1 files changed, 9 insertions, 3 deletions
diff --git a/tests.py b/tests.py
index bb2d97b..1d23620 100644
--- a/tests.py
+++ b/tests.py
@@ -16,14 +16,17 @@
from django.test import TestCase
from django.utils import timezone
-from .models import Course, Group, Timetable
+from .models import Course, Group, Timetable, Year
class CourseTestCase(TestCase):
def setUp(self):
dt = timezone.now()
- self.timetable = Timetable(name="Test timetable 2", url="http://example.org/", slug="test-timetable2")
+ 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)
@@ -53,7 +56,10 @@ class CourseTestCase(TestCase):
class GroupTestCase(TestCase):
def setUp(self):
- self.timetable = Timetable(name="Test timetable", url="http://example.com/", slug="test-timetable")
+ 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)