def test_basic(self): u = URL('http://www.w3af.com') req = HTTPRequest(u) self.assertEqual(req.get_full_url(), 'http://www.w3af.com/') self.assertEqual(req.get_uri().url_string, 'http://www.w3af.com/')
class TestSed(unittest.TestCase): def setUp(self): create_temp_dir() self.plugin = sed() self.url = URL("http://www.w3af.com/") self.request = HTTPRequest(self.url) def tearDown(self): self.plugin.end() def test_blank_body(self): body = "" headers = Headers([("content-type", "text/html")]) response = HTTPResponse(200, body, headers, self.url, self.url, _id=1) option_list = self.plugin.get_options() option_list["expressions"].set_value("qh/User/NotLuser/") self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(self.request) mod_response = self.plugin.mangle_response(response) self.assertEqual(mod_request.get_headers(), self.request.get_headers()) self.assertEqual(mod_response.get_headers(), response.get_headers()) self.assertEqual(mod_request.get_uri(), self.request.get_uri()) self.assertEqual(mod_response.get_uri(), response.get_uri()) self.assertEqual(mod_response.get_body(), response.get_body()) def test_response_body(self): body = "hello user!" headers = Headers([("content-type", "text/html")]) response = HTTPResponse(200, body, headers, self.url, self.url, _id=1) option_list = self.plugin.get_options() option_list["expressions"].set_value("sb/user/notluser/") self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(self.request) mod_response = self.plugin.mangle_response(response) self.assertEqual(mod_request.get_headers(), self.request.get_headers()) self.assertEqual(mod_response.get_headers(), response.get_headers()) self.assertEqual(mod_request.get_uri(), self.request.get_uri()) self.assertEqual(mod_response.get_uri(), response.get_uri()) self.assertEqual(mod_response.get_body(), "hello notluser!") def test_request_headers(self): headers = Headers([("content-type", "text/html")]) request = HTTPRequest(self.url, headers=headers) option_list = self.plugin.get_options() option_list["expressions"].set_value("qh/html/xml/") self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(request) value, _ = mod_request.get_headers().iget("content-type") self.assertEqual(value, "text/xml") self.assertIs(mod_request, request)
class TestSed(unittest.TestCase): def setUp(self): create_temp_dir() self.plugin = sed() self.url = URL('http://www.w3af.com/') self.request = HTTPRequest(self.url) def tearDown(self): self.plugin.end() def test_blank_body(self): body = '' headers = Headers([('content-type', 'text/html')]) response = HTTPResponse(200, body, headers, self.url, self.url, _id=1) option_list = self.plugin.get_options() option_list['expressions'].set_value('qh/User/NotLuser/') self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(self.request) mod_response = self.plugin.mangle_response(response) self.assertEqual(mod_request.get_headers(), self.request.get_headers()) self.assertEqual(mod_response.get_headers(), response.get_headers()) self.assertEqual(mod_request.get_uri(), self.request.get_uri()) self.assertEqual(mod_response.get_uri(), response.get_uri()) self.assertEqual(mod_response.get_body(), response.get_body()) def test_response_body(self): body = 'hello user!' headers = Headers([('content-type', 'text/html')]) response = HTTPResponse(200, body, headers, self.url, self.url, _id=1) option_list = self.plugin.get_options() option_list['expressions'].set_value('sb/user/notluser/') self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(self.request) mod_response = self.plugin.mangle_response(response) self.assertEqual(mod_request.get_headers(), self.request.get_headers()) self.assertEqual(mod_response.get_headers(), response.get_headers()) self.assertEqual(mod_request.get_uri(), self.request.get_uri()) self.assertEqual(mod_response.get_uri(), response.get_uri()) self.assertEqual(mod_response.get_body(), 'hello notluser!') def test_request_headers(self): headers = Headers([('content-type', 'text/html')]) request = HTTPRequest(self.url, headers=headers) option_list = self.plugin.get_options() option_list['expressions'].set_value('qh/html/xml/') self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(request) value, _ = mod_request.get_headers().iget('content-type') self.assertEqual(value, 'text/xml')
class TestSed(unittest.TestCase): def setUp(self): create_temp_dir() self.plugin = sed() self.url = URL('http://www.w3af.com/') self.request = HTTPRequest(self.url) def tearDown(self): self.plugin.end() def test_blank_body(self): body = '' headers = Headers([('content-type', 'text/html')]) response = HTTPResponse(200, body, headers, self.url, self.url, _id=1) option_list = self.plugin.get_options() option_list['expressions'].set_value('qh/User/NotLuser/') self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(self.request) mod_response = self.plugin.mangle_response(response) self.assertEqual(mod_request.get_headers(), self.request.get_headers()) self.assertEqual(mod_response.get_headers(), response.get_headers()) self.assertEqual(mod_request.get_uri(), self.request.get_uri()) self.assertEqual(mod_response.get_uri(), response.get_uri()) self.assertEqual(mod_response.get_body(), response.get_body()) def test_response_body(self): body = 'hello user!' headers = Headers([('content-type', 'text/html')]) response = HTTPResponse(200, body, headers, self.url, self.url, _id=1) option_list = self.plugin.get_options() option_list['expressions'].set_value('sb/user/notluser/') self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(self.request) mod_response = self.plugin.mangle_response(response) self.assertEqual(mod_request.get_headers(), self.request.get_headers()) self.assertEqual(mod_response.get_headers(), response.get_headers()) self.assertEqual(mod_request.get_uri(), self.request.get_uri()) self.assertEqual(mod_response.get_uri(), response.get_uri()) self.assertEqual(mod_response.get_body(), 'hello notluser!') def test_request_headers(self): headers = Headers([('content-type', 'text/html')]) request = HTTPRequest(self.url, headers=headers) option_list = self.plugin.get_options() option_list['expressions'].set_value('qh/html/xml/') self.plugin.set_options(option_list) mod_request = self.plugin.mangle_request(request) value, _ = mod_request.get_headers().iget('content-type') self.assertEqual(value, 'text/xml') self.assertIs(mod_request, request)