sl@0: // Copyright (c) 2006-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: #include <e32test.h>
sl@0: #include <bautils.h>
sl@0: #include <sqldb.h>
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: #define UNUSED_VAR(a) (a) = (a)
sl@0: 
sl@0: RTest TheTest(_L("t_sqltrans test"));
sl@0: 
sl@0: _LIT(KTestDir, "c:\\test\\");
sl@0: _LIT(KTestDbName, "c:\\test\\t_sqltrans.db");
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: void DeleteTestFiles()
sl@0: 	{
sl@0: 	RSqlDatabase::Delete(KTestDbName);
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: //Test macros and functions
sl@0: void Check1(TInt aValue, TInt aLine, TBool aPrintThreadName = EFalse)
sl@0: 	{
sl@0: 	if(!aValue)
sl@0: 		{
sl@0: 		DeleteTestFiles();
sl@0: 		if(aPrintThreadName)
sl@0: 			{
sl@0: 			RThread th;
sl@0: 			TName name = th.Name();
sl@0: 			RDebug::Print(_L("*** Thread %S, Line %d\r\n"), &name, aLine);
sl@0: 			}
sl@0: 		else
sl@0: 			{
sl@0: 			RDebug::Print(_L("*** Line %d\r\n"), aLine);
sl@0: 			}
sl@0: 		TheTest(EFalse, aLine);
sl@0: 		}
sl@0: 	}
sl@0: void Check2(TInt aValue, TInt aExpected, TInt aLine, TBool aPrintThreadName = EFalse)
sl@0: 	{
sl@0: 	if(aValue != aExpected)
sl@0: 		{
sl@0: 		DeleteTestFiles();
sl@0: 		if(aPrintThreadName)
sl@0: 			{
sl@0: 			RThread th;
sl@0: 			TName name = th.Name();
sl@0: 			RDebug::Print(_L("*** Thread %S, Line %d Expected error: %d, got: %d\r\n"), &name, aLine, aExpected, aValue);
sl@0: 			}
sl@0: 		else
sl@0: 			{
sl@0: 			RDebug::Print(_L("*** Line %d, Expected error: %d, got: %d\r\n"), aLine, aExpected, aValue);
sl@0: 			}
sl@0: 		TheTest(EFalse, aLine);
sl@0: 		}
sl@0: 	}
sl@0: #define TEST(arg) ::Check1((arg), __LINE__)
sl@0: #define TEST2(aValue, aExpected) ::Check2(aValue, aExpected, __LINE__)
sl@0: #define TTEST(arg) ::Check1((arg), __LINE__, ETrue)
sl@0: #define TTEST2(aValue, aExpected) ::Check2(aValue, aExpected, __LINE__, ETrue)
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: void CreateTestDir()
sl@0:     {
sl@0:     RFs fs;
sl@0: 	TInt err = fs.Connect();
sl@0: 	TEST2(err, KErrNone);
sl@0: 
sl@0: 	err = fs.MkDir(KTestDir);
sl@0: 	TEST(err == KErrNone || err == KErrAlreadyExists);
sl@0: 	
sl@0: 	fs.Close();
sl@0: 	}
sl@0: 
sl@0: ///////////////////////////////////////////////////////////////////////////////////////
sl@0: 
sl@0: _LIT8(KTestSql1, "INSERT INTO A(Id) VALUES(1); INSERT INTO A(Id) VALUES(2);");
sl@0: 
sl@0: const TPtrC8 KSqls[] = {KTestSql1()};
sl@0: 
sl@0: static RCriticalSection ThreadCritSect;
sl@0: static RCriticalSection MainCritSect;
sl@0: 
sl@0: static TInt TheSqlIdx = 0;
sl@0: 
sl@0: _LIT(KPanicCategory, "TransFail");
sl@0: const TInt KPanicCode = 0x1234;
sl@0: 
sl@0: //Test thread function
sl@0: TInt ThreadFunc1(void*)
sl@0: 	{
sl@0: 	__UHEAP_MARK;
sl@0: 	
sl@0: 	CTrapCleanup* tc = CTrapCleanup::New();
sl@0: 	TTEST(tc != NULL);
sl@0: 
sl@0: 	__ASSERT_ALWAYS(TheSqlIdx >= 0 && TheSqlIdx < (TInt)(sizeof(KSqls) / sizeof(KSqls[0])), User::Invariant());
sl@0: 	const TPtrC8 sql = KSqls[TheSqlIdx];
sl@0: 
sl@0: 	//Open test database
sl@0: 	RSqlDatabase db;
sl@0: 	TInt err = db.Open(KTestDbName);
sl@0: 	TTEST2(err, KErrNone);
sl@0: 
sl@0: 	RDebug::Print(_L("---:WorkThread: Begin transaction. Exec SQL...\r\n"));
sl@0: 	
sl@0: 	//Begin a transaction
sl@0: 	_LIT8(KBeginTrans, "BEGIN");
sl@0: 	err = db.Exec(KBeginTrans);
sl@0: 	TTEST(err >= 0);	
sl@0: 
sl@0: 	//Execute the SQL statement(s)
sl@0: 	err = db.Exec(sql);
sl@0: 	TTEST(err >= 0);	
sl@0: 
sl@0: 	RDebug::Print(_L("---:WorkThread: Notify the main thread about the SQL statement execution\r\n"));
sl@0: 	MainCritSect.Signal();
sl@0: 
sl@0: 	RDebug::Print(_L("---:WorkThread: Wait for permisson to continue...\r\n"));
sl@0: 	ThreadCritSect.Wait();
sl@0: 
sl@0: 	User::SetJustInTime(EFalse);	// disable debugger panic handling
sl@0: 
sl@0: 	//Panic current thread without commiting the transaction
sl@0: 	RDebug::Print(_L("---:WorkThread: Panic!\r\n"));
sl@0: 	User::Panic(KPanicCategory, KPanicCode);
sl@0: 
sl@0: 	delete tc;	
sl@0: 	
sl@0: 	__UHEAP_MARKEND;
sl@0: 	
sl@0: 	return KErrNone;		
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID			SYSLIB-SQL-CT-1623
sl@0: @SYMTestCaseDesc		Transaction atomicity test.
sl@0: 						Create a test database with a table. 
sl@0: 						Create a worker thread and make some "insert record" operations in a transaction from 
sl@0: 						that thread. Before commiting the transaction notify the main thread that the
sl@0: 						insert operation completed and wait for a notification from the main thread.
sl@0: 						The main thread notifies the worker thread to panic and checks the test table 
sl@0: 						content. No records should be found there.
sl@0: @SYMTestPriority		High
sl@0: @SYMTestActions			Transaction atomicity test.
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ					REQ5792
sl@0:                         REQ5793
sl@0: */	
sl@0: void TransactionTest1()
sl@0: 	{
sl@0: 	RDebug::Print(_L("+++:MainThread: Create critical sections\r\n"));
sl@0: 	TEST2(ThreadCritSect.CreateLocal(), KErrNone);
sl@0: 	ThreadCritSect.Wait();
sl@0: 	TEST2(MainCritSect.CreateLocal(), KErrNone);
sl@0: 	MainCritSect.Wait();
sl@0: 	
sl@0: 	RDebug::Print(_L("+++:MainThread: Create test database\r\n"));
sl@0: 	(void)RSqlDatabase::Delete(KTestDbName);
sl@0: 	RSqlDatabase db;
sl@0: 	TInt err = db.Create(KTestDbName);
sl@0: 	TEST2(err, KErrNone);
sl@0: 
sl@0: 	RDebug::Print(_L("+++:MainThread: Create a table in the test database\r\n"));
sl@0: 	_LIT8(KCreateSql, "CREATE TABLE A(Id INTEGER)");
sl@0: 	err = db.Exec(KCreateSql);
sl@0: 	TEST(err >= 0);	
sl@0: 	
sl@0: 	db.Close();
sl@0: 
sl@0: 	RDebug::Print(_L("+++:MainThread: Create the worker thread\r\n"));
sl@0: 	_LIT(KThreadName, "WorkThrd");
sl@0: 	RThread thread;
sl@0: 	TheSqlIdx = 0;
sl@0: 	TEST2(thread.Create(KThreadName, &ThreadFunc1, 0x2000, 0x1000, 0x10000, NULL, EOwnerProcess), KErrNone);
sl@0: 	TRequestStatus status;
sl@0: 	thread.Logon(status);
sl@0: 	TEST2(status.Int(), KRequestPending);
sl@0: 	thread.Resume();
sl@0: 
sl@0: 	RDebug::Print(_L("+++:MainThread: Wait SQL statement(s) to be executed...\r\n"));
sl@0: 	MainCritSect.Wait();
sl@0: 
sl@0: 	RDebug::Print(_L("+++:MainThread: Notify the worker thread to panic...\r\n"));
sl@0: 	ThreadCritSect.Signal();
sl@0: 	
sl@0: 	User::WaitForRequest(status);
sl@0: 	User::SetJustInTime(ETrue);	// enable debugger panic handling
sl@0: 
sl@0: 	TEST2(thread.ExitType(), EExitPanic);
sl@0: 	TEST2(thread.ExitReason(), KPanicCode);
sl@0: 	
sl@0: 	thread.Close();
sl@0: 
sl@0: 	RDebug::Print(_L("+++:MainThread: Check the database content...\r\n"));
sl@0: 	err = db.Open(KTestDbName);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	_LIT8(KSelectSql, "SELECT COUNT(*) AS CNT FROM A");
sl@0: 	RSqlStatement stmt;
sl@0: 	err = stmt.Prepare(db, KSelectSql);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	err = stmt.Next();
sl@0: 	TEST2(err, KSqlAtRow);
sl@0: 	TInt val = stmt.ColumnInt(0);
sl@0: 	TEST(val == 0);
sl@0: 	stmt.Close();
sl@0: 	db.Close();
sl@0: 
sl@0: 	err = RSqlDatabase::Delete(KTestDbName);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID			SYSLIB-SQL-CT-1624
sl@0: @SYMTestCaseDesc		Transaction consistency test.
sl@0: 						Create a test database with a table with a field with a CHECK constraint. 
sl@0: 						Try to insert some records in a transaction violating the CHECK constraint.
sl@0: 						The transaction should fail.
sl@0: 						No records should be found in the test table.
sl@0: @SYMTestPriority		High
sl@0: @SYMTestActions			Transaction atomicity test.
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ					REQ5792
sl@0:                         REQ5793
sl@0: */	
sl@0: void TransactionTest2()
sl@0: 	{
sl@0: 	//Create a test database
sl@0: 	(void)RSqlDatabase::Delete(KTestDbName);
sl@0: 	RSqlDatabase db;
sl@0: 	TInt err = db.Create(KTestDbName);
sl@0: 	TEST2(err, KErrNone);
sl@0: 
sl@0: 	//Create a test table
sl@0: 	_LIT8(KCreateSql, "CREATE TABLE A(Id INTEGER, CHECK(Id > 10 AND Id <= 20))");
sl@0: 	err = db.Exec(KCreateSql);
sl@0: 	TEST(err >= 0);	
sl@0: 
sl@0: 	//Begin a transaction
sl@0: 	_LIT8(KBeginTrans, "BEGIN");
sl@0: 	err = db.Exec(KBeginTrans);
sl@0: 	TEST(err >= 0);	
sl@0: 	
sl@0: 	//Exec SQL, viloate constraint.
sl@0: 	_LIT8(KInsertSql, "INSERT INTO A(Id) VALUES(15); INSERT INTO A(Id) VALUES(38);");
sl@0: 	err = db.Exec(KInsertSql);
sl@0: 	TEST2(err, KSqlErrConstraint);
sl@0: 
sl@0: 	//Rollback transaction
sl@0: 	_LIT8(KRollbackTrans, "ROLLBACK");
sl@0: 	err = db.Exec(KRollbackTrans);
sl@0: 	TEST(err >= 0);	
sl@0: 
sl@0: 	//Check the database content
sl@0: 	_LIT8(KSelectSql, "SELECT COUNT(*) AS CNT FROM A");
sl@0: 	RSqlStatement stmt;
sl@0: 	err = stmt.Prepare(db, KSelectSql);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	err = stmt.Next();
sl@0: 	TEST2(err, KSqlAtRow);
sl@0: 	TInt val = stmt.ColumnInt(0);
sl@0: 	TEST2(val, 0);
sl@0: 	stmt.Close();
sl@0: 	
sl@0: 	db.Close();
sl@0: 
sl@0: 	err = RSqlDatabase::Delete(KTestDbName);
sl@0: 	TEST2(err, KErrNone);
sl@0: 	}
sl@0: 
sl@0: void DoTests()
sl@0: 	{
sl@0: 	TheTest.Start(_L(" @SYMTestCaseID:SYSLIB-SQL-CT-1623 Transaction test 1 "));
sl@0: 	TransactionTest1();
sl@0: 
sl@0: 	TheTest.Next(_L(" @SYMTestCaseID:SYSLIB-SQL-CT-1624 Transaction test 2 "));
sl@0: 	TransactionTest2();
sl@0: 	}
sl@0: 
sl@0: TInt E32Main()
sl@0: 	{
sl@0: 	TheTest.Title();
sl@0: 	
sl@0: 	CTrapCleanup* tc = CTrapCleanup::New();
sl@0: 	
sl@0: 	__UHEAP_MARK;
sl@0: 	
sl@0: 	CreateTestDir();
sl@0: 	DeleteTestFiles();
sl@0: 	DoTests();
sl@0: 	DeleteTestFiles();
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: 	}