示例#1
0
    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)
示例#2
0
    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)