def test_expose_whitelisted_headers(self): headers = dict((h, '42') for h in ConnectionInfo._exposed_headers) c = ConnectionInfo(None, None, {}, headers, None) for h in headers.keys(): self.assertTrue(c.get_header(h) is not None)
def test_dont_expose_unknown_headers(self): c = ConnectionInfo(None, None, {}, {'StrangeHeader': '42'}, None) self.assertTrue(c.get_header('StrangeHeader') is None)