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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0.
from awscrt.http import HttpHeaders, HttpRequest
import awscrt.io
from test import NativeResourceTest
import unittest
class TestHttpHeaders(NativeResourceTest):
def test_add(self):
h = HttpHeaders()
h.add('Host', 'example.org')
self.assertEqual('example.org', h.get('Host'))
self.assertEqual(['example.org'], list(h.get_values('Host')))
def test_add_multi_values(self):
h = HttpHeaders()
h.add('Cookie', 'a=1')
h.add('Cookie', 'b=2')
self.assertEqual('a=1', h.get('Cookie'))
self.assertEqual(['a=1', 'b=2'], list(h.get_values('Cookie')))
def test_add_pairs(self):
h = HttpHeaders()
h.add_pairs([
('Host', 'example.org'),
('Cookie', 'a=1'),
('Cookie', 'b=2'),
])
self.assertEqual('example.org', h.get('Host'))
self.assertEqual(['a=1', 'b=2'], list(h.get_values('Cookie')))
def test_set(self):
h = HttpHeaders()
# create
h.set('Host', 'example.org')
self.assertEqual(['example.org'], list(h.get_values('Host')))
# replace
h.set('Host', 'example2.org')
self.assertEqual(['example2.org'], list(h.get_values('Host')))
# replace many
h.add('Host', 'example3.org')
h.add('Host', 'example4.org')
h.set('Host', 'example5.org')
self.assertEqual(['example5.org'], list(h.get_values('Host')))
def test_unicode(self):
# test adding unicode values in all the different ways
h = HttpHeaders([('a', 'แด')])
self.assertEqual('แด', h.get('a'))
h.set('b', '๐ฆ')
self.assertEqual('๐ฆ', h.get('b'))
h.add('c', '๐๐๐')
self.assertEqual('๐๐๐', h.get('c'))
h.add_pairs([('d', 'โคลฃแธโปโฝ')])
self.assertEqual('โคลฃแธโปโฝ', h.get('d'))
def test_get_none(self):
h = HttpHeaders()
self.assertIsNone(h.get('Non-Existent'))
self.assertEqual('Banana', h.get('Non-Existent', 'Banana'))
self.assertEqual([], list(h.get_values('Non-Existent')))
def test_get_is_case_insensitive(self):
h = HttpHeaders()
h.set('Cookie', 'a=1')
h.add_pairs([('cookie', 'b=2'), ('COOKIE', 'c=3')])
h.add(u'CoOkIe', 'd=4') # note: unicode
self.assertEqual('a=1', h.get(u'COOKIE'))
self.assertEqual(['a=1', 'b=2', 'c=3', 'd=4'], list(h.get_values('Cookie')))
def test_iter(self):
# test that we iterate over everything we put in
src = [('Host', 'example.org'), ('Cookie', 'a=1')]
h = HttpHeaders(src)
for pair in h:
src.remove(pair)
self.assertEqual(0, len(src))
def test_iter_order(self):
# test that headers with multiple values are iterated in insertion order
src = [('Cookie', 'a=1'), ('cookie', 'b=2')]
h = HttpHeaders(src)
gather = [pair for pair in h]
# note this also compares that we preserved case of the names
self.assertEqual(src, gather)
def test_remove(self):
h = HttpHeaders()
self.assertRaises(KeyError, h.remove, 'Non-Existent')
h.add('Host', 'example.org')
h.remove('Host')
self.assertIsNone(h.get('Host'))
def test_remove_value(self):
h = HttpHeaders()
self.assertRaises(ValueError, h.remove_value, 'Non-Existent', 'Nope')
# header with 1 value
h.add('Host', 'example.org')
self.assertRaises(ValueError, h.remove_value, 'Host', 'wrong-value')
h.remove_value('Host', 'example.org')
self.assertIsNone(h.get('Host'))
# pluck out a duplicate value [1,2,2] -> [1,2]
h.add_pairs([('Dupes', '1'), ('DUPES', '2'), ('dupes', '2')])
h.remove_value('Dupes', '2')
self.assertEqual(['1', '2'], list(h.get_values('Dupes')))
def test_clear(self):
h = HttpHeaders([('Host', 'example.org'), ('Cookie', 'a=1'), ('cookie', 'b=2')])
h.clear()
self.assertEqual([], [pair for pair in h])
class TestHttpMessage(NativeResourceTest):
def test_request_create_default(self):
request = HttpRequest()
self.assertEqual("GET", request.method)
self.assertEqual("/", request.path)
self.assertEqual([], list(request.headers))
self.assertIsNone(request.body_stream)
def test_request_create_nondefault(self):
src_headers = [('Cookie', 'a=1'), ('Cookie', 'b=2')]
body_stream = open('test/test_http_headers.py', 'rb')
request = HttpRequest(method="PUT",
path="/upload",
headers=HttpHeaders(src_headers),
body_stream=body_stream)
self.assertEqual("PUT", request.method)
self.assertEqual("/upload", request.path)
self.assertEqual(src_headers, list(request.headers))
self.assertIsNotNone(request.body_stream)
body_stream.close()
def test_headers_live_after_message_del(self):
request = HttpRequest()
headers = request.headers
del request
headers.add('Cookie', 'a=1')
self.assertEqual([('Cookie', 'a=1')], list(headers))
def test_unicode(self):
request = HttpRequest(path='/แด')
self.assertEqual('/แด', request.path)
if __name__ == '__main__':
unittest.main()
|
