devel/libmdw/trunk/libmdw/src/unitTest.cpp

/***************************************************************************
 *   Copyright (C) 2010 by Jean-Baptiste Wons   *
 *   wonsjb@gmail.com   *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU Library General Public License as       *
 *   published by the Free Software Foundation; either version 2 of the    *
 *   License, or (at your option) any later version.                       *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU Library General Public     *
 *   License along with this program; if not, write to the                 *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif


#include <cstdlib>
#include <cassert>

#include "CommandHandler.h"
#include "TcpClientStream.h"
#include "TcpServer.h"
#include "TelnetProtocol.h"
#include "LineBufferProtocol.h"
#include "CommandManager.h"
#include "Tokeniser.h"
#include "CommandProtocol.h"
#include "Logger.h"
#include "IoStream.h"
#include "Exception.h"

using namespace mdw;

namespace
{

static bool _success = false;

class CommandHandlerTest : public CommandHandler
{
public:
    virtual void execute (const std::vector<std::string> args, int startArgs, Stream &output)
    {
        if (args[startArgs] == "true")
        {
            _success = true;
        }
    }
};

void testTelnetCommands()
{
    TcpServer aServer(35555);
    TelnetProtocol aTelnet(aServer);
    LineBufferProtocol aLineBuff(aTelnet);
    Tokeniser aTokeniser(aLineBuff);
    CommandProtocol aCommand(aTokeniser);
    CommandManager &cm = CommandManager::getInstance();
    cm.createNode("set").createNode("test").createNode("result", new CommandHandlerTest());
    TcpClientStream aClient("localhost", 35555);
    aCommand.getPendingCommands(1, 0);
    aClient << "set " << "te" << "st result " << "true\n";
    aCommand.getPendingCommands(1, 0);
    assert(_success);
}

}

int main (int argc, char *argv[])
{
    Logger::initialise(mdw::cout);
    try
    {
        CommandManager::initialise();
        testTelnetCommands();

        cout << "Success!" << endl;
        return 0;
    }
    catch (mdw::Exception &e)
    {
        e.log(cout);
        return -1;
    }
    catch (std::exception &e)
    {
        cout << "Error std::exception : " << e.what() << endl;
        return -1;
    }
    catch (...)
    {
        cout << "Unknown exception" << endl;
        return -1;
    }
}

---