/*

* Copyright (c) 2009-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:

* This is the source file for testing Wlan Qt Utilities library.

*/

#include <QApplication>

#include <QtCore>

#include <QTest>

#include <QSignalSpy>

#include <QSharedPointer>

#include <rconnmon.h>

#include <nifvar.h>

#include <cmmanagerdefines_shim.h>

#include "wlanqtutilsap.h"

#include "wlanqtutilsiap.h"

#include "wlanqtutils.h"

#include "wlanqtutils_p.h"

#include "wlanqtutilsconmonwrapper.h"

#include "wlanqtutilsconmonwrapperinfo_symbian.h"

#include "testwlanqtutils.h"

#include "wlanqtutilstestcontext.h"

WlanQtUtilsTestContext testContext;

const QString TestWlanQtUtils::commsDatDefault_ = "default.cre";

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

// FRAMEWORK FUNCTIONS

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

/**

 * Test main function. Runs all test cases.

 */

int main(int argc, char *argv[])

{

    QApplication app(argc, argv);

    char *pass[3];

    pass[0] = argv[0];

    pass[1] = "-o"; 

    pass[2] = "c:\\data\\wlanqtutils_qtest_log.txt";

    TestWlanQtUtils tc;

    int res = QTest::qExec(&tc, 3, pass);

    return res;

}

/**

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

 */

void TestWlanQtUtils::initTestCase()

{

    wlanQtUtils_ = NULL;

    mSignalScanReady = NULL;

    mSignalScanApReady = NULL;

    mSignalScanDirectReady = NULL;

    signalWlanNetworkOpened_ = NULL;

    signalWlanNetworkClosed_ = NULL;

    signalIctResult_ = NULL;

    //If Active scheduler exists then don't install a new one as it will cause panic

    if (CActiveScheduler::Current() == NULL) {

        CActiveScheduler *scheduler = new CActiveScheduler();

        CActiveScheduler::Install(scheduler);

    }

}

/**

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

 */

void TestWlanQtUtils::cleanupTestCase()

{

    // CommsDat file is initialized.

    subTestLoadCommsDatFile(commsDatDefault_);

}

/**

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

 */

void TestWlanQtUtils::init()

{

    // CommsDat file is initialized.

    subTestLoadCommsDatFile(commsDatDefault_);

    testContext.initialize();

    QVERIFY(wlanQtUtils_ == NULL);

    QVERIFY(mSignalScanReady == NULL);

    QVERIFY(mSignalScanApReady == NULL);

    QVERIFY(mSignalScanDirectReady == NULL);

    QVERIFY(signalWlanNetworkOpened_ == NULL);

    QVERIFY(signalWlanNetworkClosed_ == NULL);

    QVERIFY(signalIctResult_ == NULL);

    subTestNewWlanQtUtils();

}

/**

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

 */

void TestWlanQtUtils::cleanup()

{

    delete wlanQtUtils_;

    wlanQtUtils_ = NULL;

    QCOMPARE(mSignalScanReady->count(), 0);

    delete mSignalScanReady;

    mSignalScanReady = NULL;

    QCOMPARE(mSignalScanApReady->count(), 0);

    delete mSignalScanApReady;

    mSignalScanApReady = NULL;

    QCOMPARE(mSignalScanDirectReady->count(), 0);

    delete mSignalScanDirectReady;

    mSignalScanDirectReady = NULL;

    QCOMPARE(signalWlanNetworkOpened_->count(), 0);

    delete signalWlanNetworkOpened_;

    signalWlanNetworkOpened_ = NULL;

    QCOMPARE(signalWlanNetworkClosed_->count(), 0);

    delete signalWlanNetworkClosed_;

    signalWlanNetworkClosed_ = NULL;

    QCOMPARE(signalIctResult_->count(), 0);

    delete signalIctResult_;

    signalIctResult_ = NULL;

}

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

// TEST CASES

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

