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

// Name of test harness: T_SMTSND

// Component: IMCM

// Owner: KP

// Brief description of test harness:

// Tests SMTP SendAs functionality - creates SMTP messages; adds recipients; 

// store the body text; saves message; adds/removes attachments;

// Detailed description of test harness:

// As above.

// Input files required to run test harness:

// \MSG\IMCMTSRC\rfc82201.txt		<DRIVE>:\msgtest\imcm\rfc82201.txt

// Intermediate files produced while running test harness:

// <DRIVE>:\msglogs\T_SMTSND\T_IMCM.log

// Output files produced by running test harness:

// <DRIVE>:\msglogs\T_SMTSND.<PLATFORM>.<VARIANT>.LOG.txt

// Description of how to build test harness:

// cd \msg\imcm\

// bldmake bldfiles

// abld test build

// Description of how to run test harness:

// The following instructions are the same for all platforms:

// 1. Build T_DB test harness from COMMDB component:

// cd \commdb\group

// bldmake bldfiles

// abld test build t_db

// 2. Build the test utilities:

// cd \msg\testutils\group\

// bldmake bldfiles

// abld build

// WINS running instructions:

// 1. \epoc32\release\wins\<VARIANT>\T_SMTSND.exe can be used at the command prompt

// or executable can be run from Windows Explorer.

// All other platform running instructions:

// 1. Copy \epoc32\release\<PLATFORM>\<VARIANT>\T_SMTSND.exe onto the other platform

// 2. Copy \epoc32\release\<PLATFORM>\<VARIANT>\MSVTESTUTILS.DLL into 

// <DRIVE>:\system\libs on the other platform

// 3. Copy \epoc32\release\<PLATFORM>\<VARIANT>\EMAILTESTUTILS.DLL into 

// <DRIVE>:\system\libs on the other platform 

// 4. Run T_SMTSND.exe on the other platform

// 

//

#include "emailtestutils.h"

#include <e32uid.h>

#include <txtrich.h>

#include <msvids.h>

#include "smtcmtm.h"	// CSmtpClientMtm

#include "smtpset.h"	// CImSmtpSettings

#include <miutmsg.h>

#include "miutlog.h"	// CImLog

#include <mtmuids.h>

#include <mtmdef.h>

#include <cemailaccounts.h>

	#include <mmsvattachmentmanager.h>

	#include <sendas2.h>

RTest	test(_L("SMTP SendAs Testrig"));

LOCAL_D CTrapCleanup* theCleanup;

LOCAL_D CEmailTestUtils* testUtils;

LOCAL_D CImLog* log;

#define KRecipient _L("test@psion9.demon.co.uk")

#define KRecipientAlias _L("Test Account")

//

//

LOCAL_C void InitL()

	{

	CActiveScheduler* scheduler = new (ELeave) CActiveScheduler;

	CActiveScheduler::Install(scheduler);

	CleanupStack::PushL(scheduler);

	testUtils = CEmailTestUtils::NewLC(test);

	testUtils->CreateAllTestDirectories();

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

	testUtils->CleanMessageFolderL();

 	testUtils->ClearEmailAccountsL();

	testUtils->GoServerSideL();

	test.Next(_L("Create Data Component FileStores"));

	testUtils->CreateSmtpServerMtmRegL();

	//	Loading the DLLs

	testUtils->CreateRegistryObjectAndControlL();

	testUtils->InstallSmtpMtmGroupL();

	log = CImLog::NewL(_L("c:\\logs\\email\\T_IMCM.log"), EAppend);

	CleanupStack::PushL(log);

	log->AppendComment(_L8("******** T_SMTSND Test (SMTP) Send As API ********"));

	TBuf8<80> buf;

#if defined(__WINS__)

	buf.Append(_L8("WINS "));

#else

	buf.Append(_L8("MARM "));

#endif

#if defined(_UNICODE)

	buf.Append(_L8("U"));

#endif

#if defined(_DEBUG)

	buf.Append(_L8("DEB"));

#else

	buf.Append(_L8("REL"));

#endif

	log->AppendComment(buf);

	}

LOCAL_C void Closedown()

	{

	log->AppendComment(_L8("**********    T_SMTSND Tests Complete    **********"));

	log->AppendComment(_L8(""));

	CleanupStack::PopAndDestroy(3);  //testUtils, log, scheduler

	}

//

