Coverage for tests/test_datastructures.py: 100%

1import io 5 ctx1abcde

2from pathlib import Path 5 ctx1abcde

3from typing import List 5 ctx1abcde

4 

5import pytest 5 ctx1abcde

6from fastapi import FastAPI, UploadFile 5 ctx1abcde

7from fastapi.datastructures import Default 5 ctx1abcde

8from fastapi.testclient import TestClient 5 ctx1abcde

9 

10 

11# TODO: remove when deprecating Pydantic v1 

12def test_upload_file_invalid(): 5 ctx1abcde

13 with pytest.raises(ValueError): 5 ctx1abcde

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

15 

16 

17def test_upload_file_invalid_pydantic_v2(): 5 ctx1abcde

18 with pytest.raises(ValueError): 5 ctx1abcde

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

20 

21 

22def test_default_placeholder_equals(): 5 ctx1abcde

23 placeholder_1 = Default("a") 5 ctx1abcde

24 placeholder_2 = Default("a") 5 ctx1abcde

25 assert placeholder_1 == placeholder_2 5 ctx1abcde

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

27 

28 

29def test_default_placeholder_bool(): 5 ctx1abcde

30 placeholder_a = Default("a") 5 ctx1abcde

31 placeholder_b = Default("") 5 ctx1abcde

32 assert placeholder_a 5 ctx1abcde

33 assert not placeholder_b 5 ctx1abcde

34 

35 

36def test_upload_file_is_closed(tmp_path: Path): 5 ctx1abcde

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 ctx1abcde

62async def test_upload_file(): 5 ctx1abcde

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

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

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

66 assert file.size == 4 5 ctx1abcde

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

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

69 assert file.size == 19 5 ctx1abcde

70 await file.seek(0) 5 ctx1abcde

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

72 await file.close() 5 ctx1abcde