sl@0: // Copyright (c) 2010 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: #include <e32test.h>
sl@0: #include <bautils.h>
sl@0: #include <sqldb.h>
sl@0: #include <e32math.h>
sl@0: #include "SqlBur.h"
sl@0: 
sl@0: //CSqlSrvTestBurInterface - test implementation of the MSqlSrvBurInterface, implemented in the production code by the SQL server.
sl@0: class CSqlSrvTestBurInterface : public CBase, public MSqlSrvBurInterface
sl@0: 	{
sl@0: 	public:
sl@0: 		static CSqlSrvTestBurInterface* New();
sl@0: 		virtual ~CSqlSrvTestBurInterface();
sl@0: 		virtual RFs& Fs();
sl@0: 		virtual void GetBackUpListL(TSecureId aUid, TDriveNumber aDrive, RArray<HBufC*>& aFileList);
sl@0: 		
sl@0: 	private:
sl@0: 		void Construct();
sl@0: 		
sl@0: 	private:
sl@0: 		RFs iFs;
sl@0: 		
sl@0: 	};
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: RTest TheTest(_L("t_sqlbur2"));
sl@0: 
sl@0: #ifdef _DEBUG
sl@0: static const TInt KBurstRate = 100;
sl@0: #endif
sl@0: 
sl@0: TDriveNumber KTestDrive = EDriveC;
sl@0: _LIT(KTestDir, "c:\\test\\");
sl@0: CActiveScheduler* TheScheduler = NULL;
sl@0: CSqlSrvTestBurInterface* TheSqlSrvTestBurInterface = NULL;
sl@0: TInt TestModeSqlBurError = KErrNone;//The CSqlBurEventMonitor code will set the error here
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: void DestroyTestEnv()
sl@0: 	{
sl@0: 	delete TheSqlSrvTestBurInterface;
sl@0: 	TheSqlSrvTestBurInterface = NULL;		
sl@0: 	
sl@0: 	delete TheScheduler;
sl@0: 	TheScheduler = NULL;
sl@0: 	
sl@0: 	(void)RProperty::Delete(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey);  
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: //Test macros and functions
sl@0: void Check(TInt aValue, TInt aLine)
sl@0: 	{
sl@0: 	if(!aValue)
sl@0: 		{
sl@0: 		DestroyTestEnv();
sl@0: 		RDebug::Print(_L("*** Boolean expression evaluated to false.\r\n"));
sl@0: 		TheTest(EFalse, aLine);
sl@0: 		}
sl@0: 	}
sl@0: void Check(TInt aValue, TInt aExpected, TInt aLine)
sl@0: 	{
sl@0: 	if(aValue != aExpected)
sl@0: 		{
sl@0: 		DestroyTestEnv();
sl@0: 		RDebug::Print(_L("*** Expected error: %d, got: %d.\r\n"), aExpected, aValue);
sl@0: 		TheTest(EFalse, aLine);
sl@0: 		}
sl@0: 	}
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: CSqlSrvTestBurInterface* CSqlSrvTestBurInterface::New()
sl@0: 	{
sl@0: 	CSqlSrvTestBurInterface* self = new CSqlSrvTestBurInterface;
sl@0: 	TEST(self != NULL);
sl@0: 	self->Construct();
sl@0: 	return self;
sl@0: 	}
sl@0: 
sl@0: void CSqlSrvTestBurInterface::Construct()
sl@0: 	{
sl@0: 	TInt err = iFs.Connect();
sl@0: 	TEST2(err, KErrNone);
sl@0: 
sl@0: 	err = iFs.MkDir(KTestDir);
sl@0: 	TEST(err == KErrNone || err == KErrAlreadyExists);
sl@0: 
sl@0: 	err = iFs.CreatePrivatePath(KTestDrive);
sl@0: 	TEST(err == KErrNone || err == KErrAlreadyExists);
sl@0: 	}
sl@0: 
sl@0: CSqlSrvTestBurInterface::~CSqlSrvTestBurInterface()
sl@0: 	{
sl@0: 	iFs.Close();
sl@0: 	}
sl@0: 
sl@0: RFs& CSqlSrvTestBurInterface::Fs()
sl@0: 	{
sl@0: 	return iFs;
sl@0: 	}
sl@0: 
sl@0: //No-op. Not needed in this test app.
sl@0: void CSqlSrvTestBurInterface::GetBackUpListL(TSecureId, TDriveNumber, RArray<HBufC*>&)
sl@0: 	{
sl@0: 	TEST(EFalse);
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: void CreateTestEnv()
sl@0:     {
sl@0: 	TheScheduler = new CActiveScheduler;
sl@0: 	TEST(TheScheduler != NULL);
sl@0: 	CActiveScheduler::Install(TheScheduler);
sl@0: 	
sl@0: 	TheSqlSrvTestBurInterface = CSqlSrvTestBurInterface::New();
sl@0: 	TEST(TheSqlSrvTestBurInterface != NULL);
sl@0: 	
sl@0: 	TInt err = RProperty::Define(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey, 0);  
sl@0: 	TEST(err == KErrNone || err == KErrAlreadyExists);
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID			PDS-SQL-UT-4233
sl@0: @SYMTestCaseDesc		CSqlBurEventMonitor object creation - OOM test
sl@0: 						The test runs CSqlBurEventMonitor::NewL() in an OOM loop.
sl@0: @SYMTestActions			CSqlBurEventMonitor object creation - OOM test
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMTestPriority		High
sl@0: */
sl@0: void SqlBurEventMonitorOomTest()
sl@0: 	{
sl@0: 	TInt err = KErrNoMemory;
sl@0: 	TInt failingAllocationNo = 0;
sl@0: 	TheTest.Printf(_L("Iteration:\r\n"));
sl@0: 	while(err == KErrNoMemory)
sl@0: 		{
sl@0: 		TheTest.Printf(_L(" %d"), ++failingAllocationNo);
sl@0: 		
sl@0: 		TInt startProcessHandleCount;
sl@0: 		TInt startThreadHandleCount;
sl@0: 		RThread().HandleCount(startProcessHandleCount, startThreadHandleCount);
sl@0: 		__UHEAP_MARK;
sl@0: 		__UHEAP_SETBURSTFAIL(RAllocator::EBurstFailNext, failingAllocationNo, KBurstRate);
sl@0: 
sl@0: 		CSqlBurEventMonitor* monitor = NULL;
sl@0: 		TRAP(err, monitor = CSqlBurEventMonitor::NewL(*TheSqlSrvTestBurInterface));
sl@0: 		delete monitor;
sl@0: 		
sl@0: 		__UHEAP_RESET;
sl@0: 		__UHEAP_MARKEND;
sl@0: 		TInt endProcessHandleCount;
sl@0: 		TInt endThreadHandleCount;
sl@0: 		RThread().HandleCount(endProcessHandleCount, endThreadHandleCount);
sl@0: 		
sl@0: 		TEST2(startProcessHandleCount, endProcessHandleCount);
sl@0: 		TEST2(startThreadHandleCount, endThreadHandleCount);
sl@0: 		}
sl@0: 	TEST2(err, KErrNone);
sl@0: 	TheTest.Printf(_L("\r\n=== OOM Test succeeded at heap failure rate of %d ===\r\n"), failingAllocationNo);
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID			PDS-SQL-UT-4234
sl@0: @SYMTestCaseDesc		CSqlBurEventMonitor functional test
sl@0: 						The test sets the backup & restore property status and then checks
sl@0: 						how the backup & restore property monitor (CSqlBurEventMonitor) reacts to the event.
sl@0: @SYMTestActions			CSqlBurEventMonitor functional test
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMTestPriority		High
sl@0: */
sl@0: void SqlBurEventMonitorFunctionalTest()
sl@0: 	{
sl@0: 	CSqlBurEventMonitor* monitor = NULL;
sl@0: 	TRAPD(err, monitor = CSqlBurEventMonitor::NewL(*TheSqlSrvTestBurInterface));
sl@0: 	TEST2(err, KErrNone);
sl@0: 	TEST(!monitor->ActiveBackupClient());
sl@0: 	TEST(!monitor->SqlBurCallback());
sl@0: 	//Set the property to conn::EBURBackupFull, conn::EBURBackupPartial, conn::EBURRestoreFull, conn::EBURRestorePartial, 
sl@0: 	//then start the scheduler. CSqlBurEventMonitor::RunL() gets called and 
sl@0: 	//CSqlBurCallback and CActiveBackupClient interfaces get created (if the interfaces do exist, they are destroyed first).
sl@0: 	TInt burPropertyStatus[] = {conn::EBURBackupFull, conn::EBURBackupPartial, conn::EBURRestoreFull, conn::EBURRestorePartial};
sl@0: 	for(TInt i=0;i<(sizeof(burPropertyStatus)/sizeof(burPropertyStatus[0]));++i)
sl@0: 		{
sl@0: 		err = RProperty::Set(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey, burPropertyStatus[i]);
sl@0: 		TEST2(err, KErrNone);
sl@0: 		TestModeSqlBurError = KErrNone;
sl@0: 		if(i > 0)
sl@0: 			{
sl@0: 			__UHEAP_MARK;
sl@0: 			}
sl@0: 		CActiveScheduler::Start();
sl@0: 		if(i > 0)
sl@0: 			{
sl@0: 			__UHEAP_MARKEND;
sl@0: 			}
sl@0: 		TEST2(TestModeSqlBurError, KErrNone);
sl@0: 		TEST(monitor->ActiveBackupClient() != NULL);
sl@0: 		TEST(monitor->SqlBurCallback() != NULL);
sl@0: 		}
sl@0: 	//Set the property to conn::EBURUnset, start the scheduler. CSqlBurEventMonitor::RunL() gets called 
sl@0: 	//and CSqlBurCallback interface gets destroyed.
sl@0: 	err = RProperty::Set(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey, conn::EBURUnset);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	TestModeSqlBurError = KErrNone;
sl@0: 	CActiveScheduler::Start();
sl@0: 	TEST2(TestModeSqlBurError, KErrNone);
sl@0: 	TEST(!monitor->ActiveBackupClient());
sl@0: 	TEST(!monitor->SqlBurCallback());
sl@0: 	//Set the property to conn::EBURNormal, start the scheduler. CSqlBurEventMonitor::RunL() gets called. 
sl@0: 	//CSqlBurCallback interface has been destroyed alread. No memory deallocations should be made during this call.
sl@0: 	err = RProperty::Set(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey, conn::EBURNormal);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	__UHEAP_MARK;
sl@0: 	TestModeSqlBurError = KErrNone;
sl@0: 	CActiveScheduler::Start();
sl@0: 	TEST2(TestModeSqlBurError, KErrNone);
sl@0: 	TEST(!monitor->ActiveBackupClient());
sl@0: 	TEST(!monitor->SqlBurCallback());
sl@0: 	__UHEAP_MARKEND;
sl@0: 	//Set the property, then delete it. CSqlBurEventMonitor::RunL() should get called, but the call should
sl@0: 	//fail because the property does not exist.
sl@0: 	err = RProperty::Set(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey, conn::EBURBackupFull);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	err = RProperty::Delete(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey);  
sl@0: 	TEST2(err, KErrNone);
sl@0: 	__UHEAP_MARK;
sl@0: 	TestModeSqlBurError = KErrNone;
sl@0: 	CActiveScheduler::Start();
sl@0: 	TEST2(TestModeSqlBurError, KErrNotFound);
sl@0: 	TEST(!monitor->ActiveBackupClient());
sl@0: 	TEST(!monitor->SqlBurCallback());
sl@0: 	__UHEAP_MARKEND;
sl@0: 	//Restore the property
sl@0: 	err = RProperty::Define(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey, 0);  
sl@0: 	TEST2(err, KErrNone);
sl@0: 	//
sl@0: 	delete monitor;
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID			PDS-SQL-UT-4235
sl@0: @SYMTestCaseDesc		CSqlBurEventMonitor::RunL() - OOM test
sl@0: 						The test sets the backup & restore property status and then checks
sl@0: 						how the backup & restore property monitor (CSqlBurEventMonitor) reacts to the event.
sl@0: 						The test is performed in an OOM loop.
sl@0: @SYMTestActions			CSqlBurEventMonitor::RunL() - OOM test
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMTestPriority		High
sl@0: */
sl@0: void SqlBurEventMonitorRunOomTest()
sl@0: 	{
sl@0: 	CSqlBurEventMonitor* monitor = NULL;
sl@0: 	TRAPD(err, monitor = CSqlBurEventMonitor::NewL(*TheSqlSrvTestBurInterface));
sl@0: 	
sl@0: 	err = KErrNoMemory;
sl@0: 	TInt failingAllocationNo = 0;
sl@0: 	TheTest.Printf(_L("Iteration:\r\n"));
sl@0: 	while(err == KErrNoMemory)
sl@0: 		{
sl@0: 		TheTest.Printf(_L(" %d"), ++failingAllocationNo);
sl@0: 		
sl@0: 		TInt startProcessHandleCount;
sl@0: 		TInt startThreadHandleCount;
sl@0: 		RThread().HandleCount(startProcessHandleCount, startThreadHandleCount);
sl@0: 		__UHEAP_MARK;
sl@0: 		__UHEAP_SETBURSTFAIL(RAllocator::EBurstFailNext, failingAllocationNo, KBurstRate);
sl@0: 
sl@0: 		TEST(!monitor->ActiveBackupClient());
sl@0: 		TEST(!monitor->SqlBurCallback());
sl@0: 		//Set the property, start the scheduler. CSqlBurEventMonitor::RunL() gets called and CSqlBurCallback
sl@0: 		//interface gets created.
sl@0: 		err = RProperty::Set(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey, conn::EBURBackupFull);
sl@0: 		if(err == KErrNone)
sl@0: 			{
sl@0: 			TestModeSqlBurError = KErrNone;
sl@0: 			CActiveScheduler::Start();
sl@0: 			err = TestModeSqlBurError;
sl@0: 			if(err == KErrNone)
sl@0: 				{
sl@0: 				TEST(monitor->ActiveBackupClient() != NULL);
sl@0: 				TEST(monitor->SqlBurCallback() != NULL);
sl@0: 				//Destroy the SQL backup & restore callback
sl@0: 				err = RProperty::Set(KSqlBurPropertyCategoryUid, KSqlBurBackupRestoreKey, conn::EBURNormal);
sl@0: 				TestModeSqlBurError = KErrNone;
sl@0: 				CActiveScheduler::Start();
sl@0: 				err = TestModeSqlBurError;
sl@0: 				if(err == KErrNone)
sl@0: 					{
sl@0: 					TEST(!monitor->ActiveBackupClient());
sl@0: 					TEST(!monitor->SqlBurCallback());
sl@0: 					}
sl@0: 				}
sl@0: 			}
sl@0: 		
sl@0: 		__UHEAP_RESET;
sl@0: 		__UHEAP_MARKEND;
sl@0: 		TInt endProcessHandleCount;
sl@0: 		TInt endThreadHandleCount;
sl@0: 		RThread().HandleCount(endProcessHandleCount, endThreadHandleCount);
sl@0: 		
sl@0: 		TEST2(startProcessHandleCount, endProcessHandleCount);
sl@0: 		TEST2(startThreadHandleCount, endThreadHandleCount);
sl@0: 		}
sl@0: 	TEST2(err, KErrNone);
sl@0: 	TheTest.Printf(_L("\r\n=== OOM Test succeeded at heap failure rate of %d ===\r\n"), failingAllocationNo);
sl@0: 	delete monitor;
sl@0: 	}
sl@0: 
sl@0: void DoTestsL()
sl@0: 	{
sl@0: 	TheTest.Start(_L(" @SYMTestCaseID:PDS-SQL-UT-4233 CSqlBurEventMonitor object creation - OOM test"));
sl@0: 	SqlBurEventMonitorOomTest();
sl@0: 	TheTest.Next(_L(" @SYMTestCaseID:PDS-SQL-UT-4234 CSqlBurEventMonitor functional test"));
sl@0: 	SqlBurEventMonitorFunctionalTest();
sl@0: 	TheTest.Next(_L(" @SYMTestCaseID:PDS-SQL-UT-4235 CSqlBurEventMonitor::RunL() - OOM test"));
sl@0: 	SqlBurEventMonitorRunOomTest();
sl@0: 	}
sl@0: 
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: 	__UHEAP_MARK;
sl@0: 
sl@0: 	CreateTestEnv();
sl@0: 	TRAPD(err, DoTestsL());
sl@0: 	DestroyTestEnv();
sl@0: 	TEST2(err, KErrNone);
sl@0: 
sl@0: 	__UHEAP_MARKEND;
sl@0: 
sl@0: 	TheTest.End();
sl@0: 	TheTest.Close();
sl@0: 
sl@0: 	delete tc;
sl@0: 
sl@0: 	User::Heap().Check();
sl@0: 	return KErrNone;
sl@0: 	}