#include <e32test.h>

#include <f32file.h>

#include <s32file.h>

#include <cntdb.h>

#include <cntitem.h>

#include <cntfield.h>

#include <cntfldst.h>

#include <bautils.h>

#include <barsc.h>

#include <barsread.h>

#include <cntmodel.rsg>

#include <hal.h>

#pragma warning(disable:4239) // group label warning

CCntTest* CntTest=NULL;

LOCAL_D RTest test(_L("T_GROUPS"));

TInt CCntClient::iObservers = 0;

const TPtrC KDatabaseFileName=_L("C:T_GROUP");

const TPtrC KTestName=_L("Test Name No %d");

const TPtrC KTestAddress=_L("Test Address No %d");

const TPtrC KTestFamilyName=_L("Test Family Name No %d");

const TPtrC KLongLabel = _L("New Long Label");

const TPtrC KGroupLabel = _L("Group Label No%d");

const TPtrC KTestEmail = _L("Test@Email.com No %d");

const TPtrC KTestCountry = _L("Test Country No %d");

const TPtrC KTestCompany = _L("Test Company No %d");

const TInt KLargeSizeRecords=50;

const TInt KFamilySizeRecords=20;

#define KCardGivenName _L("Given Name #%d")

#define KCardFamilyName _L("Family Name #%d")

#define KCardFieldText _L("Card id %d, text field #%d")

#define KCardFieldEditedText _L("UPDATED Card id %d, text field #%d")

//

//	SUPPORT MODULES 

//

GLDEF_C void Panic(TInt aPanic)

//

// Panic the thread with CNTMODEL as the category

//

	{

	User::Panic(_L("T_GROUP"),aPanic);

	}

LOCAL_C void SetNameL(CContactItem& aItem,TUid aType,const TDesC& aName)

//

// Set the contents of a text field, creating the field if required

//

	{

	CContactItemFieldSet& fieldSet=aItem.CardFields();

	const TInt pos=fieldSet.Find(aType);

	if (pos!=KErrNotFound)

		fieldSet[pos].TextStorage()->SetTextL(aName);

	else

		Panic(KErrGeneral);

	}

LOCAL_C void PopulateDatabaseL(TInt aNumberToPopulate, TBool aPhoneNumbers)

//

// Create and populate the database

//

	{

	TTime before;

	before.UniversalTime();

	CContactItem* templ = CntTest->Db()->ReadContactL(0);

	CleanupStack::PushL(templ);

	for (TInt ii=0;ii<aNumberToPopulate;ii++)

		{

		CContactItem* item=CContactCard::NewLC(templ);

		TBuf<16> name;

		name.Format(KTestName,ii);

   		SetNameL(*item,KUidContactFieldGivenName,name);

		if (aPhoneNumbers)

			{

			TBuf<20> number;

			switch(ii%3)

				{

				case 0:

					number.Format(_L("0171-%03d %04d"),(ii*9)%1000,((ii+11)*23)%10000);

					break;

				case 1:

					number.Format(_L("%04d:%04d:%04d:%04d"),(ii*123)%10000,(ii*666)%10000,(ii*234)%10000);

					break;

				case 2:

					number.Format(_L("+00%d-%03d %04d"),(ii*123)%100,(ii*13)%1000,((ii+13)*17)%10000);

					break;

				}

   			SetNameL(*item,KUidContactFieldPhoneNumber,number);

			if (!(ii%2))

				{

				number.Format(_L("0181-%03d %04d"),(ii*8)%1000,((ii+11)*22)%10000);

	   			SetNameL(*item,KUidContactFieldPhoneNumber,number);

				number.Format(_L("01734-%06d"),(ii*123456)%1000000);

	   			SetNameL(*item,KUidContactFieldPhoneNumber,number);

				}

			}

		TBuf<32> address;

		address.Format(KTestAddress,ii);

		SetNameL(*item,KUidContactFieldAddress,address);

//

		TBuf<32> familyName;

		familyName.Format(KTestFamilyName,ii);

		SetNameL(*item,KUidContactFieldFamilyName,familyName);

//

		TBuf<32> email;

		email.Format(KTestEmail,ii);

		SetNameL(*item,KUidContactFieldEMail,email);

//

		TBuf<32> country;

		country.Format(KTestCountry,ii);

		SetNameL(*item,KUidContactFieldCountry,country);

//	

		TBuf<32> company;

		company.Format(KTestCompany,ii);

		SetNameL(*item,KUidContactFieldCompanyName,company);

//	

		CntTest->Db()->AddNewContactL(*item); //templ

		CleanupStack::PopAndDestroy(); // item

//	

		if (ii%100==0)

			{

			test.Printf(_L("."));	// Just to show some life

			CntTest->Db()->CompactL();

			}

		}

	CleanupStack::PopAndDestroy();

	CntTest->Db()->SetDateFormatTextL(_L("%E%D%X%N%Y %1 %2 %3"));

	CntTest->Db()->CompactL();

	TTime after;

	after.UniversalTime();

	TTimeIntervalSeconds secondsTaken;

	after.SecondsFrom(before,secondsTaken);

	test.Printf(_L(" TIME: %d  Secs"),secondsTaken.Int());

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

	//test.Getch();

	}

