def test_set_existing_header():
headers = Headers({'foo': [{
'key': 'foo',
'value': 'orig-foo-value',
}]})
assert headers.get_value(name='foo') == 'orig-foo-value'
headers.set(name='foo', value='new-foo-value')
assert headers.get_value(name='foo') == 'new-foo-value'
def test_Headers_overlays_given_data():
orig_data = {
'foo': [{
'key': 'foo',
'value': 'orig-foo-value',
}]
}
headers = Headers(data=orig_data)
assert headers.get_value(name='foo') == 'orig-foo-value'
headers.set(name='foo', value='new-foo-value')
assert headers.get_value(name='foo') == 'new-foo-value'
assert orig_data['foo'][0]['value'] == 'new-foo-value'
def test_get_value_existing_header_returns_value_field():
headers = Headers(
{'header-name': [{
'key': 'header-name',
'value': 'some-value',
}]})
assert headers.get_value('header-name') == '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_value_missing_header_with_default_returns_given_default():
headers = Headers()
assert headers.get_value(name='missing', default='abc') == 'abc'
def test_get_value_missing_header_no_default_returns_None():
headers = Headers()
assert headers.get_value(name='missing') == None
