// Copyright (c) 2004-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:
// sendas2test.cpp
//
#ifndef __SENDAS2TEST_H__
#define __SENDAS2TEST_H__

#include <sendas2.h>
#include <csendasaccounts.h>
#include <csendasmessagetypes.h>
#include <e32base.h>
#include "msvtestutilsbase.h"
#include <mtclreg.h>
#include <msvids.h>
#include <msvuids.h>
#include <msvapi.h>
#include <e32uid.h>
#include <msvruids.h>
#ifdef SYMBIAN_ENABLE_SPLIT_HEADERS  
#include "msvconsts.h"
#endif

// to be defined by test harness
GLREF_D RTest test; 
GLREF_D void doTestsL();

// forward declarations
class CSendTestUtils;

// globals
TInt SchedulerError;
CSendTestUtils* theUtils = NULL;
CTrapCleanup* theCleanup = NULL;

const TInt KNumberOfTestMtms		=5; 
const TInt KNumberOfTestMtmsCanSend	=4;

_LIT(KMtmDataDir, "c:\\system\\mtm\\");
_LIT(KMtmDataFile1, "c:\\system\\mtm\\SENDAS2MTM1.dat");
_LIT(KMtmDataFile2, "c:\\system\\mtm\\SENDAS2MTM2.dat");
_LIT(KMtmDataFile3, "c:\\system\\mtm\\SENDAS2MTM3.dat");
_LIT(KMtmDataFile4, "c:\\system\\mtm\\SENDAS2MTM4.dat");
_LIT(KMtmDataFile5, "c:\\system\\mtm\\SENDAS2MTM5.dat");
_LIT(KMtmDataFile6, "c:\\system\\mtm\\SENDAS2MTM6.dat");

_LIT(KMtmName1, "test MTM 1");
_LIT(KMtmName2, "test MTM 2");
_LIT(KMtmName3, "test MTM 3 with longer name");
_LIT(KMtmName4, "test MTM 4");
_LIT(KMtmName5, "test MTM 5");
_LIT(KMtmName6, "SMS Test CLIENT");

const TUid KUidTestMtm1={0x100014A4};
const TUid KUidTestMtm2={0x100014A5};
const TUid KUidTestMtm3={0x100014A6};
const TUid KUidTestMtm4={0x100014A7};
const TUid KUidTestMtm5={0x100014A8};
const TUid KUidTestMtm6={0x1000102C};

_LIT(KTestMtmFileName1, "sendas2mtm1.dll");
_LIT(KTestMtmFileName2, "sendas2mtm2.dll");
_LIT(KTestMtmFileName3, "sendas2mtm3.dll");
_LIT(KTestMtmFileName4, "sendas2mtm4.dll");
_LIT(KTestMtmFileName5, "sendas2mtm5.dll");
_LIT(KTestMtmFileName6, "smcm.dll");

_LIT(KMtmRegistryFile, "c:\\private\\1000484b\\MTM Registry v2");

const TUid KUidRenderingUid={0x10001000};

TBuf<32> KMtm1Service1  = _L("a new service");
TBuf<64> KMtm1Service2  = _L("the last service (alphabetically)");
TBuf<32> KMtm1Service3  = _L("last created, but in the middle");
TInt theMtm1Service1Id,theMtm1Service2Id,theMtm1Service3Id;

TBuf<32> KMtm3Service1  = _L("zzzzzzzz");
TBuf<32> KMtm3Service2  = _L("aaaaaaaaahhhhhhhhh");
TInt theMtm3Service1Id,theMtm3Service2Id;

TBuf<32> KMtm4Service1  = _L("lalala");
TBuf<32> KMtm6Service1  = _L("SMS Client");
TInt theMtm4Service1Id;

// Attachment names
_LIT(KErrAttachmentName1, "attachment1.txt");
_LIT(KErrAttachmentName2, "attachment2.txt");
_LIT(KAttachmentName, "Attach.txt");


TInt theErr;
#define REPORT_IF_ERROR(a) {theErr=a; if (theErr) test(0);}

//**********************************
// CTestScheduler
//**********************************

class CTestScheduler : public CActiveScheduler
	{
public:
	void Error (TInt aError) const;
	};

void CTestScheduler::Error( TInt anError ) const
	{
	SchedulerError = anError;
	}

//**********************************
// CDummyObserver
//**********************************

class CDummyObserver : public CBase, public MMsvSessionObserver
	{
public:
	void HandleSessionEventL(TMsvSessionEvent, TAny*, TAny*, TAny*) {};
	};

//**********************************
// CSendTestUtils
//**********************************

class CSendTestUtils : public CTestUtils
	{
public:
	static CSendTestUtils* NewL(RTest& aTest);
protected:
	CSendTestUtils(RTest& aTest);
public:
	void Panic(TInt aPanic);
	void Start(const TDesC& aDes);
	void Complete();
	TUid FindMessageTypeL(RSendAs& aSendAs, const TDesC& aName);
	void SetupMtmsAndServicesL();
	void CreateDataFilesL(const TDesC& aFilename, const TDesC& aMtmName, TUid aUid, const TDesC& aServerMtmFileName, const TDesC& aClientMtmFileName);
	void DeleteServicesL(CMsvEntry& aEntry, TUid aUid);
	TMsvId CreateServiceL(CMsvEntry& aEntry, const TMsvEntry& aService);
	void RemoveMailFolderL(TInt aDrive);
private:
	TInt iTestNum;
	};

CSendTestUtils* CSendTestUtils::NewL(RTest& aTest)
	{
	CSendTestUtils* self = new(ELeave)CSendTestUtils(aTest);
	CleanupStack::PushL(self);
	self->ConstructL();
	CleanupStack::Pop(self);
	return self;
	}

CSendTestUtils::CSendTestUtils(RTest& aTest) :
	CTestUtils(aTest)
	{
	}

void CSendTestUtils::Panic(TInt)
	{
	test(EFalse);
	}

void CSendTestUtils::Start(const TDesC& aDes)
	{
	if (iTestNum == 0)
		{
		test.Start(aDes);
		}
	else
		{
		test.Next(aDes);
		}

	TestStart(++iTestNum, aDes);
	}

void CSendTestUtils::Complete()
	{
	TestFinish(iTestNum, KErrNone);
	}

TUid CSendTestUtils::FindMessageTypeL(RSendAs& aSendAs, const TDesC& aName)
	{
	// get list of available mtms
	CSendAsMessageTypes* messageTypes = CSendAsMessageTypes::NewL();
	CleanupStack::PushL(messageTypes);
	aSendAs.FilteredMessageTypesL(*messageTypes);

	// get count of mtms
	TUid mtmUid(KNullUid);
	TInt count = messageTypes->Count();
	while (count--)
		{
		if (messageTypes->AvailableMessageTypes().MdcaPoint(count) == aName)
			{
			mtmUid = messageTypes->MessageTypeUid(count);
			break;
			}
		}
	test(mtmUid != KNullUid);
	return mtmUid;
	}

