/*

 * 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:

 */

// System includes

#include <HbApplication>

#include <HbDocumentLoader>

#include <HbStackedWidget>

#include <HbRadioButtonList>

#include <HbAction>

#include <HbLineEdit>

#include <HbLabel>

#include <HbCheckBox>

#include <QGraphicsWidget>

#include <QObjectList>

#include <QtCore>

#include <QTest>

#include <QDebug>

#include <QList>

#include <HbListWidget>

#include <HbListWidgetItem>

#include <HbParameterLengthLimiter>

// User includes

#include "testwlanwizardcontext.h"

#include "hbautotest.h"

#include "wlanwizard.h"

#include "wlanwizard_p.h"

#include "wlanwizardpageinternal.h"

#include "wlanwizardhelper.h"

#include "wlanqtutils_context.h"

#include "wlanqtutils.h"

#include "wlanqtutilsap.h"

#include "eapqtconfiginterface_context.h"

#include "eapwizard.h"

#include "eapwizard_p.h"

#include "eapwizardpage.h"

#include "wpswizardpage.h"

#include "wlanmgmtclient_context.h"

#include "wpswizardstepfour.h"

#include "wpswizardstepfour_p.h"

#include "cwpsactiverunner.h"

// External function prototypes

// Local constants

class TestRadioButtonList: public HbRadioButtonList

{

public:

    void emitActivated(const QModelIndex &modelIndex)

    { HbRadioButtonList::emitActivated(modelIndex); }

};

// ---------------------------------------------------------

// FRAMEWORK FUNCTIONS

// ---------------------------------------------------------

ContextWlanApList::ContextWlanApList()

{

}

ContextWlanApList::~ContextWlanApList()

{

    clear();

}

void ContextWlanApList::Add(QString name, int netMode, int secMode, bool wpaPskInUse,

    bool wpsSupported, int signalStrength)

{

    QSharedPointer<WlanQtUtilsAp> temp = QSharedPointer<WlanQtUtilsAp>(new WlanQtUtilsAp());

    temp->setValue(WlanQtUtilsAp::ConfIdSsid, name);

    temp->setValue(WlanQtUtilsAp::ConfIdConnectionMode, netMode);

    temp->setValue(WlanQtUtilsAp::ConfIdSecurityMode, secMode);

    temp->setValue(WlanQtUtilsAp::ConfIdWpaPskUse, wpaPskInUse);

    temp->setValue(WlanQtUtilsAp::ConfIdWpsSupported, wpsSupported);

    temp->setValue(WlanQtUtilsAp::ConfIdSignalStrength, signalStrength);

    mList.append(temp);

}

void ContextWlanApList::clear()

{

    mList.clear();

}

TestWlanWizardContext::TestWlanWizardContext() :

    mView(NULL),

    mMainWindow(NULL),

    mWlanQtUtilsContext(NULL),

    mEapQtUtilsContext(NULL),

    mWlanMgmtClientContext(NULL),

    mApList(NULL),

    mApOpenList(NULL)   

{

}

TestWlanWizardContext::~TestWlanWizardContext()

{

}

/**

 * This function will be called before the first test function is executed.

 */

void TestWlanWizardContext::initTestCase()

{

    qDebug("TestWlanWizardContext::initTestCase");

    mMainWindow = new HbAutoTestMainWindow;

    mView = new TestView();

    mMainWindow->addView(mView);

    mMainWindow->setCurrentView(mView);

    mMainWindow->installEventFilter(this);

    mMainWindow->show();

    QTest::qWait(1);

    while (!mEvent) {

        QTest::qWait(WaitTimeForUi);

    }

    mEvent = false;

    mApList = new ContextWlanApList();

    mApOpenList = new ContextWlanApList();

}

/**

 * This function will be called after the last test function was executed.

 */

void TestWlanWizardContext::cleanupTestCase()

{

    qDebug("TestWlanWizardContext::cleanupTestCase");

    qDebug("delete mApList");

    delete mApList;

    mApList = NULL;

    delete mApOpenList;

    mApOpenList = NULL;

    qDebug("delete mMainWindow");

    mMainWindow->setAttribute( Qt::WA_DeleteOnClose, true );

    mMainWindow->close();

    mMainWindow = NULL;

    QTest::qWait(WaitTimeForUi);

    qDebug("TestWlanWizardContext::cleanupTestCase exit");

}

/**

 * This function will be called before each test function is executed.

 */

void TestWlanWizardContext::init()

