为模型测试伪造语言环境字段

Question

我只是在这样的模型中有一组语言：

from django.conf.global_settings import LANGUAGES

class AvailableLanguage(models.Model):
    code = models.CharField(choices=LANGUAGES, unique=True)

在 factories.py 中，我想伪造 code 字段。我试图从 LANGUAGES 随机获取语言环境代码，但失败了。它没有创建 AvailableLanguage 的一个实例，而是遍历各种语言并迅速引发问题，因为 code 设置为 unique.

所以我想创建一个Provider作为单例，但同样的原因失败了。它会生成所有区域设置代码，而不是只提供一个 !


class Provider(BaseProvider):

    lang_code = NotImplemented

    def __init__(self, generator):
        self.langs = LANGUAGES
        self.generator = generator

    def __getattribute__(self, attrib):
        if attrib == 'lang_code':
            self.lang_code = self.langs.pop()[0]
            return self.lang_code
        else:
            return super().__getattribute__(attrib)


fake.add_provider(Provider)


class AvailableLanguageFactory(DjangoModelFactory):

    code = Faker('lang_code')

    class Meta:
        model = AvailableLanguage

Answer 1

你可以尝试使用Iterator 来自工厂男孩。

from django.conf.global_settings import LANGUAGES
import factory


class AvailableLanguageFactory(factory.DjangoModelFactory):
    code = factory.Iterator(LANGUAGES, getter=lambda l: l[0])

    class Meta:
        model = AvailableLanguage

为模型测试伪造语言环境字段

Fake a locale field for model testing

python

testing

django

faker