1 /* $Id: test_filter_query_rewrite.cpp,v 1.15 2008-02-20 15:07:53 adam Exp $
2 Copyright (c) 2005-2007, Index Data.
4 This file is part of Metaproxy.
6 Metaproxy is free software; you can redistribute it and/or modify it under
7 the terms of the GNU General Public License as published by the Free
8 Software Foundation; either version 2, or (at your option) any later
11 Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY
12 WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
16 You should have received a copy of the GNU General Public License
17 along with Metaproxy; see the file LICENSE. If not, write to the
18 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
26 #include "filter_query_rewrite.hpp"
28 #include "router_chain.hpp"
29 #include "session.hpp"
30 #include "package.hpp"
32 #define BOOST_AUTO_TEST_MAIN
33 #define BOOST_TEST_DYN_LINK
34 #include <boost/test/auto_unit_test.hpp>
36 using namespace boost::unit_test;
38 namespace mp = metaproxy_1;
39 using namespace mp::util;
41 class FilterBounceZ3950: public mp::filter::Base {
43 void process(mp::Package & package) const {
45 if (package.session().is_closed())
47 std::cout << "Got Close.\n";
51 Z_GDU *gdu = package.request().get();
53 if (gdu && gdu->which == Z_GDU_Z3950
54 && gdu->u.z3950->which == Z_APDU_initRequest)
56 std::cout << "Got Z3950 Init PDU\n";
57 //Z_InitRequest *req = gdu->u.z3950->u.initRequest;
58 //package.request() = gdu;
61 else if (gdu && gdu->which == Z_GDU_Z3950
62 && gdu->u.z3950->which == Z_APDU_searchRequest)
64 std::cout << "Got Z3950 Search PDU\n";
65 //Z_SearchRequest *req = gdu->u.z3950->u.searchRequest;
66 //package.request() = gdu;
69 else if (gdu && gdu->which == Z_GDU_Z3950
70 && gdu->u.z3950->which == Z_APDU_scanRequest)
72 std::cout << "Got Z3950 Scan PDU\n";
73 //Z_ScanRequest *req = gdu->u.z3950->u.scanRequest;
74 //package.request() = gdu;
82 void check_query_rewrite_init(mp::RouterChain &router)
84 //std::cout << "QUERY REWRITE INIT\n";
86 // Create package with Z39.50 init request in it
90 Z_APDU *apdu = zget_APDU(odr, Z_APDU_initRequest);
92 pack.request() = apdu;
93 // Done creating query.
96 pack.router(router).move();
98 // Inspect bounced back request
99 //yazpp_1::GDU *gdu = &pack.response();
100 yazpp_1::GDU *gdu = &pack.request();
102 Z_GDU *z_gdu = gdu->get();
104 //std::cout << "Z_GDU " << z_gdu << "\n";
107 BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
108 BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_initRequest);
112 void check_query_rewrite_search(mp::RouterChain &router,
113 std::string query_in,
114 std::string query_expect)
116 //std::cout << "QUERY REWRITE SEARCH "
117 // << query_in << " " << query_expect << "\n";
119 // Create package with Z39.50 search request in it
123 Z_APDU *apdu = zget_APDU(odr, Z_APDU_searchRequest);
125 // create package PQF query here
126 mp::util::pqf(odr, apdu, query_in);
128 // create package PDF database info (needed!)
129 apdu->u.searchRequest->num_databaseNames = 1;
130 apdu->u.searchRequest->databaseNames
131 = (char**)odr_malloc(odr, sizeof(char *));
132 apdu->u.searchRequest->databaseNames[0] = odr_strdup(odr, "Default");
134 // Done creating request package
135 pack.request() = apdu;
138 pack.router(router).move();
140 // Inspect bounced back request
141 //yazpp_1::GDU *gdu = &pack.response();
142 yazpp_1::GDU *gdu = &pack.request();
144 Z_GDU *z_gdu = gdu->get();
145 //std::cout << "Z_GDU " << z_gdu << "\n";
149 BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
150 BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_searchRequest);
152 // take query out of package again and check rewrite
153 std::string query_changed
154 = zQueryToString(z_gdu->u.z3950->u.searchRequest->query);
155 BOOST_CHECK_EQUAL(query_expect, query_changed);
160 BOOST_AUTO_TEST_CASE( test_filter_query_rewrite_1 )
164 mp::filter::QueryRewrite f_query_rewrite;
171 BOOST_AUTO_TEST_CASE( test_filter_query_rewrite2 )
175 mp::RouterChain router;
177 mp::filter::QueryRewrite f_query_rewrite;
178 //FilterBounceZ3950 f_bounce_z3950;
180 router.append(f_query_rewrite);
181 //router.append(f_bounce_z3950);
183 check_query_rewrite_init(router);
184 check_query_rewrite_search(router,
185 "@attrset Bib-1 @attr 1=4 the",
186 "@attrset Bib-1 @attr 1=4 the");
195 BOOST_AUTO_TEST_CASE( test_filter_query_rewrite3 )
201 mp::RouterChain router;
204 std::string xmlconf =
205 "<?xml version='1.0'?>\n"
206 "<filter xmlns='http://indexdata.com/metaproxy'\n"
207 " id='qrw1' type='query_rewrite'>\n"
211 //std::cout << xmlconf << std::endl;
213 // reading and parsing XML conf
214 xmlDocPtr doc = xmlParseMemory(xmlconf.c_str(), xmlconf.size());
216 xmlNode *root_element = xmlDocGetRootElement(doc);
218 // creating and configuring filter
219 mp::filter::QueryRewrite f_query_rewrite;
220 f_query_rewrite.configure(root_element, true);
222 // remeber to free XML DOM
225 // add only filter to router
226 router.append(f_query_rewrite);
229 check_query_rewrite_init(router);
230 check_query_rewrite_search(router,
231 "@attrset Bib-1 @attr 1=4 the",
232 "@attrset Bib-1 @attr 1=4 the");
236 catch (std::exception &e) {
237 std::cout << e.what() << "\n";
249 * indent-tabs-mode: nil
250 * c-file-style: "stroustrup"
252 * vim: shiftwidth=4 tabstop=8 expandtab