// Copyright (c) 2005-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 <testexecutelog.h>

#include "t_test_decode_step.h"

#include "MIUTMSG.H"

#include <utf.h>

RTest test(_L("Decode Test"));

const TInt KImapPort = 143;

CTestDecode::~CTestDecode()

	{

	delete iConsole;

	delete iSpoofServer;	

	delete iTestUtils;

	delete iImapClient;	

	delete iScheduler;

	iSession->CloseMessageServer( );

	delete iSession;

	delete iSessionObserver;

	delete iExpectedPartsBuffer;

	delete iActualPartsBuffer;

	iFs.Close( );

	}

CTestDecode::CTestDecode()

	{

	// Call base class method to set up the human readable name for logging

	SetTestStepName( KTestDecode );

	iFs.Connect( );

	}

void CTestDecode::TestComplete( TInt aErrorCode )

	{

	CActiveScheduler::Stop();	

	if( aErrorCode != KErrNone && aErrorCode != KErrEof )

		{

		ERR_PRINTF2( _L( "Error: %d" ), aErrorCode );	

		}

	}

TBufferType CTestDecode::ConvertToBufferType(const TDesC& aBufferTypeDesc)

	{

	if ( aBufferTypeDesc.CompareF( _L( "TBufferType::EBodyBuffer" ) ) == KErrNone )

		{

		return ( TBufferType ) EBodyBuffer;

		}

	else if ( aBufferTypeDesc.CompareF( _L( "TBufferType::EAttachmentBuffer" ) ) == KErrNone )

		{

		return ( TBufferType ) EAttachmentBuffer;

		}

	return ( TBufferType ) EUndefinedBuffer;

	}

void CTestDecode::GetDetailsFromIniFileL()

	{

	//get the script file path

	if( !GetStringFromConfig( ConfigSection( ) , 

							  _L( "ScriptFile" ),

							  iScriptFile ) )

		{

		ERR_PRINTF1( _L( "Error reading script file path from ini file" ) );

		User::Leave( KErrArgument );	

		}

	TPtrC parameter;

	if( GetStringFromConfig( ConfigSection( ),

							 _L( "BufferFile" ),

							 parameter ) )

		{

		TLex parser( parameter );

		parser.SkipSpace( );

		while ( !parser.Eos( ) )

			{

			TBuffer* buffer = new ( ELeave ) TBuffer();

			CleanupStack::PushL( buffer );

			TPtrC arg;

			parser.SkipSpace( );

			if (parser.Eos())

				{

				ERR_PRINTF1( _L( "Invalid Argument for BufferFile" ) );

				User::Leave( KErrArgument );	

				}

			arg.Set( parser.NextToken( ) );

			buffer->iBufferType = ConvertToBufferType( arg );

			arg.Set( parser.NextToken( ) );

			parser.SkipSpace( );

			User::LeaveIfError( parser.Val( buffer->iAllowJunkAtEnd ) );

			RFile file;

			file.Open( iFs, arg, EFileRead );

			CleanupClosePushL( file );

			file.Read( buffer->iBuffer );

			iExpectedPartsBuffer->AppendL( *buffer );

			CleanupStack::PopAndDestroy( 1 );	//file

			CleanupStack::Pop( buffer );

			parser.SkipSpace( );

			}

		}

	else

		{

		ERR_PRINTF1( _L( "Error reading buffer file path from ini file" ) );

		User::Leave( KErrArgument );

		}

	//get the script file path

	if( !GetBoolFromConfig( ConfigSection( ),

							_L( "FetchWholeMessage" ),

							iFetchWholeMessage ) )

		{

		ERR_PRINTF1( _L( "Error reading \"FetchWholeMessage\"-argument from ini file" ) );

		User::Leave( KErrArgument );

		}

	}

