/*

* 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: This file contains testclass implementation.

*

*/

// change the level of RVCT compiler warning 'non-POD class passed thru ellipses' to a remark.

#ifdef __MARM__

#pragma diag_remark 1446

#endif

// [INCLUDE FILES] - do not remove

#include "harvesterplugintester.h"

#include <e32svr.h>

#include <f32file.h> 

#include <stifparser.h>

#include <stiftestinterface.h>

#include <favouritesdb.h>

#include <msvapi.h>

#include <bautils.h>

#include <csearchdocument.h>

#include "harvesterobserver.h"

#include "cmessageplugin.h"

#include "cmessagemonitor.h"

#include "cmessageharvester.h"

#include "cmessagesessionobserver.h" //For CMsvSession

#include "ccpixsearcher.h"

#include "bookmarksplugin.h"

#include "applicationsplugin.h"

#include "ccontactsplugin.h"

#include "ccalendarplugin.h"

#include "ccalendarobserver.h"

#include "mediaplugin.h"

#include <harvesterclient.h>

#include "mdsitementity.h"

#include "cpixmdedbmanager.h"

#include "cfolderrenamedharvester.h"

#include "videoplugin.h"

#include "imageplugin.h"

#include "cemailplugin.h"

#include "cmessagedatahandler.h"

_LIT(KAppBasePath,"@c:root file content");

_LIT(KEmailAppBasePath,"@c:root msg email");

_LIT(KAppBaseFolderFilePath,"@c:root file folder");

const TInt KMsgPluginBaseAppClassMaxLen = 64;

// For Notes --Start

#include <d32dbms.h>

#include <utf.h>

#include "notesplugin.h"

_LIT( KemailId, "1" );

_LIT(KMsgBaseAppClassGeneric, "root msg email");

#define MEDIA_QBASEAPPCLASS   "@c:root media audio"

#define LMEDIA_QBASEAPPCLASS  L"@c:root media audio"

#define VIDEO_QBASEAPPCLASS   "@c:root media video"

#define LVIDEO_QBASEAPPCLASS  L"@c:root media video"

#define IMAGE_QBASEAPPCLASS   "@c:root media image"

#define LIMAGE_QBASEAPPCLASS  L"@c:root media image"

// For Notes --End

// ============================ MEMBER FUNCTIONS ===============================

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

// Delete here all resources allocated and opened from test methods. 

// Called from destructor. 

void CHarvesterPluginTester::Delete() 

    {

    }

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

// Run specified method. Contains also table of test mothods and their names.