/**

 * Test available WLAN APs when scan is triggered by client.

 */

void TestWlanQtUtils::testAvailableWlan()

{

    testContext.mScan.createDefaultWlanScanApResultList(4);

    // Request a scan to get result signal

    wlanQtUtils_->scanWlans();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsIap> > iaps;

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlans(iaps, aps);

    testContext.mScan.verifyWlanScanApResultList(aps);

    QCOMPARE(iaps.count(), 0);

    aps.clear();

}

/**

 * Test available WLAN APs when scan is triggered by client.

 * Two scan results have the same SSID and the latter one of those will be removed.

 * Two scan results have the same SSID but different security mode

 * Two scan results have the same SSID and security mode but different WpaPsk value

 * and both are included in the results.

 * Two scan results have the same SSID but different connection mode

 */

void TestWlanQtUtils::testAvailableWlanDuplicates()

{

    testContext.mScan.createDefaultWlanScanApResultList(11);

    testContext.mScan.mWlanScanApResultList[0]->setValue(WlanQtUtilsAp::ConfIdSsid, "Same SSID");

    testContext.mScan.mWlanScanApResultList[2]->setValue(WlanQtUtilsAp::ConfIdSsid, "Same SSID");

    testContext.mScan.mWlanScanApResultList[5]->setValue(WlanQtUtilsAp::ConfIdSsid, "Same SSID, different SecMode");

    testContext.mScan.mWlanScanApResultList[5]->setValue(WlanQtUtilsAp::ConfIdSecurityMode,CMManagerShim::WlanSecModeWpa);

    testContext.mScan.mWlanScanApResultList[6]->setValue(WlanQtUtilsAp::ConfIdSsid, "Same SSID, different SecMode");

    testContext.mScan.mWlanScanApResultList[6]->setValue(WlanQtUtilsAp::ConfIdSecurityMode,CMManagerShim::WlanSecModeOpen);

    testContext.mScan.mWlanScanApResultList[7]->setValue(WlanQtUtilsAp::ConfIdSsid, "Same SSID & SecMode diff Psk");

    testContext.mScan.mWlanScanApResultList[7]->setValue(WlanQtUtilsAp::ConfIdSecurityMode,CMManagerShim::WlanSecModeWpa2);

    testContext.mScan.mWlanScanApResultList[9]->setValue(WlanQtUtilsAp::ConfIdSsid, "Same SSID & SecMode diff Psk");

    testContext.mScan.mWlanScanApResultList[9]->setValue(WlanQtUtilsAp::ConfIdSecurityMode,CMManagerShim::WlanSecModeWpa2);

    testContext.mScan.mWlanScanApResultList[9]->setValue(WlanQtUtilsAp::ConfIdWpaPskUse,true);

    testContext.mScan.mWlanScanApResultList[10]->setValue(WlanQtUtilsAp::ConfIdSsid, "Same SSID");

    testContext.mScan.mWlanScanApResultList[10]->setValue(WlanQtUtilsAp::ConfIdConnectionMode, CMManagerShim::Adhoc);

    // Request a scan to get result signal

    wlanQtUtils_->scanWlans();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsIap> > iaps;

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlans(iaps, aps);

    // First remove AP that has duplicate SSID.

    testContext.mScan.mWlanScanApResultList.removeAt(2);

    testContext.mScan.verifyWlanScanApResultList(aps);

    QCOMPARE(iaps.count(), 0);

    aps.clear();

}

/**

 * Test available WLAN APs when scan is triggered by client.

 * Include all supported security modes.

 */

void TestWlanQtUtils::testAvailableWlanSecModes()

{

    testContext.mScan.createDefaultWlanScanApResultList(11);

    testContext.mScan.mWlanScanApResultList[0]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeOpen);

    testContext.mScan.mWlanScanApResultList[0]->setValue(WlanQtUtilsAp::ConfIdConnectionMode, CMManagerShim::Adhoc);

    testContext.mScan.mWlanScanApResultList[1]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWep);

    testContext.mScan.mWlanScanApResultList[2]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWep);

    testContext.mScan.mWlanScanApResultList[3]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecMode802_1x);

    testContext.mScan.mWlanScanApResultList[4]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWpa);

    testContext.mScan.mWlanScanApResultList[5]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWpa);

    testContext.mScan.mWlanScanApResultList[5]->setValue(WlanQtUtilsAp::ConfIdWpaPskUse, true);

    testContext.mScan.mWlanScanApResultList[6]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWpa2);

    testContext.mScan.mWlanScanApResultList[7]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWpa2);

    testContext.mScan.mWlanScanApResultList[7]->setValue(WlanQtUtilsAp::ConfIdWpaPskUse, true);

    testContext.mScan.mWlanScanApResultList[8]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWapi);

    testContext.mScan.mWlanScanApResultList[9]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWapi);

    // Let's trick the stub to return WapiPsk by putting WpaPskUse on *

    testContext.mScan.mWlanScanApResultList[9]->setValue(WlanQtUtilsAp::ConfIdWpaPskUse, true);

    testContext.mScan.mWlanScanApResultList[10]->setValue(WlanQtUtilsAp::ConfIdSecurityMode, CMManagerShim::WlanSecModeWpa2);

    testContext.mScan.mWlanScanApResultList[10]->setValue(WlanQtUtilsAp::ConfIdWpaPskUse, true);

    testContext.mScan.mWlanScanApResultList[10]->setValue(WlanQtUtilsAp::ConfIdWpsSupported, true);

    // Request a scan to get result signal

    wlanQtUtils_->scanWlans();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsIap> > iaps;

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlans(iaps, aps);

    // Remove the trick *. WapiPsk and Wapi both map to just Wapi

    testContext.mScan.mWlanScanApResultList[9]->setValue(WlanQtUtilsAp::ConfIdWpaPskUse,false);

    testContext.mScan.verifyWlanScanApResultList(aps);

    QCOMPARE(iaps.count(), 0);

    aps.clear();

}

/**

 * Test available WLAN APs when scan is triggered by client.

 * Include some exotic ssids.

 */

void TestWlanQtUtils::testAvailableWlanSsids()

{

    testContext.mScan.createDefaultWlanScanApResultList(6);

    testContext.mScan.mWlanScanApResultList[1]->setValue(WlanQtUtilsAp::ConfIdSsid, "           ");

    testContext.mScan.mWlanScanApResultList[2]->setValue(WlanQtUtilsAp::ConfIdSsid, "    whitespace around ssid  ");

    testContext.mScan.mWlanScanApResultList[3]->setValue(WlanQtUtilsAp::ConfIdSsid, QString(QChar()));

    testContext.mScan.mWlanScanApResultList[4]->setValue(WlanQtUtilsAp::ConfIdSsid, "maximum length ssid that is 32 c");

    testContext.mScan.mWlanScanApResultList[5]->setValue(WlanQtUtilsAp::ConfIdSsid, "A");

    // Request a scan to get result signal

    wlanQtUtils_->scanWlans();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsIap> > iaps;

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlans(iaps, aps);

    // Empty ssids are removed from results

    testContext.mScan.mWlanScanApResultList.removeAt(3);

    testContext.mScan.mWlanScanApResultList.removeAt(1);

    testContext.mScan.verifyWlanScanApResultList(aps);

    QCOMPARE(iaps.count(), 0);

}

/**

 * Test available WLAN APs when there are also WLAN IAPs available.

 */

void TestWlanQtUtils::testAvailableWlanWithIaps()

