sl@0: // Copyright (c) 1998-2009 Nokia Corporation and/or its subsidiary(-ies).
sl@0: // All rights reserved.
sl@0: // This component and the accompanying materials are made available
sl@0: // under the terms of "Eclipse Public License v1.0"
sl@0: // which accompanies this distribution, and is available
sl@0: // at the URL "http://www.eclipse.org/legal/epl-v10.html".
sl@0: //
sl@0: // Initial Contributors:
sl@0: // Nokia Corporation - initial contribution.
sl@0: //
sl@0: // Contributors:
sl@0: //
sl@0: // Description:
sl@0: //
sl@0: 
sl@0: // MSVC++ up to 5.0 has problems with expanding inline functions
sl@0: // This disables the mad warnings for the whole project
sl@0: #if defined(NDEBUG) && defined(__VC32__) && _MSC_VER<=1100
sl@0: #pragma warning(disable : 4710)			// function not expanded. MSVC 5.0 is stupid
sl@0: #endif
sl@0: 
sl@0: #include <d32dbms.h>
sl@0: #include <s32file.h>
sl@0: #include <e32test.h>
sl@0: #include <e32math.h>
sl@0: #include <hal.h>
sl@0: 
sl@0: static RTest TheTest(_L("t_dbcomp"));
sl@0: static RFs   TheFs;
sl@0: 
sl@0: //T_BENCH.DB file is created by T_BENCH test and is used by the current test (T_COMP).
sl@0: //T_COMP test will delete T_BENCH.DB at the end as it is no more needed.
sl@0: //If you want to rerun T_COMP test again, you have to ensure that T_BENCH.DB file exists -
sl@0: //run T_BENCH test again.
sl@0: TFileName TheBenchDbFileName;
sl@0: TFileName TheCompressedFileName;
sl@0: TFileName TheDecompressedFileName;
sl@0: 
sl@0: RDbNamedDatabase TheCompDb;
sl@0: RDbNamedDatabase TheCopyDb;
sl@0: RDbTable 		 TheTable1, TheTable2;
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: static void CloseAll()
sl@0: 	{
sl@0: 	TheTable2.Close();
sl@0: 	TheTable1.Close();
sl@0: 	TheCopyDb.Close();
sl@0: 	TheCompDb.Close();
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: //Delete "aFullName" file.
sl@0: static void DeleteFile(const TDesC& aFullName)
sl@0: 	{
sl@0: 	RFs fsSession;
sl@0: 	TInt err = fsSession.Connect();
sl@0: 	if(err == KErrNone)
sl@0: 		{
sl@0: 		TEntry entry;
sl@0: 		if(fsSession.Entry(aFullName, entry) == KErrNone)
sl@0: 			{
sl@0: 			err = fsSession.SetAtt(aFullName, 0, KEntryAttReadOnly);
sl@0: 			if(err != KErrNone) 
sl@0: 				{
sl@0: 				TheTest.Printf(_L("Error %d changing \"%S\" file attributes.\n"), err, &aFullName);
sl@0: 				}
sl@0: 			err = fsSession.Delete(aFullName);
sl@0: 			if(err != KErrNone) 
sl@0: 				{
sl@0: 				TheTest.Printf(_L("Error %d deleting \"%S\" file.\n"), err, &aFullName);
sl@0: 				}
sl@0: 			}
sl@0: 		fsSession.Close();
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		TheTest.Printf(_L("Error %d connecting file session. File: %S.\n"), err, &aFullName);
sl@0: 		}
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: static void DestroyTestEnv()
sl@0: 	{
sl@0: 	CloseAll();
sl@0: 	DeleteFile(TheDecompressedFileName);
sl@0: 	DeleteFile(TheCompressedFileName);
sl@0: 	DeleteFile(TheBenchDbFileName);
sl@0: 	TheFs.Close();
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: //Tests macros and functions.
sl@0: //If (!aValue) then the test will be panicked, the test data files will be deleted.
sl@0: static void Check(TInt aValue, TInt aLine)
sl@0: 	{
sl@0: 	if(!aValue)
sl@0: 		{
sl@0: 		TheTest.Printf(_L("*** Boolean expression evaluated to false!\r\n"));
sl@0: 		DestroyTestEnv();
sl@0: 		TheTest(EFalse, aLine);
sl@0: 		}
sl@0: 	}
sl@0: //If (aValue != aExpected) then the test will be panicked, the test data files will be deleted.
sl@0: static void Check(TInt aValue, TInt aExpected, TInt aLine)
sl@0: 	{
sl@0: 	if(aValue != aExpected)
sl@0: 		{
sl@0: 		TheTest.Printf(_L("*** Expected error: %d, got: %d\r\n"), aExpected, aValue);
sl@0: 		DestroyTestEnv();
sl@0: 		TheTest(EFalse, aLine);
sl@0: 		}
sl@0: 	}
sl@0: //Use these to test conditions.
sl@0: #define TEST(arg) ::Check((arg), __LINE__)
sl@0: #define TEST2(aValue, aExpected) ::Check(aValue, aExpected, __LINE__)
sl@0: 
sl@0: //////////////////////////////////////////////////////
sl@0: 
sl@0: static TInt TheCounterFreq = -10000000;
sl@0: const TInt KMicroSecIn1Sec = 1000000;
sl@0: 
sl@0: TUint32 CalcTickDiff(TUint32 aStartTicks, TUint32 aEndTicks)
sl@0: 	{
sl@0: 	TInt64 diffTicks = (TInt64)aEndTicks - (TInt64)aStartTicks;
sl@0: 	if(diffTicks < 0)
sl@0: 		{
sl@0: 		diffTicks = KMaxTUint32 + diffTicks + 1;
sl@0: 		}
sl@0: 	return (TUint32)diffTicks;
sl@0: 	}
sl@0: 
sl@0: //Prints aFastCount parameter (converted to us)
sl@0: void PrintFcDiffAsUs(const TDesC& aFormatStr, TUint32 aFastCount)
sl@0: 	{
sl@0: 	if(TheCounterFreq <= 0)
sl@0: 		{
sl@0: 		TEST2(HAL::Get(HAL::EFastCounterFrequency, TheCounterFreq), KErrNone);
sl@0: 		TheTest.Printf(_L("Counter frequency=%d Hz\r\n"), TheCounterFreq);
sl@0: 		}
sl@0: 	double v = ((double)aFastCount * KMicroSecIn1Sec) / (double)TheCounterFreq;
sl@0: 	TInt v2 = (TInt)v;
sl@0: 	TheTest.Printf(aFormatStr, v2);
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: //
sl@0: // Prepare the test directory.
sl@0: //
sl@0: void CreateTestEnv()
sl@0:     {
sl@0: 	TInt err = TheFs.Connect();
sl@0: 	TheTest(err == KErrNone);
sl@0: 
sl@0: 	err = TheFs.MkDirAll(TheBenchDbFileName);
sl@0: 	TEST(err == KErrNone || err == KErrAlreadyExists);
sl@0: 	
sl@0: 	TUint dummy;
sl@0: 	err = TheFs.Att(TheBenchDbFileName, dummy);
sl@0: 	if(err != KErrNone)
sl@0: 		{
sl@0: 		TheTest.Printf(_L("*** The %S file does not exist, err=%d. Run \"T_BENCH\" test first!\r\n"), &TheBenchDbFileName, err);
sl@0: 		TEST2(err, KErrNone);
sl@0: 		}
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: //
sl@0: // Compress the database
sl@0: //
sl@0: void CompressL(const TDesC& aSource, const TDesC& aTarget, TBool aCompress)
sl@0: 	{
sl@0: 	CFileMan* man = CFileMan::NewL(TheFs);
sl@0: 	TInt err = man->Copy(aSource, aTarget);
sl@0: 	delete man;
sl@0: 	User::LeaveIfError(err);
sl@0: 	
sl@0: 	User::LeaveIfError(TheFs.SetAtt(aTarget, 0, KEntryAttReadOnly));
sl@0: 	CFileStore* store = CFileStore::OpenLC(TheFs, aTarget, EFileRead|EFileWrite);
sl@0: 	
sl@0: 	TUint32 fc = User::FastCounter();
sl@0: 	if(aCompress)
sl@0: 		{
sl@0: 		RDbStoreDatabase::CompressL(*store, store->Root());
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		RDbStoreDatabase::DecompressL(*store, store->Root());
sl@0: 		}
sl@0: 	TUint32 time = CalcTickDiff(fc, User::FastCounter());
sl@0: 	PrintFcDiffAsUs(_L(" %d us\r\n"), time);
sl@0: 	
sl@0: 	store->CompactL();
sl@0: 	store->CommitL();
sl@0: 	
sl@0: 	CleanupStack::PopAndDestroy(store);
sl@0: 	}
sl@0: 
sl@0: void CheckTableL(RDbDatabase& aDatabase, RDbDatabase& aCopy, const TDesC& aTable)
sl@0: 	{
sl@0: 	TheTest.Printf(_L("Processing table %S\n"), &aTable);
sl@0: 	
sl@0: 	TInt err = TheTable1.Open(aDatabase, aTable, RDbRowSet::EReadOnly);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	
sl@0: 	err = TheTable2.Open(aCopy, aTable, RDbRowSet::EReadOnly);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	
sl@0: 	TInt columns = TheTable1.ColCount();
sl@0: 	while(TheTable1.NextL())
sl@0: 		{
sl@0: 		TheTable1.GetL();
sl@0: 		TEST(TheTable2.NextL());
sl@0: 		TheTable2.GetL();
sl@0: 		for(TInt ii=1;ii<=columns;++ii)
sl@0: 			{
sl@0: 			if(TDbCol::IsLong(TheTable1.ColType(ii)))
sl@0: 				{
sl@0: 				TInt len = TheTable1.ColSize(ii);
sl@0: 				TEST2(len, TheTable2.ColSize(ii));
sl@0: 				RDbColReadStream strm1;
sl@0: 				strm1.OpenLC(TheTable1, ii);
sl@0: 				RDbColReadStream strm2;
sl@0: 				strm2.OpenLC(TheTable2, ii);
sl@0: 				TBuf8<512> buf1;
sl@0: 				TBuf8<512> buf2;
sl@0: 				while(len)
sl@0: 					{
sl@0: 					TInt block = Min(512, len);
sl@0: 					strm1.ReadL(buf1, block);
sl@0: 					strm2.ReadL(buf2, block);
sl@0: 					TEST(buf1 == buf2);
sl@0: 					len -= block;
sl@0: 					}
sl@0: 				CleanupStack::PopAndDestroy(2);
sl@0: 				}
sl@0: 			else
sl@0: 				{
sl@0: 				TEST(TheTable1.ColDes8(ii) == TheTable2.ColDes8(ii));
sl@0: 				}
sl@0: 			}
sl@0: 		}
sl@0: 	TheTable2.Close();
sl@0: 	TheTable1.Close();
sl@0: 	}
sl@0: 
sl@0: void CheckL(const TDesC& aSource, const TDesC& aTarget)
sl@0: 	{
sl@0: 	TInt err = TheCompDb.Open(TheFs, aSource, KNullDesC, RDbNamedDatabase::EReadOnly);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	
sl@0: 	err = TheCopyDb.Open(TheFs, aTarget, KNullDesC, RDbNamedDatabase::EReadOnly);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	
sl@0: 	CDbTableNames* tables = TheCompDb.TableNamesL();
sl@0: 	for(TInt ii=0;ii<tables->Count();++ii)
sl@0: 		{
sl@0: 		CheckTableL(TheCompDb, TheCopyDb, (*tables)[ii]);
sl@0: 		}
sl@0: 	delete tables;
sl@0: 	TheCopyDb.Close();
sl@0: 	TheCompDb.Close();
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-DBMS-CT-0593
sl@0: @SYMTestCaseDesc        Database compression tests.
sl@0: @SYMTestPriority        Medium
sl@0: @SYMTestActions         Tests for RDbStoreDatabase::CompressL(),RDbStoreDatabase::DecompressL() functions
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: void Test(const TDesC& aSource, const TDesC& aTarget, TBool aCompress)
sl@0: 	{
sl@0: 	TRAPD(err, CompressL(aSource, aTarget, aCompress));
sl@0: 	TEST2(err, KErrNone);
sl@0: 	TheTest.Printf(_L("Checking database"));
sl@0: 	TRAP(err, CheckL(aSource, aTarget));
sl@0: 	TEST2(err, KErrNone);
sl@0: 	}
sl@0: 
sl@0: void DoTests()
sl@0: 	{
sl@0: 	TheTest.Start(_L(" @SYMTestCaseID:SYSLIB-DBMS-CT-0593 Compressing..."));
sl@0: 	Test(TheBenchDbFileName, TheCompressedFileName, ETrue);
sl@0: 	
sl@0: 	TheTest.Next(_L("Decompressing..."));
sl@0: 	Test(TheCompressedFileName, TheDecompressedFileName, EFalse);
sl@0: 	}
sl@0: 
sl@0: //Usage: "t_comp [<drive letter>:]]"
sl@0: TInt E32Main()
sl@0:     {
sl@0: 	TheTest.Title();
sl@0: 
sl@0: 	CTrapCleanup* tc = CTrapCleanup::New();
sl@0: 	TheTest(tc != NULL);
sl@0: 	
sl@0: 	TBuf<256> cmdline;
sl@0: 	User::CommandLine(cmdline);
sl@0: 
sl@0: 	TParse parse;
sl@0: 
sl@0: 	_LIT(KBenchDbFile, "C:\\DBMS-TST\\T_BENCH.DB");
sl@0: 	parse.Set(cmdline, &KBenchDbFile, 0);
sl@0: 	TheBenchDbFileName.Copy(parse.FullName());
sl@0: 
sl@0: 	_LIT(KCompressedFile, "C:\\DBMS-TST\\T_COMP.DB1");
sl@0: 	parse.Set(cmdline, &KCompressedFile, 0);
sl@0: 	TheCompressedFileName.Copy(parse.FullName());
sl@0: 
sl@0: 	_LIT(KDecompressedFile, "C:\\DBMS-TST\\T_COMP.DB2");
sl@0: 	parse.Set(cmdline, &KDecompressedFile, 0);
sl@0: 	TheDecompressedFileName.Copy(parse.FullName());
sl@0: 	
sl@0: 	__UHEAP_MARK;
sl@0: 	
sl@0: 	CreateTestEnv();
sl@0: 	DoTests();
sl@0: 	DestroyTestEnv();
sl@0: 
sl@0: 	delete tc;
sl@0: 
sl@0: 	__UHEAP_MARKEND;
sl@0: 	
sl@0: 	TheTest.End();
sl@0: 	TheTest.Close();
sl@0: 	
sl@0: 	return 0;
sl@0:     }