// Copyright (c) 2007-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:

// T_PLAINTEXTSTORE.CPP

// 

//

#include "t_CImplaintextstore.h"

#include "cactivewaiter.h"

LOCAL_D TMsvId pop3Service;

LOCAL_D TMsvId smtpService;

LOCAL_D TMsvId newMessageId;

LOCAL_D CTrapCleanup* theCleanup;

#define KTestMessageDir			_L("")

RTest test(KCImPlainBodyTest);

class CImPlainBodyText;

CTestCImPlainTextStore::CTestCImPlainTextStore()

	{

		iCImPlainBodyTextStore= NULL;

		iRequestCount = 0;

		iActiveWaiter = NULL;

	}

/**

Initialises the tests.

*/	

void CTestCImPlainTextStore::SetupL()

	{

	iActiveWaiter = new (ELeave) CActiveWaiter();

	testUtils = CEmailTestUtils::NewL(test);

	testUtils->CreateAllTestDirectories();

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

	testUtils->ClearEmailAccountsL();

	testUtils->CleanMessageFolderL();

	testUtils->GoServerSideL();

	TBuf<KMaxFilePathLength>emailbodyfile(KEmail40kBodyFile);

	RFs fsSession;

	RFile infile; 

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

 	User::LeaveIfError(infile.Open(fsSession, emailbodyfile, EFileRead));

	TInt size = 0;

	infile.Size(size);

 	//read the 8 bit data from the file

	TInt chunkLength = size / 3; 

 	HBufC8* tempBuf1 = HBufC8::NewL(chunkLength);	

 	TPtr8 tempBufPtr1(tempBuf1->Des());

 	infile.Read(tempBufPtr1,chunkLength);

	iFilebuf1 = HBufC16::NewL(chunkLength);  

	TPtr16 ptr1=iFilebuf1->Des();	  

	ptr1.Copy(tempBufPtr1);

	HBufC8* tempBuf2 = HBufC8::NewL(chunkLength);	

 	TPtr8 tempBufPtr2(tempBuf2->Des());

 	infile.Read(tempBufPtr2,chunkLength);

	iFilebuf2 = HBufC16::NewL(chunkLength);  

	TPtr16 ptr2=iFilebuf2->Des();	  

	ptr2.Copy(tempBufPtr2);

	HBufC8* tempBuf3 = HBufC8::NewL(chunkLength);	

 	TPtr8 tempBufPtr3(tempBuf3->Des());

	infile.Read(tempBufPtr3,chunkLength);

	iFilebuf3 = HBufC16::NewL(chunkLength);  

	TPtr16 ptr3=iFilebuf3->Des();	  

	ptr3.Copy(tempBufPtr3);

	infile.Close();

	testUtils->GoClientSideL();

	testUtils->ClearEmailAccountsL();

	smtpService = testUtils->CreateSmtpServiceL();

	pop3Service = testUtils->CreatePopServiceL();

	testUtils->GoServerSideL();

	testUtils->FileSession().SetSessionPath(_L("c:\\"));

	test.Printf(_L("\nPerforming Tests\n"));

	testUtils->GoClientSideL();

	CEmailAccounts* accounts = CEmailAccounts::NewLC();

	TSmtpAccount smtpAccount;

	accounts->GetSmtpAccountL(smtpService, smtpAccount);

	accounts->SetDefaultSmtpAccountL(smtpAccount);

	CleanupStack::PopAndDestroy(accounts);

	TParse parsedFileName;

	TFileName name(KTestMessageDir);

	testUtils->ResolveLogFile(name, parsedFileName);

	// Get the default charset

	CCnvCharacterSetConverter* characterConverter = CCnvCharacterSetConverter::NewL();

	CleanupStack::PushL(characterConverter);

	CImConvertCharconv* charConv = CImConvertCharconv::NewL(*characterConverter,fsSession );

	CleanupStack::PushL(charConv);

	iDefaultCharset =	charConv->SystemDefaultCharset();

	CleanupStack::PopAndDestroy(2);

	fsSession.Close();

	}

void CTestCImPlainTextStore::TearDownL()