{

    qDebug("TestWlanWizardContext::init()");

    mView->createWizard();

    mWlanQtUtilsContext = new WlanQtUtilsContext(mView->mWizard->d_ptr->mWlanQtUtils.data());

    mEapQtUtilsContext = new EapQtConfigInterfaceContext();

    mEapQtUtilsContext->setObject(getEapQtConfig());

    mWlanMgmtClientContext = new WlanMgmtClientContext();

    //mWlanMgmtClientContext->setObject(mWlanMgmtClient);

    WpsPageStepFour* wps = (WpsPageStepFour*)(mView->mWizard->d_ptr->mPages[WpsWizardPage::PageWpsWizardStep4]);

    Q_ASSERT(wps);

    mWlanMgmtClientContext->setObject(wps->d_ptr->mWpsActiveRunner->iWLANMgmtClient);

}

/**

 * This function will be called after each test function is executed.

 */

void TestWlanWizardContext::cleanup()

{

    qDebug("TestWlanWizardContext::cleanup()");

    delete mWlanQtUtilsContext;

    delete mEapQtUtilsContext;

    delete mWlanMgmtClientContext;

    mView->deleteWizard();

    mApList->clear();

    mApOpenList->clear();

    QTest::qWait(1);

}

/*!

 * Filter to catch focus event to the text editor widget.

 */

bool TestWlanWizardContext::eventFilter(QObject *obj, QEvent *event)

{

    if (obj == mMainWindow && event->type() == QEvent::Show) {

        mMainWindow->removeEventFilter(this);

        mEvent = true;

    }

    return false;

}

QGraphicsWidget* TestWlanWizardContext::findChildItem(const QString &itemName, QGraphicsItem *widget)

{

    QList<QGraphicsItem*> list = widget->childItems();

    for (int i = 0; i < list.size(); i++) {

        QGraphicsItem* item = list[i];

        QGraphicsWidget* widget = NULL;

        if (item->isWidget()) {

            widget = (QGraphicsWidget*)item;

        }

        if (widget && widget->objectName() == itemName) {

            return widget;

        }

        else if ((item = findChildItem(itemName, item))) {

            return (QGraphicsWidget*)item;

        }

    }

    return 0;

}

bool TestWlanWizardContext::verifyStatus(WizardStatusSignal status, int iapId)

{

    return mView->verifyStatus(status, iapId);

}

bool TestWlanWizardContext::verifyCurrentPage(int pageId, int retries, int wait_ms)

{

    bool ret = true;

    for (int i = 0; i < retries; i++) {

        QTest::qWait(wait_ms);

        WlanWizardPrivate *pPrivate = mView->mWizard->d_ptr;

        HbWidget* current = qobject_cast<HbWidget*> (pPrivate->mStackedWidget->currentWidget());

        WlanWizardPage *page = pPrivate->mPageMapper[current];

        // TODO: verify title in this method

        if (page == pPrivate->mPages[pageId]) {

            switch(pageId) {

            case WlanWizardPageInternal::PageNetworkMode:

                return verifyDialogText("dialog_6", hbTrId("txt_occ_dialog_select_network_mode_and_status"));

            case WlanWizardPageInternal::PageNetworkSecurity:

                return verifyDialogText("dialog_6", hbTrId("txt_occ_dialog_select_network_security_mode"));

            case WlanWizardPageInternal::PageSsid:

                return verifyDialogText("dialog", hbTrId("txt_occ_dialog_insert_the_name_of_the_new_wlan_net"));

            case WlanWizardPageInternal::PageSummary:

                if (pPrivate->mTitle->plainText() != hbTrId("txt_occ_title_wlan_setup_wizard_summary")){

                    qWarning("TestWlanWizardContext::verifyCurrentPage: Invalid title");

                    return false;

                }

                return true;

            case EapWizardPage::PageOuterType:

                return verifyDialogText("title", hbTrId("txt_occ_title_select_eap_type"));

            case EapWizardPage::PageCertificateCa:

                return verifyDialogText("title", hbTrId("txt_occ_title_select_authority_certificate"));

            case EapWizardPage::PageCertificateUser:

                return verifyDialogText("title", hbTrId("txt_occ_title_select_user_certificate"));

            case EapWizardPage::PageNewPacStorePassword:

                ret &= verifyDialogText("setlabel_55", hbTrId("txt_occ_setlabel_new_pac_store_password"));

                ret &= verifyDialogText("setlabel_56", hbTrId("txt_occ_setlabel_confirm_password"));

                return ret;

            case EapWizardPage::PagePromptPacStorePassword:

                return verifyDialogText("setlabel_59", hbTrId("txt_occ_setlabel_eap_pac_store_password"));

            case WpsWizardPage::PageWpsWizardStep2:

                return verifyDialogText("label", hbTrId("txt_occ_dialog_selected_network_supports_wifi_pro"));

                break;

            case WpsWizardPage::PageWpsWizardStep3_Button:

                return verifyDialogText("label_heading", hbTrId("txt_occ_dialog_first_press_button_on_the_wireless"));

                break;

            case WpsWizardPage::PageWpsWizardStep3_Number:

                // TODO: something better here??

                return true;

            case WpsWizardPage::PageWpsWizardStep4:

                return verifyDialogText("label_heading", hbTrId("txt_occ_dialog_negotiating"));

            case WpsWizardPage::PageWpsWizardStep5:

                return verifyDialogText("label", hbTrId("txt_occ_dialog_settings_received_for_multiple_wlan"));

            default:

                return false;

            }

        }

    }

    qWarning("verifyCurrentPage: expected: %d", pageId);

    return false;

}

