def test_get_existing_header_returns_array_of_dict(): headers = Headers( {'header-name': [{ 'key': 'header-name', 'value': 'some-value', }]}) assert headers.get(name='header-name') is not None assert isinstance(headers.get(name='header-name'), type([])) assert headers.get(name='header-name')[0]['key'] == 'header-name' assert headers.get(name='header-name')[0]['value'] == 'some-value'
def test_set_new_header(): headers = Headers({ 'existing-header': [{ 'key': 'existing-header', 'value': 'existing-value', }] }) headers.set(name='foo', value='bar') assert headers.get(name='foo') == [{'key': 'foo', 'value': 'bar'}] assert headers.get_value(name='existing-header') == 'existing-value'
def test_get_missing_header_with_default_returns_given_default(): headers = Headers() assert headers.get(name='missing', default='abc') == 'abc'
def test_get_missing_header_no_default_returns_None(): headers = Headers() assert headers.get(name='missing') == None