diff -r 08008ce8a6df -r bde600d88860 messagingfw/sendas/test/unit/inc/sendas2test.h
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/messagingfw/sendas/test/unit/inc/sendas2test.h	Fri Jun 04 10:32:16 2010 +0100
@@ -0,0 +1,473 @@
+// 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