1 /* $Id: test_filter_query_rewrite.cpp,v 1.10 2006-06-10 14:29:12 adam Exp $
2 Copyright (c) 2005-2006, Index Data.
4 See the LICENSE file for details
11 #include "filter_query_rewrite.hpp"
13 #include "router_chain.hpp"
14 #include "session.hpp"
15 #include "package.hpp"
17 #define BOOST_AUTO_TEST_MAIN
18 #include <boost/test/auto_unit_test.hpp>
20 using namespace boost::unit_test;
22 namespace mp = metaproxy_1;
23 using namespace mp::util;
25 class FilterBounceZ3950: public mp::filter::Base {
27 void process(mp::Package & package) const {
29 if (package.session().is_closed())
31 std::cout << "Got Close.\n";
35 Z_GDU *gdu = package.request().get();
37 if (gdu && gdu->which == Z_GDU_Z3950
38 && gdu->u.z3950->which == Z_APDU_initRequest)
40 std::cout << "Got Z3950 Init PDU\n";
41 //Z_InitRequest *req = gdu->u.z3950->u.initRequest;
42 //package.request() = gdu;
45 else if (gdu && gdu->which == Z_GDU_Z3950
46 && gdu->u.z3950->which == Z_APDU_searchRequest)
48 std::cout << "Got Z3950 Search PDU\n";
49 //Z_SearchRequest *req = gdu->u.z3950->u.searchRequest;
50 //package.request() = gdu;
53 else if (gdu && gdu->which == Z_GDU_Z3950
54 && gdu->u.z3950->which == Z_APDU_scanRequest)
56 std::cout << "Got Z3950 Scan PDU\n";
57 //Z_ScanRequest *req = gdu->u.z3950->u.scanRequest;
58 //package.request() = gdu;
66 void check_query_rewrite_init(mp::RouterChain &router)
68 //std::cout << "QUERY REWRITE INIT\n";
70 // Create package with Z39.50 init request in it
74 Z_APDU *apdu = zget_APDU(odr, Z_APDU_initRequest);
76 pack.request() = apdu;
77 // Done creating query.
80 pack.router(router).move();
82 // Inspect bounced back request
83 //yazpp_1::GDU *gdu = &pack.response();
84 yazpp_1::GDU *gdu = &pack.request();
86 Z_GDU *z_gdu = gdu->get();
88 //std::cout << "Z_GDU " << z_gdu << "\n";
91 BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
92 BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_initRequest);
96 void check_query_rewrite_search(mp::RouterChain &router,
98 std::string query_expect)
100 //std::cout << "QUERY REWRITE SEARCH "
101 // << query_in << " " << query_expect << "\n";
103 // Create package with Z39.50 search request in it
107 Z_APDU *apdu = zget_APDU(odr, Z_APDU_searchRequest);
109 // create package PQF query here
110 mp::util::pqf(odr, apdu, query_in);
112 // create package PDF database info (needed!)
113 apdu->u.searchRequest->num_databaseNames = 1;
114 apdu->u.searchRequest->databaseNames
115 = (char**)odr_malloc(odr, sizeof(char *));
116 apdu->u.searchRequest->databaseNames[0] = odr_strdup(odr, "Default");
118 // Done creating request package
119 pack.request() = apdu;
122 pack.router(router).move();
124 // Inspect bounced back request
125 //yazpp_1::GDU *gdu = &pack.response();
126 yazpp_1::GDU *gdu = &pack.request();
128 Z_GDU *z_gdu = gdu->get();
129 //std::cout << "Z_GDU " << z_gdu << "\n";
133 BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
134 BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_searchRequest);
136 // take query out of package again and check rewrite
137 std::string query_changed
138 = zQueryToString(z_gdu->u.z3950->u.searchRequest->query);
139 BOOST_CHECK_EQUAL(query_expect, query_changed);
144 BOOST_AUTO_UNIT_TEST( test_filter_query_rewrite_1 )
148 mp::filter::QueryRewrite f_query_rewrite;
155 BOOST_AUTO_UNIT_TEST( test_filter_query_rewrite2 )
159 mp::RouterChain router;
161 mp::filter::QueryRewrite f_query_rewrite;
162 //FilterBounceZ3950 f_bounce_z3950;
164 router.append(f_query_rewrite);
165 //router.append(f_bounce_z3950);
167 check_query_rewrite_init(router);
168 check_query_rewrite_search(router,
169 "@attrset Bib-1 @attr 1=4 the",
170 "@attrset Bib-1 @attr 1=4 the");
179 BOOST_AUTO_UNIT_TEST( test_filter_query_rewrite3 )
185 mp::RouterChain router;
188 std::string xmlconf =
189 "<?xml version='1.0'?>\n"
190 "<filter xmlns='http://indexdata.dk/yp2/config/1'\n"
191 " id='qrw1' type='query_rewrite'>\n"
195 //std::cout << xmlconf << std::endl;
197 // reading and parsing XML conf
198 xmlDocPtr doc = xmlParseMemory(xmlconf.c_str(), xmlconf.size());
200 xmlNode *root_element = xmlDocGetRootElement(doc);
202 // creating and configuring filter
203 mp::filter::QueryRewrite f_query_rewrite;
204 f_query_rewrite.configure(root_element);
206 // remeber to free XML DOM
209 // add only filter to router
210 router.append(f_query_rewrite);
213 check_query_rewrite_init(router);
214 check_query_rewrite_search(router,
215 "@attrset Bib-1 @attr 1=4 the",
216 "@attrset Bib-1 @attr 1=4 the");
220 catch (std::exception &e) {
221 std::cout << e.what() << "\n";
233 * indent-tabs-mode: nil
234 * c-file-style: "stroustrup"
236 * vim: shiftwidth=4 tabstop=8 expandtab