// Copyright (c) 2005-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 "bkmrkdb.h"
#include "bkmrk.h"
#include "bkmrkfolder.h"
#include "testutils.h"

LOCAL_D CTestWrapper* gTestWrapper;
LOCAL_D Bookmark::TAttachmentId gIconId1;
LOCAL_D Bookmark::TAttachmentId gIconId2;

_LIT(KFilename, "t_cbookmarkdb.cpp");

_LIT(KTxtFolder1, "BkmrkTEST Folder 1");
_LIT(KTxtFolder2, "BkmrkTEST Folder 2");
_LIT(KTxtFolder3, "BkmrkTEST Folder 3");

_LIT(KTxtNewFolder, "New Folder 1");
_LIT(KTxtNewFolder2, "New Folder 2");
_LIT(KTxtNewFolder3, "New Folder 3");

_LIT(KTxtBookmark1, "BkmrkTEST Bookmark 1");
_LIT(KTxtBookmark2, "BkmrkTEST Bookmark 2");
_LIT(KTxtBookmark3, "BkmrkTEST Bookmark 3");
_LIT(KTxtBookmark4, "BkmrkTEST Bookmark 4");

_LIT8(KIconData, 	"123456789");
_LIT8(KIconData2,	"999999999");

_LIT(KTxtDesc, "This is the home page text");
_LIT8(KTxtUri, "sip:user:pass@location.com:1666;transport=tcp;x=123;df=223?asd=b");

const TUint32 KProxyId 	= 0x7CFFFFFF;
const TUint32 KNapId 	= 0x6CFFF000;

LOCAL_D void GeneralTestsL()
	{
	CBookmarkDb* db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] Icons"));
	Bookmark::TAttachmentId id = db->CreateIconL(KIconData);
	gTestWrapper->TEST(db->IconL(id).Compare(KIconData) == 0);
	gTestWrapper->TEST(db->IconL(id).Compare(KIconData2) != 0);
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] Home"));
	CBookmark& bookmark = db->CreateBookmarkL(db->RootL());
	gTestWrapper->TEST(!db->Home());
	db->SetHome(&bookmark);
	gTestWrapper->TEST(db->Home()->Id() == bookmark.Id());
	db->SetHome(NULL);
	gTestWrapper->TEST(!db->Home());
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] home page text"));
	db->SetHomePageTextL(KTxtDesc);
	gTestWrapper->TEST(db->HomePageText().Compare(KTxtDesc) == 0);
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] search uri"));
	db->SetSearchUriL(KTxtUri);
	gTestWrapper->TEST(db->SearchUri().Compare(KTxtUri) == 0);

	gTestWrapper->Next(_L("[CBookmarkDb tests] default proxy and nap"));
	gTestWrapper->TEST(db->DefaultProxy() != KProxyId);
	db->SetDefaultProxy(KProxyId);
	gTestWrapper->TEST(db->DefaultProxy() == KProxyId);
	
	gTestWrapper->TEST(db->DefaultNap() != KNapId);
	db->SetDefaultNap(KNapId);
	gTestWrapper->TEST(db->DefaultNap() == KNapId);
	
	CleanupStack::PopAndDestroy(db);
	}

