如何测试模型方法django

Question

我将测试写入我的应用程序，但我不知道如何测试此模型方法：

def has_recipe(self):
    if self.type in [101, 102, 103, 104, 201, 301, 302, 303, 304]:
        return True
    else:
        return False

当我写这样的测试时：

def test_has_recipe(self):
    book = Book(type in [101, 102, 103, 104, 201, 301, 302, 303, 304])
    self.assertTrue(book, [101, 102, 103, 104, 201, 301, 302, 303, 304])

测试通过，但没有覆盖。

Answer 1

您需要在测试中创建一本书，然后调用其has_recipe方法。

您可以尝试以下方式：

def test_has_recipe(self):
    book = Book(type=101)
    self.assertTrue(book.has_recipe())