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

//

#include "tbmtests.h"

#include <cauthentication.h>

#include "testutils.h"

_LIT ( KTxtFolder1, "Folder 1" );

_LIT ( KTxtFolder1_1, "Folder 1.1" );

_LIT ( KTxtFolder1_2, "Folder 1.2" );

_LIT ( KTxtFolder1_2_1, "Folder 1.2.1" );

_LIT ( KTxtFolder1_2_2, "Folder 1.2.2" );

_LIT ( KTxtFolder1_3, "Folder 1.3" );

_LIT ( KTxtFolder2, "Folder 2" );

_LIT ( KTxtFolder2_1, "Folder 2.1" );

_LIT ( KTxtFolder2_2, "Folder 2.2" );

_LIT ( KTxtFolder2_2_1, "Folder 2.2.1" );

_LIT ( KTxtFolder2_2_2, "Folder 2.2.2" );

_LIT ( KTxtFolder2_2_3, "Folder 2.2.3" );

_LIT ( KTxtFolder2_2_4, "Folder 2.2.4" );

_LIT ( KTxtFolder3, "Folder 3" );

_LIT ( KTxtFolder3_1, "Folder 3.1" );

_LIT ( KTxtFolder3_1_1, "Folder 3.1.1" );

_LIT ( KTxtFolder3_1_2, "Folder 3.1.2" );

_LIT ( KTxtFolder3_1_3, "Folder 3.1.3" );

_LIT ( KTxtFolder3_1_4, "Folder 3.1.4" );

_LIT ( KTxtFolder3_2, "Folder 3.2" );

_LIT8 ( KAuthUri, "sip:user:pass@location.com:1666;transport=tcp;x=123;df=223?asd=b" );

_LIT8 ( KAuthName2, "joe" );

_LIT8 ( KAuthPassword2, "pswd1234" );

_LIT8 (KIconData,		"****4644372131**7*&879");

_LIT8 (KIconData2,		"****abcdefghij**7*&xyz");

const TUid KUid1 = {0x111};

const TUid KUid2 = {0x222};

const TUid KUid3 = {0x333};

const TUid KUid4 = {0x444};

const TUid KUid5 = {0x555};

const TUid KUid6 = {0x666};

const TUid KUid7 = {0x777};

const TUid KUid8 = {0x888};

const TUid KUid9 = {0x999};

const TUid KUid10 = {0xAAA};

const TUid KUid11 = {0xBBB};

const TUid KUid12 = {0xCCC};

const TUid KUid13 = {0x1111};

const TUid KUid14 = {0x2222};

const TUid KUid15 = {0x3333};

const TUid KUid16 = {0x4444};

const TUid KUid17 = {0x5555};

const TUid KUid18 = {0x6666};

const TUid KUid19 = {0x7777};

const TUid KUid20 = {0x8888};

const TUid KUid21 = {0x9999};

const TUid KUid22 = {0xAAAA};

const TUid KUid23 = {0xBBBB};

const TUid KUid24 = {0xCCCC};

const TInt KMaxTestLength = 50;

TInt intValue1 = 1200;

TReal realValue1 = 67.456;

TInt intValue2 = 2000;

TReal realValue2 = 255.123;

TInt intValue3 = 3500;

TReal realValue3 = 289.567;

TInt intValue4 = 1200;

TReal realValue4 = 67.456;

TInt intValue5 = 2000;

TReal realValue5 = 255.123;

TInt intValue6 = 3500;

TReal realValue6 = 289.567;

_LIT8 ( KDbValue8, "Database 8-bit value1" );

_LIT ( KDbValue, "Database 16-bit value1" );

_LIT8 ( KFldrValue8, "Folder 8-bit value1" );

_LIT ( KFldrValue, "Folder 16-bit value1" );

_LIT8 ( KBkmrkValue8, "Bookmark 8-bit value1" );

_LIT ( KBkmrkValue, "Bookmark 16-bit value1" );

_LIT8 ( KDb2Value8, "Database 8-bit value2" );

