Django測試form.is_valid（）

Question

我的視圖在url中具有可選參數。 我傳遞此參數（ <int:jobNr> / None ），以形成在get_form_kwargs

class AddHoursView(LoginRequiredMixin, generic.CreateView):

    form_class = AddHoursForm

    def get_form_kwargs(self):
        # pass "jobNr" keyword argument from current url to form
        kwargs = super(AddHoursView, self).get_form_kwargs()
        kwargs[JOB_PARAM] = self.kwargs.get(JOB_PARAM)
        return kwargs

    def form_valid(self, form):
        self.object = form.save(commit = False)

        # If url/jobNR take jobNr from URL, if not take from form
        try:
            jobNr = self.kwargs[JOB_PARAM]
        except KeyError:
            jobNr = form.cleaned_data[JOB_WORKER].jobNr

        job = models.Job.objects.get(jobNr = jobNr)

        jobWorker = models.JobWorker.objects.get_or_create(job = job,
                                                user = self.request.user)
        self.object.jobWorker = jobWorker[0]
        self.object.save()
        return HttpResponseRedirect(reverse('worktime:myjobs'))

格式為__init__如果url /foo/jobNr/使用url參數創建self.jobNr 。 如果是url /foo/創建新字段。

class AddHoursForm(forms.ModelForm):

    def __init__(self, *args, **kwargs):   
        #Add jobworker field to from Worktime model if any jobNr pass in url
        #When in url will be parameter. Job foreignkey will be set by automat.
        self.jobNr = kwargs.pop(JOB_PARAM, None)
        super(AddHoursForm, self).__init__(*args, **kwargs)
        if not self.jobNr:
            self.fields[JOB_WORKER] = forms.ModelChoiceField(queryset=Job.objects.all())

    def clean(self):

        cleaned_data = super().clean()
        date = cleaned_data.get('date')

        if self.jobNr:
            jobDate = Job.objects.get(jobNr=self.jobNr).start

        elif not cleaned_data.get(JOB_WORKER).start: 
            raise forms.ValidationError("Dat work don't start yet")

        else:
            jobDate = cleaned_data.get(JOB_WORKER).start

        if date<jobDate:
            raise forms.ValidationError("Wrong date")

        return cleaned_data

現在我想為表單編碼測試。 這就是我嘗試過的

    def test_no_jobNr_in_url_and_no_date(self):

        job = Job.objects.create(jobNr = 1, street = 'a', city = 'a',
                            zip = 'a')

        form_data = {'description': 'description', 'hours': 20}
        form = AddHoursForm(form_data)
        # Override field jobWorker after __init__ form
        form.fields['jobWorker'] = job


        self.assertFalse(form.is_valid())

這就是我得到的AttributeError: 'Job' object has no attribute 'disabled'之后，我需要幫助。 我不知道了

Answer 1

您無法將job分配給表單字段（這就是form.fields['jobWorker']是）。 如果要分配初始值，則應執行以下操作：

form.fields['jobWorker'].initial = job

但是我認為您希望通過job作為所選模型，您應該只在數據中執行此操作：

form_data = {'description': 'description', 'hours': 20, 'jobWorker': job.pk}
form = AddHoursForm(form_data)
self.assertFalse(form.is_valid)