延迟评估Django模型的假设策略

时间:2017-08-09 17:12:55

标签: python django python-hypothesis

我有一个Django模型。除此之外,还有ForeignKeyUser

class MyModel(models.Model):
    foo = models.BooleanField()
    bar = models.ForeignKey(User)

我的测试中有一个方法可以生成一个形状合适的User - generate_test_user()

我想用Hypothesis编写一些测试来断言关于我的模型实例的各种属性。

我的第一个实现看起来像这样:

class MyTestCase(TestCase):

    @hypothesis.given(models(MyModel, bar=just(generate_test_user())))
    def test_my_model(self, mymodel):
        pass

但是这会失败,因为generate_test_user在导入时被调用,因此在Django迁移等运行之前尝试创建模型。

制定策略的好方法是,在正确的时间评估正确的事情/延迟评估just电话/类似事件?

2 个答案:

答案 0 :(得分:1)

听起来你需要draw the strategy interactively,而不是在@given()中指定它:

@given(data())
def test_my_model(self, data):
    mymodel = data.draw(models(MyModel, bar=just(generate_test_user())))
    # Run assertions on mymodel

答案 1 :(得分:1)

TL; DR我希望builds不是just

来自hypothesis.strategies

def just(value):
    """Return a strategy which only generates value.

    Note: value is not copied. Be wary of using mutable values.

    """

而(为简洁而编辑):

def builds(target, *args, **kwargs):
    """Generates values by drawing from ``args`` and ``kwargs`` and passing
    them to ``target`` in the appropriate argument position.

    e.g. ``builds(target, integers(), flag=booleans())`` would draw an
    integer ``i`` and a boolean ``b`` and call ``target(i, flag=b)``.

因此我原来的

@hypothesis.given(models(MyModel, bar=just(generate_test_user())))
def test_my_model(self, mymodel):
    pass

成了

@hypothesis.given(models(MyModel, bar=builds(generate_test_user)))
def test_my_model(self, mymodel):
    pass

并且一切都如我所愿,generate_test_user在每次试运行时恰好在恰当的时间召唤