如何测试模型方法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())