def test_constructor(): expected_headers = [('A', 'B'), ('C', 'D')] headers = HeaderSet(('A: B', 'C: D')) assert expected_headers == headers headers = HeaderSet({'A': 'B', 'C': 'D'}) assert expected_headers == headers
def test_copy(): headers = HeaderSet([('A', 'B'), ('C', 'D')]) headers_copy = headers.copy() assert isinstance(headers_copy, HeaderSet) headers_copy.remove('A') assert headers is not headers_copy assert 'A' in headers assert 'A' not in headers_copy
def test_setitem_getitem_delitem(): expected_headers = [('A', 'B'), ('C', 'D')] # setitem headers = HeaderSet(['A: F']) headers[0] = 'A: B' headers['C'] = 'D' assert expected_headers == headers # getitem assert expected_headers[0] == headers[0] assert 'B' == headers['A'] assert 'B' == headers['a'] # delitem del headers[0] del headers['C'] assert [] == headers
def test_remove(): headers = HeaderSet([('A', 'B'), ('C', 'D')]) headers.remove('A') headers.remove('C: D') assert [] == headers
def test_regex_match(): headers = HeaderSet(['Content-Type: application/json;utf-8']) assert re.compile('^content-type: .*', re.I) in headers
def test_extend(): expected_headers = [('A', 'B'), ('C', 'D')] headers = HeaderSet(['A: B']) headers.extend(['C: D']) assert expected_headers == headers
def test_in_operator(): headers = HeaderSet(['A: F']) assert 'A' in headers assert 'a' in headers assert 'A: F' in headers
def test_insert(): expected_headers = [('A', 'B'), ('C', 'D')] headers = HeaderSet() headers.insert(0, ('C', 'D')) headers.insert(0, ('A', 'B')) assert expected_headers == headers headers = HeaderSet() headers.insert(0, 'C: D') headers.insert(0, 'A: B') assert expected_headers == headers headers = HeaderSet() headers.insert(0, 'C', 'D') headers.insert(0, 'A', 'B') assert expected_headers == headers
def test_append(): expected_headers = [('A', 'B'), ('C', 'D')] headers = HeaderSet() headers.append(('A', 'B')) headers.append(('C', 'D')) assert expected_headers == headers headers = HeaderSet() headers.append('A: B') headers.append('C: D') assert expected_headers == headers headers = HeaderSet() headers.append('A', 'B') headers.append('C', 'D') assert expected_headers == headers