TInt CHarvesterPluginTester::RunMethodL( 

    CStifItemParser& aItem ) 

    {

    static TStifFunctionInfo const KFunctions[] =

        {  

        // Copy this line for every implemented function.

        // First string is the function name used in TestScripter script file.

        // Second is the actual implementation member function. 

        ENTRY( "TestStartHarvester", CHarvesterPluginTester::TestStartHarvesterL ),        

        ENTRY( "TestFormBaseAppClass", CHarvesterPluginTester::TestFormBaseAppClassL ),

        ENTRY( "TestDatabasePath", CHarvesterPluginTester::TestDatabasePathL ),

        ENTRY( "TestCreateIndexItemL_Add", CHarvesterPluginTester::TestCreateIndexItemL ),

        ENTRY( "TestCreateIndexItemL_Update", CHarvesterPluginTester::TestCreateIndexItemL ),

        ENTRY( "TestCreateIndexItemL_Delete", CHarvesterPluginTester::TestCreateIndexItemL ),

        ENTRY( "TestFolderCreate", CHarvesterPluginTester::TestFolderCreateL ),

        ENTRY( "TestFolderRename", CHarvesterPluginTester::TestFolderRenameL ),

        ENTRY( "TestMessaging", CHarvesterPluginTester::TestMessageHarvesterL ),

        ENTRY( "TestMessageHarvesting", CHarvesterPluginTester::TestMessageHarvesterWithMessageL ),

        ENTRY( "TestMessageDriveChange", CHarvesterPluginTester::TestMessageHarvesterChangeDriveL ),

        ENTRY( "TestMessageRunErrorL", CHarvesterPluginTester::TestMessageRunErrorL ),

        ENTRY( "TestMessageIsMediaRemovableL", CHarvesterPluginTester::TestMessageIsMediaRemovableL ),

        ENTRY( "TestMessageMountNoIndexerL", CHarvesterPluginTester::TestMessageMountNoIndexerL ),        

        ENTRY( "TestQualifiedBaseAppClass", CHarvesterPluginTester::TestQualifiedBaseAppClassL ),

        ENTRY( "TestHandleMsgMoved", CHarvesterPluginTester::TestHandleMsgMovedL ),

        ENTRY( "TestMessageDataHandler", CHarvesterPluginTester::TestMessageDataHandlerL ),

        ENTRY( "TestMsgUnMount", CHarvesterPluginTester::TestMsgUnMountL ),

        ENTRY( "TestFormBaseAppClass", CHarvesterPluginTester::TestFormFormBaseAppClassL ),

        ENTRY( "TestMsgDelete", CHarvesterPluginTester::TestMsgDeleteL ),

        ENTRY( "TestMsgHandlesession", CHarvesterPluginTester::TestMsgHandlesessionL ),

        ENTRY( "TestMsgHandlesessionPanic", CHarvesterPluginTester::TestMsgHandlesessionPanicL ),

        ENTRY( "TestMsgOverwriteOrAddToQuere", CHarvesterPluginTester::TestMsgOverwriteOrAddToQuereL ),

        ENTRY( "TestMsgMessageHarvester", CHarvesterPluginTester::TestMsgMessageHarvesterL ),

        ENTRY( "TestStartBookmarksHarvesterL", CHarvesterPluginTester::TestStartBookmarksHarvesterL ),

        ENTRY( "TestAddBookmarkL", CHarvesterPluginTester::TestAddBookmarkL ),

        ENTRY( "TestAddGetDomainL", CHarvesterPluginTester::TestAddGetDomainL ),

        ENTRY( "TestStartApplicationsHarvesterL", CHarvesterPluginTester::TestStartApplicationsHarvesterL ),        

        ENTRY( "TestCreateApplicationsIndexItemL", CHarvesterPluginTester::TestCreateApplicationsIndexItemL ),

		ENTRY( "TestStartNotesHarvesterL", CHarvesterPluginTester::TestStartNotesHarvesterL ),

        ENTRY( "TestAddNoteL", CHarvesterPluginTester::TestAddNoteL ),

        ENTRY( "TestAddLongNoteL", CHarvesterPluginTester::TestAddLongNoteL ),

        ENTRY( "TestAddAlphaNumericNoteL", CHarvesterPluginTester::TestAddAlphaNumericNoteL ),

        ENTRY( "TestDeleteNoteL", CHarvesterPluginTester::TestDeleteNoteL ),

        ENTRY( "TestUpdateNoteL", CHarvesterPluginTester::TestUpdateNoteL ),                

        ENTRY( "TestContactsHarvesting", CHarvesterPluginTester::TestStartContactsHarvesterL ),

        ENTRY( "TestCreateContactIndexItemL_Add", CHarvesterPluginTester::TestCreateContactIndexItemL ),

        ENTRY( "TestCreateAllContactFields", CHarvesterPluginTester::TestCreateAllContactFieldsL ),

        ENTRY( "TestCreateContactIndexItemL_Edit", CHarvesterPluginTester::TestCreateContactIndexItemL ),

        ENTRY( "TestCreateContactIndexItemL_Delete", CHarvesterPluginTester::TestCreateContactIndexItemL ),

        ENTRY( "TestCreateContactIndexItemL_NoIndexer", CHarvesterPluginTester::TestCreateContactIndexItemNoIndexerL ),

        ENTRY( "TestContactGetDateL", CHarvesterPluginTester::TestContactGetDateL ),

        ENTRY( "TestContactDelayedErrorL", CHarvesterPluginTester::TestContactDelayedErrorL ),

        ENTRY( "TestContactDelayedCallbackNoIContactsL", CHarvesterPluginTester::TestContactDelayedCallbackNoIContactsL ),        

        ENTRY( "TestContactDelayedCallbackNoIObserverL", CHarvesterPluginTester::TestContactDelayedCallbackNoIObserverL ),        

        ENTRY( "TestContactDelayedCallbackWrongCountL", CHarvesterPluginTester::TestContactDelayedCallbackWrongCountL ),        

        ENTRY( "TestCreateContactGroup", CHarvesterPluginTester::TestCreateContactGroupL ),

        ENTRY( "TestCalenderHarvesting", CHarvesterPluginTester::TestStartCalenderHarvesterL ),

        ENTRY( "TestCalenderEntry",CHarvesterPluginTester::TestCalenderEntryL ),

        ENTRY( "TestCreateMMS",CHarvesterPluginTester::TestCreateMmsL ),

        ENTRY( "TestCreateEmail",CHarvesterPluginTester::TestCreateEmailL ),

		ENTRY( "TestMdsSyncController",CHarvesterPluginTester::TestMdsSyncControllerL ),

		//ENTRY( "TestBlacklistPlugin",CHarvesterPluginTester::TestBlacklistPluginL ),

		//ENTRY( "TestBlacklistPluginVersion",CHarvesterPluginTester::TestBlacklistPluginVersionL ),

		ENTRY( "TestImageMMCEventL",CHarvesterPluginTester::TestImageMMCEventL ),

		ENTRY( "TestStartEmailPlugin",CHarvesterPluginTester::TestStartEmailPluginL ),

		ENTRY( "TestHandleEmailDoc",CHarvesterPluginTester::TestHandleEmailDocL ),

        //ADD NEW ENTRY HERE

        // [test cases entries] - Do not remove

        };

    const TInt count = sizeof( KFunctions ) / 

                        sizeof( TStifFunctionInfo );

    return RunInternalL( KFunctions, count, aItem );

    }