void CTestDecode::TraverseL( const TMsvId aTMsvId )

	{

	//open the entry, dont sort

	TMsvSelectionOrdering ordering;	

	ordering.SetSorting( EMsvSortByNone );

	CMsvEntry* entry = CMsvEntry::NewL( *iSession, aTMsvId,ordering );

	CleanupStack::PushL( entry );

	//retrieve the entries children

	CMsvEntrySelection* msvEntrySelection;

	msvEntrySelection=entry->ChildrenL( );

	CleanupStack::PushL( msvEntrySelection );

	TInt count = entry->Count( );

	CMsvEntry* child;

	TMsvEntry childDetails;

	//for each child test if its a message entry

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

		{

		child=iSession->GetEntryL( ( *msvEntrySelection ) [ i ] );

		CleanupStack::PushL( child );

		childDetails=child->Entry( );

		if( childDetails.iType == KUidMsvMessageEntry )

			{

			//First Check the body

			CParaFormatLayer* paraFormatLayer = CParaFormatLayer::NewL( );

			CleanupStack::PushL( paraFormatLayer );

			CCharFormatLayer* charFormatLayer = CCharFormatLayer::NewL( );

			CleanupStack::PushL( charFormatLayer );

			CRichText* richText = CRichText::NewL( paraFormatLayer, charFormatLayer );

			CleanupStack::PushL( richText );

			CImEmailMessage* emailMessage = CImEmailMessage::NewLC( *child );

			emailMessage->GetBodyTextL( child->EntryId( ), 

										CImEmailMessage::EThisMessageOnly,

										*richText, 

										*paraFormatLayer, 

										*charFormatLayer );

			emailMessage->GetAttachmentsListL( child->EntryId( ),

											   CImEmailMessage::EAllAttachments,

											   CImEmailMessage::EThisMessageOnly );

			TDes* buffer16;

			buffer16 = new( ELeave ) TBuf<5000>( );

			CleanupStack::PushL( buffer16 );

			richText->Extract( *buffer16, 0 );

			TBuf8<KMaxBufferSize> buffer8;

			TInt unconverted =  CnvUtfConverter::ConvertFromUnicodeToUtf8( buffer8, 

																	   	   *buffer16 );

			iActualPartsBuffer->AppendL( buffer8 );

			CleanupStack::PopAndDestroy( buffer16 );

			MMsvAttachmentManager& attachmentMgr = emailMessage->AttachmentManager();

			RPointerArray<CMsvAttachment> attachmentInfoArray;

			attachmentInfoArray.Reset();

			CMsvAttachment* attachmentInfo;

			for (TInt i= 0; i< attachmentMgr.AttachmentCount();i++)

				{

				attachmentInfo = attachmentMgr.GetAttachmentInfoL(i);

				attachmentInfoArray.Append(attachmentInfo);

				TDes8* buffer;									

				buffer = new( ELeave ) TBuf8<5000>( );

				CleanupStack::PushL( buffer );

				RFile file;

				file.Open( iFs, attachmentInfo->FilePath( ), EFileRead );

				CleanupClosePushL( file );

				file.Read( *buffer );

				iActualPartsBuffer->AppendL( *buffer );

				CleanupStack::PopAndDestroy( 2, buffer );	//file, buffer

			}

			//if this is the first timestamp the us iActualTimeStamp, else use iActualTimeStamp2

			CleanupStack::PopAndDestroy( 4, paraFormatLayer );	

			}

		//recursive call

		TraverseL( childDetails.Id( ) );

		CleanupStack::PopAndDestroy( child );

		}

	CleanupStack::PopAndDestroy( msvEntrySelection );

	CleanupStack::PopAndDestroy( entry );	

	}

TVerdict CTestDecode::doTestStepPreambleL( )

	{

	SetTestStepResult( EFail );

	iScheduler = new ( ELeave ) CActiveScheduler;

	CActiveScheduler::Install( iScheduler );

	iTestUtils = CEmailTestUtils::NewL( test );

	iTestUtils->FileSession( ).SetSessionPath( _L( "C:\\" ) );

	iTestUtils->CleanMessageFolderL( );

	iTestUtils->ClearEmailAccountsL( );

	iTestUtils->GoClientSideL( );

	//initialize the buffers

	iExpectedPartsBuffer = new( ELeave) CArrayFixSeg< TBuffer >( 1 );	// granularity of 1, very often there 

	iActualPartsBuffer = new( ELeave) CDesC8ArraySeg( 1 );				// will beonly 1 message part

	//start the imap server

	GetDetailsFromIniFileL( );

	iConsole=Console::NewL( _L( "Decode Test" ),TSize( KConsFullScreen, KConsFullScreen ) );

	iSpoofServer = CSpoofServer::NewL(*this,iScriptFile);

	iSpoofServer->StartL(KImapPort);

	//start the imap client

	iImapClient = CActiveImapClient::NewL(this,iFetchWholeMessage);

	iImapClient->StartL();

	CActiveScheduler::Start();

	SetTestStepResult( EPass );

	return TestStepResult();

	}

TVerdict CTestDecode::CheckBuffer( const TDesC8& aActualBuffer, 

								   const TDesC8& aExpectedBuffer, 

								   TBool aAllowJunkAtEnd )

	{

	TVerdict result = EFail;

	TInt compareResult = aExpectedBuffer.Compare( aActualBuffer );

	if ( compareResult == 0)

		{

		INFO_PRINTF1( _L( "Decoded buffer does match reference file" ) );

		result = EPass;

		}

	else if ( compareResult == aExpectedBuffer.Length( )-aActualBuffer.Length( ) &&

			  aAllowJunkAtEnd )

		{

		WARN_PRINTF1( _L( "Decoded buffer math reference file, but contains extra \"junk\" at the end" ) );

		result = EPass;

		}

	else

		{

		INFO_PRINTF1( _L( "Decoded buffer does not match reference file" ) );

		}

	return result;

	}

TVerdict CTestDecode::doTestStepL( )

	{

	TVerdict result = EPass;

	// Session observer. Needed to create a session

	iSessionObserver = new (ELeave) TDummySessionObserver;

	// Session. Needed to create a client registry.

	iSession=CMsvSession::OpenSyncL(*iSessionObserver);

	// Client registry. Needed to get the MTM component

	TraverseL(iImapClient->GetAccount().iImapService);	

	// Check every message-part (we expect them to be in the same order 

	// as defined in the .ini configuration file)

	TInt partsCount = iExpectedPartsBuffer->Count( );

	if ( partsCount != iActualPartsBuffer->Count( ) )

		{

		ERR_PRINTF1( _L( "Number of expected parts does not mach actual number of parts" ) );

		User::Leave( KErrGeneral );

		}

	// Stop as soon as a test fail, we don't care about the rest.

	for ( TInt i=0; ( ( i < partsCount ) && ( result != EFail ) ); i++ )

		{

		TBuffer expectedPart = iExpectedPartsBuffer->At( i );

		result = CheckBuffer( iActualPartsBuffer->MdcaPoint( i ), 

							  expectedPart.iBuffer,

							  expectedPart.iAllowJunkAtEnd );

		}

	SetTestStepResult( result );

	return TestStepResult( );

	}

TVerdict CTestDecode::doTestStepPostambleL()

	{

	return TestStepResult( );

	}