senf/Utils/Console/UDPServer.test.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2009
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 //
   6 // The contents of this file are subject to the Fraunhofer FOKUS Public License
   7 // Version 1.0 (the "License"); you may not use this file except in compliance
   8 // with the License. You may obtain a copy of the License at
   9 // http://senf.berlios.de/license.html
  10 //
  11 // The Fraunhofer FOKUS Public License Version 1.0 is based on,
  12 // but modifies the Mozilla Public License Version 1.1.
  13 // See the full license text for the amendments.
  14 //
  15 // Software distributed under the License is distributed on an "AS IS" basis,
  16 // WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  17 // for the specific language governing rights and limitations under the License.
  18 //
  19 // The Original Code is Fraunhofer FOKUS code.
  20 //
  21 // The Initial Developer of the Original Code is Fraunhofer-Gesellschaft e.V.
  22 // (registered association), Hansastraße 27 c, 80686 Munich, Germany.
  23 // All Rights Reserved.
  24 //
  25 // Contributor(s):
  26 //   Stefan Bund <g0dil@berlios.de>
  27
  28 /** \file
  29     \brief UDPServer.test unit tests */
  30
  31 //#include "UDPServer.test.hh"
  32 //#include "UDPServer.test.ih"
  33
  34 // Custom includes
  35 #include "Console.hh"
  36 #include <senf/Scheduler/Scheduler.hh>
  37
  38 #include <senf/Utils/auto_unit_test.hh>
  39 #include <boost/test/test_tools.hpp>
  40
  41 #define prefix_
  42 //-/////////////////////////////////////////////////////////////////////////////////////////////////
  43
  44 namespace {
  45
  46     void timeout()
  47     {
  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     unsigned nread (0);
  66     unsigned flags (0);
  67     std::string data;
  68
  69     void read(senf::ConnectedUDPv4ClientSocketHandle socket, int ev)
  70     {
  71         flags |= ev;
  72         ++nread;
  73         data = socket.read();
  74     }
  75
  76     senf::ClockService::clock_type delay(unsigned ms)
  77     {
  78         return senf::ClockService::now() + senf::ClockService::milliseconds(ms);
  79     }
  80
  81 }
  82
  83 SENF_AUTO_UNIT_TEST(udpServer)
  84 {
  85     senf::console::UDPServer server (senf::INet4SocketAddress(localhost4str(0)));
  86     senf::ConnectedUDPv4ClientSocketHandle socket (senf::INet4SocketAddress(localhost4str(0)));
  87     senf::scheduler::TimerEvent timer ("udpServer test timer", &timeout);
  88     senf::scheduler::FdEvent fdev ("udpServer test fd", boost::bind(&read, socket, _1),
  89                                    socket, senf::scheduler::FdEvent::EV_READ);
  90
  91     nread = 0;
  92     flags = 0;
  93     data = "";
  94     socket.write(std::string("ll"));
  95     timer.timeout(delay(300));
  96     senf::scheduler::process();
  97     BOOST_CHECK_EQUAL( nread, 1 );
  98     BOOST_CHECK_EQUAL( flags, senf::scheduler::FdEvent::EV_READ );
  99     BOOST_CHECK_EQUAL( data, "sys/                        \n" );
 100
 101     // Check exception handling
 102     nread = 0;
 103     flags = 0;
 104     data = "";
 105     socket.write(std::string("sys"));
 106     timer.timeout(delay(300));
 107     senf::scheduler::process();
 108     BOOST_CHECK_EQUAL( nread, 1 );
 109     BOOST_CHECK_EQUAL( flags, senf::scheduler::FdEvent::EV_READ );
 110     BOOST_CHECK_EQUAL( data, "invalid command\n" "at <unknown>:1:4\n" );
 111
 112     // switch directory
 113     nread = 0;
 114     flags = 0;
 115     data = "";
 116     socket.write(std::string("cd sys"));
 117     timer.timeout(delay(300));
 118     senf::scheduler::process();
 119     BOOST_CHECK_EQUAL( nread, 1 );
 120     BOOST_CHECK_EQUAL( flags, senf::scheduler::FdEvent::EV_READ );
 121     BOOST_CHECK_EQUAL( data, std::string(1, '\0') );
 122
 123     // Check that we go back to the root dir for every packet
 124     nread = 0;
 125     flags = 0;
 126     data = "";
 127     socket.write(std::string("ll"));
 128     timer.timeout(delay(300));
 129     senf::scheduler::process();
 130     BOOST_CHECK_EQUAL( nread, 1 );
 131     BOOST_CHECK_EQUAL( flags, senf::scheduler::FdEvent::EV_READ );
 132     BOOST_CHECK_EQUAL( data, "sys/                        \n" );
 133 }
 134
 135 //-/////////////////////////////////////////////////////////////////////////////////////////////////
 136 #undef prefix_
 137
 138 \f
 139 // Local Variables:
 140 // mode: c++
 141 // fill-column: 100
 142 // comment-column: 40
 143 // c-file-style: "senf"
 144 // indent-tabs-mode: nil
 145 // ispell-local-dictionary: "american"
 146 // compile-command: "scons -u test"
 147 // End: