def test_dict_conversion(self, d):
# Also tested in connectionpool, needs to preserve case
hdict = {
'Content-Length': '0',
'Content-type': 'text/plain',
'Server': 'TornadoServer/1.2.3'
}
h = dict(HTTPHeaderDict(hdict).items())
assert hdict == h
assert hdict == dict(HTTPHeaderDict(hdict))
def test_create_from_headerdict(self):
headers = [('ab', 'A'), ('cd', 'B'), ('cookie', 'C'), ('cookie', 'D'),
('cookie', 'E')]
org = HTTPHeaderDict(headers)
h = HTTPHeaderDict(org)
assert len(h) == 3
assert 'ab' in h
clist = h.getlist('cookie')
assert len(clist) == 3
assert clist[0] == 'C'
assert clist[-1] == 'E'
assert h is not org
assert h == org
def test_headerdict(self):
default_headers = HTTPHeaderDict(a='b')
proxy_headers = HTTPHeaderDict()
proxy_headers.add('foo', 'bar')
http = proxy_from_url(self.proxy_url,
headers=default_headers,
proxy_headers=proxy_headers)
self.addCleanup(http.clear)
request_headers = HTTPHeaderDict(baz='quux')
r = http.request('GET',
'%s/headers' % self.http_url,
headers=request_headers)
returned_headers = json.loads(r.data.decode())
self.assertEqual(returned_headers.get('Foo'), 'bar')
self.assertEqual(returned_headers.get('Baz'), 'quux')
def test_create_from_list(self):
headers = [('ab', 'A'), ('cd', 'B'), ('cookie', 'C'), ('cookie', 'D'),
('cookie', 'E')]
h = HTTPHeaderDict(headers)
assert len(h) == 3
assert 'ab' in h
clist = h.getlist('cookie')
assert len(clist) == 3
assert clist[0] == 'C'
assert clist[-1] == 'E'
def test_string_enforcement(self, d):
# This currently throws AttributeError on key.lower(), should
# probably be something nicer
with pytest.raises(Exception):
d[3] = 5
with pytest.raises(Exception):
d.add(3, 4)
with pytest.raises(Exception):
del d[3]
with pytest.raises(Exception):
HTTPHeaderDict({3: 3})
def test_not_equal(self, d):
b = HTTPHeaderDict(cookie='foo, bar')
c = NonMappingHeaderContainer(cookie='foo, bar')
assert not (d != b)
assert not (d != c)
assert d != 2
def test_equal(self, d):
b = HTTPHeaderDict(cookie='foo, bar')
c = NonMappingHeaderContainer(cookie='foo, bar')
assert d == b
assert d == c
assert d != 2
def test_getlist_after_copy(self, d):
assert d.getlist('cookie') == HTTPHeaderDict(d).getlist('cookie')
def test_extend_from_headerdict(self, d):
h = HTTPHeaderDict(Cookie='foo', e='foofoo')
d.extend(h)
assert d['cookie'] == 'foo, bar, foo'
assert d['e'] == 'foofoo'
assert len(d) == 2
def test_create_from_iterator(self):
teststr = 'urllib3ontherocks'
h = HTTPHeaderDict((c, c * 5) for c in teststr)
assert len(h) == len(set(teststr))
def test_create_from_dict(self):
h = HTTPHeaderDict(dict(ab=1, cd=2, ef=3, gh=4))
assert len(h) == 4
assert 'ab' in h
def d():
header_dict = HTTPHeaderDict(Cookie='foo')
header_dict.add('cookie', 'bar')
return header_dict