如何使用FactoryBoy指定列表大小

时间:2015-07-31 08:47:20

标签: python python-2.7 testing mocking factory-boy

我们假设我有这个模型:

class FooContainerModel(object):
    def __init__(self, foos):
        self.foos = foos

我希望能够决定在创建时创建多少个foos,例如

model = FooContainerFactory(count=15)

尝试像

这样的工厂
class FooContainerFactory(factory.Factory):
    class Meta:
        model = FooContainerModel

    foos = factory.List([Foo() for _ in xrange(20)]) # fixed amount
    foos = factory.lazy_attribute(lambda o: [Foo() for _ in xrange(20)]) # same

当然我可以手动创建具有所需长度的Foo()列表并实例化FooContainerModel,但这不是我想要的。任何解决方案?

2 个答案:

答案 0 :(得分:1)

我不知道FactoryBoy是如何工作的,但是当我需要Django模型中的字段的动态属性时,这个策略通常适用于我:

def FooContainerFactory(factory.Factory):

    def __init__(self, count=20, *args, **kwargs):
        self.foos = factory.List([Foo() for _ in range(20)])
        super(FooContainerFactory, self).__init__(*args, **kwargs)

    class Meta:
        model = FooContainerModel

答案 1 :(得分:0)

您可以使用create_batch方法进行操作。

models = FooContainerFactory.create_batch(size=15)

如果要创建15个绝对不同的模型,则可能需要使用SequencesFaker(另外)