/*
* Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description: RmsExtensionPlugin unit tests with Java Captain
*
*/

#include <string>
#include "TestHarness.h"
#include "comms.h"
#include "commsmessage.h"
#include "commsclientendpoint.h"
#include "monitor.h"

#include "rmsextensionplugin.h"

using namespace java::rms;
using namespace java::util;
using namespace java::comms;

class NotifyListener : public CommsListener
{
public:
    NotifyListener(Monitor& aMonitor, int aMsgCount) : mListenerCountMsg(0), mListenerCount(0),
            mMonitor(aMonitor), mMsgCount(aMsgCount) {};
    virtual void processMessage(CommsMessage& aMessage)
    {
        int id = aMessage.getMessageId();
        switch (id)
        {
        case MSG_ID_RECORD_CHANGE:
            if (--mMsgCount == 0)
            {
                mMonitor.notify();
            }
            break;
        case MSG_ID_LISTENER_COUNT:
        {
            mListenerCountMsg++;
            std::string name;
            aMessage >> name >> mListenerCount;
            break;
        }
        default:
            break;
        }
    }

    int mListenerCountMsg;
    int mListenerCount;
private:
    Monitor& mMonitor;
    int mMsgCount;
};


CommsMessage createAddListenerMsg()
{
    CommsMessage addMsg;
    addMsg.setModuleId(PLUGIN_ID_RMS_C);
    addMsg.setMessageId(MSG_ID_LISTENER_CHANGE);
    addMsg << "aaa" << ADD_LISTENER;
    return addMsg;
}

CommsMessage createRemoveListenerMsg()
{
    CommsMessage removeMsg;
    removeMsg.setModuleId(PLUGIN_ID_RMS_C);
    removeMsg.setMessageId(MSG_ID_LISTENER_CHANGE);
    removeMsg << "aaa" << REMOVE_LISTENER;
    return removeMsg;
}

CommsMessage createRecordChangedMsg()
{
    CommsMessage recordMsg;
    recordMsg.setModuleId(PLUGIN_ID_RMS_C);
    recordMsg.setMessageId(MSG_ID_RECORD_CHANGE);
    recordMsg << "aaa";
    return recordMsg;
}

TEST_GROUP(TestPlugin)
{
    TEST_SETUP()
    {
    }

    TEST_TEARDOWN()
    {
    }
};


TEST(TestPlugin, oneListener)
{
    CommsClientEndpoint a;
    std::auto_ptr<Monitor> monitor(Monitor::createMonitor());
    NotifyListener listener(*monitor, 1);
    a.registerDefaultListener(&listener);

    int rc = a.connect(IPC_ADDRESS_JAVA_CAPTAIN_C);
    CHECK(rc == 0);
    CommsMessage msg = createAddListenerMsg();
    CommsMessage receivedMsg;
    rc = a.sendReceive(msg, receivedMsg, WAIT_FOR_EVER);
    CHECK(rc == 0);
    listener.processMessage(receivedMsg);
    CHECK(listener.mListenerCountMsg == 1);

    CommsClientEndpoint dummy;
    dummy.connect(IPC_ADDRESS_JAVA_CAPTAIN_C);
    msg = createRecordChangedMsg();
    rc = dummy.send(msg);
    CHECK(rc == 0);

    monitor->wait();

    msg = createRemoveListenerMsg();
    rc = a.send(msg);
    CHECK(rc == 0);
}

TEST(TestPlugin, multipleListeners)
{
    // 1st listener
    CommsClientEndpoint a;
    std::auto_ptr<Monitor> monitorA(Monitor::createMonitor());
    NotifyListener listenerA(*monitorA, 2);
    a.registerDefaultListener(&listenerA);
    int rc = a.connect(IPC_ADDRESS_JAVA_CAPTAIN_C);
    CHECK(rc == 0);

    // 2nd listener
    CommsClientEndpoint b;
    std::auto_ptr<Monitor> monitorB(Monitor::createMonitor());
    NotifyListener listenerB(*monitorB, 2);
    b.registerDefaultListener(&listenerB);
    rc = b.connect(IPC_ADDRESS_JAVA_CAPTAIN_C);
    CHECK(rc == 0);

    // 3rd listener
    CommsClientEndpoint c;
    std::auto_ptr<Monitor> monitorC(Monitor::createMonitor());
    NotifyListener listenerC(*monitorC, 2);
    c.registerDefaultListener(&listenerC);
    rc = c.connect(IPC_ADDRESS_JAVA_CAPTAIN_C);
    CHECK(rc == 0);

    // add listeners
    CommsMessage msg = createAddListenerMsg();
    rc = a.send(msg);
    CHECK(rc == 0);
    rc = b.send(msg);
    CHECK(rc == 0);
    rc = c.send(msg);
    CHECK(rc == 0);

    // send record modified
    msg = createRecordChangedMsg();
    rc = a.send(msg);
    CHECK(rc == 0);
    rc = b.send(msg);
    CHECK(rc == 0);
    rc = c.send(msg);
    CHECK(rc == 0);

    monitorA->wait();
    monitorB->wait();
    monitorC->wait();

    CHECK(listenerA.mListenerCountMsg == 3);
    CHECK(listenerB.mListenerCountMsg == 2);
    CHECK(listenerC.mListenerCountMsg == 1);

    CHECK(listenerA.mListenerCount == 3);
    CHECK(listenerB.mListenerCount == 3);
    CHECK(listenerC.mListenerCount == 3);

    // remove listeners
    msg = createRemoveListenerMsg();
    rc = a.send(msg);
    CHECK(rc == 0);
    rc = b.send(msg);
    CHECK(rc == 0);
    rc = c.send(msg);
    CHECK(rc == 0);
}