{

    // Form WLAN IAP scan result list

    testContext.mScan.createDefaultWlanScanIapResultList(3);

    testContext.mScan.mWlanScanIapResultList.removeAt(1);

    testContext.mScan.mWlanScanIapResultList[0]->setValue(WlanQtUtilsIap::ConfIdIapId, 5);

    testContext.mScan.mWlanScanIapResultList[1]->setValue(WlanQtUtilsIap::ConfIdIapId, 7);

    // Form WLAN AP scan result list

    testContext.mScan.createDefaultWlanScanApResultList(7);

    testContext.mScan.mWlanScanApResultList[1]->setValue(WlanQtUtilsAp::ConfIdSsid, "WLAN IAP 3");

    testContext.mScan.mWlanScanApResultList[1]->setValue(WlanQtUtilsAp::ConfIdSecurityMode,CMManagerShim::WlanSecModeWpa);    

    testContext.mScan.mWlanScanApResultList[1]->setValue(WlanQtUtilsAp::ConfIdWpaPskUse,true);    

    testContext.mScan.mWlanScanApResultList[5]->setValue(WlanQtUtilsAp::ConfIdSsid, "WLAN IAP 1");

    // SSID is found in Internet Snap, but security mode does not match:

    testContext.mScan.mWlanScanApResultList[6]->setValue(WlanQtUtilsAp::ConfIdSsid, "WLAN IAP 2");

    testContext.mScan.mWlanScanApResultList[6]->setValue(WlanQtUtilsAp::ConfIdSecurityMode,CMManagerShim::WlanSecModeWpa);

    // Request a scan to get result signal

    wlanQtUtils_->scanWlans();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsIap> > iaps;

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlans(iaps, aps);

    // Verify WLAN IAP list

    testContext.mScan.verifyWlanScanIapResultList(iaps);

    // Verify WLAN AP list. First, remove scan results that will not appear

    // because they are WLAN IAPs.

    testContext.mScan.mWlanScanApResultList.removeAt(5);

    testContext.mScan.mWlanScanApResultList.removeAt(1);

    testContext.mScan.verifyWlanScanApResultList(aps);

    iaps.clear();

    aps.clear();

}

/**

 * Test refereshing of WLAN APs when client requests sequential scans.

 */

void TestWlanQtUtils::testAvailableWlanSequence()

{

    // **************** Before 1st scan ********************

    QList< QSharedPointer<WlanQtUtilsIap> > iaps;

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlans(iaps, aps);

    // Verify we have no results

    QCOMPARE(aps.count(), 0);

    QCOMPARE(iaps.count(), 0);

    // **************** 1st scan ********************

    // 6 APs are required for this scan

    testContext.mScan.createDefaultWlanScanApResultList(6);

    wlanQtUtils_->scanWlans();

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get the results for the 1st scan

    wlanQtUtils_->availableWlans(iaps, aps);

    // Verify the results for the scan

    testContext.mScan.verifyWlanScanApResultList(aps);

    QCOMPARE(iaps.count(), 0);

    aps.clear();

    // **************** 2nd scan ********************

    // 2 APs are required for this scan

    testContext.mScan.createDefaultWlanScanApResultList(2);

    wlanQtUtils_->scanWlans();

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get the results for the 2nd scan

    wlanQtUtils_->availableWlans(iaps, aps);

    // Verify the results for the scan

    testContext.mScan.verifyWlanScanApResultList(aps);

    QCOMPARE(iaps.count(), 0);

    aps.clear();

    // **************** 3rd scan ********************

    // 4 APs are required for the 3rd scan

    testContext.mScan.createDefaultWlanScanApResultList(4);

    wlanQtUtils_->scanWlans();

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get the results for the 3rd scan

    wlanQtUtils_->availableWlans(iaps, aps);

    // Verify the results for the scan

    testContext.mScan.verifyWlanScanApResultList(aps);

    QCOMPARE(iaps.count(), 0);

    aps.clear();

}

/**

 * Test availability scan that finds nothing.

 */

void TestWlanQtUtils::testAvailableWlanScanEmpty()