void CSendTestUtils::SetupMtmsAndServicesL()
	{
	FileSession().Delete(KMtmDataFile1);
	FileSession().Delete(KMtmDataFile2);
	FileSession().Delete(KMtmDataFile3);
	FileSession().Delete(KMtmDataFile4);
	FileSession().Delete(KMtmDataFile5);
	FileSession().Delete(KMtmRegistryFile);
	FileSession().Delete(KMtmDataFile6);
	
	CDummyObserver* ob = new(ELeave) CDummyObserver;
	CleanupStack::PushL(ob);
	CMsvSession* session = CMsvSession::OpenSyncL(*ob);
	CleanupStack::PushL(session);

	// MTMs
	CClientMtmRegistry* reg = CClientMtmRegistry::NewL(*session);
	TInt initialMtmCount = reg->NumRegisteredMtmDlls();
	delete reg; 
	reg=NULL;

	// Create an mtm directory if not already exists.
	TBool isOpen(EFalse);
	if (FileSession().IsFileOpen(KMtmDataDir, isOpen) != KErrNone)
		{
		TInt err = FileSession().MkDirAll(KMtmDataDir);
		if (err != KErrAlreadyExists)
			{
			User::Leave(err);
			}
		}

	TInt installedTestMtms = 0;
	CreateDataFilesL(KMtmDataFile1, KMtmName1, KUidTestMtm1, KTestMtmFileName1, KTestMtmFileName1); 
	TInt err = session->InstallMtmGroup(KMtmDataFile1);
	if( err != KErrAlreadyExists )
		{
		++installedTestMtms;
		}
	CreateDataFilesL(KMtmDataFile2, KMtmName2, KUidTestMtm2, KTestMtmFileName2, KTestMtmFileName2); 
	err = session->InstallMtmGroup(KMtmDataFile2);
	if( err != KErrAlreadyExists )
		{
		++installedTestMtms;
		}
	CreateDataFilesL(KMtmDataFile3, KMtmName3, KUidTestMtm3, KTestMtmFileName3, KTestMtmFileName3); 
	err = session->InstallMtmGroup(KMtmDataFile3);
	if( err != KErrAlreadyExists )
		{
		++installedTestMtms;
		}
	CreateDataFilesL(KMtmDataFile4, KMtmName4, KUidTestMtm4, KTestMtmFileName4, KTestMtmFileName4); 
	err = session->InstallMtmGroup(KMtmDataFile4);
	if( err != KErrAlreadyExists )
		{
		++installedTestMtms;
		}
	CreateDataFilesL(KMtmDataFile5, KMtmName5, KUidTestMtm5, KTestMtmFileName5, KTestMtmFileName5); 
	err = session->InstallMtmGroup(KMtmDataFile5);
	if( err != KErrAlreadyExists )
		{
		++installedTestMtms;
		}
	CreateDataFilesL(KMtmDataFile6, KMtmName6, KUidTestMtm6, KTestMtmFileName6, KTestMtmFileName6); 
	err = session->InstallMtmGroup(KMtmDataFile6);
	if( err != KErrAlreadyExists )
		{
		++installedTestMtms;
		}

	//
	reg = CClientMtmRegistry::NewL(*session);
	test(reg->NumRegisteredMtmDlls()==initialMtmCount+installedTestMtms);
	delete reg; reg=NULL;
	test.Printf(TRefByValue<const TDesC>(_L("created %d test mtms\n")),installedTestMtms);
	test.Printf(TRefByValue<const TDesC>(_L("-> %d test mtms already installed\n")),KNumberOfTestMtms-installedTestMtms);
	
	//
	// services
	CMsvEntry* entry = session->GetEntryL(KMsvRootIndexEntryId);
	CleanupStack::PushL(entry);
	DeleteServicesL(*entry, KUidTestMtm1);
	DeleteServicesL(*entry, KUidTestMtm3);
	DeleteServicesL(*entry, KUidTestMtm4);
	DeleteServicesL(*entry, KUidTestMtm5);
	DeleteServicesL(*entry, KUidTestMtm6);
	//
	TMsvEntry service;
	service.iType = KUidMsvServiceEntry;
	service.iMtm = KUidTestMtm1;
	service.iDetails.Set(KMtm1Service1);
	theMtm1Service1Id = CreateServiceL(*entry, service);
	//
	service.iDetails.Set(KMtm1Service2);
	theMtm1Service2Id = CreateServiceL(*entry, service);
	//
	service.iDetails.Set(KMtm1Service3);
	theMtm1Service3Id = CreateServiceL(*entry, service);
	//
	service.iMtm = KUidTestMtm3;
	service.iDetails.Set(KMtm3Service1);
	theMtm3Service1Id = CreateServiceL(*entry, service);
	//
	service.iDetails.Set(KMtm3Service2);
	theMtm3Service2Id = CreateServiceL(*entry, service);
	//
	service.iMtm = KUidTestMtm4;
	service.iDetails.Set(KMtm4Service1);
	theMtm4Service1Id = CreateServiceL(*entry, service);

	service.iMtm = KUidTestMtm6;
	service.iDetails.Set(KMtm6Service1);
	theMtm4Service1Id = CreateServiceL(*entry, service);

	CleanupStack::PopAndDestroy(3); // ob, session, entry
	}


