测试表单字段小部件类型

时间:2018-06-01 12:10:20

标签: django django-forms django-testing django-widget

我正在以Django模型形式编写测试:

class TemplateFieldTextForm(forms.ModelForm):
    class Meta:
        model = TemplateFieldText
        fields = []

    def __init__(self, *args, **kwargs):
        super(TemplateFieldTextForm, self).__init__(*args, **kwargs)
        field = self.instance.template_field
        label = field.label or var_to_proper(field.identifier)

        if field.input_type == 'text_area':
            self.fields['text'] = forms.CharField(label=label, required=field.required, widget=forms.Textarea)
        else:
            self.fields['text'] = forms.CharField(label=label, required=field.required)

我正在尝试测试text字段的正确小部件是否已设置,如下所示:

def test_submit_with_char_field(self):
    template_field = mommy.make(TemplateField, label='char_field')
    template_field_text = mommy.make(TemplateFieldText, template_field=template_field,
                                     award_process=self.award_process)
    form_data = {
        'text': 'some text'
    }
    form = TemplateFieldTextForm(form_data, instance=template_field_text)
    self.assertEqual(form.fields['text'].widget.__class__.__name__, 'TextInput')

但是,小部件似乎始终是Textarea小部件,而不是默认的TextInput。当我在实际渲染模板中检查表单时,表单确实按预期更改。关于我在测试中出错的地方有什么想法吗?

1 个答案:

答案 0 :(得分:0)

事实证明我设置的模型字段值不正确。

template_field = mommy.make(TemplateField, label='char_field')

应该是:

template_field = mommy.make(TemplateField, input_type='char_field')