Coverage for tests/test_datastructures.py: 100%

1import io 5 ctx1fghij

2from pathlib import Path 5 ctx1fghij

3from typing import List 5 ctx1fghij

4 

5import pytest 5 ctx1fghij

6from fastapi import FastAPI, UploadFile 5 ctx1fghij

7from fastapi.datastructures import Default 5 ctx1fghij

8from fastapi.testclient import TestClient 5 ctx1fghij

9 

10 

11# TODO: remove when deprecating Pydantic v1 

12def test_upload_file_invalid(): 5 ctx1fghij

13 with pytest.raises(ValueError): 5 ctx1zABCD

14 UploadFile.validate("not a Starlette UploadFile") 5 ctx1zABCD

15 

16 

17def test_upload_file_invalid_pydantic_v2(): 5 ctx1fghij

18 with pytest.raises(ValueError): 5 ctx1EFGHI

19 UploadFile._validate("not a Starlette UploadFile", {}) 5 ctx1EFGHI

20 

21 

22def test_default_placeholder_equals(): 5 ctx1fghij

23 placeholder_1 = Default("a") 5 ctx1pqrst

24 placeholder_2 = Default("a") 5 ctx1pqrst

25 assert placeholder_1 == placeholder_2 5 ctx1pqrst

26 assert placeholder_1.value == placeholder_2.value 5 ctx1pqrst

27 

28 

29def test_default_placeholder_bool(): 5 ctx1fghij

30 placeholder_a = Default("a") 5 ctx1uvwxy

31 placeholder_b = Default("") 5 ctx1uvwxy

32 assert placeholder_a 5 ctx1uvwxy

33 assert not placeholder_b 5 ctx1uvwxy

34 

35 

36def test_upload_file_is_closed(tmp_path: Path): 5 ctx1fghij

37 path = tmp_path / "test.txt" 5 ctx1abcde

38 path.write_bytes(b"<file content>") 5 ctx1abcde

39 app = FastAPI() 5 ctx1abcde

40 

41 testing_file_store: List[UploadFile] = [] 5 ctx1abcde

42 

43 @app.post("/uploadfile/") 5 ctx1abcde

44 def create_upload_file(file: UploadFile): 5 ctx1abcde

45 testing_file_store.append(file) 5 ctx1abcde

46 return {"filename": file.filename} 5 ctx1abcde

47 

48 client = TestClient(app) 5 ctx1abcde

49 with path.open("rb") as file: 5 ctx1abcde

50 response = client.post("/uploadfile/", files={"file": file}) 5 ctx1abcde

51 assert response.status_code == 200, response.text 5 ctx1abcde

52 assert response.json() == {"filename": "test.txt"} 5 ctx1abcde

53 

54 assert testing_file_store 5 ctx1abcde

55 assert testing_file_store[0].file.closed 5 ctx1abcde

56 

57 

58# For UploadFile coverage, segments copied from Starlette tests 

59 

60 

61@pytest.mark.anyio 5 ctx1fghij

62async def test_upload_file(): 5 ctx1fghij

63 stream = io.BytesIO(b"data") 5 ctx1klmno

64 file = UploadFile(filename="file", file=stream, size=4) 5 ctx1klmno

65 assert await file.read() == b"data" 5 ctx1klmno

66 assert file.size == 4 5 ctx1klmno

67 await file.write(b" and more data!") 5 ctx1klmno

68 assert await file.read() == b"" 5 ctx1klmno

69 assert file.size == 19 5 ctx1klmno

70 await file.seek(0) 5 ctx1klmno

71 assert await file.read() == b"data and more data!" 5 ctx1klmno

72 await file.close() 5 ctx1klmno