Метод тестирования, который возвращает атрибут модели отношения

Question

У меня есть следующие модели:

class Course(Base):

    title = models.CharField('Título', max_length=200)
    subcategory = models.ForeignKey(SubCategory, on_delete=models.CASCADE)

    def get_category(self):
        return self.subcategory.category


class SubCategory(Base):
    title = models.CharField('Título', max_length=80)
    category = models.ForeignKey(Category, on_delete=models.CASCADE)


class Category(Base):
    title = models.CharField('Título', max_length=80)

Как я могу проверить метод get_category?

Я пытаюсь это сделать, но тест не передает отчет о покрытии.

Answer 1

Как то так.

class CourseModelTest(TestCase):
    def test_get_category(self):
        category = CategoryFactory()
        subcategory = SubCategoryFactory(category=category)
        course = CourseFactory(subcategory=subcategory)

        self.assertEqual(course.get_category().id, category.id)