/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the Qt Mobility Components.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used 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.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QObject>

#include <QTest>

#include <QSharedPointer>

#include <QDebug>

#include <QTimer>

#include <QEventLoop>

#include <QFile>

#include "qmessagestore.h"

#include "../support/support.h"

#if (defined(Q_WS_MAEMO_5) || defined(Q_WS_MAEMO_6))

# if defined(TESTDATA_DIR)

#  undef TESTDATA_DIR

# endif

# define TESTDATA_DIR "/var/tmp"

#elif (defined(Q_OS_SYMBIAN) || defined(Q_OS_WIN) && defined(_WIN32_WCE))

# if defined(TESTDATA_DIR)

#  undef TESTDATA_DIR

# endif

# define TESTDATA_DIR "."

#endif

//TESTED_CLASS=

//TESTED_FILES=

QTM_USE_NAMESPACE

#if defined(Q_OS_WIN)

const QByteArray defaultCharset("UTF-16");

const QByteArray alternateCharset("UTF-8");

#else

const QByteArray defaultCharset("UTF-8");

const QByteArray alternateCharset("UTF-16");

#endif

template<typename T1, typename T2>

void approximateCompare(const T1 &actual, const T2 &expected, const T2 &variance, int line)

{

    if (!((expected - variance) <= actual) ||

        !(actual <= (expected + variance))) {

        qWarning() << "at line:" << line;

        qWarning() << "\tactual:" << actual;

        qWarning() << "\texpected:" << expected;

        qWarning() << "\tvariance:" << variance;

    }

    QVERIFY((expected - variance) <= actual);

    QVERIFY(actual <= (expected + variance));

}

#define QAPPROXIMATECOMPARE(a,e,v) approximateCompare(a,e,v,(__LINE__))

class SignalCatcher : public QObject

{

    Q_OBJECT

public:

    typedef QPair<QMessageId, QMessageManager::NotificationFilterIdSet> Notification;

    QList<Notification> added;

    QList<Notification> updated;

    QList<Notification> removed;

public slots:

    void messageAdded(const QMessageId &id, const QMessageManager::NotificationFilterIdSet &filterIds)

    {

        added.append(qMakePair(id, filterIds));

    }

    void messageUpdated(const QMessageId &id, const QMessageManager::NotificationFilterIdSet &filterIds)

    {

        updated.append(qMakePair(id, filterIds));

    }

    void messageRemoved(const QMessageId &id, const QMessageManager::NotificationFilterIdSet &filterIds)

    {

        removed.append(qMakePair(id, filterIds));

    }

};

class FilterRegistration

{

public:

    QMessageManager::NotificationFilterId id;

    QMessageManager *manager;

    FilterRegistration(const QMessageFilter &filter, QMessageManager *mgr)

        : id(0),

          manager(mgr)

    {

        id = manager->registerNotificationFilter(filter);

    }

    ~FilterRegistration()

    {

        manager->unregisterNotificationFilter(id);

    }

};

/*

    Unit test for QMessageManager class.

*/

class tst_QMessageStore : public QObject

{

    Q_OBJECT

public:

    tst_QMessageStore();

    virtual ~tst_QMessageStore();

private slots:

    void initTestCase();

    void cleanup();

    void cleanupTestCase();

    void testAccount_data();

    void testAccount();

    void testFolder_data();

    void testFolder();

    void testMessage_data();

    void testMessage();

private:

    QMessageManager *manager;

};

QTEST_MAIN(tst_QMessageStore)

#include "tst_qmessagestore.moc"

tst_QMessageStore::tst_QMessageStore()

    : manager(0)

{

}

tst_QMessageStore::~tst_QMessageStore()

{

    delete manager;

}

void tst_QMessageStore::initTestCase()

{

#if defined(Q_OS_WIN) && !defined(Q_OS_WINCE)

    if (!Support::mapiAvailable())

        QSKIP("Skipping tests because a MAPI subsystem does not appear to be available", SkipAll);

#endif

    Support::clearMessageStore();

    manager = new QMessageManager;

}

void tst_QMessageStore::cleanup()

{

}

void tst_QMessageStore::cleanupTestCase()

{

}

void tst_QMessageStore::testAccount_data()

{

    QTest::addColumn<QString>("name");

    QTest::addColumn<QString>("fromAddress");

    QTest::newRow("1") << "Test Account #1" << "tester1@example.com";

    QTest::newRow("2") << "Test Account #2" << "tester2@example.com";

}

void tst_QMessageStore::testAccount()

{

    QFETCH(QString, name);

    QFETCH(QString, fromAddress);

    Support::Parameters p;

    p.insert("name", name);

    p.insert("fromAddress", fromAddress);

    int originalCount = manager->countAccounts();

    QMessageAccountId accountId(Support::addAccount(p));

    QVERIFY(accountId.isValid());

    QVERIFY(accountId != QMessageAccountId());

    QCOMPARE(manager->countAccounts(), originalCount + 1);

    QMessageAccount account(accountId);

    QCOMPARE(account.id(), accountId);

    QCOMPARE(account.name(), name);

    QCOMPARE(account.messageTypes(), QMessage::Email);

    QCOMPARE(QMessageAccount(account).id(), accountId);

    QVERIFY(!(accountId < accountId));

    QVERIFY((QMessageAccountId() < accountId) || (accountId < QMessageAccountId()));

    QMessageAccountIdList accountIds(manager->queryAccounts());

    QVERIFY(accountIds.contains(accountId));

    QVERIFY(QMessageAccount::defaultAccount(QMessage::Email).isValid());

}

void tst_QMessageStore::testFolder_data()

{

    QTest::addColumn<QString>("path");

    QTest::addColumn<QString>("name");

    QTest::addColumn<QString>("parentFolderPath");

    QTest::addColumn<QString>("nameResult");

	// Note: on Win CE, we can't use 'Inbox' 'Drafts' etc., becuase they're added automatically by the system

    QTest::newRow("Inbox") << "Unbox" << "Unbox" << "" << "Unbox";

#if !defined(Q_OS_SYMBIAN) && !defined(Q_WS_MAEMO_5) && !defined(Q_WS_MAEMO_6)

    // Symbian does not currently support paths

    QTest::newRow("Drafts") << "Crafts" << "" << "" << "Crafts";

    QTest::newRow("Archived") << "Unbox/Archived" << "Archived" << "Unbox" << "Archived";

    QTest::newRow("Backup") << "Unbox/Archived/Backup" << "Backup" << "Unbox/Archived" << "Backup";

#endif    

}

void tst_QMessageStore::testFolder()

{

    // Ensure we have an account to link these folders to

    static const QString testAccountName("testAccount");

    QMessageAccountId testAccountId;

    QMessageAccountIdList accountIds(manager->queryAccounts(QMessageAccountFilter::byName(testAccountName)));

    if (accountIds.isEmpty()) {

        Support::Parameters p;

        p.insert("name", testAccountName);

        p.insert("fromAddress", "someone@example.com");

        testAccountId = Support::addAccount(p);

    } else {

        testAccountId = accountIds.first();

    }

    QVERIFY(testAccountId.isValid());

    QFETCH(QString, path);

    QFETCH(QString, name);

    QFETCH(QString, parentFolderPath);

    QFETCH(QString, nameResult);

    Support::Parameters p;

    p.insert("path", path);

    p.insert("name", name);

    p.insert("parentAccountName", testAccountName);

    p.insert("parentFolderPath", parentFolderPath);

#if defined(Q_OS_SYMBIAN) || defined(Q_WS_MAEMO_5) || defined(Q_WS_MAEMO_6)

    int originalCount = manager->countFolders(QMessageFolderFilter::byParentAccountId(testAccountId));

#else

    int originalCount = manager->countFolders();

#endif    

    QMessageFolderId folderId(Support::addFolder(p));

    QVERIFY(folderId.isValid());

    QVERIFY(folderId != QMessageFolderId());

#if defined(Q_OS_SYMBIAN) || defined(Q_WS_MAEMO_5) || defined(Q_WS_MAEMO_6)

    QCOMPARE(manager->countFolders(QMessageFolderFilter::byParentAccountId(testAccountId)), originalCount + 1);

#else

    QCOMPARE(manager->countFolders(), originalCount + 1);

#endif    

    QMessageFolder folder(folderId);

    QCOMPARE(folder.id(), folderId);

    QCOMPARE(folder.path(), path);

    QCOMPARE(folder.name(), nameResult);

    QCOMPARE(folder.parentAccountId(), testAccountId);

    QCOMPARE(QMessageFolder(folder).id(), folderId);

    QVERIFY(!(folderId < folderId));

    QVERIFY((QMessageFolderId() < folderId) || (folderId < QMessageFolderId()));

    if (!parentFolderPath.isEmpty()) {

        QMessageFolderFilter filter(QMessageFolderFilter::byPath(parentFolderPath) & QMessageFolderFilter::byParentAccountId(testAccountId));

        QMessageFolderIdList list(manager->queryFolders(filter));

        QMessageFolderId parentFolderId(list.first());

        QCOMPARE(folder.parentFolderId(), parentFolderId);

    }

    QMessageFolderIdList folderIds(manager->queryFolders());

    QVERIFY(folderIds.contains(folderId));

}