LOCAL_C void TestSMTSendL(CMsvEntry& aEntry, TMsvId aDestination)

	{

	// Get count of children

	aEntry.SetEntryL(KMsvRootIndexEntryId);

	aEntry.SetEntryL(aDestination);

	TInt children = aEntry.Count();

	CTestTimer* timer = CTestTimer::NewL();

	timer->After(0x200000);

	CActiveScheduler::Start();

	delete timer;

	// rich text

	CParaFormatLayer* paraFormatLayer=CParaFormatLayer::NewL();

	CleanupStack::PushL(paraFormatLayer);

	CCharFormatLayer* charFormatLayer=CCharFormatLayer::NewL(); 

	CleanupStack::PushL(charFormatLayer);

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

	CleanupStack::PushL(bodyText);

	TBuf<100> bodyContents = _L("Test body contents so that we can test the SendAs API.  We \06 can try using a delimiter");

	bodyText->InsertL(0, bodyContents);

	RSendAs sa;

	TInt err = sa.Connect();

	User::LeaveIfError(err);

	CleanupClosePushL(sa);

	RSendAsMessage sa_msg;

	sa_msg.CreateL(sa, KUidMsgTypeSMTP);

	CleanupClosePushL(sa_msg);

	sa_msg.AddRecipientL(KRecipient, RSendAsMessage::ESendAsRecipientTo);

	sa_msg.AddRecipientL(KRecipient, KRecipientAlias, RSendAsMessage::ESendAsRecipientTo);

	sa_msg.SetSubjectL(_L("t_smtsnd message test 1"));

	sa_msg.SetBodyTextL(*bodyText);

	sa_msg.SaveMessageAndCloseL();

	CleanupStack::Pop(); // sa_msg

	CleanupStack::PopAndDestroy(); // sa

	// Check child count has increased

	aEntry.SetEntryL(KMsvRootIndexEntryId);

	aEntry.SetEntryL(aDestination);

	test(aEntry.Count() == children + 1);

	// Check the body text

	// ok, need to find the message entry

	TMsvId message = 0x00000000;

	TMsvEntry msgEntry;

	TBool found = EFalse;

	CMsvEntrySelection* sel = aEntry.ChildrenL();

	CleanupStack::PushL(sel);

	// Loop through the sub-entries

	for(TInt i=0;i<sel->Count();i++)

	{

		// Reset the entry to the child[i]

		aEntry.SetEntryL(sel->At(i));

		msgEntry = aEntry.Entry();

		if (msgEntry.iDescription.Compare(_L("t_smtsnd message test 1")) == 0)

		{

			message = aEntry.EntryId();

			found = ETrue;

			break;

		}

	}

	CleanupStack::PopAndDestroy(); // sel

	// make sure we actually found our message

	if (!found)

		User::Leave(KErrNotFound);

	// Check we've got what we wanted

	aEntry.SetEntryL(message);

	test(aEntry.Entry().Parent() == aDestination);

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

	CleanupStack::PushL(validRichText1);

	CMsvOperationWait* iWait=CMsvOperationWait::NewLC();

	CImEmailMessage* imEmailMessage = CImEmailMessage::NewLC(*testUtils->iMsvEntry);

	imEmailMessage->GetBodyTextL(iWait->iStatus, msgEntry.Id(), CImEmailMessage::EThisMessageOnly, *validRichText1, *paraFormatLayer, *charFormatLayer);

	iWait->Start();

	CActiveScheduler::Start();	// wait for the asynch operation to complete

	TBuf<100> bufa = bodyText->Read(0);

	TBuf<100> bufb = validRichText1->Read(0);

	test((bufa.Compare(bufb))==0);

	aEntry.SetEntryL(msgEntry.Id());

	CMsvStore* store = aEntry.ReadStoreL();

	CleanupStack::PushL(store);

	CImHeader* header = CImHeader::NewLC();

	header->RestoreL(*store);

	// check that there are 2 recipients

	test(header->ToRecipients().Count() == 2);

	// check that the alias in the second recipient is not lost!

	TBuf<50> recipient;

	recipient.Append('\"');

	recipient.Append(KRecipientAlias);

	recipient.Append('\"');

	recipient.Append(' ');

	recipient.Append('<');

	recipient.Append(KRecipient);

	recipient.Append('>');

	TPtrC toRecipient = header->ToRecipients()[1];

	test(toRecipient.CompareF(recipient)==0);

	CleanupStack::PopAndDestroy(header);

	CleanupStack::PopAndDestroy(store);

	CleanupStack::PopAndDestroy(2); //imEmailMessage, iWait

	CleanupStack::PopAndDestroy(4);	//bodyText, validRichText1, paraFormatLayer, charFormatLayer

	}