//virtual 

	{

	delete iCImPlainBodyTextStore;

	delete iFilebuf1;

	delete iFilebuf2;

	delete iFilebuf3;

	delete testUtils;

	delete iActiveWaiter;

	test.Close();

	}

/**

Create new Plaintext message

*/

void CTestCImPlainTextStore::CreateNewPlaintextMessageL()

	{

	TMsvEmailTypeList msvEmailTypeList = 0;

	TMsvPartList partList = (KMsvMessagePartBody | KMsvMessagePartAttachments);

	CImEmailOperation* emailOperation = CImEmailOperation::CreateNewL(iActiveWaiter->iStatus, *(testUtils->iMsvSession), KMsvGlobalOutBoxIndexEntryId,KMsvNullIndexEntryId, partList, msvEmailTypeList, KUidMsgTypeSMTP, CActive::EPriorityStandard, ETrue);

	CleanupStack::PushL(emailOperation);

	iActiveWaiter->WaitActive();

	//check progres

	TMsvId temp;	

	TPckgC<TMsvId> paramPack(temp);

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

	paramPack.Set(progBuf);

	newMessageId=paramPack();

	CleanupStack::PopAndDestroy(emailOperation); 

	}

void CTestCImPlainTextStore::TestStoreRestore16BitChunkSynchL()

	{

	CreateNewPlaintextMessageL();

	CMsvEntry* entry = testUtils->iMsvSession->GetEntryL(newMessageId);

	CImEmailMessage* imEmailMessage = CImEmailMessage::NewLC(*entry);

	iCImPlainBodyTextStore = imEmailMessage->OpenPlainBodyTextForWriteL();

	iCImPlainBodyTextStore->StoreChunkL(iFilebuf1->Des());

	TInt bodyTextSize = iFilebuf1->Length();

	iCImPlainBodyTextStore->StoreChunkL(iFilebuf2->Des());

	bodyTextSize += iFilebuf2->Length();

	iCImPlainBodyTextStore->StoreChunkL(iFilebuf3->Des());

	bodyTextSize += iFilebuf3->Length();

	iCImPlainBodyTextStore->CommitL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	delete iCImPlainBodyTextStore;

	iCImPlainBodyTextStore = NULL;

	//Restore the data in chunks.	

	HBufC16* filebuf11 = HBufC16::NewL(100+KTestStringLength );

	TInt origSize = bodyTextSize;

	TInt sizeOfRestoredText = 0;

	TPtr bufPtr = filebuf11->Des();

	TInt pos = 0;

	iCImPlainBodyTextStore = imEmailMessage->OpenPlainBodyTextForReadL(CImEmailMessage::EThisMessageOnly,100);

	iCImPlainBodyTextStore->NextChunkL(bufPtr);

	RArray<TPtr> chunkArray;

	chunkArray.Append(bufPtr);

	sizeOfRestoredText += bufPtr.Length();

	while(bufPtr.Length() > 0 && bufPtr.Length() == 100)

		{

		iCImPlainBodyTextStore->NextChunkL(bufPtr);

		// put retrived buffer in a array.

		chunkArray.Append(bufPtr);

		sizeOfRestoredText += bufPtr.Length();

		}

	ASSERT_EQUALS(bodyTextSize, sizeOfRestoredText);

	for(TInt j=chunkArray.Count() - 1;j >0; --j )

		{

		bufPtr.SetLength(0);

		iCImPlainBodyTextStore->PreviousChunkL(bufPtr);

		// Compare chunk retrieved with chunk retrieved by NextChunkL

		TInt comp = bufPtr.Compare(chunkArray[j-1]);

		ASSERT_EQUALS(comp, 0);

		}

	CleanupStack::PopAndDestroy(imEmailMessage); 

	}

