/***************************************************************************
**
** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).
** All rights reserved.
** Contact: Nokia Corporation (directui@nokia.com)
**
** This file is part of applauncherd
**
** If you have questions regarding the use of this file, please contact
** Nokia at directui@nokia.com.
**
** This library is free software; you can redistribute it and/or
** modify it under the terms of the GNU Lesser General Public
** License version 2.1 as published by the Free Software Foundation
** and appearing in the file LICENSE.LGPL included in the packaging
** of this file.
**
****************************************************************************/

#include "ut_connection.h"
#include "connection.h"
#include <sys/socket.h>
#include <sys/un.h>
#include <errno.h>

// Redefine some methods for Connection class
class MyConnection : public Connection
{
public:
    int nextMsg;
    char* nextStr;

    MyConnection(const string socketId, bool testMode);

private:
    bool recvMsg(uint32_t *msg);
    const char* recvStr();
};

MyConnection::MyConnection(const string socketId, bool testMode) :
    Connection(socketId, testMode),
    nextMsg(0),
    nextStr(NULL)
{}

bool MyConnection::recvMsg(uint32_t *msg)
{
    *msg = nextMsg;
    return true;
}

const char* MyConnection::recvStr()
{
    return nextStr;
}

Ut_Connection::Ut_Connection()
{}

Ut_Connection::~Ut_Connection()
{}

void Ut_Connection::initTestCase()
{}

void Ut_Connection::cleanupTestCase()
{}

/*
 * Check that socket gets initialized for provided socket id
 */
void Ut_Connection::testInitConnection()
{
    unsigned int prevNum = Connection::socketPool.size();
    Connection::initSocket("aaa");
    Connection::initSocket("bbb");
    Connection::initSocket("bbb");
    Connection::initSocket("aaa");

    QVERIFY(Connection::socketPool.size() == prevNum + 2);
    QVERIFY(Connection::findSocket("aaa") != -1);
    QVERIFY(Connection::findSocket("ccc") == -1);
    QVERIFY(Connection::findSocket("bbb") != -1);

    unlink("aaa");
    unlink("bbb");
}

/* 
 * Test that socket creation / closing works.
 */
void Ut_Connection::testSocket()
{
    const char* socketName = "testAccept";

    Connection::initSocket(socketName);
    MyConnection* conn = new MyConnection(socketName, false);
    conn->m_fd = 1000;

    QVERIFY(conn->m_fd > 0);

    conn->close();
    QVERIFY(conn->m_fd == -1);

    unlink(socketName);
}

/* 
 * Check that env variable passed from invoker will 
 * be set in launcher process.
 *
 * Run in the test mode (no sockets really created).
 */
void Ut_Connection::testGetEnv()
{
    QVERIFY(getenv("MY_TEST_ENV_VAR") == NULL);
    QVERIFY(getenv("PATH") != NULL);

    const char* socketName = "testGetEnv";
    MyConnection* conn = new MyConnection(socketName, true);

    char* envVar = strdup("MY_TEST_ENV_VAR=3");

    conn->nextMsg = 1;
    conn->nextStr = envVar; 

    QVERIFY(conn->receiveEnv() == true);
    QVERIFY(getenv("MY_TEST_ENV_VAR") != NULL);
    QVERIFY(getenv("PATH") != NULL);

    delete envVar;
}

/*
 * Check getAppName() function correctness
 *
 * Run in the test mode (no sockets really created).
 */
void Ut_Connection::testGetAppName()
{
    const char* socketName = "testGetAppName";
    MyConnection* conn = new MyConnection(socketName, true);

    // Wrong type of message
    conn->nextMsg = INVOKER_MSG_EXEC;
    string wrongStr = conn->receiveAppName();
    QVERIFY(wrongStr.empty());

    // Empty app name
    conn->nextMsg = INVOKER_MSG_NAME;
    conn->nextStr = NULL;
    string emptyName = conn->receiveAppName();
    QVERIFY(emptyName.empty());

    // Real name
    string realName("looooongApplicationName");
    char* dupName = strdup(realName.c_str());

    conn->nextMsg = INVOKER_MSG_NAME;
    conn->nextStr = dupName;

    string resName = conn->receiveAppName();
    QVERIFY(!resName.empty());
    QVERIFY(resName.compare(realName) == 0);
}

QTEST_APPLESS_MAIN(Ut_Connection);