//

//	TEST MODULES 

//

LOCAL_C void CreateBasicContactGroups()

	{

	test.Next(_L("Create Basic Groups"));

	__ASSERT_ALWAYS(CntTest->Db()->GroupCount()==0,Panic(KErrGeneral));

	CContactItem* newGroup = CntTest->Db()->CreateContactGroupL(_L("New Group"));

	CContactItem* newGroup2 = CntTest->Db()->CreateContactGroupL();

	CleanupStack::PushL(newGroup);

	CleanupStack::PushL(newGroup2);

	CntTest->CloseDatabase();

	CntTest->OpenDatabaseL();

	test(CntTest->Db()->GroupCount()==2);

	CleanupStack::PopAndDestroy(2); // newGroup newGroup2

	}

LOCAL_C void CreateGroupFromAClientL(CCntClient& aCntClient)

	{

	test.Next(_L("Create a basic group"));

	CContactItem* tempGroup = aCntClient.Db()->CreateContactGroupL(_L("New Group"));

	delete tempGroup ;

	CActiveScheduler::Start();

	}

LOCAL_C void DeleteBasicContactGroups()

	{

	test.Next(_L("Delete Basic Groups"));

	TInt groupCount = CntTest->Db()->GroupCount();

	__ASSERT_ALWAYS(groupCount==2,Panic(KErrGeneral));

	CContactIdArray* groupIdList = CntTest->Db()->GetGroupIdListL();

	CleanupStack::PushL(groupIdList);

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

//

	for (TInt ii=0; ii< groupIdList->Count();ii++)

		CntTest->Db()->DeleteContactL((*groupIdList)[ii]);

	test(CntTest->Db()->CountL()==0); //own card

	CContactIdArray* newGroupIdList = CntTest->Db()->GetGroupIdListL();

	test(newGroupIdList->Count()==0);

	delete newGroupIdList;

	CleanupStack::PopAndDestroy(); //groupIdList;

	}	

LOCAL_C void CheckBasicPopulatedGroup()

	{

	test.Next(_L("Populate Basic Group"));

	CntTest->CloseDatabase();

	CntTest->DeleteDatabaseL();

	CntTest->CreateDatabaseL();

	TRAP_IGNORE(PopulateDatabaseL(KFamilySizeRecords,ETrue));

	CContactItem* newGroup = CntTest->Db()->CreateContactGroupLC(_L("Family Group"));

	TContactItemId groupId = newGroup->Id();

	for(TInt ii=2;ii<12;ii++)

		{

		CntTest->Db()->AddContactToGroupL(ii,groupId);

		}

	CntTest->CloseDatabase();

	CntTest->OpenDatabaseL();

//

	CContactItem* familyGroup = CntTest->Db()->ReadContactLC(groupId);

	CContactIdArray* memberArray = STATIC_CAST(CContactGroup*,familyGroup)->ItemsContainedLC();

	test(memberArray->Count()==10);

//

	CContactItem* familyMember = CntTest->Db()->ReadContactLC(2);

	CContactIdArray* groups = STATIC_CAST(CContactCard*, familyMember)->GroupsJoinedLC();

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

	TContactItemId memberShip = (*groups)[0];

	test(memberShip==groupId);

	CleanupStack::PopAndDestroy(5); 

	// newGroup familyGroup memberArray familyMember groups

	}

