def test_merge_headers_with_list_of_tuples(self):
session = ClientSession(headers={
"h1": "header1",
"h2": "header2"
},
loop=self.loop)
headers = session._prepare_headers([("h1", "h1")])
self.assertIsInstance(headers, CIMultiDict)
self.assertEqual(headers, CIMultiDict([("h2", "header2"),
("h1", "h1")]))
session.close()
def test_merge_headers(self):
# Check incoming simple dict
session = ClientSession(headers={
"h1": "header1",
"h2": "header2"
},
loop=self.loop)
headers = session._prepare_headers({"h1": "h1"})
self.assertIsInstance(headers, CIMultiDict)
self.assertEqual(headers, CIMultiDict([("h2", "header2"),
("h1", "h1")]))
session.close()
def test_merge_headers_with_multi_dict(self):
session = ClientSession(
headers={
"h1": "header1",
"h2": "header2"
}, loop=self.loop)
headers = session._prepare_headers(MultiDict([("h1", "h1")]))
self.assertIsInstance(headers, CIMultiDict)
self.assertEqual(headers, CIMultiDict([
("h1", "h1"),
("h2", "header2")
]))
def test_merge_headers(self):
# Check incoming simple dict
session = ClientSession(
headers={
"h1": "header1",
"h2": "header2"
}, loop=self.loop)
headers = session._prepare_headers({
"h1": "h1"
})
self.assertIsInstance(headers, CIMultiDict)
self.assertEqual(headers, CIMultiDict([
("h1", "h1"),
("h2", "header2")
]))
# Check incoming multi dict
session = ClientSession(
headers={
"h1": "header1",
"h2": "header2"
}, loop=self.loop)
headers = session._prepare_headers(MultiDict([("h1", "h1")]))
self.assertIsInstance(headers, CIMultiDict)
self.assertEqual(headers, CIMultiDict([
("h1", "h1"),
("h2", "header2")
]))
# Check incoming list of tuples
session = ClientSession(
headers={
"h1": "header1",
"h2": "header2"
}, loop=self.loop)
headers = session._prepare_headers([("h1", "h1")])
self.assertIsInstance(headers, CIMultiDict)
self.assertEqual(headers, CIMultiDict([
("h1", "h1"),
("h2", "header2")
]))
def test_merge_headers_with_list_of_tuples_duplicated_names(self):
session = ClientSession(
headers={
"h1": "header1",
"h2": "header2"
}, loop=self.loop)
headers = session._prepare_headers([("h1", "v1"),
("h1", "v2")])
self.assertIsInstance(headers, CIMultiDict)
self.assertEqual(headers, CIMultiDict([
("H2", "header2"),
("H1", "v1"),
("H1", "v2"),
]))
session.close()
