繁体   English   中英

Django中带有ImageField的单元测试表单

[英]Unit test form with ImageField in django

我有ImageField模型。 我正在尝试将单元测试写入表单,但是它总是会引发错误。 有人可以告诉我我的错误在哪里吗?

从错误中我知道该表格无效。 print str(form.errors)返回我这个:

<ul class="errorlist"><li>image<ul class="errorlist"><li>This field is required.</li></ul></li></ul>

错误

Traceback (most recent call last):
  File "/home/nurzhan/CA/slider/tests.py", in test_form_valid
    self.assertTrue(form.is_valid())
AssertionError: False is not true

tests.py:

def test_form_valid(self):
        self.image_file = open(
            os.path.join(BASE_DIR, 'static/images/test.jpg'), "rb"
        )
        data = {
            'title': 'TEST',
        }
        image_data = {
            'image': self.file_valid
        }
        form = SlideForm(data=data, files=image_data)
        self.assertTrue(form.is_valid())

print data返回我:

{'image': <open file '/home/nurzhan/CA/static/images/test.jpg', mode 'rb' at 0x7efe24b47b70>}

我也尝试使用这个:

data = {
    'image': SimpleUploadedFile(
        self.image_file.name,
        self.image_file.read()
    )
}

在这种情况下, print data返回{'image': <SimpleUploadedFile: test.jpg (text/plain)>}

form.py

class ArticleForm(forms.ModelForm):
    class Meta:
    model = Article
    fields = ('image',)

def __init__(self, *args, **kwargs):
    super(SlideForm, self).__init__(*args, **kwargs)
    self.fields['image'].widget.attrs = {
        'accept': 'image/png, image/jpeg, image/gif',
        'id': 'image',
    }

models.py:

from django.db.models.signals import pre_save
from django.dispatch import receiver

class Article(models.Model):
    image = models.ImageField(
        upload_to='article/images/%Y/%m/%d/',
        blank=False,
    )


@receiver(pre_save, sender=Article)
def delete_old_article_image(sender, instance, *args, **kwargs):
    if instance.pk:
        article = article.objects.get(pk=instance.pk)
        if instance.image and article.image != instance.image:
            article.image.delete(False)

回答:

科伦特单元测试是:

  def test_form_valid(self):
        data = {
            'title': 'TEST',
        }
        files_data = {
            'image': SimpleUploadedFile(
                self.image_file.name,
                self.image_valid.read()
            )
        }
        form = ArticleForm(data=data, files=files_data)
        self.assertTrue(form.is_valid())

我查找了Django的源代码并发现了这一点: https : //github.com/django/django/blob/master/django/forms/forms.py#L74-L76

您可能需要将文件作为files=data传递。

祝好运!

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM