// **********************************************************************
//
// Copyright (c) 2003-present ZeroC, Inc. All rights reserved.
//
// **********************************************************************

#include <IceUtil/IceUtil.h>
#include <Ice/Ice.h>
#include <Glacier2/Glacier2.h>

#include <TestHelper.h>

#include <iostream>
#include <iomanip>
#include <list>

#include <Callback.h>

using namespace std;
using namespace Test;

namespace
{

class CallbackReceiverI : public Test::CallbackReceiver
{
public:

    CallbackReceiverI() : _received(false)
    {
    }

    virtual void callback(const Ice::Current&)
    {
        IceUtil::Monitor<IceUtil::Mutex>::Lock lock(_monitor);
        _received = true;
        _monitor.notify();
    }

    void waitForCallback()
    {
        IceUtil::Monitor<IceUtil::Mutex>::Lock lock(_monitor);
        while(!_received)
        {
            _monitor.wait();
        }
        _received = false;
    }

    IceUtil::Monitor<IceUtil::Mutex> _monitor;
    bool _received;
};
ICE_DEFINE_PTR(CallbackReceiverIPtr, CallbackReceiverI);

class Application : public Glacier2::Application
{
public:

    Application() : _restart(0), _destroyed(false), _receiver(ICE_MAKE_SHARED(CallbackReceiverI))
    {
    }

    virtual Glacier2::SessionPrxPtr
    createSession()
    {
        return ICE_UNCHECKED_CAST(Glacier2::SessionPrx, router()->createSession("userid", "abc123"));
    }

    virtual int
    runWithSession(int, char*[])
    {
        test(router());
        test(!categoryForClient().empty());
        test(objectAdapter());

        if(_restart == 0)
        {
            cout << "testing Glacier2::Application restart... " << flush;
        }
        Ice::ObjectPrxPtr base = communicator()->stringToProxy(
            "callback:" + TestHelper::getTestEndpoint(communicator()->getProperties()));
        CallbackPrxPtr callback = ICE_UNCHECKED_CAST(CallbackPrx, base);
        if(++_restart < 5)
        {
            CallbackReceiverPrxPtr receiver = ICE_UNCHECKED_CAST(CallbackReceiverPrx, addWithUUID(_receiver));
            callback->initiateCallback(receiver);
            _receiver->waitForCallback();
            restart();
        }
        cout << "ok" << endl;

        cout << "testing server shutdown... " << flush;
        callback->shutdown();
        cout << "ok" << endl;

        return 0;
    }

    virtual void sessionDestroyed()
    {
        _destroyed = true;
    }

    int _restart;
    bool _destroyed;
    CallbackReceiverIPtr _receiver;
};

} // anonymous namespace end

class Client : public Test::TestHelper
{
public:

    void run(int, char**);
};

void
Client::run(int argc, char** argv)
{
    Application app;
    Ice::InitializationData initData;
    initData.properties = createTestProperties(argc, argv);
    initData.properties->setProperty("Ice.Warn.Connections", "0");
    initData.properties->setProperty("Ice.Default.Router",
                                     "Glacier2/router:" + TestHelper::getTestEndpoint(initData.properties, 50));
    int status = app.main(argc, argv, initData);
    if(status != 0)
    {
        test(false);
    }

    initData.properties->setProperty("Ice.Default.Router", "");

    Ice::CommunicatorHolder communicator = initialize(argc, argv, initData);

    cout << "testing stringToProxy for process object... " << flush;
    Ice::ObjectPrxPtr processBase = communicator->stringToProxy("Glacier2/admin -f Process:" + getTestEndpoint(51));
    cout << "ok" << endl;

    cout << "testing checked cast for admin object... " << flush;
    Ice::ProcessPrxPtr process = ICE_CHECKED_CAST(Ice::ProcessPrx, processBase);
    test(process != 0);
    cout << "ok" << endl;

    cout << "testing Glacier2 shutdown... " << flush;
    process->shutdown();

    try
    {
        process->ice_ping();
        test(false);
    }
    catch(const Ice::LocalException&)
    {
        cout << "ok" << endl;
    }

    test(app._restart == 5);
    test(app._destroyed);
}

DEFINE_TEST(Client)