_LIT( KNoErrorString, "No Error" );

//_LIT( KErrorString, " *** Error ***" );

void doLog( CStifLogger* logger, TInt error, const TDesC& errorString )

    {

    if( KErrNone == error ) logger->Log( KNoErrorString );

    else logger->Log( errorString );

    }

CSearchDocument* CHarvesterPluginTester::prepareemaildocument()

    {    

    CSearchDocument* doc = CSearchDocument::NewL( KemailId, 

            KMsgBaseAppClassGeneric ); 

    return doc;

    }

// Example test method function.

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

TInt CHarvesterPluginTester::TestStartHarvesterL( CStifItemParser& /*aItem*/ )

    {

    // Print to UI

    _LIT( KHarvesterPluginTester, "HarvesterPluginTester: %S" );

    _LIT( KExample, "In TestStartHarvesterL" );

    TestModuleIf().Printf( 0, KHarvesterPluginTester, KExample );

    // Print to log file

    iLog->Log( KExample );

    CFilePlugin* filePlugin = CFilePlugin::NewL();

    CHarvesterObserver* iPluginTester = CHarvesterObserver::NewL( filePlugin );

    filePlugin->StartPluginL(); //Calls Add

    filePlugin->StartHarvestingL( KAppBasePath );

    iPluginTester->iWaitForHarvester->Start();//Wait here till Harvesting is complete.

    delete filePlugin;

    delete iPluginTester;

    iPluginTester = NULL;

    doLog( iLog, KErrNone, KNoErrorString );

    return KErrNone;

    }