_LIT ( KDb2Value, "Database 16-bit value2" );

_LIT8 ( KFldr2Value8, "Folder 8-bit value2" );

_LIT ( KFldr2Value, "Folder 16-bit value2" );

_LIT8 ( KBkmrk2Value8, "Bookmark 8-bit value2" );

_LIT ( KBkmrk2Value, "Bookmark 16-bit value2" );

CBmTests* CBmTests::NewL ( CTestWrapper& aTestWrapper )

	{

	CBmTests* self = new ( ELeave )CBmTests ( aTestWrapper );

	return self;

	}

CBmTests::CBmTests ( CTestWrapper& aTestWrapper )

: iTestWrapper ( aTestWrapper )

	{

	}

CBmTests::~CBmTests ()

	{

	CloseDb ();

	}

void CBmTests::OpenDbL ()

	{

	iBkDatabase.OpenL ();

	}

void CBmTests::CloseDb ()

	{

	iBkDatabase.Close ();

	}

RBkFolder CBmTests::CreateFolderLC ( const TDesC& aFolderName, RBkFolder* aParent )

	{

	RBkFolder folder;

	TRAPD ( err, folder = iBkDatabase.CreateFolderL ( aFolderName, aParent ) );

	User::LeaveIfError ( err );

	CleanupClosePushL ( folder );

	return folder;

	}

RBkFolder CBmTests::OpenFolderLC ( const TDesC& aFolderName )

	{

	RBkFolder folder = iBkDatabase.OpenFolderL ( aFolderName );

	CleanupClosePushL ( folder );

	return folder;

	}

void CBmTests::CommitDbL ()

	{

	iBkDatabase.CommitL ();

	}

void CBmTests::SetBkmrkDbPropertiesL ()

	{

	RBkBookmark home = iBkDatabase.CreateBookmarkL ();

	CleanupClosePushL ( home );

	home.SetTitleL ( _L ( "Home bookmark" ) );

	iBkDatabase.SetHome ( home );	

	iBkDatabase.SetHomePageTextL ( _L ( "My home page text" ) );

	iBkDatabase.SetSearchUriL ( _L8 ( "www.searchuri.com" ) );

	CleanupStack::PopAndDestroy ( &home ); // home

	}

void CBmTests::UpdateBkmrkDbPropertiesL ()

	{

	RBkBookmark home = iBkDatabase.CreateBookmarkL ();

	CleanupClosePushL ( home );

	home.SetTitleL ( _L ( "New home bookmark" ) );

	iBkDatabase.SetHome ( home );	

	iBkDatabase.SetHomePageTextL ( _L ( "My new home page text" ) );

	iBkDatabase.SetSearchUriL ( _L8 ( "www.newsearchuri.com" ) );

	CleanupStack::PopAndDestroy ( &home ); // home	

	}

