src/test_filter_query_rewrite.cpp

   1 /* $Id: test_filter_query_rewrite.cpp,v 1.2 2006-01-20 22:38:12 marc Exp $
   2    Copyright (c) 2005, Index Data.
   3
   4 %LICENSE%
   5  */
   6
   7 #include "config.hpp"
   8 #include <iostream>
   9 #include <stdexcept>
  10
  11 #include "filter_query_rewrite.hpp"
  12 #include "util.hpp"
  13 #include "router_chain.hpp"
  14 #include "session.hpp"
  15 #include "package.hpp"
  16
  17 #define BOOST_AUTO_TEST_MAIN
  18 #include <boost/test/auto_unit_test.hpp>
  19
  20 using namespace boost::unit_test;
  21
  22 class FilterBounceZ3950: public yp2::filter::Base {
  23 public:
  24     void process(yp2::Package & package) const {
  25
  26         if (package.session().is_closed())
  27         {
  28             std::cout << "Got Close.\n";
  29         }
  30
  31         Z_GDU *gdu = package.request().get();
  32
  33         if (gdu && gdu->which == Z_GDU_Z3950
  34             && gdu->u.z3950->which == Z_APDU_initRequest)
  35         {
  36             std::cout << "Got Z3950 Init PDU\n";
  37             //Z_InitRequest *req = gdu->u.z3950->u.initRequest;
  38             //package.request() = gdu;
  39         }
  40         else if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
  41             Z_APDU_searchRequest)
  42         {
  43             std::cout << "Got Z3950 Search PDU\n";
  44             //Z_SearchRequest *req = gdu->u.z3950->u.searchRequest;
  45             //package.request() = gdu;
  46         }
  47         else if (gdu && gdu->which == Z_GDU_Z3950 && gdu->u.z3950->which ==
  48             Z_APDU_scanRequest)
  49         {
  50             std::cout << "Got Z3950 Scan PDU\n";
  51             //Z_ScanRequest *req = gdu->u.z3950->u.scanRequest;
  52             //package.request() = gdu;
  53         }
  54
  55         package.move();
  56     };
  57 };
  58
  59 void check_query_rewrite_init(yp2::RouterChain &router)
  60 {
  61     //std::cout << "QUERY REWRITE INIT\n";
  62
  63     // Create package with Z39.50 init request in it
  64     yp2::Package pack;
  65
  66     yp2::odr odr;
  67     Z_APDU *apdu = zget_APDU(odr, Z_APDU_initRequest);
  68
  69     pack.request() = apdu;
  70     // Done creating query.
  71
  72     // Put it in router
  73     pack.router(router).move();
  74
  75     // Inspect bounced back request
  76     //yazpp_1::GDU *gdu = &pack.response();
  77     yazpp_1::GDU *gdu = &pack.request();
  78
  79     Z_GDU *z_gdu = gdu->get();
  80
  81     //std::cout << "Z_GDU " << z_gdu << "\n";
  82     BOOST_CHECK(z_gdu);
  83     if (z_gdu) {
  84         BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
  85         BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_initRequest);
  86     }
  87 }
  88
  89 void check_query_rewrite_search(yp2::RouterChain &router,
  90                                 std::string query_in,
  91                                 std::string query_expect)
  92 {
  93     //std::cout << "QUERY REWRITE SEARCH "
  94     //          << query_in << " " << query_expect << "\n";
  95
  96     // Create package with Z39.50 search request in it
  97     yp2::Package pack;
  98
  99     yp2::odr odr;
 100     Z_APDU *apdu = zget_APDU(odr, Z_APDU_searchRequest);
 101
 102     // create package PQF query here
 103     yp2::util::pqf(odr, apdu, query_in);
 104
 105     // create package PDF database info (needed!)
 106     apdu->u.searchRequest->num_databaseNames = 1;
 107     apdu->u.searchRequest->databaseNames
 108         = (char**)odr_malloc(odr, sizeof(char *));
 109     apdu->u.searchRequest->databaseNames[0] = odr_strdup(odr, "Default");
 110
 111     // Done creating request package
 112     pack.request() = apdu;
 113
 114     // Put it in router
 115     pack.router(router).move();
 116
 117     // Inspect bounced back request
 118     //yazpp_1::GDU *gdu = &pack.response();
 119     yazpp_1::GDU *gdu = &pack.request();
 120
 121     Z_GDU *z_gdu = gdu->get();
 122     //std::cout << "Z_GDU " << z_gdu << "\n";
 123
 124     BOOST_CHECK(z_gdu);
 125     if (z_gdu) {
 126         BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
 127         BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_searchRequest);
 128
 129         // take query out of package again and check rewrite
 130         //std::string query_changed = yp2::util::apduToPqf(apdu);
 131         std::string query_changed
 132             = yp2::util::zQueryToString(z_gdu->u.z3950->u.searchRequest->query);
 133
 134         //BOOST_CHECK_EQUAL(query_in, query_expect);
 135
 136         //std::cout << "'" << query_expect << "'\n";
 137         //std::cout << "'" << query_changed << "'\n";
 138
 139         // need ugly whitespace here ..
 140         //BOOST_CHECK_EQUAL(query_expect, std::string(" ") + query_changed);
 141         BOOST_CHECK_EQUAL(query_expect, query_changed);
 142     }
 143 }
 144
 145
 146 BOOST_AUTO_UNIT_TEST( test_filter_query_rewrite_1 )
 147 {
 148     try
 149     {
 150         yp2::filter::QueryRewrite f_query_rewrite;
 151     }
 152     catch ( ... ) {
 153         BOOST_CHECK (false);
 154     }
 155 }
 156
 157 BOOST_AUTO_UNIT_TEST( test_filter_query_rewrite2 )
 158 {
 159     try
 160     {
 161         yp2::RouterChain router;
 162
 163         yp2::filter::QueryRewrite f_query_rewrite;
 164         //FilterBounceZ3950 f_bounce_z3950;
 165
 166         router.append(f_query_rewrite);
 167         //router.append(f_bounce_z3950);
 168
 169         check_query_rewrite_init(router);
 170         check_query_rewrite_search(router,
 171                                    "@attrset Bib-1 @attr 1=4 the",
 172                                    "@attrset Bib-1 @attr 1=4 the");
 173
 174     }
 175     catch ( ... ) {
 176         BOOST_CHECK (false);
 177     }
 178 }
 179
 180 /*
 181  * Local variables:
 182  * c-basic-offset: 4
 183  * indent-tabs-mode: nil
 184  * c-file-style: "stroustrup"
 185  * End:
 186  * vim: shiftwidth=4 tabstop=8 expandtab
 187  */