Django-Factoryboy:使用stringformating的__str__调用返回对象

时间:2019-11-15 17:15:35

标签: django factory-boy

在使用factory-boy生成模型实例的同时,我正在为Django应用创建许多测试。但是,我的测试失败了,但是我不确定为什么会这样。

    def test__str__(self) -> None:
        self.assertEqual(
            str(self.academy_settings),
            "Academy settings for {}".format(self.academy_settings.academy),
        )

如您所见,我正在一个django模型上验证__str__方法,但是当我运行此测试时,出现以下错误。

Failure
Expected :"<AcademySettingsFactory for <class 'core.models.AcademySettings'>>"
Actual   :'Academy settings for <factory.declarations.SubFactory object at 0x1105ce438>'

这些错误令我有些困惑。很明显,实际结果与我期望的结果完全不同,但是可以为期望的结果设置相同的结果。

正如您在下面的模型定义中所看到的,它们两个都应返回简单的字符串对象。

class Academy(models.Model):
    name = models.CharField(
        max_length=400, unique=True
    )

    def __str__(self) -> str:
        return "{}".format(self.name)


class AcademySettings(models.Model):

    academy = models.ForeignKey(Academy, on_delete=models.CASCADE)
    (...)

    def __str__(self) -> str:
        return "Academy settings for {}".format(self.academy)

现在,正如您在下面的工厂中看到的那样,我编写了一些自定义lazy_attribute来保持name字段唯一,但是即使没有该代码,错误也会不断出现。所以我不确定这里的真正原因是什么-有人知道我在做什么错吗?

class AcademyFactory(factory.DjangoModelFactory):
    class Meta:
        model = 'core.Academy'

    @factory.lazy_attribute
    def name(self):
        """
        The faker module has an habit of generating too many duplicate names
        every now and then. We solve this by appending the name with the
         instance.id
        """
        return '{} {}'.format(
            factory.Faker('company').generate(),
            str(factory.Sequence(lambda n: ' {}'.format(n))._creation_counter),
        )

class AcademySettingsFactory(factory.DjangoModelFactory):
    class Meta:
        model = 'core.AcademySettings'

    academy = factory.SubFactory(AcademyFactory)


1 个答案:

答案 0 :(得分:1)

很可能是在定义self.academy_settings = AcademySettingsFactory,因此引用了 class 而不是获取实例。

可以通过以下方法解决此问题:

def test__str__(self):
    academy_settings = AcademySettingsFactory()
    self.assertEqual(
        "Academy settings for {}".format(academy_settings.academy),
        str(academy_settings),
    )

(请注意:调用为self.assertEqual(expected, computed):将期望值放在第一位,然后将计算值放在最后)。