src/test_filter_query_rewrite.cpp

   1 /* $Id: test_filter_query_rewrite.cpp,v 1.1 2006-01-19 12:18:09 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 && gdu->u.z3950->which ==
  34             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 void check_query_rewrite_search(yp2::RouterChain &router,
  89                                 std::string query_in,
  90                                 std::string query_out)
  91 {
  92     std::cout << "QUERY REWRITE SEARCH "
  93               << query_in << " " << query_out << "\n";
  94
  95     // Create package with Z39.50 search request in it
  96     yp2::Package pack;
  97
  98     yp2::odr odr;
  99     Z_APDU *apdu = zget_APDU(odr, Z_APDU_searchRequest);
 100
 101     // create package PQF query here
 102     yp2::util::pqf(odr, apdu, query_in);
 103
 104
 105     //apdu->u.searchRequest->num_databaseNames = 1;
 106     //apdu->u.searchRequest->databaseNames = (char**)
 107     //odr_malloc(odr, sizeof(char *));
 108     //apdu->u.searchRequest->databaseNames[0] = odr_strdup(odr, "Default");
 109
 110
 111     pack.request() = apdu;
 112     // Done creating query.
 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         // take query out of package again
 129         BOOST_CHECK_EQUAL(query_in, query_out);
 130     }
 131 }
 132
 133
 134 BOOST_AUTO_UNIT_TEST( test_filter_query_rewrite_1 )
 135 {
 136     try
 137     {
 138         yp2::filter::QueryRewrite f_query_rewrite;
 139     }
 140     catch ( ... ) {
 141         BOOST_CHECK (false);
 142     }
 143 }
 144
 145 BOOST_AUTO_UNIT_TEST( test_filter_query_rewrite2 )
 146 {
 147     try
 148     {
 149         yp2::RouterChain router;
 150
 151         yp2::filter::QueryRewrite f_query_rewrite;
 152         //FilterBounceZ3950 f_bounce_z3950;
 153
 154         router.append(f_query_rewrite);
 155         //router.append(f_bounce_z3950);
 156
 157         check_query_rewrite_init(router);
 158         check_query_rewrite_search(router, "@attr 1=4 the", "@attr 1=4 the");
 159
 160     }
 161     catch ( ... ) {
 162         BOOST_CHECK (false);
 163     }
 164 }
 165
 166 /*
 167  * Local variables:
 168  * c-basic-offset: 4
 169  * indent-tabs-mode: nil
 170  * c-file-style: "stroustrup"
 171  * End:
 172  * vim: shiftwidth=4 tabstop=8 expandtab
 173  */