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
117
118
119
120
121
122
123
124
125
126
127
128
|
import pytest
from urllib3.fields import RequestField
from urllib3.filepost import encode_multipart_formdata, iter_fields
from six import b, u
BOUNDARY = "!! test boundary !!"
class TestIterfields(object):
def test_dict(self):
for fieldname, value in iter_fields(dict(a="b")):
assert (fieldname, value) == ("a", "b")
assert list(sorted(iter_fields(dict(a="b", c="d")))) == [("a", "b"), ("c", "d")]
def test_tuple_list(self):
for fieldname, value in iter_fields([("a", "b")]):
assert (fieldname, value) == ("a", "b")
assert list(iter_fields([("a", "b"), ("c", "d")])) == [("a", "b"), ("c", "d")]
class TestMultipartEncoding(object):
@pytest.mark.parametrize(
"fields", [dict(k="v", k2="v2"), [("k", "v"), ("k2", "v2")]]
)
def test_input_datastructures(self, fields):
encoded, _ = encode_multipart_formdata(fields, boundary=BOUNDARY)
assert encoded.count(b(BOUNDARY)) == 3
@pytest.mark.parametrize(
"fields",
[
[("k", "v"), ("k2", "v2")],
[("k", b"v"), (u("k2"), b"v2")],
[("k", b"v"), (u("k2"), "v2")],
],
)
def test_field_encoding(self, fields):
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + b(BOUNDARY) + b"\r\n"
b'Content-Disposition: form-data; name="k"\r\n'
b"\r\n"
b"v\r\n"
b"--" + b(BOUNDARY) + b"\r\n"
b'Content-Disposition: form-data; name="k2"\r\n'
b"\r\n"
b"v2\r\n"
b"--" + b(BOUNDARY) + b"--\r\n"
)
assert encoded == expected
assert content_type == "multipart/form-data; boundary=" + str(BOUNDARY)
def test_filename(self):
fields = [("k", ("somename", b"v"))]
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + b(BOUNDARY) + 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"--" + b(BOUNDARY) + b"--\r\n"
)
assert encoded == expected
assert content_type == "multipart/form-data; boundary=" + str(BOUNDARY)
def test_textplain(self):
fields = [("k", ("somefile.txt", b"v"))]
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + b(BOUNDARY) + 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"--" + b(BOUNDARY) + b"--\r\n"
)
assert encoded == expected
assert content_type == "multipart/form-data; boundary=" + str(BOUNDARY)
def test_explicit(self):
fields = [("k", ("somefile.txt", b"v", "image/jpeg"))]
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + b(BOUNDARY) + 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"--" + b(BOUNDARY) + b"--\r\n"
)
assert encoded == expected
assert content_type == "multipart/form-data; boundary=" + str(BOUNDARY)
def test_request_fields(self):
fields = [
RequestField(
"k",
b"v",
filename="somefile.txt",
headers={"Content-Type": "image/jpeg"},
)
]
encoded, content_type = encode_multipart_formdata(fields, boundary=BOUNDARY)
expected = (
b"--" + b(BOUNDARY) + b"\r\n"
b"Content-Type: image/jpeg\r\n"
b"\r\n"
b"v\r\n"
b"--" + b(BOUNDARY) + b"--\r\n"
)
assert encoded == expected
|