LOCAL_D void TreeTestsL()
	{
	CBookmarkDb* db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] Construct tree and root"));
	db->ConstructTreeL();
	CBookmarkBase* rootPtr = db->FindItem(Bookmark::KRootItemID);
	gTestWrapper->TEST(rootPtr != NULL);
	gTestWrapper->TEST(db->RootL().Id() == rootPtr->Id());
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] Create items"));
	CBookmarkFolder& folder1 = db->CreateFolderL(KTxtFolder1, db->RootL());
	CBookmark& bookmark1 = db->CreateBookmarkL(folder1);
	CBookmarkFolder& folder2 = db->CreateFolderL(KTxtFolder2, folder1);
	CBookmark& bookmark2 = db->CreateBookmarkL(folder2);
	CBookmark& bookmark3 = db->CreateBookmarkL(folder2);
	CBookmarkFolder& folder3 = db->CreateFolderL(KTxtFolder3, folder2);
	CBookmark& bookmark4 = db->CreateBookmarkL(folder3);
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] Find items"));
	CBookmarkBase* item = db->FindItem(folder1.Id());
	CBookmarkFolder* folder = db->FindFolder(KTxtFolder1);
	gTestWrapper->TEST(item != NULL);
	gTestWrapper->TEST(item == folder);
	gTestWrapper->TEST(item->Parent()->Id() == Bookmark::KRootItemID);
	item = db->FindItem(bookmark1.Id());
	gTestWrapper->TEST(item != NULL);
	gTestWrapper->TEST(item->Parent()->Id() == folder1.Id());
	item = db->FindItem(folder2.Id());
	folder = db->FindFolder(KTxtFolder2);
	gTestWrapper->TEST(item != NULL);
	gTestWrapper->TEST(item == folder);
	gTestWrapper->TEST(item->Parent()->Id() == folder1.Id());
	item = db->FindItem(bookmark2.Id());
	gTestWrapper->TEST(item != NULL);
	gTestWrapper->TEST(item->Parent()->Id() == folder2.Id());
	item = db->FindItem(bookmark3.Id());
	gTestWrapper->TEST(item != NULL);
	gTestWrapper->TEST(item->Parent()->Id() == folder2.Id());
	item = db->FindItem(folder3.Id());
	folder = db->FindFolder(KTxtFolder3);
	gTestWrapper->TEST(item != NULL);
	gTestWrapper->TEST(item == folder);
	gTestWrapper->TEST(item->Parent()->Id() == folder2.Id());
	item = db->FindItem(bookmark4.Id());
	gTestWrapper->TEST(item != NULL);
	gTestWrapper->TEST(item->Parent()->Id() == folder3.Id());
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] Delete items"));
	CBookmarkFolder& root = db->RootL();
	gTestWrapper->TEST(root.Count() == 1);
	gTestWrapper->TEST(folder1.Count() == 2);
	gTestWrapper->TEST(folder2.Count() == 3);
	gTestWrapper->TEST(folder3.Count() == 1);
	db->DeleteItemL(folder2.Id(), EFalse);
	folder = db->FindFolder(KTxtFolder2);
	gTestWrapper->TEST(!folder);
	gTestWrapper->TEST(root.Count() == 4);
	gTestWrapper->TEST(folder1.Count() == 1);
	gTestWrapper->TEST(folder3.Count() == 1);
	db->DeleteItemL(folder1.Id(), ETrue);
	folder = db->FindFolder(KTxtFolder1);
	gTestWrapper->TEST(!folder);
	gTestWrapper->TEST(root.Count() == 3);
	db->DeleteItemL(bookmark2.Id(), EFalse);
	folder = db->FindFolder(KTxtBookmark2);
	gTestWrapper->TEST(folder == NULL);
	db->DeleteItemL(bookmark4.Id(), ETrue);
	folder = db->FindFolder(KTxtBookmark4);
	gTestWrapper->TEST(!folder);
	gTestWrapper->TEST(root.Count() == 2);
	gTestWrapper->TEST(folder3.Count() == 0);
	
	CleanupStack::PopAndDestroy(db);
	}

LOCAL_D void CreateTreeL(CBookmarkDb* aDb)
	{
	gTestWrapper->Next(_L("[CBookmarkDb tests] Create tree"));
	CBookmarkFolder& folder1 = aDb->CreateFolderL(KTxtFolder1, aDb->RootL());
	CBookmark& bookmark1 = aDb->CreateBookmarkL(folder1);
	bookmark1.SetTitleL(KTxtBookmark1);
	CBookmarkFolder& folder2 = aDb->CreateFolderL(KTxtFolder2, folder1);
	CBookmark& bookmark2 = aDb->CreateBookmarkL(folder2);
	bookmark2.SetTitleL(KTxtBookmark2);
	CBookmark& bookmark3 = aDb->CreateBookmarkL(folder2);
	bookmark3.SetTitleL(KTxtBookmark3);
	CBookmarkFolder& folder3 = aDb->CreateFolderL(KTxtFolder3, folder2);
	CBookmark& bookmark4 = aDb->CreateBookmarkL(folder3);
	bookmark4.SetTitleL(KTxtBookmark4);
	
	// create a couple of icons
	gIconId1 = aDb->CreateIconL(KIconData);
	gIconId2 = aDb->CreateIconL(KIconData2);
	CBkmrkExtendedProperties& properties1 = bookmark1.BkmrkExtendedPropertiesL();
	properties1.SetIconIdL(gIconId1);
	CBkmrkExtendedProperties& properties2 = bookmark2.BkmrkExtendedPropertiesL();
	properties2.SetIconIdL(gIconId2);
	
	CBookmarkFolder& root = aDb->RootL();
	gTestWrapper->TEST(root.Count() == 1);
	gTestWrapper->TEST(folder1.Count() == 2);
	gTestWrapper->TEST(folder2.Count() == 3);
	gTestWrapper->TEST(folder3.Count() == 1);
	}
	
