senf/PPI/SocketSink.test.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2007
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
   6 //     Stefan Bund <g0dil@berlios.de>
   7 //
   8 // This program is free software; you can redistribute it and/or modify
   9 // it under the terms of the GNU General Public License as published by
  10 // the Free Software Foundation; either version 2 of the License, or
  11 // (at your option) any later version.
  12 //
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17 //
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the
  20 // Free Software Foundation, Inc.,
  21 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22
  23 /** \file
  24     \brief SocketSink unit tests */
  25
  26 //#include "SocketSink.test.hh"
  27 //#include "SocketSink.test.ih"
  28
  29 // Custom includes
  30 #include <senf/Socket/Protocols/INet/UDPSocketHandle.hh>
  31 #include <senf/Socket/Protocols/INet/ConnectedUDPSocketHandle.hh>
  32 #include "SocketSource.hh"
  33 #include "DebugModules.hh"
  34 #include "SocketSink.hh"
  35 #include "Setup.hh"
  36
  37 #include <senf/Utils/auto_unit_test.hh>
  38 #include <boost/test/test_tools.hpp>
  39
  40 #define prefix_
  41 //-/////////////////////////////////////////////////////////////////////////////////////////////////
  42 namespace ppi = senf::ppi;
  43 namespace module = ppi::module;
  44 namespace debug = module::debug;
  45
  46 namespace {
  47     void timeout() {
  48         senf::scheduler::terminate();
  49     }
  50
  51     int base_pid = 0;
  52
  53     unsigned port(unsigned i)
  54     {
  55         if (! base_pid)
  56             base_pid = ::getpid();
  57         return 23456u + (((base_pid^(base_pid>>8)^(base_pid>>16)^(base_pid>>24))&0xff)<<2) + i;
  58     }
  59
  60     std::string localhost4str(unsigned i)
  61     {
  62         return (boost::format("localhost:%d") % port(i)).str();
  63     }
  64
  65     std::string localhost6str(unsigned i)
  66     {
  67         return (boost::format("[::1]:%d") % port(i)).str();
  68     }
  69 }
  70
  71 SENF_AUTO_UNIT_TEST(passiveSocketSink)
  72 {
  73     senf::ConnectedUDPv4ClientSocketHandle outputSocket (
  74         senf::INet4SocketAddress(localhost4str(0)));
  75     module::PassiveSocketSink<> udpSink(outputSocket);
  76     debug::ActiveSource source;
  77     ppi::connect(source, udpSink);
  78
  79     std::string data ("TEST");
  80     senf::Packet p (senf::DataPacket::create(data));
  81
  82     senf::UDPv4ClientSocketHandle inputSocket;
  83     inputSocket.bind(senf::INet4SocketAddress(localhost4str(0)));
  84     senf::ppi::init();
  85     source.submit(p);
  86
  87     std::string input (inputSocket.read());
  88     BOOST_CHECK_EQUAL( data, input );
  89 }
  90
  91 SENF_AUTO_UNIT_TEST(activeSocketSink)
  92 {
  93     senf::ConnectedUDPv4ClientSocketHandle outputSocket (
  94         senf::INet4SocketAddress(localhost4str(0)));
  95     module::ActiveSocketSink<> udpSink(outputSocket);
  96     debug::PassiveSource source;
  97     ppi::connect(source, udpSink);
  98
  99     std::string data ("TEST");
 100     senf::Packet p (senf::DataPacket::create(data));
 101
 102     senf::UDPv4ClientSocketHandle inputSocket;
 103     inputSocket.bind(senf::INet4SocketAddress(localhost4str(0)));
 104     senf::scheduler::TimerEvent timer (
 105         "activeSocketSink test timer", &timeout,
 106         senf::ClockService::now() + senf::ClockService::milliseconds(100));
 107     source.submit(p);
 108     senf::ppi::run();
 109
 110     std::string input (inputSocket.read());
 111     BOOST_CHECK_EQUAL( data, input );
 112 }
 113
 114 //-/////////////////////////////////////////////////////////////////////////////////////////////////
 115 #undef prefix_
 116
 117 \f
 118 // Local Variables:
 119 // mode: c++
 120 // fill-column: 100
 121 // comment-column: 40
 122 // c-file-style: "senf"
 123 // indent-tabs-mode: nil
 124 // ispell-local-dictionary: "american"
 125 // compile-command: "scons -u test"
 126 // End: