def test_remove_headers(): headers = {'hello': ['goodbye'], 'secret': ['header']} request = Request('GET', 'http://google.com', '', headers) _remove_headers(request, ['secret']) assert request.headers == {'hello': 'goodbye'}
def test_remove_headers(): request = mock.Mock(headers=[('hello','goodbye'),('secret','header')]) assert _remove_headers(request, ['secret']).headers == frozenset([('hello','goodbye')])
def test_remove_headers_empty(): headers = {'hello': 'goodbye', 'secret': 'header'} request = Request('GET', 'http://google.com', '', headers) _remove_headers(request, []) assert request.headers == headers