Unit-Tests eine Django-Form mit einem FileField

Ich habe eine form, wie:

#forms.py
from django import forms

class MyForm(forms.Form):
    title = forms.CharField()
    file = forms.FileField()


#tests.py
from django.test import TestCase
from forms import MyForm

class FormTestCase(TestCase)
    def test_form(self):
        upload_file = open('path/to/file', 'r')
        post_dict = {'title': 'Test Title'}
        file_dict = {} #??????
        form = MyForm(post_dict, file_dict)
        self.assertTrue(form.is_valid())

Wie gestalte ich die file_dict übergeben upload_file, um die form?

InformationsquelleAutor Jason Christa | 2010-03-18
  1. 79

    So weit ich gefunden habe, dieser Weg funktioniert

    from django.core.files.uploadedfile import SimpleUploadedFile
 ...
def test_form(self):
        upload_file = open('path/to/file', 'rb')
        post_dict = {'title': 'Test Title'}
        file_dict = {'file': SimpleUploadedFile(upload_file.name, upload_file.read())}
        form = MyForm(post_dict, file_dict)
        self.assertTrue(form.is_valid())
    • Je nachdem, was Ihr hochladen, der code funktioniert, müssen Sie möglicherweise um die content_type Attribut upload_file, zu
    • Wo content_type ist ein string, wie 'image/jpeg' für eine Bilddatei im jpg-Format zum Beispiel.
    • Ich bin immer TypeError: 'str' does not support the buffer interface.
    • Ich möchte sagen, dass dies getan werden sollte mit der with Konstrukt zu vermeiden, schließen Sie die upload_file ausdrücklich.
    InformationsquelleAutor Jason Christa
  2. 21

    Werden kann, das ist nicht ganz korrekt, aber ich bin der Erstellung von Bild-Datei im unit-test mit StringIO:

    imgfile = StringIO('GIF87a\x01\x00\x01\x00\x80\x01\x00\x00\x00\x00ccc,\x00'
                     '\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;')
imgfile.name = 'test_img_file.gif'

response = self.client.post(url, {'file': imgfile})
    • Ich bin versuchen, um zu testen das Formular separat aus dem Blick. Eigentlich habe ich nicht selbst gemacht, die anzeigen noch.
    • Ok, scheint, dass SimpleUploadedFile ist eine ganz klare Lösung für die Prüfung von
    • +1 für nicht den Umweg über eine externe Datei
    InformationsquelleAutor dragoon
  3. 5

    Hier ist eine andere Weise, die nicht erfordern, dass Sie ein Bild.

    EDIT: Update für Python 3.

    from PIL import Image
from io import BytesIO # Python 2: from StringIO import StringIO
from django.core.files.uploadedfile import InMemoryUploadedFile

...

def test_form(self):
    im = Image.new(mode='RGB', size=(200, 200)) # create a new image using PIL
    im_io = BytesIO() # a BytesIO object for saving image
    im.save(im_io, 'JPEG') # save the image to im_io
    im_io.seek(0) # seek to the beginning

    image = InMemoryUploadedFile(
        im_io, None, 'random-name.jpg', 'image/jpeg', len(im_io.getvalue()), None
    )

    post_dict = {'title': 'Test Title'}
    file_dict = {'picture': image}

    form = MyForm(data=post_dict, files=file_dict)
    • Derzeit existieren die folgenden, aber: TypeError: string argument expected, got 'bytes'. Ich löste es mit BytesIO statt StringIO, ohne im_io.len weil ein BytesIO Objekte havent Länge-Attribut.
    • Vielen Dank für das heads-up. Getestet habe ich den oben genannten code in Python 2.7 und es funktioniert gut. Könnten Sie mir bitte sagen, welche version von Python und Django verwenden Sie? Ich möchte korrigieren Sie den code oben für zukünftige Leser. Danke.
    • Ich bin mit django 1.9 und python 3.5.
    InformationsquelleAutor xyres
  4. 0

    Kombiniert ein paar Ideen und andere Beiträge Stack mit Python 3. Keine externe Datei oder Bibliotheken benötigt.

    from django.core.files.uploadedfile import SimpleUploadedFile

png_hex = ['\x89', 'P', 'N', 'G', '\r', '\n', '\x1a', '\n', '\x00',
           '\x00', '\x00', '\r', 'I', 'H', 'D', 'R', '\x00',
           '\x00', '\x00', '\x01', '\x00', '\x00', '\x00', '\x01',
           '\x08', '\x02', '\x00', '\x00', '\x00', '\x90',
           'w', 'S', '\xde', '\x00', '\x00', '\x00', '\x06', 'b', 'K',
           'G', 'D', '\x00', '\x00', '\x00', '\x00',
           '\x00', '\x00', '\xf9', 'C', '\xbb', '\x7f', '\x00', '\x00',
           '\x00', '\t', 'p', 'H', 'Y', 's', '\x00',
           '\x00', '\x0e', '\xc3', '\x00', '\x00', '\x0e', '\xc3',
           '\x01', '\xc7', 'o', '\xa8', 'd', '\x00', '\x00',
           '\x00', '\x07', 't', 'I', 'M', 'E', '\x07', '\xe0', '\x05',
           '\r', '\x08', '%', '/', '\xad', '+', 'Z',
           '\x89', '\x00', '\x00', '\x00', '\x0c', 'I', 'D', 'A', 'T',
           '\x08', '\xd7', 'c', '\xf8', '\xff', '\xff',
           '?', '\x00', '\x05', '\xfe', '\x02', '\xfe', '\xdc', '\xcc',
           'Y', '\xe7', '\x00', '\x00', '\x00', '\x00',
           'I', 'E', 'N', 'D', '\xae', 'B', '`', '\x82']

valid_png_bin = str.encode("".join(png_hex))
png = SimpleUploadedFile("test.png", valid_png_bin)

post_dict = {'title': 'Test Title'}
file_dict = {'picture': png}

form = MyForm(data=post_dict, files=file_dict)
    InformationsquelleAutor smooth-texan
Schreibe einen Kommentar