LOCAL_C void CheckRemoveMember()

	{

	test.Next(_L("Remove Group Member a check Persist"));

//

	CContactItem* familyMember = CntTest->Db()->ReadContactLC(2);

	CContactIdArray* groups = STATIC_CAST(CContactCard*, familyMember)->GroupsJoinedLC();

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

	TContactItemId memberShip = (*groups)[0];

	CContactItem* group = CntTest->Db()->ReadContactLC(memberShip);

	CContactIdArray* memberArray = STATIC_CAST(CContactGroup*,group)->ItemsContainedLC();

	TInt memberCount1 = memberArray->Count();

//

	CntTest->Db()->RemoveContactFromGroupL(*familyMember, *group);

	CleanupStack::PopAndDestroy(); // group

//

	CContactItem* changedGroup = CntTest->Db()->ReadContactLC(memberShip);

	CContactIdArray* newMemberArray = STATIC_CAST(CContactGroup*,changedGroup)->ItemsContainedLC();

	test(newMemberArray->Count()==memberCount1-1);

	CleanupStack::PopAndDestroy(5); // familyMember groups changedGroup newMemberArray memberarray

	}

LOCAL_C void CheckRemoveGroup()

	{

	test.Next(_L("Remove Group check Persist"));

//

	CContactIdArray* groupIdList = CntTest->Db()->GetGroupIdListL();

	CleanupStack::PushL(groupIdList);

	CContactItem* group = CntTest->Db()->ReadContactLC((*groupIdList)[0]);

	CContactIdArray* memberArray = STATIC_CAST(CContactGroup*,group)->ItemsContainedLC();

//

	CContactItem* member = CntTest->Db()->ReadContactLC((*memberArray)[0]);

	CContactIdArray* memberShipArray = STATIC_CAST(CContactCard*, member)->GroupsJoinedLC();

	__ASSERT_ALWAYS(memberShipArray->Count()==1,Panic(KErrGeneral));

	test((*memberShipArray)[0]==(*groupIdList)[0]);

//

	CntTest->Db()->DeleteContactL((*groupIdList)[0]);

//

	CContactItem* newMember = CntTest->Db()->ReadContactLC((*memberArray)[0]);

	CContactIdArray* newMemberShipArray = STATIC_CAST(CContactCard*, newMember)->GroupsJoinedLC();

	test(newMemberShipArray->Count()==0);

//

	CleanupStack::PopAndDestroy(7); // groupidList group memberArray member memberShipArray newMember newMemberShipArray

	}