LOCAL_C void TestSMTSendL()

	{

	TMsvDummyObserver* ob = new(ELeave) TMsvDummyObserver;

	CleanupStack::PushL(ob);

	CMsvSession* session = CMsvSession::OpenSyncL(*ob);

	CleanupStack::PushL(session);

	CMsvEntry* cEntry = session->GetEntryL(KMsvRootIndexEntryId);

	CleanupStack::PushL(cEntry);

	// with the new SendAs2 API messages are always created in Drafts

	TestSMTSendL(*cEntry, KMsvDraftEntryId);

	CleanupStack::PopAndDestroy(3); // cEntry, session, ob

	}

LOCAL_C void TestCapabilityL()

	{

	RSendAs sa;

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

	CleanupClosePushL(sa);

	sa.FilterAgainstCapability(KUidMtmQuerySupportedBody);

	sa.FilterAgainstCapability(KUidMtmQuerySupportAttachments);

	sa.FilterAgainstCapability(KUidMtmQuerySupportSubject);

	sa.FilterAgainstCapability(KUidMtmQueryCanSendMsg);

	CSendAsMessageTypes* msgTypes = CSendAsMessageTypes::NewL();

	CleanupStack::PushL(msgTypes);

	sa.FilteredMessageTypesL(*msgTypes);

	// make sure SMTP is still in the list

	TInt found = 0;

	TInt count = msgTypes->Count();

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

	{

		if (msgTypes->MessageTypeUid(i) == KUidMsgTypeSMTP)

		{

			found++;

			break;

		}

	}

	test(found > 0);

	// now set a capability not supported by SMTP

	msgTypes->Reset();

	sa.FilterAgainstCapability(KUidMtmQuerySupportsFolder);

	sa.FilteredMessageTypesL(*msgTypes);

	// make sure SMTP is no longer in the list

	found = EFalse;

	count = msgTypes->Count();

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

	{

		if (msgTypes->MessageTypeUid(i) == KUidMsgTypeSMTP)

		{

			found = ETrue;

			break;

		}

	}

	test(found == 0);

	CleanupStack::PopAndDestroy(2); // sa, msgTypes

	}

LOCAL_C void TestValidationL()

	{

// Looks line validation functionality is not part of the new SendAs2 API, so we skip this test	

	}