void CBmTests::CreateTreeL ()

	{

	// Create bookmarks under root

	RBkBookmark bookmark1 = iBkDatabase.CreateBookmarkL ();

	CleanupClosePushL ( bookmark1 );

	RBkBookmark bookmark2 = iBkDatabase.CreateBookmarkL ();

	CleanupClosePushL ( bookmark2 );

	bookmark1.SetTitleL ( _L("Google") );

	bookmark2.SetTitleL ( _L("Hotmail") );

	bookmark1.SetUriL ( KAuthUri );

	CAuthentication* auth = CAuthentication::NewL(KAuthName2, KAuthPassword2, CAuthentication::EBasic);

	CleanupStack::PushL(auth);

	bookmark2.SetAuthenticationL(*auth);

	CleanupStack::PopAndDestroy ( auth );

	CleanupStack::PopAndDestroy ( 2 ); // bookmark1, bookmark2

	// Create folder1

	RBkFolder folder1 = iBkDatabase.CreateFolderL ( KTxtFolder1() );

	CleanupClosePushL ( folder1 );

	// Create sub-folders under folder1

	RBkFolder folder1_1 = iBkDatabase.CreateFolderL ( KTxtFolder1_1(), &folder1 );

	CleanupClosePushL ( folder1_1 );

	RBkFolder folder1_2 = iBkDatabase.CreateFolderL ( KTxtFolder1_2(), &folder1 );

	CleanupClosePushL ( folder1_2 );

	RBkFolder folder1_3 = iBkDatabase.CreateFolderL ( KTxtFolder1_3(), &folder1 );

	CleanupClosePushL ( folder1_3 );

	RBkFolder folder1_2_1 = iBkDatabase.CreateFolderL ( KTxtFolder1_2_1(), &folder1_2 );

	CleanupClosePushL ( folder1_2_1 );

	RBkFolder folder1_2_2 = iBkDatabase.CreateFolderL ( KTxtFolder1_2_2(), &folder1_2 );

	CleanupClosePushL ( folder1_2_2 );

	// Create bookmarks under folder1

	bookmark1 = iBkDatabase.CreateBookmarkL ( &folder1 );

	CleanupClosePushL ( bookmark1 );

	bookmark2 = iBkDatabase.CreateBookmarkL ( &folder1 );

	CleanupClosePushL ( bookmark2 );

	bookmark1.SetTitleL ( _L ( "ESPN Sports" ) );

	bookmark2.SetTitleL ( _L ( "Star Sports" ) );

	Bookmark::TAttachmentId iconId = iBkDatabase.CreateIconL ( KIconData );

	bookmark1.SetIconIdL ( iconId );

	CleanupStack::PopAndDestroy ( 2 ); // bookmark1, bookmark2

	CleanupStack::PopAndDestroy ( 6 ); 	// folder1, folder1_1, folder1_2, folder1_3,

										// folder1_2_1, folder1_2_2

	RBkFolder folder2 = iBkDatabase.CreateFolderL ( KTxtFolder2() );

	CleanupClosePushL ( folder2 );

	RBkFolder folder2_1 = iBkDatabase.CreateFolderL ( KTxtFolder2_1(), &folder2 );

	CleanupClosePushL ( folder2_1 );

	RBkFolder folder2_2 = iBkDatabase.CreateFolderL ( KTxtFolder2_2(), &folder2 );	

	CleanupClosePushL ( folder2_2 );		

	RBkFolder folder2_2_1 = iBkDatabase.CreateFolderL ( KTxtFolder2_2_1(), &folder2_2 );

	CleanupClosePushL ( folder2_2_1 );

	RBkFolder folder2_2_2 = iBkDatabase.CreateFolderL ( KTxtFolder2_2_2(), &folder2_2 );

	CleanupClosePushL ( folder2_2_2 );

	CleanupStack::PopAndDestroy ( 5 ); 	// folder2, folder2_1, folder2_2, 

										// folder2_2_1, folder2_2_2	

	RBkFolder folder3 = iBkDatabase.CreateFolderL ( KTxtFolder3() );

	CleanupClosePushL ( folder3 );

	Bookmark::TAttachmentId iconId2 = iBkDatabase.CreateIconL ( KIconData2 );

	folder3.SetIconIdL ( iconId2 );

	RBkFolder folder3_1 = iBkDatabase.CreateFolderL ( KTxtFolder3_1(), &folder3 );

	CleanupClosePushL ( folder3_1 );

	RBkFolder folder3_2 = iBkDatabase.CreateFolderL ( KTxtFolder3_2(), &folder3 );

	CleanupClosePushL ( folder3_2 );	

	RBkFolder folder3_1_1 = iBkDatabase.CreateFolderL ( KTxtFolder3_1_1(), &folder3_1 );

	CleanupClosePushL ( folder3_1_1 );

	RBkFolder folder3_1_4 = iBkDatabase.CreateFolderL ( KTxtFolder3_1_4(), &folder3_1 );

	CleanupClosePushL ( folder3_1_4 );

	CleanupStack::PopAndDestroy ( 5 ); // folder3, folder3_1, folder3_2, folder3_1_1, folder3_1_4

	}

