Coverage for tests/test_multipart_installation.py: 100%

1import warnings 6 ctx1abcdef

2 

3import pytest 6 ctx1abcdef

4from fastapi import FastAPI, File, Form, UploadFile 6 ctx1abcdef

5from fastapi.dependencies.utils import ( 6 ctx1abcdef

6 multipart_incorrect_install_error, 

7 multipart_not_installed_error, 

8) 

9 

10 

11def test_incorrect_multipart_installed_form(monkeypatch): 6 ctx1abcdef

12 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1ghijkl

13 with warnings.catch_warnings(record=True): 6 ctx1ghijkl

14 warnings.simplefilter("always") 6 ctx1ghijkl

15 monkeypatch.delattr("multipart.multipart.parse_options_header", raising=False) 6 ctx1ghijkl

16 with pytest.raises(RuntimeError, match=multipart_incorrect_install_error): 6 ctx1ghijkl

17 app = FastAPI() 6 ctx1ghijkl

18 

19 @app.post("/") 6 ctx1ghijkl

20 async def root(username: str = Form()): 6 ctx1ghijkl

21 return username # pragma: nocover 

22 

23 

24def test_incorrect_multipart_installed_file_upload(monkeypatch): 6 ctx1abcdef

25 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1mnopqr

26 with warnings.catch_warnings(record=True): 6 ctx1mnopqr

27 warnings.simplefilter("always") 6 ctx1mnopqr

28 monkeypatch.delattr("multipart.multipart.parse_options_header", raising=False) 6 ctx1mnopqr

29 with pytest.raises(RuntimeError, match=multipart_incorrect_install_error): 6 ctx1mnopqr

30 app = FastAPI() 6 ctx1mnopqr

31 

32 @app.post("/") 6 ctx1mnopqr

33 async def root(f: UploadFile = File()): 6 ctx1mnopqr

34 return f # pragma: nocover 

35 

36 

37def test_incorrect_multipart_installed_file_bytes(monkeypatch): 6 ctx1abcdef

38 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1stuvwx

39 with warnings.catch_warnings(record=True): 6 ctx1stuvwx

40 warnings.simplefilter("always") 6 ctx1stuvwx

41 monkeypatch.delattr("multipart.multipart.parse_options_header", raising=False) 6 ctx1stuvwx

42 with pytest.raises(RuntimeError, match=multipart_incorrect_install_error): 6 ctx1stuvwx

43 app = FastAPI() 6 ctx1stuvwx

44 

45 @app.post("/") 6 ctx1stuvwx

46 async def root(f: bytes = File()): 6 ctx1stuvwx

47 return f # pragma: nocover 

48 

49 

50def test_incorrect_multipart_installed_multi_form(monkeypatch): 6 ctx1abcdef

51 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1yzABCD

52 with warnings.catch_warnings(record=True): 6 ctx1yzABCD

53 warnings.simplefilter("always") 6 ctx1yzABCD

54 monkeypatch.delattr("multipart.multipart.parse_options_header", raising=False) 6 ctx1yzABCD

55 with pytest.raises(RuntimeError, match=multipart_incorrect_install_error): 6 ctx1yzABCD

56 app = FastAPI() 6 ctx1yzABCD

57 

58 @app.post("/") 6 ctx1yzABCD

59 async def root(username: str = Form(), password: str = Form()): 6 ctx1yzABCD

60 return username # pragma: nocover 

61 

62 

63def test_incorrect_multipart_installed_form_file(monkeypatch): 6 ctx1abcdef

64 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1EFGHIJ

65 with warnings.catch_warnings(record=True): 6 ctx1EFGHIJ

66 warnings.simplefilter("always") 6 ctx1EFGHIJ

67 monkeypatch.delattr("multipart.multipart.parse_options_header", raising=False) 6 ctx1EFGHIJ

68 with pytest.raises(RuntimeError, match=multipart_incorrect_install_error): 6 ctx1EFGHIJ

69 app = FastAPI() 6 ctx1EFGHIJ

70 

71 @app.post("/") 6 ctx1EFGHIJ

72 async def root(username: str = Form(), f: UploadFile = File()): 6 ctx1EFGHIJ