TInt CHarvesterPluginTester::TestFormBaseAppClassL( CStifItemParser& aItem )

    {

    // Print to UI

    _LIT( KFilePluginBaseAppClass, "FormBaseAppClass" );

    _LIT( KExample, "In TestFormBaseAppClass" );

    TestModuleIf().Printf( 0, KFilePluginBaseAppClass, KExample );

    TInt driveNumber;

    TBuf<100> formedBaseAppClass;

    TBuf<100> expectedBaseAppClass;

    TChar driveChar;

    CFilePlugin* filePlugin = CFilePlugin::NewL();

    CHarvesterObserver* iPluginTester = CHarvesterObserver::NewL( filePlugin );

    filePlugin->StartPluginL();

    TInt error = KErrNone;

    TInt errorNumber = 0;

    _LIT( KFileBaseAppClassFolder, "root file folder");

    _LIT( KBaseAppClassFormatString, "@%c:root file folder");

    _LIT( KTestFormBaseAppClassNoError, "TestFormBaseAppClass: No Error" );

    if( aItem.GetNextInt ( driveNumber ) == KErrNone  && error == KErrNone )

        {

        filePlugin->FormBaseAppClass(  static_cast<TDriveNumber>( driveNumber ),KFileBaseAppClassFolder, formedBaseAppClass );

        if( aItem.GetNextChar ( driveChar ) == KErrNone )

            {

            expectedBaseAppClass.Format( KBaseAppClassFormatString, driveChar );

            error = expectedBaseAppClass.Compare( formedBaseAppClass );

            if( aItem.GetNextInt ( errorNumber ) == KErrNone )

                error = ( KErrNone == error || -1 == errorNumber ) ? KErrNone : KErrGeneral ;

            }

        }

    delete filePlugin;

    delete iPluginTester;

    iPluginTester = NULL;

    doLog( iLog, error, KTestFormBaseAppClassNoError );

    return error;

    }

TInt CHarvesterPluginTester::TestDatabasePathL( CStifItemParser& aItem )

    {

    TInt driveNumber;

    CFilePlugin* filePlugin = CFilePlugin::NewL();

    CHarvesterObserver* iPluginTester = CHarvesterObserver::NewL( filePlugin );

    filePlugin->StartPluginL();

    HBufC* buffer;

    TBuf<KMaxFileName> returnedDbPath;

    TBuf<KMaxFileName> expectedDbPath;

    TInt error = KErrNone;

    TInt errorNumber;

    TChar driveChar;

    _LIT( KTestDatabasePathNoError, "TestDatabasePathL: No Error" );

    _LIT(KPathFileContent, "\\root\\file\\content");

    _LIT( KDbFormatString, "%c:\\Private\\2001f6f7\\indexing\\indexdb\\root\\file\\content" );

    if( aItem.GetNextInt ( driveNumber ) == KErrNone  && error == KErrNone )

        {

        buffer = filePlugin->DatabasePathLC( static_cast<TDriveNumber>( driveNumber ), KPathFileContent );

        returnedDbPath.Copy( *buffer );

        if( aItem.GetNextChar ( driveChar ) == KErrNone )

            {

            expectedDbPath.Format( KDbFormatString, driveChar );

            error = expectedDbPath.Compare( returnedDbPath );

            if( aItem.GetNextInt ( errorNumber ) == KErrNone )

                error = ( KErrNone == error || -1 == errorNumber ) ? KErrNone : KErrGeneral ;

            }

        }

    CleanupStack::PopAndDestroy( buffer );

    delete filePlugin;

    delete iPluginTester;

    iPluginTester = NULL;

    doLog( iLog, error, KTestDatabasePathNoError );

    return error;

    }

//Helper functions

enum TSearchType

    {

    ESearchTypeResultsExpected=0,

    ESearchTypeNoResultsExpected

    };

TInt doSearchL( const TDesC& aSearchString, const TDesC& aBaseAppClass, TSearchType aSearchType )

    {

    _LIT(KHeaderPrefix, "$prefix(\"");

    _LIT(KTrailerPrefix, "\")");

    RSearchServerSession session;

    User::LeaveIfError( session.Connect() );

    CCPixSearcher* searcher = CCPixSearcher::NewLC( session );

    //Form prefix search query

    searcher->OpenDatabaseL( aBaseAppClass );

    HBufC* searchbuf = HBufC::NewLC(KHeaderPrefix().Length() + aSearchString.Length() + KTrailerPrefix().Length());

    TPtr searchstring = searchbuf->Des();

    searchstring.Append(KHeaderPrefix);

    searchstring.Append(aSearchString);

    searchstring.Append(KTrailerPrefix);

    TInt docCount = searcher->SearchL( searchstring, KNullDesC);

    CleanupStack::PopAndDestroy( searchbuf );

    CleanupStack::PopAndDestroy( searcher );

    session.Close();

    if( aSearchType == ESearchTypeResultsExpected )

        {

        return docCount > 0 ? KErrNone : KErrNotFound;

        }

    else

        {

        return 0 == docCount ? KErrNone : KErrNotFound;

        }

    }