void CBmTests::ModifyTreeL ()

	{

	RBkFolder folder2_2 = iBkDatabase.OpenFolderL ( KTxtFolder2_2() );

	CleanupClosePushL ( folder2_2 );

	RBkFolder folder2_2_3 = iBkDatabase.CreateFolderL ( KTxtFolder2_2_3() );

	CleanupClosePushL ( folder2_2_3 );

	RBkFolder folder2_2_4 = iBkDatabase.CreateFolderL ( KTxtFolder2_2_4() );

	CleanupClosePushL ( folder2_2_4 );

	folder2_2.AppendL ( folder2_2_3 );

	folder2_2.AppendL ( folder2_2_4 );

	CleanupStack::PopAndDestroy ( 3 ); // folder2_2, folder2_2_3, folder2_2_4

	RBkFolder folder3_1 = iBkDatabase.OpenFolderL ( KTxtFolder3_1() );

	CleanupClosePushL ( folder3_1 );

	RBkFolder folder3_1_2 = iBkDatabase.CreateFolderL ( KTxtFolder3_1_2() );

	CleanupClosePushL ( folder3_1_2 );

	RBkFolder folder3_1_3 = iBkDatabase.CreateFolderL ( KTxtFolder3_1_3() );

	CleanupClosePushL ( folder3_1_3 );

	folder3_1.InsertL ( folder3_1_2, 1 );

	folder3_1.InsertL ( folder3_1_3, 2 );

	CleanupStack::PopAndDestroy ( 3 ); // folder3_1, folder3_1_2, folder3_3

	}

void CBmTests::CreateCustomPropertiesL ()

	{

	iBkDatabase.RegisterDatabasePropertyL ( KUid1, Bookmark::EDataTypeInt );

	iBkDatabase.RegisterDatabasePropertyL ( KUid2, Bookmark::EDataTypeReal );

	iBkDatabase.RegisterDatabasePropertyL ( KUid3, Bookmark::EDataTypeDes8 );

	iBkDatabase.RegisterDatabasePropertyL ( KUid4, Bookmark::EDataTypeDes16 );

	iBkDatabase.RegisterBookmarkPropertyL ( KUid5, Bookmark::EDataTypeInt );

	iBkDatabase.RegisterBookmarkPropertyL ( KUid6, Bookmark::EDataTypeReal );

	iBkDatabase.RegisterBookmarkPropertyL ( KUid7, Bookmark::EDataTypeDes8 );

	iBkDatabase.RegisterBookmarkPropertyL ( KUid8, Bookmark::EDataTypeDes16 );

	iBkDatabase.RegisterFolderPropertyL ( KUid9, Bookmark::EDataTypeInt );

	iBkDatabase.RegisterFolderPropertyL ( KUid10, Bookmark::EDataTypeReal );

	iBkDatabase.RegisterFolderPropertyL ( KUid11, Bookmark::EDataTypeDes8 );

	iBkDatabase.RegisterFolderPropertyL ( KUid12, Bookmark::EDataTypeDes16 );

	iBkDatabase.RegisterDatabasePropertyL ( KUid13, Bookmark::EDataTypeInt );

	iBkDatabase.RegisterDatabasePropertyL ( KUid14, Bookmark::EDataTypeReal );

	iBkDatabase.RegisterDatabasePropertyL ( KUid15, Bookmark::EDataTypeDes8 );

	iBkDatabase.RegisterDatabasePropertyL ( KUid16, Bookmark::EDataTypeDes16 );

	iBkDatabase.RegisterBookmarkPropertyL ( KUid17, Bookmark::EDataTypeInt );

	iBkDatabase.RegisterBookmarkPropertyL ( KUid18, Bookmark::EDataTypeReal );

	iBkDatabase.RegisterBookmarkPropertyL ( KUid19, Bookmark::EDataTypeDes8 );

	iBkDatabase.RegisterBookmarkPropertyL ( KUid20, Bookmark::EDataTypeDes16 );

	iBkDatabase.RegisterFolderPropertyL ( KUid21, Bookmark::EDataTypeInt );

	iBkDatabase.RegisterFolderPropertyL ( KUid22, Bookmark::EDataTypeReal );

	iBkDatabase.RegisterFolderPropertyL ( KUid23, Bookmark::EDataTypeDes8 );

	iBkDatabase.RegisterFolderPropertyL ( KUid24, Bookmark::EDataTypeDes16 );

	}

