Django使用ForeignKey进行单元测试

Question

我有一个包含一些ModelForm的ForeignKey，对User对象说，但是对其他任何模型也可以。我对此表单有一个单元测试类，但是当我尝试传递数据时，出现了Select a valid choice. That choice is not one of the available choices错误。测试看起来像这样:

class Monkey(Model):
    user = models.ForeignKey(User)
    ...

class MyForm(ModelForm):
    class Meta:
        model = Monkey
        fields = ['user', ...]

def test_my_form_with_a_user(self):
    ...
    data = {'user': User.objects.get(pk=1), ... } #  Nope.
    data = {'user': [u'1'], ... } #  Nope.
    data = {'user': [u'JaneDoe'], ... } #  Nope.
    form = MyForm(data, ...)
    self.assertTrue(form.is_valid(), form.errors)
    ...

我尝试了对 user进行任意数量的排列，但是却遇到了相同的错误。

我想念什么？

Answer 1

您应该可以使用以下方法为测试中的用户字段分配一个值:

def test_my_form_with_a_user(self):
    user_pk = User.objects.get(pk=1).pk
    data = {'user': user_pk}
    ...