LOCAL_C void CheckLargePopulatedGroup()

	{

	test.Next(_L("Create MANY Contacts"));

	CntTest->CloseDatabase();

	CntTest->DeleteDatabaseL();

	CntTest->CreateDatabaseL();

	TRAP_IGNORE(PopulateDatabaseL(KLargeSizeRecords,ETrue));

	CContactItem* newGroup = CntTest->Db()->CreateContactGroupL(_L("Large Group"));

	CleanupStack::PushL(newGroup);

	TContactItemId groupId = newGroup->Id();

	test.Next(_L("Add MANY Contacts to a Group"));

	TTime before;

	before.UniversalTime();

	for(TInt ii=1;ii<KLargeSizeRecords+1;ii++)

		{

		CntTest->Db()->AddContactToGroupL(ii, groupId); //*tempContact,*newGroup);

		if (ii%100==0)

			test.Printf(_L("."));	// Just to show some life

		}

	test(TestGroupStateL(CntTest->Db(),1,KLargeSizeRecords));

	TTime after;

	after.UniversalTime();

	TTimeIntervalSeconds secondsTaken;

	after.SecondsFrom(before,secondsTaken);

	test.Printf(_L(" TIME: %d  Secs"),secondsTaken.Int());

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

	//test.Getch();

	CntTest->CloseDatabase();

	CntTest->OpenDatabaseL();

//

	CContactItem* familyGroup = CntTest->Db()->ReadContactL(groupId);

	CContactIdArray* memberArray = STATIC_CAST(CContactGroup*,familyGroup)->ItemsContainedLC();

	test(memberArray->Count()==KLargeSizeRecords);

	delete familyGroup;

//

	CContactItem* familyMember = CntTest->Db()->ReadContactL(2);

	CContactIdArray* groups = STATIC_CAST(CContactCard*, familyMember)->GroupsJoinedLC();

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

	TContactItemId memberShip = (*groups)[0];

	test(memberShip==groupId);

	delete familyMember;

	CleanupStack::PopAndDestroy(3); // newGroup groups memberArray

	}

LOCAL_C void DeleteGroupMembersCheckPersist()

	{

	test.Next(_L("Delete Group Members Check Persist"));

//

	TTime before;

	before.UniversalTime();

	for(TInt ii=1;ii<KLargeSizeRecords+1;ii++)

		{

		CntTest->Db()->DeleteContactL(ii);

		if (ii%100==0)

			test.Printf(_L("."));	// Just to show some life

		}

	TTime after;

	after.UniversalTime();

	TTimeIntervalSeconds secondsTaken;

	after.SecondsFrom(before,secondsTaken);

	test.Printf(_L(" TIME: %d  Secs"),secondsTaken.Int());

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

	//test.Getch();

//

	CContactIdArray* groups = CntTest->Db()->GetGroupIdListL();

	CleanupStack::PushL(groups);

	CContactItem* group = CntTest->Db()->ReadContactL((*groups)[0]);

	CContactIdArray* memberArray = STATIC_CAST(CContactGroup*,group)->ItemsContainedLC();

	test(memberArray->Count()==0);

	delete group;

	CleanupStack::PopAndDestroy(2); // groups memberArray

	}

LOCAL_C void CreateManyGroups()

	{

	test.Next(_L("Populate Database"));

	CntTest->CloseDatabase();

	CntTest->DeleteDatabaseL();

	CntTest->CreateDatabaseL();

	// add 5 new contacts

	TRAP_IGNORE(PopulateDatabaseL(5,ETrue));

//

	test.Next(_L("Create MANY Groups"));

	TTime before;

	before.UniversalTime();

	for(TInt ii=0;ii<KLargeSizeRecords;ii++)

		{

		TBuf<32> label;

		label.Format(KGroupLabel,ii);

		CContactItem* newGroup = CntTest->Db()->CreateContactGroupL(label);

		if (ii%100==0)

				test.Printf(_L("."));	// Just to show some life

		delete newGroup;

		}

	TTime after;

	after.UniversalTime();

	TTimeIntervalSeconds secondsTaken;

	after.SecondsFrom(before,secondsTaken);

	test.Printf(_L(" TIME: %d  Secs"),secondsTaken.Int());

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

	//test.Getch();

	test(CntTest->Db()->GroupCount()==KLargeSizeRecords);

//

	test.Next(_L("Add 5 Contacts to MANY Groups"));

	before.UniversalTime();

	for(TInt jj=1;jj<6;jj++)

	// add cards to all groups

		{

		CContactItem* contact = CntTest->Db()->ReadContactL(jj);

		CleanupStack::PushL(contact);

		for(TInt gg=6;gg<KLargeSizeRecords+6;gg++)	// assume knowledge of group ids..

			{

			CntTest->Db()->AddContactToGroupL(jj,gg);	//*contact,*group);

			if (gg%100==0)

				test.Printf(_L("."));	// Just to show some life

			}

		CleanupStack::PopAndDestroy(); // contact

		}

	after.UniversalTime();

	after.SecondsFrom(before,secondsTaken);

	test.Printf(_L(" TIME: %d  Secs"),secondsTaken.Int());

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

	test(TestGroupStateL(CntTest->Db(),KLargeSizeRecords,5*KLargeSizeRecords));

	//test.Getch();

//

	test.Next(_L("Check Contacts Group membership"));

	before.UniversalTime();

	for(TInt gg=1;gg<6;gg++)

	//	check cards membership

		{

		CContactItem* contact = CntTest->Db()->ReadContactLC(gg);

		CContactIdArray* groups = STATIC_CAST(CContactCard*, contact)->GroupsJoinedLC();

		test(groups->Count()==KLargeSizeRecords);

		CleanupStack::PopAndDestroy(2); // contact groups

		}

	after.UniversalTime();

	after.SecondsFrom(before,secondsTaken);

	test.Printf(_L(" TIME: %d  Secs"),secondsTaken.Int());

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

	//test.Getch();

	}