LOCAL_D void CheckDatabaseL(CBookmarkDb& aDb)
	{
	CBookmarkFolder& root = aDb.RootL();
	gTestWrapper->TEST(root.Count() == 1);
	CBookmarkFolder* folder1 = aDb.FindFolder(KTxtFolder1);
	gTestWrapper->TEST(folder1 != NULL);
	gTestWrapper->TEST(folder1->Count() == 2);
	gTestWrapper->TEST(folder1->At(0).Type() == Bookmark::ETypeBookmark);
	CBookmark& bookmark1 = static_cast<CBookmark&>(folder1->At(0));
	CBkmrkExtendedProperties& properties1 = bookmark1.BkmrkExtendedPropertiesL();
	gTestWrapper->TEST(properties1.IconId() == gIconId1);
	CBookmarkFolder* folder2 = aDb.FindFolder(KTxtFolder2);
	gTestWrapper->TEST(folder2 != NULL);
	gTestWrapper->TEST(folder2->Count() == 3);
	gTestWrapper->TEST(folder2->At(0).Type() == Bookmark::ETypeBookmark);
	CBookmark& bookmark2 = static_cast<CBookmark&>(folder2->At(0));
	CBkmrkExtendedProperties& properties2 = bookmark2.BkmrkExtendedPropertiesL();
	gTestWrapper->TEST(properties2.IconId() == gIconId2);
	CBookmarkFolder* folder3 = aDb.FindFolder(KTxtFolder3);
	gTestWrapper->TEST(folder3 != NULL);
	gTestWrapper->TEST(folder3->Count() == 1);
	}
	
LOCAL_D void CommitTestsL()
	{
	gTestWrapper->Next(_L("[CBookmarkDb tests] Commit"));
	CBookmarkDb* db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	CreateTreeL(db);
	db->CommitL();
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	
	CheckDatabaseL(*db);
	
	CBookmarkFolder* folder = db->FindFolder(KTxtFolder1);
	db->DeleteItemL(folder->Id(), ETrue);
	gTestWrapper->TEST(db->RootL().Count() == 0);
	
	CleanupStack::PopAndDestroy(db);
	
	// The delete has not been committed so it should reload 
	db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	
	CheckDatabaseL(*db);
	
	folder = db->FindFolder(KTxtFolder1);
	db->DeleteItemL(folder->Id(), ETrue);
	gTestWrapper->TEST(db->RootL().Count() == 0);
	
	User::LeaveIfError(db->DeleteIconL(gIconId1));
	User::LeaveIfError(db->DeleteIconL(gIconId2));
	
	db->CommitL();
	
	CleanupStack::PopAndDestroy(db);
	
	// The delete has been committed so it should not reload 
	db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	CBookmarkFolder& root = db->RootL();
	gTestWrapper->TEST(root.Count() == 0);
	CleanupStack::PopAndDestroy(db);
	}

LOCAL_D void IconDataTestsL()
	{
	CBookmarkDb* db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	
	gTestWrapper->Next(_L("[CBookmarkDb tests] Icon data tests"));
	Bookmark::TAttachmentId iconId = db->CreateIconL(KIconData);
	gTestWrapper->TEST(db->IconL(iconId).Compare(KIconData) == 0);
	gTestWrapper->TEST(db->IconL(iconId).Compare(KIconData2) != 0);

	CBookmark& bookmark = db->CreateBookmarkL(db->RootL());
	Bookmark::TItemId bookmarkId = bookmark.Id();
	CBkmrkExtendedProperties& properties = bookmark.BkmrkExtendedPropertiesL();
	properties.SetIconIdL(iconId);
	
	db->CommitL();
	
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	
	db->ConstructTreeL();
	
	CBookmark* bookmarkPtr = static_cast<CBookmark*>(db->FindItem(bookmarkId));
	if (!bookmarkPtr)
		{
		User::Leave(KErrNotFound);
		}
	CBkmrkExtendedProperties& properties2 = bookmarkPtr->BkmrkExtendedPropertiesL();
	iconId = properties2.IconId();
	gTestWrapper->TEST(db->IconL(iconId).Compare(KIconData) == 0);
	gTestWrapper->TEST(db->IconL(iconId).Compare(KIconData2) != 0);
	
	// cleanup
	db->DeleteItemL(bookmarkId, ETrue);
	db->DeleteIconL(iconId);
	db->CommitL();
	
	CleanupStack::PopAndDestroy(db);
	}

LOCAL_D void CreateSecurityTreeL(CBookmarkDb& aDb)
	{
	CBookmarkFolder& folder1 = aDb.CreateFolderL(KTxtNewFolder, aDb.RootL());
	CBookmarkFolder& folder2 = aDb.CreateFolderL(KTxtNewFolder2, folder1);
	CBookmarkFolder& folder3 = aDb.CreateFolderL(KTxtNewFolder3, folder2);
	aDb.CommitL();
	
	folder3.SetPublicL(EFalse);
	
	CBookmark& bookmark1 = aDb.CreateBookmarkL(folder1);
	CBookmark& bookmark2 = aDb.CreateBookmarkL(folder2);
	CBookmark& bookmark3 = aDb.CreateBookmarkL(folder3);
	aDb.CommitL();
	
	bookmark1.SetPublicL(ETrue); // to avoid compiler warnings
	bookmark2.SetPublicL(EFalse);
	bookmark3.SetPublicL(ETrue); // to avoid compiler warnings
	aDb.CommitL();
	}
	