void CBmTests::SetCustomPropertiesValueL ()

	{	

	iBkDatabase.SetCustomPropertyL ( KUid1, intValue1 );

	iBkDatabase.SetCustomPropertyL ( KUid2, realValue1 );	

	iBkDatabase.SetCustomPropertyL ( KUid3, KDbValue8 );

	iBkDatabase.SetCustomPropertyL ( KUid4, KDbValue );	

	RBkBookmark bookmark1 = iBkDatabase.OpenBookmarkL ( 1 );

	CleanupClosePushL ( bookmark1 );

	bookmark1.SetCustomPropertyL ( KUid5, intValue2 );

	bookmark1.SetCustomPropertyL ( KUid6, realValue2 );	

	bookmark1.SetCustomPropertyL ( KUid7, KBkmrkValue8 );

	bookmark1.SetCustomPropertyL ( KUid8, KBkmrkValue );	

	CleanupStack::PopAndDestroy ( 1 ); // bookmark1

	RBkFolder folder1 = iBkDatabase.OpenFolderL ( KTxtFolder1 );

	CleanupClosePushL ( folder1 );

	folder1.SetCustomPropertyL ( KUid9, intValue3 );

	folder1.SetCustomPropertyL ( KUid10, realValue3 );	

	folder1.SetCustomPropertyL ( KUid11, KFldrValue8 );

	folder1.SetCustomPropertyL ( KUid12, KFldrValue );	

	CleanupStack::PopAndDestroy ( &folder1 );

	iBkDatabase.SetCustomPropertyL ( KUid13, intValue4 );

	iBkDatabase.SetCustomPropertyL ( KUid14, realValue4 );	

	iBkDatabase.SetCustomPropertyL ( KUid15, KDb2Value8 );

	iBkDatabase.SetCustomPropertyL ( KUid16, KDb2Value );	

	RBkBookmark bookmark2 = iBkDatabase.OpenBookmarkL ( 2 );

	CleanupClosePushL ( bookmark2 );

	bookmark2.SetCustomPropertyL ( KUid17, intValue5 );

	bookmark2.SetCustomPropertyL ( KUid18, realValue5 );	

	bookmark2.SetCustomPropertyL ( KUid19, KBkmrk2Value8 );

	bookmark2.SetCustomPropertyL ( KUid20, KBkmrk2Value );	

	CleanupStack::PopAndDestroy ( &bookmark2 );

	// Reset the custom property value for bookmark1

	bookmark1 = iBkDatabase.OpenBookmarkL ( 1 );

	CleanupClosePushL ( bookmark1 );

	bookmark1.SetCustomPropertyL ( KUid17, intValue5 );

	bookmark1.SetCustomPropertyL ( KUid18, realValue5 );	

	bookmark1.SetCustomPropertyL ( KUid19, KBkmrk2Value8 );

	bookmark1.SetCustomPropertyL ( KUid20, KBkmrk2Value );	

	CleanupStack::PopAndDestroy ( 1 ); // bookmark1

	RBkFolder folder2 = iBkDatabase.OpenFolderL ( KTxtFolder2 );

	CleanupClosePushL ( folder2 );

	folder2.SetCustomPropertyL ( KUid21, intValue6 );

	folder2.SetCustomPropertyL ( KUid22, realValue6 );	

	folder2.SetCustomPropertyL ( KUid23, KFldr2Value8 );

	folder2.SetCustomPropertyL ( KUid24, KFldr2Value );	

	CleanupStack::PopAndDestroy ( &folder2 );

	folder1 = iBkDatabase.OpenFolderL ( KTxtFolder1 );

	CleanupClosePushL ( folder1 );

	folder1.SetCustomPropertyL ( KUid21, intValue6 );

	folder1.SetCustomPropertyL ( KUid22, realValue6 );	

	folder1.SetCustomPropertyL ( KUid23, KFldr2Value8 );

	folder1.SetCustomPropertyL ( KUid24, KFldr2Value );	

	CleanupStack::PopAndDestroy ( &folder1 );

	}

