Question

有两种型号：

class Model1(models.Model):
    attribute = models.CharField(max_length=100)

class Model2(models.Model):
    target = models.ForeignKey(Model1)

单位测试tests.py：

class Model1Test(TestCase):
    def test_make_new_model_and_save():
        m = Model1()
        m.attribute = 'test values'
        m.save()

class Model2Test(TestCase):
    def test_make_new_model_and_save():
        # Make new Model1 to FK to
        m = Model1()
        m.attribute = 'test values'
        m.save()   

        m2 = Model2()
        m2.target = m
        m2.save()

我想调用Model1测试类中定义的unittest函数，而不是再次复制代码以测试Model2。我想这就像：

class Model1Test(TestCase):
    def test_make_new_model_and_save():
        m = Model1()
        m.attribute = 'test values'
        m.save()
        return m

class Model2Test(TestCase):
    def test_make_new_model_and_save():
        # Make new Model1 to FK to
        m = Model1Test('test_make_new_model_and_save')

        m2 = Model2()
        m2.target = m
        m2.save()

但这似乎不起作用，我立即发现自己处于单元测试框架的内部。我找不到很多关于这类事情的参考资料，所以对于单元测试来说，我觉得自己在某处出现了错误的转向。

是否存在重复使用单元测试的“正常”方法，或者我的方法存在缺陷？

Answer 1

这实际上只是关于Python中对象和类的问题。答案是定义一个包含所需初始化代码的公共超类，从中继承Model1Test和Model2Test，并从两者中调用它。

class CommonSetUp(TestCase):
    def make_new_model_and_save():
        m = Model1()
        m.attribute = 'test values'
        m.save()
        return m

class Model2Test(CommonSetUp):
    def test_make_new_model_and_save():
        self.make_new_model_and_save()
        self.assertTrue(...your assertions here...)

class Model2Test(CommonSetUp):
    def test_make_new_model_and_save():
        self.make_new_model_and_save()

        m2 = Model2()
        m2.target = m
        m2.save()

        self.assertTrue(...your assertions here...)

Answer 2

我不建议在另一种测试方法中调用测试方法它变得一团糟，经过一段时间的测试变得不稳定，难以修复或维护。

Django：在其他单元测试中重用单元测试

2 个答案: