src/test_filter_query_rewrite.cpp

   1 /* $Id: test_filter_query_rewrite.cpp,v 1.14 2007-11-02 17:47:41 adam Exp $
   2    Copyright (c) 2005-2007, Index Data.
   3
   4 This file is part of Metaproxy.
   5
   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
   9 version.
  10
  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
  14 for more details.
  15
  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
  19 02111-1307, USA.
  20  */
  21
  22 #include "config.hpp"
  23 #include <iostream>
  24 #include <stdexcept>
  25
  26 #include "filter_query_rewrite.hpp"
  27 #include "util.hpp"
  28 #include "router_chain.hpp"
  29 #include "session.hpp"
  30 #include "package.hpp"
  31
  32 #define BOOST_AUTO_TEST_MAIN
  33 #define BOOST_TEST_DYN_LINK
  34 #include <boost/test/auto_unit_test.hpp>
  35
  36 using namespace boost::unit_test;
  37
  38 namespace mp = metaproxy_1;
  39 using namespace mp::util;
  40
  41 class FilterBounceZ3950: public mp::filter::Base {
  42 public:
  43     void process(mp::Package & package) const {
  44
  45         if (package.session().is_closed())
  46         {
  47             std::cout << "Got Close.\n";
  48             return;
  49         }
  50
  51         Z_GDU *gdu = package.request().get();
  52
  53         if (gdu && gdu->which == Z_GDU_Z3950
  54             && gdu->u.z3950->which == Z_APDU_initRequest)
  55         {
  56             std::cout << "Got Z3950 Init PDU\n";
  57             //Z_InitRequest *req = gdu->u.z3950->u.initRequest;
  58             //package.request() = gdu;
  59             return;
  60         }
  61         else if (gdu && gdu->which == Z_GDU_Z3950
  62                  && gdu->u.z3950->which == Z_APDU_searchRequest)
  63         {
  64             std::cout << "Got Z3950 Search PDU\n";
  65             //Z_SearchRequest *req = gdu->u.z3950->u.searchRequest;
  66             //package.request() = gdu;
  67             return;
  68         }
  69         else if (gdu && gdu->which == Z_GDU_Z3950
  70                  && gdu->u.z3950->which == Z_APDU_scanRequest)
  71         {
  72             std::cout << "Got Z3950 Scan PDU\n";
  73             //Z_ScanRequest *req = gdu->u.z3950->u.scanRequest;
  74             //package.request() = gdu;
  75             return;
  76         }
  77
  78         package.move();
  79     };
  80 };
  81
  82 void check_query_rewrite_init(mp::RouterChain &router)
  83 {
  84     //std::cout << "QUERY REWRITE INIT\n";
  85
  86     // Create package with Z39.50 init request in it
  87     mp::Package pack;
  88
  89     mp::odr odr;
  90     Z_APDU *apdu = zget_APDU(odr, Z_APDU_initRequest);
  91
  92     pack.request() = apdu;
  93     // Done creating query.
  94
  95     // Put it in router
  96     pack.router(router).move();
  97
  98     // Inspect bounced back request
  99     //yazpp_1::GDU *gdu = &pack.response();
 100     yazpp_1::GDU *gdu = &pack.request();
 101
 102     Z_GDU *z_gdu = gdu->get();
 103
 104     //std::cout << "Z_GDU " << z_gdu << "\n";
 105     BOOST_CHECK(z_gdu);
 106     if (z_gdu) {
 107         BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
 108         BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_initRequest);
 109     }
 110 }
 111
 112 void check_query_rewrite_search(mp::RouterChain &router,
 113                                 std::string query_in,
 114                                 std::string query_expect)
 115 {
 116     //std::cout << "QUERY REWRITE SEARCH "
 117     //          << query_in << " " << query_expect << "\n";
 118
 119     // Create package with Z39.50 search request in it
 120     mp::Package pack;
 121
 122     mp::odr odr;
 123     Z_APDU *apdu = zget_APDU(odr, Z_APDU_searchRequest);
 124
 125     // create package PQF query here
 126     mp::util::pqf(odr, apdu, query_in);
 127
 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");
 133
 134     // Done creating request package
 135     pack.request() = apdu;
 136
 137     // Put it in router
 138     pack.router(router).move();
 139
 140     // Inspect bounced back request
 141     //yazpp_1::GDU *gdu = &pack.response();
 142     yazpp_1::GDU *gdu = &pack.request();
 143
 144     Z_GDU *z_gdu = gdu->get();
 145     //std::cout << "Z_GDU " << z_gdu << "\n";
 146
 147     BOOST_CHECK(z_gdu);
 148     if (z_gdu) {
 149         BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
 150         BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_searchRequest);
 151
 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);
 156     }
 157 }
 158
 159
 160 BOOST_AUTO_TEST_CASE( test_filter_query_rewrite_1 )
 161 {
 162     try
 163     {
 164         mp::filter::QueryRewrite f_query_rewrite;
 165     }
 166     catch ( ... ) {
 167         BOOST_CHECK (false);
 168     }
 169 }
 170
 171 BOOST_AUTO_TEST_CASE( test_filter_query_rewrite2 )
 172 {
 173     try
 174     {
 175         mp::RouterChain router;
 176
 177         mp::filter::QueryRewrite f_query_rewrite;
 178         //FilterBounceZ3950 f_bounce_z3950;
 179
 180         router.append(f_query_rewrite);
 181         //router.append(f_bounce_z3950);
 182
 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");
 187
 188     }
 189     catch ( ... ) {
 190         BOOST_CHECK (false);
 191     }
 192 }
 193
 194
 195 BOOST_AUTO_TEST_CASE( test_filter_query_rewrite3 )
 196 {
 197
 198
 199     try
 200     {
 201         mp::RouterChain router;
 202
 203
 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"
 208             "</filter>\n"
 209             ;
 210
 211         //std::cout << xmlconf  << std::endl;
 212
 213         // reading and parsing XML conf
 214         xmlDocPtr doc = xmlParseMemory(xmlconf.c_str(), xmlconf.size());
 215         BOOST_CHECK(doc);
 216         xmlNode *root_element = xmlDocGetRootElement(doc);
 217
 218         // creating and configuring filter
 219         mp::filter::QueryRewrite f_query_rewrite;
 220         f_query_rewrite.configure(root_element);
 221
 222         // remeber to free XML DOM
 223         xmlFreeDoc(doc);
 224
 225         // add only filter to router
 226         router.append(f_query_rewrite);
 227
 228         // start testing
 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");
 233
 234     }
 235
 236     catch (std::exception &e) {
 237         std::cout << e.what() << "\n";
 238         BOOST_CHECK (false);
 239     }
 240
 241     catch ( ... ) {
 242         BOOST_CHECK (false);
 243     }
 244 }
 245
 246 /*
 247  * Local variables:
 248  * c-basic-offset: 4
 249  * indent-tabs-mode: nil
 250  * c-file-style: "stroustrup"
 251  * End:
 252  * vim: shiftwidth=4 tabstop=8 expandtab
 253  */