Coverage for tests/test_datastructures.py: 100%

1import io 7 ctx1hijklmn

2from pathlib import Path 7 ctx1hijklmn

3from typing import List 7 ctx1hijklmn

4 

5import pytest 7 ctx1hijklmn

6from fastapi import FastAPI, UploadFile 7 ctx1hijklmn

7from fastapi.datastructures import Default 7 ctx1hijklmn

8from fastapi.testclient import TestClient 7 ctx1hijklmn

9 

10 

11# TODO: remove when deprecating Pydantic v1 

12def test_upload_file_invalid(): 7 ctx1hijklmn

13 with pytest.raises(ValueError): 7 ctx1JKLMNOP

14 UploadFile.validate("not a Starlette UploadFile") 7 ctx1JKLMNOP

15 

16 

17def test_upload_file_invalid_pydantic_v2(): 7 ctx1hijklmn

18 with pytest.raises(ValueError): 7 ctx1QRSTUVW

19 UploadFile._validate("not a Starlette UploadFile", {}) 7 ctx1QRSTUVW

20 

21 

22def test_default_placeholder_equals(): 7 ctx1hijklmn

23 placeholder_1 = Default("a") 7 ctx1vwxyzAB

24 placeholder_2 = Default("a") 7 ctx1vwxyzAB

25 assert placeholder_1 == placeholder_2 7 ctx1vwxyzAB

26 assert placeholder_1.value == placeholder_2.value 7 ctx1vwxyzAB

27 

28 

29def test_default_placeholder_bool(): 7 ctx1hijklmn

30 placeholder_a = Default("a") 7 ctx1CDEFGHI

31 placeholder_b = Default("") 7 ctx1CDEFGHI

32 assert placeholder_a 7 ctx1CDEFGHI

33 assert not placeholder_b 7 ctx1CDEFGHI

34 

35 

36def test_upload_file_is_closed(tmp_path: Path): 7 ctx1hijklmn

37 path = tmp_path / "test.txt" 7 ctx1abcdefg

38 path.write_bytes(b"<file content>") 7 ctx1abcdefg

39 app = FastAPI() 7 ctx1abcdefg

40 

41 testing_file_store: List[UploadFile] = [] 7 ctx1abcdefg

42 

43 @app.post("/uploadfile/") 7 ctx1abcdefg

44 def create_upload_file(file: UploadFile): 7 ctx1abcdefg

45 testing_file_store.append(file) 7 ctx1abcdefg

46 return {"filename": file.filename} 7 ctx1abcdefg

47 

48 client = TestClient(app) 7 ctx1abcdefg

49 with path.open("rb") as file: 7 ctx1abcdefg

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

51 assert response.status_code == 200, response.text 7 ctx1abcdefg

52 assert response.json() == {"filename": "test.txt"} 7 ctx1abcdefg

53 

54 assert testing_file_store 7 ctx1abcdefg

55 assert testing_file_store[0].file.closed 7 ctx1abcdefg

56 

57 

58# For UploadFile coverage, segments copied from Starlette tests 

59 

60 

61@pytest.mark.anyio 7 ctx1hijklmn

62async def test_upload_file(): 7 ctx1hijklmn

63 stream = io.BytesIO(b"data") 7 ctx1opqrstu

64 file = UploadFile(filename="file", file=stream, size=4) 7 ctx1opqrstu

65 assert await file.read() == b"data" 7 ctx1opqrstu

66 assert file.size == 4 7 ctx1opqrstu

67 await file.write(b" and more data!") 7 ctx1opqrstu

68 assert await file.read() == b"" 7 ctx1opqrstu

69 assert file.size == 19 7 ctx1opqrstu

70 await file.seek(0) 7 ctx1opqrstu

71 assert await file.read() == b"data and more data!" 7 ctx1opqrstu

72 await file.close() 7 ctx1opqrstu