void CTestCImPlainTextStore::TestStoreRestore16BitChunkAsyncL()

	{

	CreateNewPlaintextMessageL();

	CMsvEntry* entry = testUtils->iMsvSession->GetEntryL(newMessageId);

	CImEmailMessage* imEmailMessage = CImEmailMessage::NewLC(*entry);

	iCImPlainBodyTextStore = imEmailMessage->OpenPlainBodyTextForWriteL();

	iCImPlainBodyTextStore->StoreChunkL(iFilebuf1->Des(),iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	TInt bodyTextSize = iFilebuf1->Length();

	iCImPlainBodyTextStore->StoreChunkL(iFilebuf2->Des(),iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	bodyTextSize += iFilebuf2->Length();

	iCImPlainBodyTextStore->StoreChunkL(iFilebuf3->Des(),iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	bodyTextSize += iFilebuf3->Length();

	iCImPlainBodyTextStore->CommitL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	delete iCImPlainBodyTextStore;

	iCImPlainBodyTextStore = NULL;

	//Restore the data in chunks.	

	HBufC16* filebuf11 = HBufC16::NewL(100+KTestStringLength);

	TInt origSize = bodyTextSize;

	TInt sizeOfRestoredText = 0;

	TPtr bufPtr = filebuf11->Des();

	TInt pos = 0;

	iCImPlainBodyTextStore = imEmailMessage->OpenPlainBodyTextForReadL(CImEmailMessage::EThisMessageOnly,100);

	iCImPlainBodyTextStore->NextChunkL(bufPtr,iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	RArray<TPtr> chunkArray;

	chunkArray.Append(bufPtr);

	sizeOfRestoredText += bufPtr.Length();

	while(bufPtr.Length() > 0 && bufPtr.Length() == 100)

		{

		iCImPlainBodyTextStore->NextChunkL(bufPtr,iActiveWaiter->iStatus);

		iActiveWaiter->WaitActive();

		// put retrived buffer in a array.

		chunkArray.Append(bufPtr);

		sizeOfRestoredText += bufPtr.Length();

		}

	ASSERT_EQUALS(bodyTextSize, sizeOfRestoredText);

	for(TInt j=chunkArray.Count() - 1;j >0; --j )

		{

		bufPtr.SetLength(0);

		iCImPlainBodyTextStore->PreviousChunkL(bufPtr,iActiveWaiter->iStatus);

		iActiveWaiter->WaitActive();

		// Compare chunk retrieved with chunk retrieved by NextChunkL

		TInt comp = bufPtr.Compare(chunkArray[j-1]);

		ASSERT_EQUALS(comp, 0);

		}

	CleanupStack::PopAndDestroy(imEmailMessage); 

	}

void CTestCImPlainTextStore::TestStoreRichTextAsPlainTextL()

	{	

	CreateNewPlaintextMessageL();

	CMsvEntry* entry = testUtils->iMsvSession->GetEntryL(newMessageId);

	CImEmailMessage* imEmailMessage = CImEmailMessage::NewLC(*entry);

	iCImPlainBodyTextStore = imEmailMessage->OpenPlainBodyTextForWriteL();

	CParaFormatLayer* paraLayer = CParaFormatLayer::NewL();

	CCharFormatLayer* charLayer = CCharFormatLayer::NewL();

	CRichText* richText=CRichText::NewL(paraLayer, charLayer, CEditableText::EFlatStorage, KBufferGranularity);

	TBuf<50> theMessageContents = _L("Test message contents");

	richText->InsertL(0, theMessageContents);

	TInt textLength = richText->DocumentLength();

	iCImPlainBodyTextStore->StoreRichTextAsPlainTextL(*richText);

	iCImPlainBodyTextStore->CommitL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	delete richText;

	richText=CRichText::NewL(paraLayer, charLayer, CEditableText::EFlatStorage, KBufferGranularity);

	imEmailMessage->GetBodyTextL(newMessageId, CImEmailMessage::EThisMessageOnly, *richText, *paraLayer, *charLayer);	

	richText->DocumentLength();

	ASSERT_EQUALS(richText->DocumentLength(), textLength);

	delete richText;

	delete charLayer;

	delete paraLayer;

	CleanupStack::PopAndDestroy(imEmailMessage); 

	}	

CTestSuite* CTestCImPlainTextStore::CreateSuiteL(const TDesC& aName)

	{

	SUB_SUITE;

	ADD_ASYNC_TEST_STEP(TestStoreRestore16BitChunkSynchL);

	ADD_ASYNC_TEST_STEP(TestStoreRestore16BitChunkAsyncL);

	ADD_ASYNC_TEST_STEP(TestStoreRichTextAsPlainTextL);

	END_SUITE;

	}