Mô hình thử nghiệm Django với ImageField

Question

Tôi cần kiểm tra mô hình Ảnh của ứng dụng Django của mình. Làm thế nào tôi có thể thử ImageField với một tập tin hình ảnh thử nghiệm?

tests.py

class PhotoTestCase(TestCase): 

    def test_add_photo(self): 
     newPhoto = Photo() 
     newPhoto.image = # ?????? 
     newPhoto.save() 
     self.assertEqual(Photo.objects.count(), 1) 

Answer 1

Đối với người dùng trong tương lai, tôi đã giải quyết được vấn đề. Bạn có thể giả lập một ImageField bằng một cá thể SimpleUploadedFile.

test.py

from django.core.files.uploadedfile import SimpleUploadedFile 

newPhoto.image = SimpleUploadedFile(name='test_image.jpg', content=open(image_path, 'rb').read(), content_type='image/jpeg') 

Answer 2

Cho thư viện mô hình để tạo ra một đối tượng giả dựa trên Django của lớp File

import mock 
from django.core.files import File 

file_mock = mock.MagicMock(spec=File, name='FileMock') 

và sau đó sử dụng trong các thử nghiệm của bạn

newPhoto.image = file_mock 

Answer 3

Một thay thế:

from django.core.files import File 

newPhoto.image = File(open(image_path)) 

Answer 4

Giải pháp:

from StringIO import StringIO 
# in python 3: from io import StringIO 
from PIL import Image 
from django.core.files.base import File 

Và tạo ra một phương pháp tĩnh trong lớp TestCase của bạn:

@staticmethod 
def get_image_file(name='test.png', ext='png', size=(50, 50), color=(256, 0, 0)): 
    file_obj = StringIO() 
    image = Image.new("RGBA", size=size, color=color) 
    image.save(file_obj, ext) 
    file_obj.seek(0) 
    return File(file_obj, name=name) 

Ví dụ:

instance = YourModel(name=value, image=self.get_image_file()) 

Answer 5

Bạn có thể sử dụng một tập tin tạm thời, sử dụng tempfile. Vì vậy, bạn không cần một tập tin thực sự để làm bài kiểm tra của bạn.

import tempfile 

image = tempfile.NamedTemporaryFile(suffix=".jpg").name 

Nếu bạn thích làm sạch thủ công, hãy sử dụng tempfile.mkstemp() thay thế.