TInt CHarvesterPluginTester::TestCreateIndexItemL( CStifItemParser& aItem )

    {

    TInt error = KErrNone;

    TPtrC fileName;

    TPtrC createIndexAction;

    CFilePlugin* filePlugin = CFilePlugin::NewL();

    CHarvesterObserver* observer = CHarvesterObserver::NewL( filePlugin );

    filePlugin->StartPluginL();

    _LIT( KFileNameFormat, "C:\\Data\\" );

    _LIT( KActionAdd, "add" );

    _LIT( KActionDelete, "delete" );

    _LIT( KActionUpdate, "update" );

    _LIT( KCreateIndexItemNoError, "CreateIndexItemL: Error" );

    TBuf<KMaxFileName> filePathName( KFileNameFormat );

    if( aItem.GetNextString ( createIndexAction ) == KErrNone )

        {

        if( createIndexAction.Compare( KActionAdd ) == 0 )

            {

            if( aItem.GetNextString ( fileName ) == KErrNone  && error == KErrNone )

                {

                filePathName.Append( fileName );

                filePlugin->CreateContentIndexItemL( filePathName, ECPixAddAction );

                User::After( (TTimeIntervalMicroSeconds32)35000000 );

                TPtrC searchString;

                while( aItem.GetNextString ( searchString ) == KErrNone  && error == KErrNone )

                    {

                    error = doSearchL( searchString, KAppBasePath, ESearchTypeResultsExpected );

                    }

                filePlugin->CreateContentIndexItemL( filePathName, ECPixRemoveAction );

                }

            }

        if( createIndexAction.Compare( KActionDelete ) == 0 )

            {

            if( aItem.GetNextString ( fileName ) == KErrNone  && error == KErrNone )

                {

                filePathName.Append( fileName );

                filePlugin->CreateContentIndexItemL( filePathName, ECPixRemoveAction );

                TPtrC searchString;

                while( aItem.GetNextString ( searchString ) == KErrNone  && error == KErrNone )

                    {

                    error = doSearchL( searchString, KAppBasePath, ESearchTypeNoResultsExpected );

                    }

                }

            }

        if( createIndexAction.Compare( KActionUpdate ) == 0 )

            {

            if( aItem.GetNextString ( fileName ) == KErrNone  && error == KErrNone )

                {

                filePathName.Append( fileName );

                filePlugin->CreateContentIndexItemL( filePathName, ECPixUpdateAction );

                User::After( (TTimeIntervalMicroSeconds32)35000000 );

                TPtrC searchString;

                while( aItem.GetNextString ( searchString ) == KErrNone  && error == KErrNone )

                    {

                    error = doSearchL( searchString, KAppBasePath, ESearchTypeResultsExpected );

                    }

                }

            }

        }

    delete filePlugin;

    delete observer;

    doLog( iLog, error, KCreateIndexItemNoError );

    return error;

    }

TInt CHarvesterPluginTester::TestFolderCreateL( CStifItemParser& /*aItem*/ )

    {

    TInt error = KErrNone;

    CFilePlugin* filePlugin = CFilePlugin::NewL();

    CHarvesterObserver* iPluginTester = CHarvesterObserver::NewL( filePlugin );

    filePlugin->StartPluginL();

    filePlugin->StartHarvestingL( KAppBasePath );

    _LIT( KDirectoryToCreate, "C:\\data\\TestFolder\\" );    

    RFs fs;

    User::LeaveIfError( fs.Connect() );

    TBool folderExists = BaflUtils::FolderExists(fs, KDirectoryToCreate);

    if(!folderExists)

        {

        error = fs.MkDir(KDirectoryToCreate);

        }    

    iPluginTester->iWaitForHarvester->Start();//Wait here till Harvesting is complete.

    if(error == KErrNone)

        {

        error = doSearchL( _L("TestFolder"), KAppBaseFolderFilePath, ESearchTypeResultsExpected );

        fs.RmDir(KDirectoryToCreate);

        }

    fs.Close();

    delete filePlugin;

    delete iPluginTester;

    iPluginTester = NULL;

    doLog( iLog, error, _L("Error: TestFolderCreateL") );

    return error;

    }

