Coverage for tests/test_datastructures.py: 100%

1import io 6 ctx1ghijkl

2from pathlib import Path 6 ctx1ghijkl

3from typing import List 6 ctx1ghijkl

4 

5import pytest 6 ctx1ghijkl

6from fastapi import FastAPI, UploadFile 6 ctx1ghijkl

7from fastapi.datastructures import Default 6 ctx1ghijkl

8from fastapi.testclient import TestClient 6 ctx1ghijkl

9 

10 

11# TODO: remove when deprecating Pydantic v1 

12def test_upload_file_invalid(): 6 ctx1ghijkl

13 with pytest.raises(ValueError): 6 ctx1EFGHIJ

14 UploadFile.validate("not a Starlette UploadFile") 6 ctx1EFGHIJ

15 

16 

17def test_upload_file_invalid_pydantic_v2(): 6 ctx1ghijkl

18 with pytest.raises(ValueError): 6 ctx1KLMNOP

19 UploadFile._validate("not a Starlette UploadFile", {}) 6 ctx1KLMNOP

20 

21 

22def test_default_placeholder_equals(): 6 ctx1ghijkl

23 placeholder_1 = Default("a") 6 ctx1stuvwx

24 placeholder_2 = Default("a") 6 ctx1stuvwx

25 assert placeholder_1 == placeholder_2 6 ctx1stuvwx

26 assert placeholder_1.value == placeholder_2.value 6 ctx1stuvwx

27 

28 

29def test_default_placeholder_bool(): 6 ctx1ghijkl

30 placeholder_a = Default("a") 6 ctx1yzABCD

31 placeholder_b = Default("") 6 ctx1yzABCD

32 assert placeholder_a 6 ctx1yzABCD

33 assert not placeholder_b 6 ctx1yzABCD

34 

35 

36def test_upload_file_is_closed(tmp_path: Path): 6 ctx1ghijkl

37 path = tmp_path / "test.txt" 6 ctx1abcdef

38 path.write_bytes(b"<file content>") 6 ctx1abcdef

39 app = FastAPI() 6 ctx1abcdef

40 

41 testing_file_store: List[UploadFile] = [] 6 ctx1abcdef

42 

43 @app.post("/uploadfile/") 6 ctx1abcdef

44 def create_upload_file(file: UploadFile): 6 ctx1abcdef

45 testing_file_store.append(file) 6 ctx1abcdef

46 return {"filename": file.filename} 6 ctx1abcdef

47 

48 client = TestClient(app) 6 ctx1abcdef

49 with path.open("rb") as file: 6 ctx1abcdef

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

51 assert response.status_code == 200, response.text 6 ctx1abcdef

52 assert response.json() == {"filename": "test.txt"} 6 ctx1abcdef

53 

54 assert testing_file_store 6 ctx1abcdef

55 assert testing_file_store[0].file.closed 6 ctx1abcdef

56 

57 

58# For UploadFile coverage, segments copied from Starlette tests 

59 

60 

61@pytest.mark.anyio 6 ctx1ghijkl

62async def test_upload_file(): 6 ctx1ghijkl

63 stream = io.BytesIO(b"data") 6 ctx1mnopqr

64 file = UploadFile(filename="file", file=stream, size=4) 6 ctx1mnopqr

65 assert await file.read() == b"data" 6 ctx1mnopqr

66 assert file.size == 4 6 ctx1mnopqr

67 await file.write(b" and more data!") 6 ctx1mnopqr

68 assert await file.read() == b"" 6 ctx1mnopqr

69 assert file.size == 19 6 ctx1mnopqr

70 await file.seek(0) 6 ctx1mnopqr

71 assert await file.read() == b"data and more data!" 6 ctx1mnopqr

72 await file.close() 6 ctx1mnopqr