Coverage for tests / test_datastructures.py: 100%

1import io 4 ctx1defg

2from pathlib import Path 4 ctx1defg

3 

4import pytest 4 ctx1defg

5from fastapi import FastAPI, UploadFile 4 ctx1defg

6from fastapi.datastructures import Default 4 ctx1defg

7from fastapi.testclient import TestClient 4 ctx1defg

8 

9 

10def test_upload_file_invalid_pydantic_v2(): 4 ctx1defg

11 with pytest.raises(ValueError): 3 ctx1qrs

12 UploadFile._validate("not a Starlette UploadFile", {}) 3 ctx1qrs

13 

14 

15def test_default_placeholder_equals(): 4 ctx1defg

16 placeholder_1 = Default("a") 3 ctx1klm

17 placeholder_2 = Default("a") 3 ctx1klm

18 assert placeholder_1 == placeholder_2 3 ctx1klm

19 assert placeholder_1.value == placeholder_2.value 3 ctx1klm

20 

21 

22def test_default_placeholder_bool(): 4 ctx1defg

23 placeholder_a = Default("a") 3 ctx1nop

24 placeholder_b = Default("") 3 ctx1nop

25 assert placeholder_a 3 ctx1nop

26 assert not placeholder_b 3 ctx1nop

27 

28 

29def test_upload_file_is_closed(tmp_path: Path): 4 ctx1defg

30 path = tmp_path / "test.txt" 3 ctx1abc

31 path.write_bytes(b"<file content>") 3 ctx1abc

32 app = FastAPI() 3 ctx1abc

33 

34 testing_file_store: list[UploadFile] = [] 3 ctx1abc

35 

36 @app.post("/uploadfile/") 3 ctx1abc

37 def create_upload_file(file: UploadFile): 3 ctx1abc

38 testing_file_store.append(file) 3 ctx1abc

39 return {"filename": file.filename} 3 ctx1abc

40 

41 client = TestClient(app) 3 ctx1abc

42 with path.open("rb") as file: 3 ctx1abc

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

44 assert response.status_code == 200, response.text 3 ctx1abc

45 assert response.json() == {"filename": "test.txt"} 3 ctx1abc

46 

47 assert testing_file_store 3 ctx1abc

48 assert testing_file_store[0].file.closed 3 ctx1abc

49 

50 

51# For UploadFile coverage, segments copied from Starlette tests 

52 

53 

54@pytest.mark.anyio 4 ctx1defg

55async def test_upload_file(): 4 ctx1defg

56 stream = io.BytesIO(b"data") 3 ctx1hij

57 file = UploadFile(filename="file", file=stream, size=4) 3 ctx1hij

58 assert await file.read() == b"data" 3 ctx1hij

59 assert file.size == 4 3 ctx1hij

60 await file.write(b" and more data!") 3 ctx1hij

61 assert await file.read() == b"" 3 ctx1hij

62 assert file.size == 19 3 ctx1hij

63 await file.seek(0) 3 ctx1hij

64 assert await file.read() == b"data and more data!" 3 ctx1hij

65 await file.close() 3 ctx1hij