TInt CHarvesterPluginTester::TestFolderRenameL( CStifItemParser& /*aItem*/ )

    {

    TInt error = KErrNone;

    RFs fs;

    User::LeaveIfError( fs.Connect() );

    CFilePlugin* filePlugin = CFilePlugin::NewL();

    CFolderRenamedHarvester* iFolderRenameHarvester = CFolderRenamedHarvester::NewL( *filePlugin, fs);

    CHarvesterObserver* iPluginTester = CHarvesterObserver::NewL( filePlugin );

    filePlugin->StartPluginL();

    filePlugin->StartHarvestingL( KAppBasePath );    

    _LIT( KDirectoryToCreate, "C:\\data\\TestRenameFolder\\" );

    _LIT( KDirectoryRenamed, "C:\\data\\TestFolderRenamed\\" );

    TFileName oldFolderName(KDirectoryToCreate);

    TFileName newFolderName(KDirectoryRenamed);    

    if(!BaflUtils::FolderExists(fs, KDirectoryToCreate))

        {

        User::LeaveIfError(fs.MkDir(KDirectoryToCreate));        

        }

    if(BaflUtils::FolderExists(fs, KDirectoryRenamed))

        {

        User::LeaveIfError(fs.RmDir(KDirectoryRenamed));

        }

    iPluginTester->iWaitForHarvester->Start();//Wait here till Harvesting is complete.

    error = doSearchL( _L("TestRenameFolder"), KAppBaseFolderFilePath, ESearchTypeResultsExpected );        

    if(error == KErrNone)

        {

           fs.Rename(KDirectoryToCreate, KDirectoryRenamed);

           iFolderRenameHarvester->StartL( oldFolderName, newFolderName );           

        }    

    User::After( (TTimeIntervalMicroSeconds32)35000000 );

    //Search for the renamed directory

    error = doSearchL( _L("TestFolderRenamed"), KAppBaseFolderFilePath, ESearchTypeNoResultsExpected );    

    fs.RmDir(KDirectoryRenamed);    

    delete filePlugin;

    delete iFolderRenameHarvester;

    iFolderRenameHarvester = NULL;

    delete iPluginTester;

    iPluginTester = NULL;

    fs.Close();

    doLog( iLog, error, _L("Error: TestFolderRenameL") );

    return error;

    }

/**

* Message harvester test method.

* @since ?Series60_version

* @param aItem Script line containing parameters.

* @return Symbian OS error code.

*/

TInt CHarvesterPluginTester::TestMessageHarvesterL( CStifItemParser& /*aItem*/ )

    {

    // Print to UI

    _LIT( KHarvesterPluginTester, "HarvesterPluginTester" );

    _LIT( KExample, "In TestMessageHarvesterL" );

    TestModuleIf().Printf( 0, KHarvesterPluginTester, KExample );

    // Print to log file

    iLog->Log( KExample );

    CMessagePlugin* plugin = CMessagePlugin::NewL();

    CHarvesterObserver* iPluginTester = CHarvesterObserver::NewL( plugin );

    plugin->StartPluginL();

    CMessageSessionObserver* sessionobserver = CMessageSessionObserver::NewL();

    CMsvSession* msgSession = CMsvSession::OpenSyncL(*sessionobserver);

    // Form the baseappclass for this media

    TBuf<KMsgPluginBaseAppClassMaxLen> baseAppClass;

    TInt drive = msgSession->CurrentDriveL();

    CMessagePlugin::FormBaseAppClass(TDriveNumber(drive),baseAppClass);

    //Get current base app class for drive

    //CMessagePlugin::FormBaseAppClass

    plugin->StartHarvestingL( baseAppClass );

    iPluginTester->iWaitForHarvester->Start();