bool TestWlanWizardContext::verifyCurrentPageWithInfo(

    int pageId, const QString &info, int retries, int wait_ms)

{

    bool ret = true;

    for (int i = 0; i < retries; i++) {

        QTest::qWait(wait_ms);

        WlanWizardPrivate *pPrivate = mView->mWizard->d_ptr;

        HbWidget* current = qobject_cast<HbWidget*> (pPrivate->mStackedWidget->currentWidget());

        WlanWizardPage *page = pPrivate->mPageMapper[current];

        // TODO: verify title in this method

        if (page == pPrivate->mPages[pageId]) {

            switch(pageId) {

            case WlanWizardPageInternal::PageGenericError:

                return verifyDialogText("dialog", info);

            case WlanWizardPageInternal::PageKeyQuery:

                return verifyDialogText("dialog", HbParameterLengthLimiter("txt_occ_dialog_enter_key_for_1").arg(info));

            case WlanWizardPageInternal::PageProcessSettings:

                return verifyDialogText("dialog", HbParameterLengthLimiter("txt_occ_dialog_checking_connection_to_1").arg(info));

            case WlanWizardPageInternal::PageScanning:

                return verifyDialogText("dialog", HbParameterLengthLimiter("txt_occ_dialog_searching").arg(info));

            case EapWizardPage::PageUsernamePassword:

                ret &= verifyDialogText(

                    "setlabel_55",

                    HbParameterLengthLimiter("txt_occ_setlabel_user_name_for_1").arg(info));

                ret &= verifyDialogText("setlabel_56", hbTrId("txt_occ_setlabel_eap_password"));

                return ret;

            case EapWizardPage::PageIdentity:

                ret &= verifyDialogText("setlabel_53", HbParameterLengthLimiter("txt_occ_setlabel_user_name_for_1").arg(info));

                ret &= verifyDialogText("setlabel_53_val", hbTrId("txt_occ_setlabel_user_name_val_generate_automatic"));

                ret &= verifyDialogText("setlabel_54", hbTrId("txt_occ_setlabel_eap_realm"));

                ret &= verifyDialogText("setlabel_54_val", hbTrId("txt_occ_setlabel_eap_realm_val_generate_automatically"));

                return ret;

            case EapWizardPage::PageInnerTypePeap:

            case EapWizardPage::PageInnerTypeEapTtls:

                return verifyDialogText("title", HbParameterLengthLimiter("txt_occ_title_select_innear_eap_type_for_1").arg(info));

            default:

                return verifyCurrentPage(pageId, retries, wait_ms);

            }

        }

    }

    qWarning("verifyCurrentPage: expected: %d", pageId);

    return false;

}

bool TestWlanWizardContext::verifyDialogText(const QString objName, const QString text)

