python - 单元测试 Django 查询集

Question

我正在尝试使用 Django/unittest 学习单元测试。

这些是我的模型的简单版本：

class Device(models.Model):
    name = models.CharField(max_length=100)

    def get_ips(self):
        return DeviceIP.objects.filter(device=self.id)


class DeviceIP(models.Model):
    ip = models.GenericIPAddressField()
    device = models.ForeignKey(Device)

这是我想出的测试代码：

from django.test import TestCase

class DeviceTest(TestCase):

    def test_get_ips(self):
        device = Device()
        device.name = 'My Device'

        ip1 = DeviceIP()
        ip1.ip = '127.0.0.1'
        ip1.device = device
        ip1.save()

        ip2 = DeviceIP()
        ip2.ip = '127.0.0.2'
        ip2.device = device
        ip2.save()

        ip3 = DeviceIP()
        ip3.ip = '127.0.0.3'
        ip3.device = device
        ip3.save()

        self.assertEqual(device.get_ips(), [ip1, ip2, ip3])

测试结果失败，因为AssertionError即使device.get_ips()和的字符串表示[ip1, ip2, ip3]相同。

如果我尝试使用self.assertListEqual我会得到一个错误，因为device.get_ips()它是一个 QuerySet 而不是一个列表。

如果我尝试self.assertQuerySetEqual我得到一个错误说“ DeviceTest object has no attribute assertQuerySetEqual”但我不知道为什么因为DeviceTestextendsdjango.test的 TestCase。

我应该如何做这样的测试？

此外，在“真实”项目中进行如此简单的测试是否有意义？

Answer 1

实际上，djangoproject 推荐的正确方法是：

    self.assertEqual(list(device.get_ips()), [ip1, ip2, ip3])

强制sorted查询集和列表会改变你的测试场景，你不想要它。

Answer 2

该调用device.get_ips()返回一个QuerySetwhile[ip1, ip2, ip3]是一个列表。因此，它们目前不相等。

鉴于您不想测试可能无关紧要的事情（.filter()从数据库返回的行的顺序），我建议进行如下测试：

results = device.get_ips()
result_ips = [ip.ip for ip in results]
self.assertEqual(len(results), 3)
self.assertTrue(ip1.ip in result_ips)
self.assertTrue(ip2.ip in result_ips)
self.assertTrue(ip3.ip in result_ips)

这个测试：三个结果和IP是一样的。这应该让您有理由相信您正在获得相同的对象（尽管您可以根据需要添加更多断言）。

Answer 3

你输入self.assertQuerySetEqual了，它应该是self.assertQuerysetEqual，试试：

self.assertQuerysetEqual(device.get_ips(), [repr(ip1), repr(ip2), repr(ip3)], 
                         ordered=False)

或者，如果您仍想针对列表进行测试：

self.assertItemsEqual(device.get_ips(), [ip1, ip2, ip3])