25 lines
857 B
Python
25 lines
857 B
Python
from django.test import TestCase
|
|
from django.core.files.uploadedfile import SimpleUploadedFile
|
|
from .forms import DetectForm
|
|
from django.conf import settings
|
|
|
|
|
|
# Test sprawdza czy wprowadzone pliki są w odpowiednim formacie
|
|
class CanUploadFiles(TestCase):
|
|
def test_upload_img_file(self):
|
|
file = SimpleUploadedFile(
|
|
name="jpg_file.jpg",
|
|
content=open(f"{settings.MEDIA_ROOT}/test_image.jpg", "rb").read(),
|
|
content_type="image/jpg",
|
|
)
|
|
form = DetectForm(files={"image": file})
|
|
print(form.errors)
|
|
self.assertTrue(form.is_valid())
|
|
|
|
def test_upload_other_file(self):
|
|
file = SimpleUploadedFile(
|
|
"txt_file.txt", content=b"some text", content_type="text/plain"
|
|
)
|
|
form = DetectForm(files={"image": file})
|
|
self.assertFalse(form.is_valid())
|