{

    HbWidget* current = qobject_cast<HbWidget*> (

        mView->mWizard->d_ptr->mStackedWidget->currentWidget());

    HbWidget* widget = qobject_cast<HbWidget*> (

        findChildItem(objName, current));

    if (widget) {

        HbLabel *label = qobject_cast<HbLabel*>(widget);

        if (label) {

            if(label->plainText() == text) {

                if (!label->isVisible()){

                    qWarning("verifyDialogText: not visible");

                    return false;

                }

                return true;

            } else {

                qWarning("verifyDialogText: not match");

                qDebug() << "expect: " << text;

                qDebug() << "actual: " << label->plainText();

                return false;

            }

        }

        HbCheckBox *checkBox = qobject_cast<HbCheckBox*>(widget);

        if (checkBox) {

            if(checkBox->text() == text) {

                if (!checkBox->isVisible()){

                    qWarning("verifyDialogText: not visible");

                    return false;

                }

                return true;

            } else {

                qWarning("verifyDialogText: not match");

                qDebug() << "expect: " << text;

                qDebug() << "actual: " << checkBox->text();

                return false;

            }

        }

    }

    qWarning("verifyDialogText: object not found");

    return false;

}

bool TestWlanWizardContext::verifyActionButtons(ButtonStatus prevStatus,

    ButtonStatus cancelStatus,

    ButtonStatus nextStatus,

    ButtonStatus finishStatus)

{

    WlanWizardPrivate *pPrivate = mView->mWizard->d_ptr;

    ButtonStatus prevReally = ButtonHidden;

    ButtonStatus cancelReally = ButtonHidden;

    ButtonStatus nextReally = ButtonHidden;

    ButtonStatus finishReally = ButtonHidden;

    if(pPrivate->mActionPrevious->isVisible()) {

        if(pPrivate->mActionPrevious->isEnabled()) {

            prevReally = ButtonEnabled;

        } else {

            prevReally = ButtonDisabled;

        }

    }

    if(pPrivate->mActionCancel->isVisible()) {

        if(pPrivate->mActionCancel->isEnabled()) {

            cancelReally = ButtonEnabled;

        } else {

            cancelReally = ButtonDisabled;

        }

    }

    if(pPrivate->mActionNext->isVisible()) {

        if(pPrivate->mActionNext->isEnabled()) {

            nextReally = ButtonEnabled;

        } else {

            nextReally = ButtonDisabled;

        }

    }

    if(pPrivate->mActionFinish->isVisible()) {

        if(pPrivate->mActionFinish->isEnabled()) {

            finishReally = ButtonEnabled;

        } else {

            finishReally = ButtonDisabled;

        }

    }

    bool ret = true;

    if (prevReally != prevStatus){

        qWarning("Previous Button: really: %d, status: %d", prevReally, prevStatus);

        ret = false;

    }

    if (cancelReally != cancelStatus){

        qWarning("Cancel Button: really: %d, status: %d", cancelReally, cancelStatus);

        ret = false;

    }

    if (nextReally != nextStatus){

        qWarning("Next Button: really: %d, status: %d", nextReally, nextStatus);

        ret = false;

    }

    if (finishReally != finishStatus){

        qWarning("Finish Button: really: %d, status: %d", finishReally, finishStatus);

        ret = false;

    }

    return ret;

}

bool TestWlanWizardContext::verifySummaryPage(

    const QString &ssid, 

    int netMode, 

    bool hidden,

    int secMode, 

    bool useWpaPsk,

    const QString &destination, 

    int outerType,

    int innerType,

    bool eapFastProvMode)

{

    bool ret = true;

    int i = 0;

    HbListWidget* listWidget = qobject_cast<HbListWidget*> (

        mView->mWizard->d_ptr->mStackedWidget->currentWidget());

    if (!listWidget){

        qWarning("verifySummaryPage: ListWidget not found");

        return false;

    }

    // VERIFY: NETWORK NAME

    HbListWidgetItem *item = listWidget->item(i++);

    if (item->text() != hbTrId("txt_occ_dblist_network_name")) {

        qWarning("verifySummaryPage: no match network name");

        ret = false;

    }

    if (item->secondaryText() != ssid){

        qWarning("verifySummaryPage: no match ssid");

        ret = false;

    }

    // VERIFY: NETWORK MODE

    item = listWidget->item(i++);

    if (item->text() != hbTrId("txt_occ_dblist_network_mode")) {

        qWarning("verifySummaryPage: no match network mode");

        ret = false;

    }

    QString netModeString(toNetworkModeString(netMode, hidden));

    if (item->secondaryText() != netModeString){

        qWarning("verifySummaryPage: no match network mode value");

        qDebug() << "Actual: " << item->secondaryText();

        qDebug() << "Expected: " << netModeString;

        ret = false;

    }

    // VERIFY: SECURITY MODE

    item = listWidget->item(i++);

    if (item->text() != hbTrId("txt_occ_dblist_security_mode")) {