X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Ftest_filter_rewrite.cpp;h=110d6144037ec4d8d29d7e92f4fb2990c22fd67f;hb=586d78659d671683f33ec55f4a7d32b28e345ccd;hp=a048c74cf9a306ba532d58ee21e2a2afbbe0cadd;hpb=aed0227951a728103325801ebae842b96f57745f;p=metaproxy-moved-to-github.git diff --git a/src/test_filter_rewrite.cpp b/src/test_filter_rewrite.cpp index a048c74..110d614 100644 --- a/src/test_filter_rewrite.cpp +++ b/src/test_filter_rewrite.cpp @@ -1,5 +1,5 @@ /* This file is part of Metaproxy. - Copyright (C) 2005-2013 Index Data + Copyright (C) Index Data Metaproxy is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free @@ -43,7 +43,6 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) { try { - std::cout << "Running non-xml config test case" << std::endl; mp::RouterChain router; mp::filter::HttpRewrite fhr; @@ -68,9 +67,11 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) " \n" " \n" " \n" + " \n" + " \n" + " \n" " \n" " \n" - " \n" " \n" @@ -80,9 +81,13 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) " \n" " \n" " \n" - " \n" + " \n" " \n" - " \n" + " \n" + " \n" + " \n" + " \n" + " \n" " \n" " \n" "\n" @@ -104,15 +109,17 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) mp::odr odr; Z_GDU *gdu_req = z_get_HTTP_Request_uri(odr, - "http://proxyhost/proxypath/targetsite/page1.html", 0, 1); - + "http://proxyhost/proxypath/targetsite/page1.html", 0, 1); + + Z_HTTP_Request *hreq = gdu_req->u.HTTP_Request; + z_HTTP_header_set(odr, &hreq->headers, + "X-Metaproxy-SkipLink", ".* skiplink.com" ); pack.request() = gdu_req; //create the http response const char *resp_buf = "HTTP/1.1 200 OK\r\n" - "Content-Length: 441\r\n" "Content-Type: text/html\r\n" "Link: ; rel=absolute\r\n" "Link: ; rel=relative\r\n" @@ -125,7 +132,8 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) "" "" "" "" "

Welcome to our website. It doesn't make it easy to get pro" @@ -135,13 +143,14 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) "" " Another abs link" "" - "" - "" + "" + "\n" + "skip\n" ""; const char *resp_expected = "HTTP/1.1 200 OK\r\n" - "Content-Length: 564\r\n" + "Content-Length: 686\r\n" "Content-Type: text/html\r\n" "Link: ; rel=absolute\r\n" "Link: ; rel=relative\r\n" @@ -154,7 +163,8 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) "" "" "" "" "

Welcome to our website. It doesn't make it easy to get pro" @@ -164,8 +174,9 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) "" " Another abs link" "" - "" - "" + "" + "\n" + "skip\n" ""; Z_GDU *gdu_res; @@ -205,7 +216,7 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) { //compare buffers std::cout << "Expected result:\n" << resp_expected << "\n"; - std::cout << "Got result:\n" << "\n"; + std::cout << "Got result:\n"; fflush(stdout); fwrite(resp_result, 1, resp_result_len, stdout); fflush(stdout); @@ -221,6 +232,150 @@ BOOST_AUTO_TEST_CASE( test_filter_rewrite_1 ) } } +BOOST_AUTO_TEST_CASE( test_filter_rewrite_2 ) +{ + try + { + mp::RouterChain router; + mp::filter::HttpRewrite fhr; + + std::string xmlconf = + "\n" + "\n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + " \n" + "\n" + ; + + std::cout << xmlconf; + + // reading and parsing XML conf + xmlDocPtr doc = xmlParseMemory(xmlconf.c_str(), xmlconf.size()); + BOOST_CHECK(doc); + xmlNode *root_element = xmlDocGetRootElement(doc); + fhr.configure(root_element, true, ""); + xmlFreeDoc(doc); + + router.append(fhr); + + // create an http request + mp::Package pack; + + mp::odr odr; + Z_GDU *gdu_req = z_get_HTTP_Request_uri(odr, + "http://proxyhost/proxypath/targetsite/page1.html", 0, 1); + + pack.request() = gdu_req; + + //create the http response + + const char *resp_buf = + "HTTP/1.1 200 OK\r\n" + "Content-Type: application/javascript\r\n" + "Link: ; rel=absolute\r\n" + "Link: ; rel=relative\r\n" + "\r\n" + "// \"\n" + "my.location = 'http://targetsite/images/bg.png';\n" + "my.other = \"http://targetsite/images/fg.png\";\n" + "my.thrd = \"other\";\n" + "// \"http://targetsite/images/bg.png\n"; + + const char *resp_expected = + "HTTP/1.1 200 OK\r\n" + "Content-Length: 195\r\n" + "Content-Type: application/javascript\r\n" + "Link: ; rel=absolute\r\n" + "Link: ; rel=relative\r\n" + "\r\n" + "// \"\n" + "my.location = 'http://proxyhost/proxypath/targetsite/images/bg.png';\n" + "my.other = \"http://proxyhost/proxypath/targetsite/images/fg.png\";\n" + "my.thrd = \"other\";\n" + "// \"http://targetsite/images/bg.png\n"; + + Z_GDU *gdu_res; + mp::odr dec(ODR_DECODE); + mp::odr enc(ODR_ENCODE); + odr_setbuf(dec, (char *) resp_buf, strlen(resp_buf), 0); + int r = z_GDU(dec, &gdu_res, 0, 0); + + BOOST_CHECK(r); + if (r) + { + BOOST_CHECK_EQUAL(gdu_res->which, Z_GDU_HTTP_Response); + + pack.response() = gdu_res; + + //feed to the router + pack.router(router).move(); + + //analyze the response + Z_GDU *gdu_res_rew = pack.response().get(); + BOOST_CHECK(gdu_res_rew); + BOOST_CHECK_EQUAL(gdu_res_rew->which, Z_GDU_HTTP_Response); + + Z_HTTP_Response *hres = gdu_res_rew->u.HTTP_Response; + BOOST_CHECK(hres); + + z_GDU(enc, &gdu_res_rew, 0, 0); + char *resp_result; + int resp_result_len; + resp_result = odr_getbuf(enc, &resp_result_len, 0); + + int equal = ((size_t) resp_result_len == strlen(resp_expected)) + && !memcmp(resp_result, resp_expected, resp_result_len); + BOOST_CHECK(equal); + + if (!equal) + { + //compare buffers + std::cout << "Expected result:\n" << resp_expected << "\n"; + std::cout << "Got result:\n"; + fflush(stdout); + fwrite(resp_result, 1, resp_result_len, stdout); + fflush(stdout); + std::cout << "\nGot result buf len: " << resp_result_len + << "\n"; + } + } + } + catch (std::exception & e) { + std::cout << e.what(); + std::cout << std::endl; + BOOST_CHECK (false); + } +} /* * Local variables: