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

**

** Copyright (C) 2009 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 <QApplication>

#include <QtTest/QtTest>

#include <QDebug>

#include <QEventLoop>

#include <QTimer>

#include "qmessagemanager.h"

#include "qmessage.h"

#ifdef Q_OS_SYMBIAN

#include <msvapi.h>

#include <msvids.h>  // KMsvGlobalInBoxIndexEntryId

#include <mtclreg.h>  // CClientMtmRegistry

#include <smsclnt.h>  // CSmsClientMtm

#include <smut.h>  // KUidMsgTypeSMS

#include <txtrich.h>  // CRichText

#include <cemailaccounts.h>

#include <mmsconst.h>

#include <MIUTMSG.H>

#include <MSVFIND.H>  

#include <IMPCMTM.H>

#include <mmsclient.h>

#endif

#include <qdebug.h>

QTM_USE_NAMESPACE

class OsNative;

#if defined(Q_OS_SYMBIAN)

class OsNative : public MMsvSessionObserver {

public:

  OsNative()

  {    

  };

  void HandleSessionEventL(TMsvSessionEvent aEvent, TAny* aArg1, TAny* aArg2, TAny* aArg3) { };

  void getFolders(RArray<TMsvId> &folders);

  TMsvId makeSMS(TMsvId folder);

  TMsvId makeIMAP4(TMsvId folder);

  CMsvSession *iSession;

  CClientMtmRegistry *iMtmRegistry;

  CSmsClientMtm *iSmsMtm;

  CImap4ClientMtm *iImap4Mtm;  

  CMmsClientMtm *iMmsMtm;

  RArray<TMsvId> iFolders;

  RArray<TMsvId> iNewMsgs;

};

TMsvId OsNative::makeSMS(TMsvId folder){

  TBuf<10> aAddress(_L("Nokia"));

  TBuf<20> aDescription(_L("Important Message"));

  _LIT(KTxt1,"Hi phone owner, how r u?");

  TBuf<150> iMessage;     

  iMessage.Copy(KTxt1);

//      m_native->iSmsMtm->SwitchCurrentEntryL(KMsvGlobalInBoxIndexEntryId); //inbox KMsvDraftEntryId

  iSmsMtm->SwitchCurrentEntryL(folder);

  iSmsMtm->CreateMessageL(KUidMsgTypeSMS.iUid);

  CSmsHeader& iHeader = iSmsMtm->SmsHeader();

  iHeader.SetFromAddressL(aAddress);

  CRichText& body = iSmsMtm->Body();

  body.Reset();

  body.InsertL(0, iMessage);

  TMsvEntry entry = iSmsMtm->Entry().Entry();

  entry.SetInPreparation(EFalse);

  entry.SetVisible(ETrue);

  entry.iDate.HomeTime();

  entry.iDescription.Set(aDescription);

  entry.iDetails.Set(aAddress);

  entry.SetUnread(EFalse);

  iSmsMtm->Entry().ChangeL(entry);

  iSmsMtm->SaveMessageL();

  return entry.Id();

}

TMsvId OsNative::makeIMAP4(TMsvId folder){

  TBuf<20> aSubject(_L("Test Subject"));

  TBuf<20> aFrom(_L("Nokia@foo.com"));

  TBuf<20> aTo(_L("Boo@bar.com"));

  TBuf<20> aDescription(_L("Test Message"));

  _LIT(KTo, "Boo@bar.com");

  _LIT(KTxt1,"Test autogenerated message by benchmark/tst_messaging for performance testing");

  TBuf<150> iMessage;     

  iMessage.Copy(KTxt1);

//      m_native->iSmsMtm->SwitchCurrentEntryL(KMsvGlobalInBoxIndexEntryId); //inbox KMsvDraftEntryId

  iImap4Mtm->SwitchCurrentEntryL(folder); 

  // Set the context to the folder in which message has to be created

  CMsvEntry* entry = CMsvEntry::NewL(*iSession,folder,TMsvSelectionOrdering());

  CleanupStack::PushL(entry);

  entry->SetEntryL(folder);

  CMsvOperationActiveSchedulerWait* waiter=CMsvOperationActiveSchedulerWait::NewLC();  

  //CleanupStack::PushL(waiter);

  TMsvEmailTypeList msvEmailTypeList = 0;

  TMsvPartList partList = (KMsvMessagePartBody | KMsvMessagePartAttachments);

  CImEmailOperation* emailOperation = CImEmailOperation::CreateNewL(waiter->iStatus, *iSession,KMsvGlobalOutBoxIndexEntryId, partList, msvEmailTypeList, KUidMsgTypeSMTP);

  CleanupStack::PushL(emailOperation);

  waiter->Start();

  TMsvId temp;

  TPckgC<TMsvId> paramPack(temp);

  const TDesC8& progBuf = emailOperation->ProgressL();

  paramPack.Set(progBuf);

  TMsvId newMessageId;

  newMessageId = paramPack();

  entry->SetEntryL(newMessageId);

  CMsvStore* store = entry->EditStoreL();

  CleanupStack::PushL(store);

  CImHeader* emailEntry = CImHeader::NewLC();

  emailEntry->RestoreL(*store);

  emailEntry->SetFromL((TDesC8&)aFrom);

  emailEntry->SetSubjectL((TDesC&)aDescription);

  emailEntry->ToRecipients().AppendL(KTo);

  // Paragraph format layer for the rich text object 

  CParaFormatLayer* paraFormatLayer = CParaFormatLayer::NewL();

  CleanupStack::PushL(paraFormatLayer);

  // Character format layer for the rich text object

  CCharFormatLayer* charFormatLayer = CCharFormatLayer::NewL(); 

  CleanupStack::PushL(charFormatLayer);

  CRichText* bodyText = CRichText::NewL(paraFormatLayer, charFormatLayer, CEditableText::EFlatStorage, 256);

  CleanupStack::PushL(bodyText);

  // Insert the contents of a buffer into the document at specified position

  bodyText->InsertL(0, KTxt1);

  store->StoreBodyTextL(*bodyText);

  emailEntry->StoreL(*store);

  // Store the changes permanently

  store->CommitL();

  CleanupStack::PopAndDestroy(8,entry); // bodyText,charFormatLayer,paraFormatLayer,emailEntry,store, emailOperationg,waiter,entry

  return entry->EntryId();

}

#else

    class OsNative {

    };

#endif

//Q_DECLARE_METATYPE(QSystemInfo::Feature);

class tst_messaging : public QObject

{

    Q_OBJECT

public:

    enum platform {

        QMessaging = 0,

        Native = 1

    };

    enum bases {

      Accounts = 0,

      AccountsImap,

      AccountsPop,

      AccountsSmtp,

      Folders,

      FoldersEmail,

      FoldersSMS,

      FoldersMMS,

      Messages,

      MessagesAlt,

    };

    enum filter {

      Id = 0,

      Type,

      Sender,

      Recipients,

      Subject,

      TimeStamp,

      Status,

      Priority,

      Size,

      AllId,

      AllSender

    };

    enum types {

      Sms = 0,

      Mms,

      Email          

    };

private slots:

    void initTestCase();

    void cleanupTestCase();

    void tst_createTime_data();

    void tst_createTime();

    void tst_counts_data();

    void tst_counts();

    void tst_query_data();

    void tst_query();

    void tst_fetch_data();

    void tst_fetch();

    void tst_fetchAll_data();

    void tst_fetchAll();

    void tst_fetchFilter_data();

    void tst_fetchFilter();

    void tst_addMessage_data();

    void tst_addMessage();

    void tst_removeMessage_data();

    void tst_removeMessage();

    void tst_removeAllMessage_data();

    void tst_removeAllMessage();

private:

    QMessage *createMessage();

    QMessage *messageTemplate();

    QMessageId addMessage(QMessage *);     

    void clearMessages();

    int countMessages();

    QMessageManager *m_mgr;

    QMessageIdList msg_list;

    QMessageAccount m_act;

    //QMessageFolder m_fol;

    OsNative *m_native;

};

Q_DECLARE_METATYPE(tst_messaging::platform);

Q_DECLARE_METATYPE(tst_messaging::bases);

Q_DECLARE_METATYPE(tst_messaging::filter);

Q_DECLARE_METATYPE(tst_messaging::types);

void tst_messaging::initTestCase()

{

#if defined(Q_WS_MAEMO_6)

    QFAIL("Maemo 6 not supported by tst_messaging");

#elif defined(Q_WS_MAEMO_5)

    QFAIL("Maemo 5 does not support tst_messageing");

#elif defined(Q_OS_SYMBIAN)

#else

    QFAIL("Platform not supported");

#endif

    m_mgr = new QMessageManager();

    QMessageAccountIdList list;

    list = m_mgr->queryAccounts();

    while(!list.empty()) {      

      QMessageAccount act = m_mgr->account(list.takeFirst());

      qDebug() << "Account: " << act.name();

#if defined(Q_OS_SYMBIAN)      

      if(act.name() == "Personal"){

        m_act = act;

        break;

      }     

#endif      

    }

//    QMessageFolderIdList flist = m_mgr->queryFolders();

//    while(!flist.empty()) {      

//      QMessageFolder fol = m_mgr->folder(flist.takeFirst());

//      qDebug() << "Folder: " << fol.name();

//#if defined(Q_OS_SYMBIAN)      

//      if(fol.name() == "Inbox"){

//        m_fol = fol;

//        break;

//      }     

//#endif      

//    }

#if defined(Q_OS_SYMBIAN)

    // define/init native features we need native

    m_native = new OsNative;

    m_native->iSession = CMsvSession::OpenSyncL(*m_native);

    m_native->iMtmRegistry = CClientMtmRegistry::NewL(*m_native->iSession);

    m_native->iSmsMtm = STATIC_CAST(CSmsClientMtm*, m_native->iMtmRegistry->NewMtmL(KUidMsgTypeSMS));

    m_native->iImap4Mtm = STATIC_CAST(CImap4ClientMtm*, m_native->iMtmRegistry->NewMtmL(KUidMsgTypeIMAP4));

    m_native->iMmsMtm = STATIC_CAST(CMmsClientMtm*, m_native->iMtmRegistry->NewMtmL(KUidMsgTypeMultimedia));

#endif

    for(int i = 0; i < 20; i++)

      createMessage();

}

void tst_messaging::cleanupTestCase()

{

#if defined(Q_OS_SYMBIAN)

  delete m_native->iMmsMtm;

  delete m_native->iImap4Mtm;

  delete m_native->iSmsMtm;

  delete m_native->iMtmRegistry;

  delete m_native->iSession;

  delete m_native;

#endif

  clearMessages();

  delete m_mgr;

}

void tst_messaging::tst_createTime_data()

{

    QTest::addColumn<tst_messaging::platform>("platform");

    QTest::newRow("Qt-Create") << tst_messaging::QMessaging;

    QTest::newRow("Native-Create") << tst_messaging::Native;

}

void tst_messaging::tst_createTime()

{

    QFETCH(tst_messaging::platform, platform);

    //qDebug() << "Platform: " << platform;

    if(platform == tst_messaging::QMessaging){

      QMessageManager *mgr = 0x0;

      QBENCHMARK {

        mgr = new QMessageManager();        

      }

      delete mgr;

    }

    else if(platform == tst_messaging::Native){

#ifdef Q_OS_SYMBIAN

      __UHEAP_MARK;

      if(m_native){

        delete m_native->iMmsMtm;

        delete m_native->iImap4Mtm;

        delete m_native->iSmsMtm;

        delete m_native->iMtmRegistry;

        delete m_native->iSession;

        delete m_native;

      }

      QBENCHMARK {

        // define/init native features we need native

        m_native = new OsNative;

        m_native->iSession = CMsvSession::OpenSyncL(*m_native);

        m_native->iMtmRegistry = CClientMtmRegistry::NewL(*m_native->iSession);

        m_native->iSmsMtm = STATIC_CAST(CSmsClientMtm*, m_native->iMtmRegistry->NewMtmL(KUidMsgTypeSMS));

        m_native->iImap4Mtm = STATIC_CAST(CImap4ClientMtm*, m_native->iMtmRegistry->NewMtmL(KUidMsgTypeIMAP4));

        m_native->iMmsMtm = STATIC_CAST(CMmsClientMtm*, m_native->iMtmRegistry->NewMtmL(KUidMsgTypeMultimedia));

      }

      __UHEAP_MARKEND;

#endif

    }

}

void tst_messaging::tst_counts_data()

{

    QTest::addColumn<tst_messaging::platform>("platform");

    QTest::addColumn<tst_messaging::bases>("base");

    QTest::newRow("Qt-Accounts") << tst_messaging::QMessaging << tst_messaging::Accounts;

//    QTest::newRow("Native-Accounts") << tst_messaging::Native << tst_messaging::Accounts;

    QTest::newRow("Qt-Folders") << tst_messaging::QMessaging << tst_messaging::Folders;

    QTest::newRow("Native-Folders") << tst_messaging::Native << tst_messaging::Folders;

    QTest::newRow("Qt-Messages") << tst_messaging::QMessaging << tst_messaging::Messages;

    QTest::newRow("Qt-MessagesAlt") << tst_messaging::QMessaging << tst_messaging::MessagesAlt;

    QTest::newRow("Native-Messages") << tst_messaging::Native << tst_messaging::Messages;

}

void tst_messaging::tst_counts()

{

    QFETCH(tst_messaging::platform, platform);

    QFETCH(tst_messaging::bases, base);

    if(platform == tst_messaging::QMessaging){

      int cnt;

      Q_UNUSED(cnt);

      if(base == tst_messaging::Accounts){

        QBENCHMARK {

          cnt = m_mgr->countAccounts();

        }

      }

      else if(base == tst_messaging::Folders){

        QBENCHMARK {

          cnt = m_mgr->countFolders();

        }

      }

      else if(base == tst_messaging::Messages){

        QBENCHMARK {

          cnt = m_mgr->countMessages();

        }        

      }

      else if(base == tst_messaging::MessagesAlt){

        QMessageIdList list;

        QBENCHMARK {

          list = m_mgr->queryMessages();

          cnt = list.count();

        }        

      }

      //qDebug() << "Got cnt: " << cnt;      

    }

    else if(platform == tst_messaging::Native){

#ifdef Q_OS_SYMBIAN

      QBENCHMARK {

        // Access the Inbox

        TMsvSelectionOrdering sort;

        CMsvEntry* inboxContext = CMsvEntry::NewL(*m_native->iSession, KMsvGlobalInBoxIndexEntryId, sort);

        CleanupStack::PushL(inboxContext);

        // Get all entries in the Inbox

        CMsvEntrySelection* entries = inboxContext->ChildrenL();

        CleanupStack::PushL(entries);

        TInt messages = entries->Count();

        CleanupStack::PopAndDestroy(2, inboxContext);

      }           

#endif

    }

}

void tst_messaging::tst_query_data()

{

    QTest::addColumn<tst_messaging::platform>("platform");

    QTest::addColumn<tst_messaging::bases>("base");

    QTest::newRow("Qt-Accounts") << tst_messaging::QMessaging << tst_messaging::Accounts;    

//    QTest::newRow("Qt-Folders") << tst_messaging::QMessaging << tst_messaging::Folders;    

    QTest::newRow("Qt-Messages") << tst_messaging::QMessaging << tst_messaging::Messages;

    QTest::newRow("Native-Accounts") << tst_messaging::Native << tst_messaging::Accounts;

    QTest::newRow("Native-Accounts-Imap") << tst_messaging::Native << tst_messaging::AccountsImap;

    QTest::newRow("Native-Accounts-Pop") << tst_messaging::Native << tst_messaging::AccountsPop;

    QTest::newRow("Native-Accounts-Smtp") << tst_messaging::Native << tst_messaging::AccountsSmtp;

    QTest::newRow("Native-Folders") << tst_messaging::Native << tst_messaging::Folders;

//    QTest::newRow("Native-Folders") << tst_messaging::Native << tst_messaging::FoldersEmail;

//    QTest::newRow("Native-Folders") << tst_messaging::Native << tst_messaging::FoldersSMS;

//    QTest::newRow("Native-Folders") << tst_messaging::Native << tst_messaging::FoldersMMS;