Coverage for tests / test_datastructures.py: 100%

1import io 4 ctx1efgh

2from pathlib import Path 4 ctx1efgh

3 

4import pytest 4 ctx1efgh

5from fastapi import FastAPI, UploadFile 4 ctx1efgh

6from fastapi.datastructures import Default 4 ctx1efgh

7from fastapi.testclient import TestClient 4 ctx1efgh

8 

9 

10def test_upload_file_invalid_pydantic_v2(): 4 ctx1efgh

11 with pytest.raises(ValueError): 4 ctx1yzAB

12 UploadFile._validate("not a Starlette UploadFile", {}) 4 ctx1yzAB

13 

14 

15def test_default_placeholder_equals(): 4 ctx1efgh

16 placeholder_1 = Default("a") 4 ctx1qrst

17 placeholder_2 = Default("a") 4 ctx1qrst

18 assert placeholder_1 == placeholder_2 4 ctx1qrst

19 assert placeholder_1.value == placeholder_2.value 4 ctx1qrst

20 

21 

22def test_default_placeholder_bool(): 4 ctx1efgh

23 placeholder_a = Default("a") 4 ctx1uvwx

24 placeholder_b = Default("") 4 ctx1uvwx

25 assert placeholder_a 4 ctx1uvwx

26 assert not placeholder_b 4 ctx1uvwx

27 

28 

29def test_upload_file_is_closed(tmp_path: Path): 4 ctx1efgh

30 path = tmp_path / "test.txt" 4 ctx1abcd

31 path.write_bytes(b"<file content>") 4 ctx1abcd

32 app = FastAPI() 4 ctx1abcd

33 

34 testing_file_store: list[UploadFile] = [] 4 ctx1abcd

35 

36 @app.post("/uploadfile/") 4 ctx1abcd

37 def create_upload_file(file: UploadFile): 4 ctx1abcd

38 testing_file_store.append(file) 4 ctx1abcd

39 return {"filename": file.filename} 4 ctx1abcd

40 

41 client = TestClient(app) 4 ctx1abcd

42 with path.open("rb") as file: 4 ctx1abcd

43 response = client.post("/uploadfile/", files={"file": file}) 4 ctx1abcd

44 assert response.status_code == 200, response.text 4 ctx1abcd

45 assert response.json() == {"filename": "test.txt"} 4 ctx1abcd

46 

47 assert testing_file_store 4 ctx1abcd

48 assert testing_file_store[0].file.closed 4 ctx1abcd

49 

50 

51# For UploadFile coverage, segments copied from Starlette tests 

52 

53 

54@pytest.mark.anyio 4 ctx1efgh

55async def test_upload_file(): 4 ctx1efgh

56 stream = io.BytesIO(b"data") 8 ctx1ijklmnop

57 file = UploadFile(filename="file", file=stream, size=4) 8 ctx1ijklmnop

58 assert await file.read() == b"data" 8 ctx1ijklmnop

59 assert file.size == 4 8 ctx1ijklmnop

60 await file.write(b" and more data!") 8 ctx1ijklmnop

61 assert await file.read() == b"" 8 ctx1ijklmnop

62 assert file.size == 19 8 ctx1ijklmnop

63 await file.seek(0) 8 ctx1ijklmnop

64 assert await file.read() == b"data and more data!" 8 ctx1ijklmnop

65 await file.close() 8 ctx1ijklmnop