LOCAL_D void CheckAllL(CBookmarkDb& aDb)
	{
	CBookmarkFolder& root = aDb.RootL();
	gTestWrapper->TEST(root.Count() == 1);
	
	CBookmarkFolder* folder = aDb.FindFolder(KTxtNewFolder);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 2);
	
	folder = aDb.FindFolder(KTxtNewFolder2);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 2);
	
	folder = aDb.FindFolder(KTxtNewFolder3);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 1);
	}

LOCAL_D void CheckAllMinusOneL(CBookmarkDb& aDb)
	{
	CBookmarkFolder& root = aDb.RootL();
	gTestWrapper->TEST(root.Count() == 3);
	
	CBookmarkFolder* folder = aDb.FindFolder(KTxtNewFolder);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 1);
	
	folder = aDb.FindFolder(KTxtNewFolder2);
	gTestWrapper->TEST(!folder);
	
	folder = aDb.FindFolder(KTxtNewFolder3);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 1);
	}

LOCAL_D void CheckPrivateL(CBookmarkDb& aDb)
	{
	CBookmarkFolder& root = aDb.RootL();
	gTestWrapper->TEST(root.Count() == 1);
	
	CBookmarkFolder* folder = aDb.FindFolder(KTxtNewFolder);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 1);
	
	folder = aDb.FindFolder(KTxtNewFolder2);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 2);
	
	folder = aDb.FindFolder(KTxtNewFolder3);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 0);
	}

LOCAL_D void CheckPublicL(CBookmarkDb& aDb)
	{
	CBookmarkFolder& root = aDb.RootL();
	gTestWrapper->TEST(root.Count() == 2);
	
	CBookmarkFolder* folder = aDb.FindFolder(KTxtNewFolder);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 2);
	
	folder = aDb.FindFolder(KTxtNewFolder2);
	gTestWrapper->TEST(folder != NULL);
	gTestWrapper->TEST(folder->Count() == 0);
	
	folder = aDb.FindFolder(KTxtNewFolder3);
	gTestWrapper->TEST(!folder);
	}

LOCAL_D void SecurityTestsL()
	{
	gTestWrapper->Next(_L("[CBookmarkDb tests] Security"));
	CBookmarkDb* db = CBookmarkDb::NewL(Bookmark::EVisibilityManager, NULL);
	CleanupStack::PushL(db);
	CreateSecurityTreeL(*db);
	CheckAllL(*db);
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	db->ConstructTreeL();
	CheckAllL(*db);
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityPublic, NULL);
	CleanupStack::PushL(db);
	db->ConstructTreeL();
	CheckPublicL(*db);
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityPrivate, NULL);
	CleanupStack::PushL(db);
	db->ConstructTreeL();
	CheckPrivateL(*db);
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityManager, NULL);
	CleanupStack::PushL(db);
	db->ConstructTreeL();
	CBookmarkFolder* publicFolder = db->FindFolder(KTxtNewFolder2);
	publicFolder->SetOwnerL(0x00001234); 
	db->CommitL();
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityAll, NULL);
	CleanupStack::PushL(db);
	db->ConstructTreeL();
	CheckAllMinusOneL(*db);
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityManager, NULL);
	CleanupStack::PushL(db);
	db->ConstructTreeL();
	CheckAllL(*db);
	CleanupStack::PopAndDestroy(db);
	
	db = CBookmarkDb::NewL(Bookmark::EVisibilityDefault, NULL);
	CleanupStack::PushL(db);
	db->ConstructTreeL();
	CBookmarkFolder* folder = db->FindFolder(KTxtNewFolder);
	gTestWrapper->TEST(folder != NULL);
	db->DeleteItemL(folder->Id(), ETrue);
	db->CommitL();
	CleanupStack::PopAndDestroy(db);
	}
	
void DoCBookmarkDbUnitL(CTestWrapper* aTest)
	{
	gTestWrapper = aTest;
	gTestWrapper->SetFile(KFilename);
	gTestWrapper->Next(_L("*** CBookmarkDb tests ***"));
	
	GeneralTestsL();
	TreeTestsL();
	CommitTestsL();
	IconDataTestsL();
	SecurityTestsL();
	}