49
49
50
50
51
51
class ReadMultipartResponseTests (TestCase ):
52
- data1 = b"\r \n \r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n Content-Type: application/json\r \n \r \n {}\r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n Content-Type: text/plain\r \n Content-Disposition: inline; filename=test_upload\r \n \r \n file_"
53
- data2 = b"to_stream\r \n --6067d4698f8d40a0a794ea7d7379d53a--\r \n \r \n "
54
- data3 = b"\r \n \r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n content-type: application/json\r \n \r \n {}\r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n content-type: text/plain\r \n content-disposition: inline; filename=test_upload\r \n \r \n file_"
52
+ multipart_response_data1 = b"\r \n \r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n Content-Type: application/json\r \n \r \n {}\r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n Content-Type: text/plain\r \n Content-Disposition: inline; filename=test_upload\r \n \r \n file_"
53
+ multipart_response_data2 = (
54
+ b"to_stream\r \n --6067d4698f8d40a0a794ea7d7379d53a--\r \n \r \n "
55
+ )
56
+ multipart_response_data_cased = b"\r \n \r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n cOntEnt-type: application/json\r \n \r \n {}\r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n Content-tyPe: text/plain\r \n conTent-dispOsition: inline; filename=test_upload\r \n \r \n file_"
55
57
56
58
redirect_data = b"\r \n \r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n Content-Type: application/json\r \n \r \n {}\r \n --6067d4698f8d40a0a794ea7d7379d53a\r \n Location: https://cdn.example.org/ab/c1/2345.txt\r \n \r \n --6067d4698f8d40a0a794ea7d7379d53a--\r \n \r \n "
57
59
@@ -104,8 +106,8 @@ def test_parse_file(self) -> None:
104
106
result , deferred , protocol = self ._build_multipart_response (249 , 250 )
105
107
106
108
# Start sending data.
107
- protocol .dataReceived (self .data1 )
108
- protocol .dataReceived (self .data2 )
109
+ protocol .dataReceived (self .multipart_response_data1 )
110
+ protocol .dataReceived (self .multipart_response_data2 )
109
111
# Close the connection.
110
112
protocol .connectionLost (Failure (ResponseDone ()))
111
113
@@ -127,8 +129,8 @@ def test_parse_file_lowercase_headers(self) -> None:
127
129
result , deferred , protocol = self ._build_multipart_response (249 , 250 )
128
130
129
131
# Start sending data.
130
- protocol .dataReceived (self .data3 )
131
- protocol .dataReceived (self .data2 )
132
+ protocol .dataReceived (self .multipart_response_data_cased )
133
+ protocol .dataReceived (self .multipart_response_data2 )
132
134
# Close the connection.
133
135
protocol .connectionLost (Failure (ResponseDone ()))
134
136
@@ -167,7 +169,7 @@ def test_too_large(self) -> None:
167
169
result , deferred , protocol = self ._build_multipart_response (UNKNOWN_LENGTH , 180 )
168
170
169
171
# Start sending data.
170
- protocol .dataReceived (self .data1 )
172
+ protocol .dataReceived (self .multipart_response_data1 )
171
173
172
174
self .assertEqual (result .getvalue (), b"file_" )
173
175
self ._assert_error (deferred , protocol )
@@ -178,11 +180,11 @@ def test_additional_data(self) -> None:
178
180
result , deferred , protocol = self ._build_multipart_response (UNKNOWN_LENGTH , 180 )
179
181
180
182
# Start sending data.
181
- protocol .dataReceived (self .data1 )
183
+ protocol .dataReceived (self .multipart_response_data1 )
182
184
self ._assert_error (deferred , protocol )
183
185
184
186
# More data might have come in.
185
- protocol .dataReceived (self .data2 )
187
+ protocol .dataReceived (self .multipart_response_data2 )
186
188
187
189
self .assertEqual (result .getvalue (), b"file_" )
188
190
self ._assert_error (deferred , protocol )
@@ -196,7 +198,7 @@ def test_content_length(self) -> None:
196
198
self .assertFalse (deferred .called )
197
199
198
200
# Start sending data.
199
- protocol .dataReceived (self .data1 )
201
+ protocol .dataReceived (self .multipart_response_data1 )
200
202
self ._assert_error (deferred , protocol )
201
203
self ._cleanup_error (deferred )
202
204
0 commit comments