void CBmTests::ReadCustomPropertiesValueL ()

	{

	HBufC* retDesC16 = HBufC::NewLC(KMaxTestLength);

	TPtr retPtr16 = retDesC16->Des();

	HBufC8* retDesC8 = HBufC8::NewLC(KMaxTestLength * 2);

	TPtr8 retPtr8 = retDesC8->Des();

	TReal realValue;

	TInt intValue;

	iBkDatabase.GetCustomPropertyL ( KUid1, intValue );	

	iTestWrapper.TEST ( intValue == intValue1 );

	iBkDatabase.GetCustomPropertyL ( KUid2, realValue );	

	iTestWrapper.TEST ( realValue == realValue1 );

	iBkDatabase.GetCustomPropertyL ( KUid3, retPtr8 );

	iTestWrapper.TEST(retPtr8.Compare(KDbValue8) == 0);

	iBkDatabase.GetCustomPropertyL ( KUid4, retPtr16 );

	iTestWrapper.TEST(retPtr16.Compare(KDbValue) == 0);

	RBkBookmark bookmark2 = iBkDatabase.OpenBookmarkL ( 2 );

	CleanupClosePushL ( bookmark2 );

	bookmark2.GetCustomPropertyL ( KUid17, intValue );

	iTestWrapper.TEST ( intValue == intValue5 );	

	bookmark2.GetCustomPropertyL ( KUid18, realValue );

	iTestWrapper.TEST ( realValue == realValue5 );	

	bookmark2.GetCustomPropertyL ( KUid19, retPtr8 );

	iTestWrapper.TEST(retPtr8.Compare ( KBkmrk2Value8 ) == 0);

	bookmark2.GetCustomPropertyL ( KUid20, retPtr16 );

	iTestWrapper.TEST(retPtr16.Compare ( KBkmrk2Value ) == 0);

	RBkFolder folder2 = iBkDatabase.OpenFolderL ( KTxtFolder2 );

	CleanupClosePushL ( folder2 );

	folder2.GetCustomPropertyL ( KUid21, intValue );

	iTestWrapper.TEST ( intValue == intValue6 );	

	folder2.GetCustomPropertyL ( KUid22, realValue );	

	iTestWrapper.TEST ( realValue == realValue6 );	

	folder2.GetCustomPropertyL ( KUid23, retPtr8 );

	iTestWrapper.TEST(retPtr8.Compare ( KFldr2Value8 ) == 0);

	folder2.GetCustomPropertyL ( KUid24, retPtr16 );	

	iTestWrapper.TEST(retPtr16.Compare ( KFldr2Value ) == 0);

	CleanupStack::PopAndDestroy ( 4 ); // retDesC16, retDesC8, bookmark2, folder2

	}

void CBmTests::UpdateTreeL ()

	{

	iBkDatabase.RefreshL ();

	}

void CBmTests::DeleteAllL ()

	{

	RBkFolder root = iBkDatabase.OpenRootL ();

	CleanupClosePushL ( root );

	TInt count = root.Count ();	

	while ( count-- )

		{

		RBkNode node = root.OpenItemL ( 0 );

		CleanupClosePushL ( node );

		TBool recursive = ETrue;

		if ( node.Type() == Bookmark::ETypeBookmark )

			{

			recursive = EFalse;

			}

		iBkDatabase.DeleteItemL ( node.Id(), recursive );

		CleanupStack::PopAndDestroy ( &node );

		}		

	CleanupStack::PopAndDestroy ( &root ); 

	}