void CSendTestUtils::CreateDataFilesL(const TDesC& aFilename, const TDesC& aMtmName, TUid aUid, const TDesC& aServerMtmFileName, const TDesC& aClientMtmFileName)
	{
	CMtmDllInfoArray* mtmdllinfoarray=new(ELeave) CMtmDllInfoArray();
	CleanupStack::PushL(mtmdllinfoarray);

	CMtmDllInfo* mtmdllinfo1=CMtmDllInfo::NewL(aMtmName, TUidType(KDynamicLibraryUid, KUidMtmServerComponent, TUid::Uid(KUidMtmDefaultSpecificVal)),aServerMtmFileName,1,TVersion(2,0,0));
	mtmdllinfoarray->AddMtmDllInfoL(mtmdllinfo1);

	CMtmDllInfo* mtmdllinfo2=CMtmDllInfo::NewL(aMtmName, TUidType(KDynamicLibraryUid, KUidMtmClientComponent, TUid::Uid(KUidMtmDefaultSpecificVal)),aClientMtmFileName,1,TVersion(2,0,0));
	mtmdllinfoarray->AddMtmDllInfoL(mtmdllinfo2);

	TCapabilitySet capSet;
	capSet.SetEmpty();
	CleanupStack::Pop(mtmdllinfoarray); // next line takes ownership
	CMtmGroupData* mtmgroupdata=CMtmGroupData::NewL(aUid,aUid, mtmdllinfoarray, capSet);

	CleanupStack::PushL(mtmgroupdata);

	CFileStore* filestore = CPermanentFileStore::ReplaceLC(theUtils->FileSession(), aFilename, EFileShareExclusive|EFileStream|EFileWrite);
	TUidType uidtype(KPermanentFileStoreLayoutUid,KUidMsvDataComponent, aUid);
	filestore->SetTypeL(uidtype);
	RStoreWriteStream out;
	TStreamId streamid=out.CreateLC(*filestore);							// Push to stack
	mtmgroupdata->ExternalizeL(out);
	out.CommitL();
	CleanupStack::PopAndDestroy(); // out
	filestore->SetRootL(streamid);
	filestore->CommitL();
	CleanupStack::PopAndDestroy(); // filestore
	
	CleanupStack::PopAndDestroy();  // mtmgroupdata
	}


void CSendTestUtils::DeleteServicesL(CMsvEntry& aEntry, TUid aUid)
	{
	CMsvEntrySelection* selection = aEntry.ChildrenWithMtmL(aUid);
	if (selection->Count())
		{
		CMsvOperationWait* wait = CMsvOperationWait::NewLC();
		wait->Start();
		CMsvOperation* op = aEntry.DeleteL(*selection, wait->iStatus);
		CActiveScheduler::Start();
		test(wait->iStatus.Int()==KErrNone||wait->iStatus.Int()==KErrNotFound);
		delete op;
		CleanupStack::PopAndDestroy(wait);
		}
	delete selection;
	}


TMsvId CSendTestUtils::CreateServiceL(CMsvEntry& aEntry, const TMsvEntry& aService)
	{
	CMsvOperationWait* wait = CMsvOperationWait::NewLC();
	wait->Start();
	CMsvOperation* op = aEntry.CreateL(aService, wait->iStatus);
	CActiveScheduler::Start();
	REPORT_IF_ERROR(wait->iStatus.Int());
	TMsvLocalOperationProgress localProg(McliUtils::GetLocalProgressL(*op));
	TMsvId id = localProg.iId;
	delete op;
	CleanupStack::PopAndDestroy(); // wait
	return id;
	}

void CSendTestUtils::RemoveMailFolderL(TInt aDrive)
	{
	CFileMan* fileMan = CFileMan::NewL(theUtils->FileSession());
	CleanupStack::PushL(fileMan);

	TDriveUnit drive(aDrive);
	TPtrC ptr(drive.Name());

	TParse parse;
	test(parse.Set(ptr, &KMsvDefaultFolder(), NULL) == KErrNone);

	TInt error = fileMan->RmDir(parse.DriveAndPath());
	test(error == KErrNone || error == KErrPathNotFound);

	CleanupStack::PopAndDestroy();
	}

//**********************************
// Global
//**********************************

void SetupSchedulerL()
	{
	CActiveScheduler::Install(new(ELeave)CTestScheduler);
	}

void CloseScheduler(TInt aRet)
	{
	if (theUtils)
		{
		if (!aRet)
			{
			theUtils->TestHarnessCompleted();
			}
		else
			{
			theUtils->TestHarnessFailed(aRet);
			}
		}
	test(aRet == KErrNone);

	delete CActiveScheduler::Current();
	}

void doMainL()
	{
	SetupSchedulerL();

	// sendas2 test utils
	theUtils = CSendTestUtils::NewL(test);

	// TODO: Use test server to delete message folder and remove CENREP entries
	//theUtils->RemoveMailFolderL(EDriveC);

	// setup mtms
  	theUtils->SetupMtmsAndServicesL();
  	
  	// do the tests
	doTestsL();
	}

GLDEF_C TInt E32Main()
	{
	__UHEAP_MARK;
	theCleanup=CTrapCleanup::New();
	TRAPD(ret,doMainL());		
	test(ret == KErrNone);
	CloseScheduler(ret);
	delete theUtils;
	delete theCleanup;	
	test.Console()->SetPos(0, 13);
	test.End();
	test.Close();
	__UHEAP_MARKEND;
	return(KErrNone);
	}

#endif