1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
from __future__ import annotations
import pytest
from urllib3.fields import RequestField
from urllib3.filepost import _TYPE_FIELDS, encode_multipart_formdata
BOUNDARY = "!! test boundary !!"
BOUNDARY_BYTES = BOUNDARY.encode()
class TestMultipartEncoding:
@pytest.mark.parametrize(
"fields", [dict(k="v", k2="v2"), [("k", "v"), ("k2", "v2")]]
)
def test_input_datastructures(self, fields: _TYPE_FIELDS) -> None:
encoded, _ = encode_multipart_formdata(fields, boundary=BOUNDARY)
assert encoded.count(BOUNDARY_BYTES) == 3
@pytest.mark.parametrize(
"fields",
[
[("k", "v"), ("k2", "v2")],
[("k", b"v"), ("k2", b"v2")],
[("k", b"v"), ("k2", "v2")],
],
)
def test_field_encoding(self, fields: _TYPE_FIELDS) -> None:
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + BOUNDARY_BYTES + b"\r\n"
b'Content-Disposition: form-data; name="k"\r\n'
b"\r\n"
b"v\r\n"
b"--" + BOUNDARY_BYTES + b"\r\n"
b'Content-Disposition: form-data; name="k2"\r\n'
b"\r\n"
b"v2\r\n"
b"--" + BOUNDARY_BYTES + b"--\r\n"
)
assert encoded == expected
assert content_type == "multipart/form-data; boundary=" + str(BOUNDARY)
def test_filename(self) -> None:
fields = [("k", ("somename", b"v"))]
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + BOUNDARY_BYTES + b"\r\n"
b'Content-Disposition: form-data; name="k"; filename="somename"\r\n'
b"Content-Type: application/octet-stream\r\n"
b"\r\n"
b"v\r\n"
b"--" + BOUNDARY_BYTES + b"--\r\n"
)
assert encoded == expected
assert content_type == "multipart/form-data; boundary=" + str(BOUNDARY)
def test_textplain(self) -> None:
fields = [("k", ("somefile.txt", b"v"))]
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + BOUNDARY_BYTES + b"\r\n"
b'Content-Disposition: form-data; name="k"; filename="somefile.txt"\r\n'
b"Content-Type: text/plain\r\n"
b"\r\n"
b"v\r\n"
b"--" + BOUNDARY_BYTES + b"--\r\n"
)
assert encoded == expected
assert content_type == "multipart/form-data; boundary=" + str(BOUNDARY)
def test_explicit(self) -> None:
fields = [("k", ("somefile.txt", b"v", "image/jpeg"))]
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + BOUNDARY_BYTES + b"\r\n"
b'Content-Disposition: form-data; name="k"; filename="somefile.txt"\r\n'
b"Content-Type: image/jpeg\r\n"
b"\r\n"
b"v\r\n"
b"--" + BOUNDARY_BYTES + b"--\r\n"
)
assert encoded == expected
assert content_type == "multipart/form-data; boundary=" + str(BOUNDARY)
def test_request_fields(self) -> None:
fields = [
RequestField(
"k",
b"v",
filename="somefile.txt",
headers={"Content-Type": "image/jpeg"},
)
]
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + BOUNDARY_BYTES + b"\r\n"
b"Content-Type: image/jpeg\r\n"
b"\r\n"
b"v\r\n"
b"--" + BOUNDARY_BYTES + b"--\r\n"
)
assert encoded == expected
|
