Plankton_Detector/PlanktonDetector/DetectionApp/tests.py

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())