Q_DECLARE_METATYPE(QList<QByteArray>)

Q_DECLARE_METATYPE(QList<int>)

typedef QMap<QString, QString> CustomFieldMap;

Q_DECLARE_METATYPE(CustomFieldMap)

void tst_QMessageStore::testMessage_data()

{

    QTest::addColumn<QString>("to");

    QTest::addColumn<QString>("from");

    QTest::addColumn<QString>("cc");

    QTest::addColumn<QString>("date");

    QTest::addColumn<QString>("subject");

    QTest::addColumn<QByteArray>("messageType");

    QTest::addColumn<QByteArray>("messageSubType");

    QTest::addColumn<int>("messageSize");

    QTest::addColumn<QString>("text");

    QTest::addColumn<QByteArray>("bodyType");

    QTest::addColumn<QByteArray>("bodySubType");

    QTest::addColumn<int>("bodySize");

    QTest::addColumn<QList<QByteArray> >("attachments");

    QTest::addColumn<QList<QByteArray> >("attachmentType");

    QTest::addColumn<QList<QByteArray> >("attachmentSubType");

    QTest::addColumn<QList<int> >("attachmentSize");

    QTest::addColumn<CustomFieldMap>("custom");

    QTest::addColumn<QString>("removeMessage");

    CustomFieldMap customData;

    customData.insert("cake", "chocolate");

    customData.insert("muffin", "blueberry");

    QTest::newRow("1")

        << "alice@example.com"

        << "bob@example.com"

        << ""

        << "1999-12-31T23:59:59Z"

        << "Last message..."

        << QByteArray("text")

        << QByteArray("plain")

#if defined(Q_OS_SYMBIAN)

        << 89

#else

#if defined(Q_OS_WIN) && defined(_WIN32_WCE)

        << 32

#else

        << 1400

#endif

#endif        

        << "...before Y2K"

        << QByteArray("text")

        << QByteArray("plain")

        << 24

        << QList<QByteArray>()

        << QList<QByteArray>()

        << QList<QByteArray>()

        << QList<int>()

        << customData

        << "byId";

    QTest::newRow("2")

        << "alice@example.com"

        << "bob@example.com"

        << "charlie@example.com, donald@example.com"

        << "1999-12-31T23:59:59Z"

        << "Last HTML message..."

        << QByteArray("text")

        << QByteArray("html")

#if defined(Q_OS_SYMBIAN)

        << 157

#else

#if defined(Q_OS_WIN) && defined(_WIN32_WCE)

        << 64

#else

        << 1536

#endif

#endif

        << "<html><p>...before <b>Y2K</b></p></html>"

        << QByteArray("text")

        << QByteArray("html")

        << 64

        << QList<QByteArray>()

        << QList<QByteArray>()

        << QList<QByteArray>()

        << QList<int>()

        << customData

        << "byFilter";

    QTest::newRow("3")

        << "alice@example.com"

        << "bob@example.com"

        << ""

        << "1999-12-31T23:59:59Z"

        << "Last message..."

        << QByteArray("multipart")

        << QByteArray("mixed")

#if defined(Q_OS_SYMBIAN)

        << 611

#else

#if defined(Q_OS_WIN) && defined(_WIN32_WCE)

        << 512

#else

        << 1536

#endif

#endif        

        << "...before Y2K"

        << QByteArray("text")

        << QByteArray("plain")

        << 24

        << ( QList<QByteArray>() << "1.txt" )

        << ( QList<QByteArray>() << "text" )

        << ( QList<QByteArray>() << "plain" )

        << ( QList<int>() << 512 )

        << customData

        << "byId";

    QTest::newRow("4")

        << "alice@example.com"

        << "bob@example.com"

        << ""

        << "1999-12-31T23:59:59Z"

        << "Last HTML message..."

        << QByteArray("multipart")

        << QByteArray("mixed")

#if defined(Q_OS_SYMBIAN)

        << 4731

#else

#if defined(Q_OS_WIN) && !defined(_WIN32_WCE)

        << 5120

#else

        << 4096

#endif

#endif

        << "<html><p>...before <b>Y2K</b></p></html>"

        << QByteArray("text")

        << QByteArray("html")

        << 64

        << ( QList<QByteArray>() << "1.txt" << "qtlogo.png" )

        << ( QList<QByteArray>() << "text" << "image" )

        << ( QList<QByteArray>() << "plain" << "png" )

        << ( QList<int>() << 512 << 4096 )

        << customData

        << "byFilter";

}

void tst_QMessageStore::testMessage()

{

    // Ensure we have an account to create messages with

    static const QString testAccountName("testAccount");

    QMessageAccountId testAccountId;

    QMessageAccountIdList accountIds(manager->queryAccounts(QMessageAccountFilter::byName(testAccountName)));

    if (accountIds.isEmpty()) {

        Support::Parameters p;

        p.insert("name", testAccountName);

        testAccountId = Support::addAccount(p);

    } else {

        testAccountId = accountIds.first();

        QMessageAccount acc(testAccountId);

    }

    QVERIFY(testAccountId.isValid());

    QMessageFolderId testFolderId;

#if !defined(Q_OS_SYMBIAN) && !defined(Q_WS_MAEMO_5) && !defined(Q_WS_MAEMO_6)

    QMessageFolderFilter filter(QMessageFolderFilter::byName("Inbox") & QMessageFolderFilter::byParentAccountId(testAccountId));

#else

    // Created Messages can not be stored into "Inbox" folder in Symbian & Meamo

    QMessageFolderFilter filter(QMessageFolderFilter::byName("Unbox") & QMessageFolderFilter::byParentAccountId(testAccountId));

#endif

    QMessageFolderIdList folderIds(manager->queryFolders(filter));

    if (folderIds.isEmpty()) {

        Support::Parameters p;

#if !defined(Q_OS_SYMBIAN) && !defined(Q_WS_MAEMO_5) && !defined(Q_WS_MAEMO_6)

        p.insert("path", "Inbox");

        p.insert("name", "Inbox");

#else

        // Created Messages can not be stored into "Inbox" folder in Symbian & Maemo

        p.insert("path", "Unbox");

        p.insert("name", "Unbox");

#endif

        p.insert("parentAccountName", testAccountName);

        testFolderId = Support::addFolder(p);

    } else {

        testFolderId = folderIds.first();

    }

    QVERIFY(testFolderId.isValid());

    QMessageFolder testFolder(testFolderId);

#if defined(Q_WS_MAEMO_5) || defined(Q_WS_MAEMO_6)

    // Wait 1/100 second to make sure that there is

    // enough time to start monitoring new folder

    {

        QEventLoop eventLoop;

        QTimer::singleShot(100, &eventLoop, SLOT(quit()));

        eventLoop.exec();

    }

    // Note: QTest::qSleep(100); can not be used

    //       because QTest::qSleep(100); blocks

    //       signaling from other threads

#endif

    SignalCatcher catcher;

    connect(manager, SIGNAL(messageAdded(QMessageId, QMessageManager::NotificationFilterIdSet)), &catcher, SLOT(messageAdded(QMessageId, QMessageManager::NotificationFilterIdSet)));

    connect(manager, SIGNAL(messageUpdated(QMessageId, QMessageManager::NotificationFilterIdSet)), &catcher, SLOT(messageUpdated(QMessageId, QMessageManager::NotificationFilterIdSet)));

    SignalCatcher removeCatcher;

    connect(manager, SIGNAL(messageRemoved(QMessageId, QMessageManager::NotificationFilterIdSet)), &removeCatcher, SLOT(messageRemoved(QMessageId, QMessageManager::NotificationFilterIdSet)));