Django Queryset單元測試

Question

我是新來的單元測試作爲一個整體，我想知道如何編寫以下函數的單元測試。

from poll.models import Poll 
def total(): 
    all_polls = Poll.objects.all() 
    return all_polls.count() 

理想情況下，我應該在上面的函數中測試什麼？

Answer 1

測試這個特殊功能沒有太大意義，因爲它沒有做任何非常有用的事情。但是，由於您將其用作示例，因此其原理是使用一些示例數據設置系統，然後檢查該函數是否會返回您所期望的內容。因此，在這種情況下，您的測試可以創建兩個Poll對象，然後斷言該函數返回值2.

您可能還想檢查邊緣情況：在這種情況下，我唯一能想到的就是它在沒有任何民意調查時不會崩潰。所以，在插入你的兩個民意調查之前，你可以調用函數並測試它返回0;然後添加兩個實例並檢查返回值2.

儘管如此，您應該避免測試Django自己的功能：這在Django自己的單元測試中已有很好的介紹。在像這樣的情況下，除了調用內置方法外，你的函數什麼也不做，因爲我們已經知道它將會返回兩個值，所以檢查count()函數是沒有意義的。