LOCAL_C void TestAttachmentL(CMsvEntry& aEntry, TMsvId aDestination)

	{

	// Get count of children

	aEntry.SetEntryL(KMsvRootIndexEntryId);

	aEntry.SetEntryL(aDestination);

	TInt childCount = aEntry.Count();

	CTestTimer* timer = CTestTimer::NewL();

	timer->After(0x200000);

	CActiveScheduler::Start();

	delete timer;

	// rich text

	CParaFormatLayer* paraFormatLayer=CParaFormatLayer::NewL();

	CleanupStack::PushL(paraFormatLayer);

	CCharFormatLayer* charFormatLayer=CCharFormatLayer::NewL(); 

	CleanupStack::PushL(charFormatLayer);

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

	CleanupStack::PushL(bodyText);

	TBuf<100> bodyContents = _L("Test body contents so that we can test the SendAs API.  We \06 can try using a delimiter");

	bodyText->InsertL(0, bodyContents);

	RSendAs sa;

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

	CleanupClosePushL(sa);

	RSendAsMessage sa_msg;

	sa_msg.CreateL(sa, KUidMsgTypeSMTP);

	CleanupClosePushL(sa_msg);

	sa_msg.AddRecipientL(KRecipient, RSendAsMessage::ESendAsRecipientTo);

	sa_msg.SetSubjectL(_L("t_smtsnd message test 2"));

	sa_msg.SetBodyTextL(*bodyText);

	TRequestStatus status;

	sa_msg.AddAttachment(_L("C:\\mailtest\\imcm\\rfc82201.txt"), status);

	User::WaitForRequest(status);

	sa_msg.SaveMessageAndCloseL();

	TMsvId message = 0x00000000;

	TMsvEntry msgEntry;

	TBool found = EFalse;

	aEntry.SetEntryL(KMsvRootIndexEntryId);

	aEntry.SetEntryL(aDestination);

	CMsvEntrySelection* sel = aEntry.ChildrenL();

	CleanupStack::PushL(sel);

	// Loop through the sub-entries

	for(TInt i=0;i<sel->Count();i++)

	{

		// Reset the entry to the child[i]

		aEntry.SetEntryL(sel->At(i));

		msgEntry = aEntry.Entry();

		if (msgEntry.iDescription.Compare(_L("t_smtsnd message test 2")) == 0)

		{

			message = aEntry.EntryId();

			found = ETrue;

			break;

		}

	}

	CleanupStack::PopAndDestroy(); // sel

	// make sure we actually found our message

	if (!found)

		User::Leave(KErrNotFound);

	// Check child count has increased

	aEntry.SetEntryL(KMsvRootIndexEntryId);

	aEntry.SetEntryL(aDestination);

	test(aEntry.Count() == childCount + 1);

	// Check we've got what we wanted

	aEntry.SetEntryL(message);

	test(aEntry.Entry().Parent() == aDestination);

	testUtils->iMsvEntry->SetEntryL(msgEntry.Id());

	CImEmailMessage* imEmailMessage = CImEmailMessage::NewLC(*testUtils->iMsvEntry);

	// Testing the message has one attachment using attachment manager

	TInt count = imEmailMessage->AttachmentManager().AttachmentCount();

	test(count == 1);

	// open the file for read, using index

	RFile file = imEmailMessage->AttachmentManager().GetAttachmentFileL(count-1);

	CleanupClosePushL(file);

	TInt size = 0;

	test(KErrNone == file.Size(size));

	HBufC8* buf8 = HBufC8::NewLC(size);

	TPtr8 des = buf8->Des();

	test(KErrNone == file.Read(des));

	CleanupStack::PopAndDestroy(2,&file); // buf8, file 

	// open the file for read, using attachment id

	// with the new API we don't know the attachment ID, so we skip this test for SendAs2

	_LIT8(KTestData,"Some test data");

	// open the file for write, using index

	file = imEmailMessage->AttachmentManager().GetAttachmentFileForWriteL(count-1);

	CleanupClosePushL(file);

	TInt filePosition = size;

	test(KErrNone == file.Write(filePosition,KTestData));

	file.Flush();

	// get the updated file size

	test(KErrNone == file.Size(size));

	CleanupStack::PopAndDestroy(&file); 

	// get the attachment info, using index

	CMsvAttachment* attachmentInfo = imEmailMessage->AttachmentManager().GetAttachmentInfoL(count-1);

	CleanupStack::PushL(attachmentInfo);

	// modify size information in attachment info

	attachmentInfo->SetSize(size);

	// Test modify the attachment info

	CTestActive* testActive = new (ELeave) CTestActive();

	CleanupStack::Pop(attachmentInfo); // pass ownership to attachment manager

	CleanupStack::PushL(testActive);

	imEmailMessage->AttachmentManager().ModifyAttachmentInfoL(attachmentInfo,testActive->iStatus);	

	testActive->StartL();

	CActiveScheduler::Start();	// wait for the asynch operation to complete

	TInt result = testActive->iStatus.Int(); 

	CleanupStack::PopAndDestroy(2,imEmailMessage); // testActive , imEmailMessage

	testUtils->iMsvEntry->SetEntryL(msgEntry.Id());

	// Check that the structure of message is:

	//	MESSAGE

	//		FOLDER_MIX

	//			TEXT

	//			ATTACHMENT

	CMsvEntrySelection* children = testUtils->iMsvEntry->ChildrenL();

	CleanupStack::PushL(children);

	test(children->Count() == 1);

	TMsvId serviceId;

	TMsvEmailEntry entry;

	testUtils->iMsvEntry->Session().GetEntry(children->At(0), serviceId, entry);

	test(entry.MessageFolderType() == EFolderTypeMixed);

	testUtils->SetEntryL(children->At(0));

	CleanupStack::PopAndDestroy(); //children

	children = testUtils->iMsvEntry->ChildrenL();

	CleanupStack::PushL(children);

	test(children->Count() == 2);

	testUtils->iMsvEntry->Session().GetEntry(children->At(0), serviceId, entry);

	test(entry.iType == KUidMsvAttachmentEntry);

	testUtils->iMsvEntry->Session().GetEntry(children->At(1), serviceId, entry);

	test(entry.iType == KUidMsvEmailTextEntry);

	CleanupStack::PopAndDestroy(children); //children

	CleanupStack::PopAndDestroy(2);	//sendAs, bodyText

	CleanupStack::PopAndDestroy(3); // ob, paraFormatLayer, charFormatLayer

	}

LOCAL_C void TestAttachmentL()

	{

	TMsvDummyObserver* ob = new(ELeave) TMsvDummyObserver;

	CleanupStack::PushL(ob);

	CMsvSession* session = CMsvSession::OpenSyncL(*ob);