73 return username # pragma: nocover 

74 

75 

76def test_no_multipart_installed(monkeypatch): 6 ctx1abcdef

77 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1KLMNOP

78 with warnings.catch_warnings(record=True): 6 ctx1KLMNOP

79 warnings.simplefilter("always") 6 ctx1KLMNOP

80 monkeypatch.delattr("multipart.__version__", raising=False) 6 ctx1KLMNOP

81 with pytest.raises(RuntimeError, match=multipart_not_installed_error): 6 ctx1KLMNOP

82 app = FastAPI() 6 ctx1KLMNOP

83 

84 @app.post("/") 6 ctx1KLMNOP

85 async def root(username: str = Form()): 6 ctx1KLMNOP

86 return username # pragma: nocover 

87 

88 

89def test_no_multipart_installed_file(monkeypatch): 6 ctx1abcdef

90 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1QRSTUV

91 with warnings.catch_warnings(record=True): 6 ctx1QRSTUV

92 warnings.simplefilter("always") 6 ctx1QRSTUV

93 monkeypatch.delattr("multipart.__version__", raising=False) 6 ctx1QRSTUV

94 with pytest.raises(RuntimeError, match=multipart_not_installed_error): 6 ctx1QRSTUV

95 app = FastAPI() 6 ctx1QRSTUV

96 

97 @app.post("/") 6 ctx1QRSTUV

98 async def root(f: UploadFile = File()): 6 ctx1QRSTUV

99 return f # pragma: nocover 

100 

101 

102def test_no_multipart_installed_file_bytes(monkeypatch): 6 ctx1abcdef

103 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1WXYZ01

104 with warnings.catch_warnings(record=True): 6 ctx1WXYZ01

105 warnings.simplefilter("always") 6 ctx1WXYZ01

106 monkeypatch.delattr("multipart.__version__", raising=False) 6 ctx1WXYZ01

107 with pytest.raises(RuntimeError, match=multipart_not_installed_error): 6 ctx1WXYZ01

108 app = FastAPI() 6 ctx1WXYZ01

109 

110 @app.post("/") 6 ctx1WXYZ01

111 async def root(f: bytes = File()): 6 ctx1WXYZ01

112 return f # pragma: nocover 

113 

114 

115def test_no_multipart_installed_multi_form(monkeypatch): 6 ctx1abcdef

116 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1234567

117 with warnings.catch_warnings(record=True): 6 ctx1234567

118 warnings.simplefilter("always") 6 ctx1234567

119 monkeypatch.delattr("multipart.__version__", raising=False) 6 ctx1234567

120 with pytest.raises(RuntimeError, match=multipart_not_installed_error): 6 ctx1234567

121 app = FastAPI() 6 ctx1234567

122 

123 @app.post("/") 6 ctx1234567

124 async def root(username: str = Form(), password: str = Form()): 6 ctx1234567

125 return username # pragma: nocover 

126 

127 

128def test_no_multipart_installed_form_file(monkeypatch): 6 ctx1abcdef

129 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx189!#$%

130 with warnings.catch_warnings(record=True): 6 ctx189!#$%

131 warnings.simplefilter("always") 6 ctx189!#$%

132 monkeypatch.delattr("multipart.__version__", raising=False) 6 ctx189!#$%

133 with pytest.raises(RuntimeError, match=multipart_not_installed_error): 6 ctx189!#$%

134 app = FastAPI() 6 ctx189!#$%

135 

136 @app.post("/") 6 ctx189!#$%

137 async def root(username: str = Form(), f: UploadFile = File()): 6 ctx189!#$%

138 return username # pragma: nocover 

139 

140 

141def test_old_multipart_installed(monkeypatch): 6 ctx1abcdef

142 monkeypatch.setattr("python_multipart.__version__", "0.0.12") 6 ctx1'()*+,

143 with warnings.catch_warnings(record=True): 6 ctx1'()*+,

144 warnings.simplefilter("always") 6 ctx1'()*+,

145 app = FastAPI() 6 ctx1'()*+,

146 

147 @app.post("/") 6 ctx1'()*+,

148 async def root(username: str = Form()): 6 ctx1'()*+,

149 return username # pragma: nocover