{

    // Form result list

    testContext.mScan.createDefaultWlanScanApResultList(0);

    // Request a direct scan to get result signal

    wlanQtUtils_->scanWlans();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsIap> > iaps;

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlans(iaps, aps);

    testContext.mScan.verifyWlanScanApResultList(aps);

    QCOMPARE(iaps.count(), 0);

}

/**

 * Test WLAN availability scan failure.

 */

void TestWlanQtUtils::testAvailableWlanScanError()

{

    // Set the scan to fail

    testContext.mScan.mApScanRetValue = KErrGeneral;

    // Start the scan

    wlanQtUtils_->scanWlans();

    // Catch & check the scan result signal

    subTestScanResultSignal(mSignalScanReady, WlanQtUtils::ScanStatusError);

}

/**

 * Test available WLAN APs when scan is triggered by client.

 */

void TestWlanQtUtils::testAvailableWlanAps()

{

    testContext.mScan.createDefaultWlanScanApResultList(4);

    // Request a scan to get result signal

    wlanQtUtils_->scanWlanAps();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanApReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlanAps(aps);

    testContext.mScan.verifyWlanScanApResultList(aps);

    aps.clear();

}

/**

 * Test a direct scan that finds nothing.

 */

void TestWlanQtUtils::testAvailableWlanApsScanEmpty()

{

    // Form result list

    testContext.mScan.createDefaultWlanScanApResultList(0);

    // Request a direct scan to get result signal

    wlanQtUtils_->scanWlanAps();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanApReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlanAps(aps);

    testContext.mScan.verifyWlanScanApResultList(aps);

}

/**

 * Test cancellation of WLAN AP scan.

 */

void TestWlanQtUtils::testStopWlanApScan()

{

    // Do not complete scan request immediately

    testContext.mScan.mCompleteWlanApScan = false;

    // Start the scan

    wlanQtUtils_->scanWlanAps();

    // Cancel the scan

    wlanQtUtils_->stopWlanScan();

    // Catch & check the scan result signal

    subTestScanResultSignal(mSignalScanApReady, WlanQtUtils::ScanStatusCancelled);

}

/**

 * Test WLAN AP scan failure.

 */

void TestWlanQtUtils::testWlanApScanError()

{

    // Set the scan to fail

    testContext.mScan.mApScanRetValue = KErrGeneral;

    // Start the scan

    wlanQtUtils_->scanWlanAps();

    // Catch & check the scan result signal

    subTestScanResultSignal(mSignalScanApReady, WlanQtUtils::ScanStatusError);

}

/**

 * Test WLAN AP scan with special characters.

 */

void TestWlanQtUtils::testAvailableWlanApsSpecialChars()

{

    testContext.mScan.createDefaultWlanScanApResultList(3);

    // Set SSID's to include special characters

    testContext.mScan.mWlanScanApResultList[0]->setValue(

        WlanQtUtilsAp::ConfIdSsid,

        "\x57\x41\x50\x49\xBD\xA7\xA4\xE4\xF6");

    testContext.mScan.mWlanScanApResultList[1]->setValue(

        WlanQtUtilsAp::ConfIdSsid,

        "\x01\x02\x03\xAB\x43\x52\x41\x50\xBB\xA7\xA4\xFE\xFF");

    testContext.mScan.mWlanScanApResultList[2]->setValue(

        WlanQtUtilsAp::ConfIdSsid,

        "12345678901234567890123456789012");  // Maximum SSID length of 32 bytes

    // Request a scan to get result signal

    wlanQtUtils_->scanWlanAps();

    // Let active object run and verify signal.

    subTestScanResultSignal(mSignalScanApReady, WlanQtUtils::ScanStatusOk);

    // Get and verify the list of available WLAN APs.

    QList< QSharedPointer<WlanQtUtilsAp> > aps;

    wlanQtUtils_->availableWlanAps(aps);

    testContext.mScan.verifyWlanScanApResultList(aps);

    aps.clear();

}