Files
website-v2/conftest.py
2022-05-29 14:52:50 -05:00

23 lines
473 B
Python

import pytest
import tempfile
from PIL import Image
from django.core.files import File as DjangoFile
pytest_plugins = [
"users.tests.fixtures",
"versions.tests.fixtures",
]
@pytest.fixture
def temp_image_file():
image = Image.new("RGB", (100, 100))
tmp_file = tempfile.NamedTemporaryFile(suffix=".jpg")
image.save(tmp_file)
tmp_file.seek(0)
file_obj = DjangoFile(open(tmp_file.name, mode="rb"), name="tmp_file")
yield file_obj.seek(0)