src/test_filter_frontend_net.cpp

   1
   2 #include "config.hpp"
   3 #include <iostream>
   4 #include <stdexcept>
   5
   6 #include "filter_frontend_net.hpp"
   7
   8 #include "router.hpp"
   9 #include "session.hpp"
  10 #include "package.hpp"
  11
  12 #define BOOST_AUTO_TEST_MAIN
  13 #include <boost/test/auto_unit_test.hpp>
  14
  15 using namespace boost::unit_test;
  16
  17 class FilterInit: public yp2::Filter {
  18 public:
  19     yp2::Package & process(yp2::Package & package) const {
  20
  21         Z_GDU *gdu = package.request().get();
  22         if (package.session().is_closed())
  23         {
  24             // std::cout << "Got Close. Sending nothing\n";
  25         }
  26
  27         if (gdu)
  28         {
  29             // std::cout << "Got PDU. Sending init response\n";
  30             ODR odr = odr_createmem(ODR_ENCODE);
  31             Z_APDU *apdu = zget_APDU(odr, Z_APDU_initResponse);
  32
  33             apdu->u.initResponse->implementationName = "YP2/YAZ";
  34
  35             package.response() = apdu;
  36             odr_destroy(odr);
  37         }
  38         return package.move();
  39     };
  40 };
  41
  42
  43 BOOST_AUTO_TEST_CASE( test_filter_frontend_net_1 )
  44 {
  45     try
  46     {
  47         {
  48             yp2::FilterFrontendNet nf;
  49         }
  50         BOOST_CHECK(true);
  51     }
  52     catch ( ... ) {
  53         BOOST_CHECK (false);
  54     }
  55 }
  56
  57 BOOST_AUTO_TEST_CASE( test_filter_frontend_net_2 )
  58 {
  59     try
  60     {
  61         {
  62             yp2::RouterChain router;
  63
  64             FilterInit tf;
  65
  66             router.rule(tf);
  67
  68             // Create package with Z39.50 init request in it
  69             yp2::Session session;
  70             yp2::Origin origin;
  71             yp2::Package pack_in(session, origin);
  72
  73             ODR odr = odr_createmem(ODR_ENCODE);
  74             Z_APDU *apdu = zget_APDU(odr, Z_APDU_initRequest);
  75
  76             pack_in.request() = apdu;
  77             odr_destroy(odr);
  78             // Done creating query.
  79
  80             // Put it in router
  81             pack_in.router(router).move();
  82
  83             // Inspect that we got Z39.50 init response
  84             yazpp_1::GDU *gdu = &pack_in.response();
  85
  86             Z_GDU *z_gdu = gdu->get();
  87             BOOST_CHECK(z_gdu);
  88             if (z_gdu) {
  89                 BOOST_CHECK_EQUAL(z_gdu->which, Z_GDU_Z3950);
  90                 BOOST_CHECK_EQUAL(z_gdu->u.z3950->which, Z_APDU_initResponse);
  91             }
  92         }
  93         BOOST_CHECK(true);
  94     }
  95     catch ( ... ) {
  96         BOOST_CHECK (false);
  97     }
  98 }
  99
 100 BOOST_AUTO_TEST_CASE( test_filter_frontend_net_3 )
 101 {
 102     try
 103     {
 104         {
 105             yp2::RouterChain router;
 106             yp2::FilterFrontendNet filter_front;
 107             filter_front.listen_address() = "unix:socket";
 108             filter_front.listen_duration() = 2;  // listen a short time only
 109             router.rule(filter_front);
 110
 111             FilterInit filter_init;
 112             router.rule(filter_init);
 113
 114             yp2::Session session;
 115             yp2::Origin origin;
 116             yp2::Package pack_in(session, origin);
 117
 118             pack_in.router(router).move();
 119         }
 120         BOOST_CHECK(true);
 121     }
 122     catch ( ... ) {
 123         BOOST_CHECK (false);
 124     }
 125 }
 126
 127 /*
 128  * Local variables:
 129  * c-basic-offset: 4
 130  * indent-tabs-mode: nil
 131  * End:
 132  * vim: shiftwidth=4 tabstop=8 expandtab
 133  */