<reponame>fcortes/django-ninja<filename>tests/test_files.py
import pytest
from ninja import NinjaAPI, File, UploadedFile
from ninja.testing import TestClient
from django.core.files.uploadedfile import SimpleUploadedFile


api = NinjaAPI()


@api.post("/file")
def file_upload(request, file: UploadedFile = File(...)):
    return {"name": file.name, "data": file.read().decode()}


client = TestClient(api)


def test_files():
    response = client.post("/file")  # no file
    assert response.status_code == 422

    file = SimpleUploadedFile("test.txt", b"data123")
    response = client.post("/file", FILES={"file": file})
    assert response.status_code == 200
    assert response.json() == {"name": "test.txt", "data": "data123"}


def test_schema():
    schema = api.get_openapi_schema()
    method = schema["paths"]["/api/file"]["post"]
    assert method["requestBody"] == {
        "content": {
            "multipart/form-data": {
                "schema": {
                    "title": "FileParams",
                    "required": ["file"],
                    "type": "object",
                    "properties": {
                        "file": {"title": "File", "type": "string", "format": "binary"}
                    },
                }
            }
        },
        "required": True,
    }


def test_invalid_file():
    with pytest.raises(ValueError):
        UploadedFile._validate("not_a_file")
