/*

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

*

*/

// INCLUDE FILES

#include <e32svr.h>

#include <StifParser.h>

#include <StifTestEventInterface.h>

#include <StifTestInterface.h>

#include "HarvesterClientTestScripter.h"

#include "mdcserializationbuffer.h"

#include "mdeharvestersession.h"

#include "mdsutils.h"

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

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

// CHarvesterClientTestScripter::Delete

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

// Called from destructor. 

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

//

void CHarvesterClientTestScripter::Delete() 

    {

    }

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

// CHarvesterClientTestScripter::RunMethodL

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

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

//

TInt CHarvesterClientTestScripter::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( "SetUp", CHarvesterClientTestScripter::SetUp ),

		ENTRY( "TearDown", CHarvesterClientTestScripter::TearDown ),

        // Test helper methods

        ENTRY( "AddObjectObserver", CHarvesterClientTestScripter::AddObjectObserverL ),

        ENTRY( "RemoveObjectObserver", CHarvesterClientTestScripter::RemoveObjectObserverL ),

        ENTRY( "SetUpHarvesting", CHarvesterClientTestScripter::BeginHarvesterSessionL ),

        ENTRY( "EndHarvesterSession", CHarvesterClientTestScripter::EndHarvesterSessionL ),

        // Harvester tests

        ENTRY( "HarvesterClientConnect", CHarvesterClientTestScripter::HarvesterClientConnectL ),

        ENTRY( "HarvesterClientClose", CHarvesterClientTestScripter::HarvesterClientCloseL ),

        ENTRY( "HarvesterClientPause", CHarvesterClientTestScripter::HarvesterClientPauseL ),

        ENTRY( "HarvesterClientResume", CHarvesterClientTestScripter::HarvesterClientResumeL ),

        ENTRY( "SetHarvesterObserver", CHarvesterClientTestScripter::SetHarvesterObserverL ),

        ENTRY( "RemoveHarvesterObserver", CHarvesterClientTestScripter::RemoveHarvesterObserverL ),

        ENTRY( "SetLocationInfo", CHarvesterClientTestScripter::SetLocationInfoL ),

        ENTRY( "HarvestFile", CHarvesterClientTestScripter::HarvestFileL ),

        ENTRY( "HarvestFileWithUID", CHarvesterClientTestScripter::HarvestFileWithUIDL ),

        ENTRY( "QueryImageObjectByIdIndex", CHarvesterClientTestScripter::QueryImageObjectByIdIndexL ),

        ENTRY( "QueryAudioObjectByIdIndex", CHarvesterClientTestScripter::QueryAudioObjectByIdIndexL ),

        ENTRY( "RemoveObject", CHarvesterClientTestScripter::RemoveObjectL ),

        ENTRY( "Results", CHarvesterClientTestScripter::Results ),     

        ENTRY( "AddHarvesterEventObserver", CHarvesterClientTestScripter::AddHarvesterEventObserverL ),

        ENTRY( "RemoveHarvesterEventObserver", CHarvesterClientTestScripter::RemoveHarvesterEventObserverL ),

        };

    const TInt count = sizeof( KFunctions ) / sizeof( TStifFunctionInfo );

    return RunInternalL( KFunctions, count, aItem );

    }

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

// CHarvesterClientTest::SetUp

// Set up test.

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

//

