def test_model_relation_course_missing(self):
course = Course.objects.create(name='test1')
lesson = Lesson() # Lesson without `course` set
with self.assertRaises(ValidationError):
lesson.full_clean()
lesson.save()
self.assertEqual(
Lesson.objects.filter(course__name='test2').count(), 0)
def test_model_relation(self):
course = Course.objects.create(name='test1')
lesson = Lesson(course=Course.objects.create(name='test2'),
subject='test1',
description='test1',
order=int('1')
)
lesson.full_clean() # `course` correctly set. This should pass
lesson.save()
self.assertEqual(
Lesson.objects.filter(course__name='test2').count(), 1)