LOCAL_C void DeleteManyGroups()

	{

//

	test.Next(_L("Delete MANY Groups"));

	TTime before;

	before.UniversalTime();

	for(TInt ii=6;ii<KLargeSizeRecords+6;ii++)

		{

		CntTest->Db()->DeleteContactL(ii);

		if (ii%100==0)

				test.Printf(_L("."));	// Just to show some life

		}

	TTime after;

	after.UniversalTime();

	TTimeIntervalSeconds secondsTaken;

	after.SecondsFrom(before,secondsTaken);

	test.Printf(_L(" TIME: %d  Secs"),secondsTaken.Int());

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

	//test.Getch();

//

	test.Next(_L("Test Card Membership Update"));

	test(CntTest->Db()->CountL()==5); // 5 contacts + own card

	for(TInt jj=1;jj<6;jj++)

	// add cards to all groups

		{

		CContactItem* contact = CntTest->Db()->ReadContactLC(jj);

		CContactIdArray* groups = STATIC_CAST(CContactCard*, contact)->GroupsJoinedLC();

		test(groups->Count()==0);

		CleanupStack::PopAndDestroy(2); // contact groups

		}

	CntTest->Db()->CompactL();

	}

LOCAL_C void AddCardToCard()

	{

//

	test.Next(_L("Add Card to Card"));

	// assumes Prior Knowldege of contacts 2 & 3 (both are cards)

	CntTest->Db()->AddContactToGroupL(2,3);	

//

	}	

LOCAL_C void AddGroupToCard()

	{

//

	test.Next(_L("Group to Card"));

	CContactItem* newGroup = CntTest->Db()->CreateContactGroupL(_L("New Group"));

	TContactItemId tempId = newGroup->Id();

	delete newGroup;

	// assumes Prior Knowldege of contact 3 (is a card)

	CntTest->Db()->AddContactToGroupL(tempId,3);	

//

	}

LOCAL_C void AddBadCardToGroup()

	{

//

	test.Next(_L("Add unknown Card to group"));

	// where id 19 does not exist (assumes knowledge of KLargeSizeRecords+6 as a group)

	CntTest->Db()->AddContactToGroupL(KLargeSizeRecords+9,KLargeSizeRecords+6);	

//

	}

LOCAL_C void AddCardToBadGroup()

	{

//

	test.Next(_L("Add Card to Unknown group"));

	// where id 19 does not exist

	CntTest->Db()->AddContactToGroupL(3,KLargeSizeRecords+9);	

//

	}

LOCAL_C void AddCardToGroupTwice()

	{

//

	test.Next(_L("Add Card to group Twice"));

	// assumes Prior Knowldege of KLargeSizeRecords+6 is a group

	CntTest->Db()->AddContactToGroupL(3,KLargeSizeRecords+6);	

	CntTest->Db()->AddContactToGroupL(3,KLargeSizeRecords+6);	

//

	}

LOCAL_C void AddFieldToGroup()

	{