TInt CHarvesterClientTestScripter::SetUp( CStifItemParser& /* aItem */ )

	{

    _LIT( KMsg1, "SetUp" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

	iHarvestingComplete = EFalse;

	iPlaceholderComplete = EFalse;

	iNewObjectComplete = EFalse;

	iObjectNotification = EFalse;

	iSetLocation = EFalse;

	iObjectIdArray.Reset();

	TRAP_IGNORE( iMdeSession = CMdESession::NewL( *this ) );

    return KErrNone;

	}

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

// CHarvesterClientTest::TearDown

// Tear down test.

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

//

TInt CHarvesterClientTestScripter::TearDown( CStifItemParser& /* aItem */ )

	{

    _LIT( KMsg1, "TearDown" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

	delete iHarvesterSession;

	iHarvesterSession = NULL;

	delete iMdeSession;

	iMdeSession = NULL;

    return KErrNone;

	}

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

// CHarvesterClientTest::HarvesterClientConnectL

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

//

TInt CHarvesterClientTestScripter::HarvesterClientConnectL( CStifItemParser& /*aItem*/ )

    {

    _LIT( KMsg1, "HarvesterClientConnectL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

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

    return KErrNone;

    }

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

// CHarvesterClientTest::HarvesterClientCloseL

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

//

TInt CHarvesterClientTestScripter::HarvesterClientCloseL( CStifItemParser& /*aItem*/ )

    {

    _LIT( KMsg1, "HarvesterClientCloseL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    iHc.Close();

    return KErrNone;

    }

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

// CHarvesterClientTest::HarvesterClientPauseL

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

//

TInt CHarvesterClientTestScripter::HarvesterClientPauseL( CStifItemParser& /*aItem*/ )

    {

    _LIT( KMsg1, "Enter HarvesterClientPauseL" );

    iLog->Log( KMsg1 );  

    RDebug::Print( KMsg1 );

    User::LeaveIfError( iHc.Pause() );

    _LIT( KMsg2, "Exit HarvesterClientPauseL" );

    iLog->Log( KMsg2 );  

    RDebug::Print( KMsg2 );

    return KErrNone;

    }

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

// CHarvesterClientTest::HarvesterClientResumeL

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

//

TInt CHarvesterClientTestScripter::HarvesterClientResumeL( CStifItemParser& /*aItem*/ )

    {

    _LIT( KMsg1, "Enter HarvesterClientResumeL" );

     iLog->Log( KMsg1 );  

     RDebug::Print( KMsg1 );

    User::LeaveIfError( iHc.Resume() );

    _LIT( KMsg2, "Exit HarvesterClientResumeL" );

    iLog->Log( KMsg2 );  

    RDebug::Print( KMsg2 );

    return KErrNone;

    }

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

// CHarvesterClientTest::SetHarvesterObserverL

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

//

TInt CHarvesterClientTestScripter::SetHarvesterObserverL( CStifItemParser& /*aItem*/ )

    {

    _LIT( KMsg1, "SetHarvesterObserverL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    iHc.SetObserver( this );

    return KErrNone;

    }

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

// CHarvesterClientTest::RemoveHarvesterObserverL

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

//

TInt CHarvesterClientTestScripter::RemoveHarvesterObserverL( CStifItemParser& /*aItem*/ )

    {

    _LIT( KMsg1, "RemoveHarvesterObserverL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    iHc.RemoveObserver( this );

    return KErrNone;

    }

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

// CHarvesterClientTest::AddObjectObserverL

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

//

TInt CHarvesterClientTestScripter::AddObjectObserverL( CStifItemParser& /* aItem */ )

    {

    _LIT( KMsg1, "AddObjectObserverL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    iMdeSession->AddObjectObserverL( *this );

    return KErrNone;

    }

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

// CHarvesterClientTest::AddObjectObserverL

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

//

TInt CHarvesterClientTestScripter::RemoveObjectObserverL( CStifItemParser& /* aItem */ )

    {

    _LIT( KMsg1, "RemoveObjectObserverL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    TRAP_IGNORE( iMdeSession->RemoveObjectObserverL( *this ) );

    return KErrNone;

    }

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

// CHarvesterClientTest::AddObjectObserverL

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

//

TInt CHarvesterClientTestScripter::SetLocationInfoL( CStifItemParser& /* aItem */ )

    {

    iSetLocation = ETrue;

    return KErrNone;

    }

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

// CHarvesterClientTest::StartHarvesterSessionL

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

//

TInt CHarvesterClientTestScripter::BeginHarvesterSessionL( CStifItemParser& aItem )

    {

    _LIT( KMsg1, "Enter BeginHarvesterSessionL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    TPtrC inputFile;

    User::LeaveIfError( aItem.GetNextString( inputFile ));

    TRAP_IGNORE( iMdeSession->RemoveObjectL(inputFile) );

    _LIT( KMsg2, "Exit BeginHarvesterSessionL" );

    iLog->Log( KMsg2 );

    RDebug::Print( KMsg2 );

    return KErrNone;

    }

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

// CHarvesterClientTest::HarvestFileTestL

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

//

TInt CHarvesterClientTestScripter::HarvestFileL( CStifItemParser& aItem )

    {

    _LIT( KMsg1, "Enter HarvestFileL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

	TPtrC inputFile;

	User::LeaveIfError( aItem.GetNextString( inputFile ));

    iAlbumIds.Append( 1 );

    iQuerySuccess = EFalse;

    iUri = inputFile;

    iHc.HarvestFile( inputFile, iAlbumIds, iSetLocation );

    _LIT( KMsg2, "Exit HarvestFileL" );

    iLog->Log( KMsg2 );

    RDebug::Print( KMsg2 );

    return KErrNone;

    }

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

// CHarvesterClientTest::HarvestFileTestL

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

//

TInt CHarvesterClientTestScripter::HarvestFileWithUIDL( CStifItemParser& aItem )

    {

    _LIT( KMsg1, "Enter HarvestFileWithUIDL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    TUid uid( KNullUid );

    TPtrC inputFile;

    User::LeaveIfError( aItem.GetNextString( inputFile ));

    iAlbumIds.Append( 1 );

    iQuerySuccess = EFalse;

    iUri = inputFile;

    iHc.HarvestFileWithUID( inputFile, iAlbumIds, EFalse, uid );

    _LIT( KMsg2, "Exit HarvestFileWithUIDL" );

    iLog->Log( KMsg2 );

    RDebug::Print( KMsg2 );

    return KErrNone;

    }

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

// CHarvesterClientTest::HarvestingCompleteL

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

//

TInt CHarvesterClientTestScripter::EndHarvesterSessionL( CStifItemParser& /* aItem */ )

    {

    _LIT( KMsg1, "EndHarvesterSessionL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    iAlbumIds.Close();

    return KErrNone;

    }

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

// CHarvesterClientTest::QueryObjectByIdIndexL

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

//

TInt CHarvesterClientTestScripter::QueryImageObjectByIdIndexL( CStifItemParser& aItem )

    {

    _LIT( KMsg1, "QueryImageObjectByIdIndexL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    TInt index;

    User::LeaveIfError( aItem.GetNextInt( index ));

    CMdENamespaceDef& defNS = iMdeSession->GetDefaultNamespaceDefL();

    CMdEObjectDef& objDef = defNS.GetObjectDefL( MdeConstants::Image::KImageObject );

    iQuery = iMdeSession->NewObjectQueryL( defNS, objDef, this );

    iQuery->SetResultMode( EQueryResultModeItem );

    iQuery->FindL();

    return KErrNone;

    }

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

// CHarvesterClientTest::QueryObjectByIdIndexL

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

//

TInt CHarvesterClientTestScripter::QueryAudioObjectByIdIndexL( CStifItemParser& aItem )

    {

    _LIT( KMsg1, "QueryAudioObjectByIdIndexL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    TInt index;

    User::LeaveIfError( aItem.GetNextInt( index ));

    CMdENamespaceDef& defNS = iMdeSession->GetDefaultNamespaceDefL();

    CMdEObjectDef& objDef = defNS.GetObjectDefL( MdeConstants::Audio::KAudioObject );

    iQuery = iMdeSession->NewObjectQueryL( defNS, objDef, this );

    iQuery->SetResultMode( EQueryResultModeItem );

    iQuery->FindL();

    return KErrNone;

    }

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

// CHarvesterClientTest::RemoveObjectL

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

//

TInt CHarvesterClientTestScripter::RemoveObjectL( CStifItemParser& aItem )

    {

    _LIT( KMsg1, "RemoveObjectL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    TPtrC inputFile;

    User::LeaveIfError( aItem.GetNextString( inputFile ));

    TRAP_IGNORE( iMdeSession->RemoveObjectL( inputFile ) );

    return KErrNone;

    }

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

// CHarvesterClientTest::Results

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

//

TInt CHarvesterClientTestScripter::Results( CStifItemParser& /*aItem*/ )

    {

    _LIT( KMsg1, "Enter Results" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    if( !iQuerySuccess )

        {

        return KErrNotFound;

        }

    if( !iHarvestingComplete )

        {

        return KErrUnknown;

        }

    _LIT( KMsg2, "Enter Results" );

    iLog->Log( KMsg2 );

    RDebug::Print( KMsg2 );

    return KErrNone;

    }

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

// CHarvesterClientTest::AddHarvesterEventObserver

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

//

TInt CHarvesterClientTestScripter::AddHarvesterEventObserverL( CStifItemParser& aItem )

    {

    _LIT( KMsg1, "AddHarvesterEventObserverL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    TInt testCount( 0 );

    User::LeaveIfError( aItem.GetNextInt( testCount ) );

    for( TInt i = 0; i < testCount; i++ )

    	{

    	TInt err = iHc.AddHarvesterEventObserver( *this, EHEObserverTypePlaceholder, 3 );

	    TL( err == KErrNone );

    	}

    return KErrNone;

    }

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

// CHarvesterClientTest::RemoveHarvesterEventObserver

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

//

TInt CHarvesterClientTestScripter::RemoveHarvesterEventObserverL( CStifItemParser& aItem )

    {

    _LIT( KMsg1, "RemoveHarvesterEventObserverL" );

    iLog->Log( KMsg1 );

    RDebug::Print( KMsg1 );

    TInt testCount( 0 );

    User::LeaveIfError( aItem.GetNextInt( testCount ) );

    for( TInt i = 0; i < testCount; i++ )

		{

		TInt err = iHc.RemoveHarvesterEventObserver( *this );

	    TL( err == KErrNone );

		}

    return KErrNone;

    }

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

// CHarvesterClientTest::HarvestingUpdated

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

//

void CHarvesterClientTestScripter::HarvestingUpdated( 

	HarvesterEventObserverType /*aHEObserverType*/, 

	HarvesterEventState /*aHarvesterEventState*/,

	TInt /*aItemsLeft*/ )

	{

    _LIT( KMsg1, "CallBck HarvestingUpdated" );

    iLog->Log( KMsg1 );     

    RDebug::Print( KMsg1 );

	}

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

// CHarvesterClientTest::HarvestingComplete

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

//

void CHarvesterClientTestScripter::HarvestingComplete( TDesC& aURI, TInt aError )

	{

    _LIT( KMsg, "CallBck HarvestingComplete - Error code : %d" );

    TBuf <100> msg;

    msg.Format(KMsg, aError);

    iLog->Log( msg );     

    RDebug::Print( msg );

    if( MdsUtils::Compare( aURI, iUri ) == 0 )

        {

        if( iObjectNotification )

            {

            iHarvestingComplete = ETrue;

            // session event

            TEventIf event( TEventIf::ESetEvent, _L("Continue") );

            TestModuleIf().Event( event );

            }

        else

            {

            iHarvestingComplete = ETrue;