def test_add_header(self): xff = x_forwarded_for() u = URL('http://www.w3af.com/') r = HTTPRequest(u) modified_request = xff.modify_request(r) modified_headers = modified_request.get_headers() self.assertIn('X-forwarded-for', modified_headers) self.assertEqual(modified_headers['X-forwarded-for'], u'163.7.70.57', modified_headers)
def test_no_modification(self): xff = x_forwarded_for() u = URL('http://www.w3af.com/') headers = Headers([('X-Forwarded-For', '127.0.0.1')]) r = HTTPRequest(u, headers=headers) modified_request = xff.modify_request( r ) modified_headers = modified_request.get_headers() self.assertIn('X-forwarded-for', modified_headers) self.assertEqual(modified_headers['X-forwarded-for'], u'127.0.0.1', modified_headers)