// 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 <e32std.h>

#include <f32file.h>

#include <e32test.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h> 

#include <hal.h>

#include <utf.h>				//CnvUtfConverter

#include "sqlite3.h"

#include "SqliteSymbian.h"

#define UNUSED_VAR(a) a = (a)

//Forward declaration

struct Mem;

static RTest 			TheTest(_L ("t_sqlite.exe"));

static CTrapCleanup*	TheTrapCleanup = NULL;

static RFs 				TheFs;

//File-local buffer used for converted to UTF16 (from UTF8) strings or

//for a temporary file name buffer.

static TUint16 TheFileName[KMaxFileName + 1];

_LIT(KTestDir, "c:\\test\\");

const char KDbName1[] = "c:\\test\\t_sqlite1.db";

const char KDbName2[] = "c:\\test\\t_sqlite2.db";

_LIT(KContactsFile, "c:\\test\\Contacts.DB");

sqlite3 *TheDb1 = NULL, *TheDb2 = NULL;

static TBuf<4096> TheBuf1;

static TBuf<4096> TheBuf2;

static TBuf8<512> TheBuf3;

static TInt TheCmpFCallCnt = 0;

static TInt TheCmpCCallCnt = 0;

_LIT(KSimpleContactsSqlFile, "z:\\test\\add_simple_contacts.sql");

//In order to be able to compile the test, the following variables are defined (used inside the OS porting layer, when _SQLPROFILER macro is defined)

#ifdef _SQLPROFILER

TInt TheSqlSrvProfilerFileRead = 0;

TInt TheSqlSrvProfilerFileWrite = 0;

TInt TheSqlSrvProfilerFileSync = 0;

TInt TheSqlSrvProfilerFileSetSize = 0;

#endif

///////////////////////////////////////////////////////////////////////////////////////

//Creates TPtrC object which points to the unicode presentation of aData.

//aData argument is expected to point to UTF8 encoded, zero terminated string.

//The function returns a TPtrC, non-zero terminated object pointing to the unicode presentation of aData.

//If the length of the returned TPtrC object is 0 - it means that unicode conversion of aData failed, 

//because the aData is too long or is NULL. 

//Max allowed aData length is KMaxFileName (excluding terminating 0 character).

static TPtrC ConvertToUtf16(const char *aData)

	{

	TPtrC ptr(0, 0);

	if(aData)

		{

		TInt len = mbstowcs(reinterpret_cast <wchar_t*> (TheFileName), aData, KMaxFileName + 1);

		//Check converted string length. If it is longer than KMaxFileName characters, then aData is too long.

		if(len > 0 && len <= KMaxFileName)

			{

			ptr.Set(TheFileName, len);

			}

		}

	return ptr;

	}

///////////////////////////////////////////////////////////////////////////////////////

static void DeleteTestFiles()

	{

   	sqlite3_close(TheDb2);

   	sqlite3_close(TheDb1);

	TheFs.Delete(ConvertToUtf16(KDbName2));

	TheFs.Delete(ConvertToUtf16(KDbName1));

	TheFs.Delete(KContactsFile);

	}

///////////////////////////////////////////////////////////////////////////////////////

///////////////////////////////////////////////////////////////////////////////////////

//Test macros and functions

static void Check(TInt aValue, TInt aLine)

	{

	if(!aValue)

		{

		DeleteTestFiles();

		TheTest(EFalse, aLine);

		}

	}

static void Check(TInt aValue, TInt aExpected, TInt aLine)

	{

	if(aValue != aExpected)

		{

		DeleteTestFiles();

		RDebug::Print(_L("*** Expected error: %d, got: %d\r\n"), aExpected, aValue);

		TheTest(EFalse, aLine);

		}

	}

#define TEST(arg) ::Check((arg), __LINE__)

#define TEST2(aValue, aExpected) ::Check(aValue, aExpected, __LINE__)

///////////////////////////////////////////////////////////////////////////////////////

static void CreatePrivateDirs()

	{

	RFs fs;

	TEST2(fs.Connect(), KErrNone);

	for(TInt i=0;i<('Z'-'A');++i)

		{

		fs.CreatePrivatePath(i);

		}

	fs.Close();

	}

///////////////////////////////////////////////////////////////////////////////////////

static void PrintConfig(const TDesC& aDbFilePath)

	{

	RDebug::Print(_L("================= Configuration ================\r\n"));	

	RDebug::Print(_L("Cache page size %dK, pages %d, total %dK\r\n"), SQLITE_DEFAULT_PAGE_SIZE/1024, SQLITE_DEFAULT_CACHE_SIZE, SQLITE_DEFAULT_PAGE_SIZE * SQLITE_DEFAULT_CACHE_SIZE/1024);	

	RDebug::Print(_L("Temp cache page size %dK, pages %d, total %dK\r\n"), SQLITE_DEFAULT_PAGE_SIZE/1024, SQLITE_DEFAULT_TEMP_CACHE_SIZE, SQLITE_DEFAULT_PAGE_SIZE * SQLITE_DEFAULT_TEMP_CACHE_SIZE/1024);	

	_LIT(K1, "On");

	_LIT(K2, "Off");

	RDebug::Print(_L("Autovacuum: %S\r\n"), SQLITE_DEFAULT_AUTOVACUUM ? &K1 : &K2);	

	#ifdef SQLITE_DEBUG

		RDebug::Print(_L("Debug: On\r\n"));	

	#else

		RDebug::Print(_L("Debug: Off\r\n"));	

	#endif

	RDebug::Print(_L("Db file: %S\r\n"), &aDbFilePath);

	RDebug::Print(_L("================================================\r\n"));	

	}

///////////////////////////////////////////////////////////////////////////////////////

static TInt callback(void */*NotUsed*/, TInt argc, char **argv, char **azColName)

	{

	TInt i;

	for(i=0; i<argc; i++)

		{

		TPtrC8 colName((const TUint8*)azColName[i]);

		TPtrC8 colVal((const TUint8*)(argv[i] ? argv[i] : "NULL"));

		TheBuf2.Copy(colName);

		TheBuf2.Append(_L(" = "));

		TheBuf1.Copy(colVal);

		TheBuf2.Append(TheBuf1);

    	RDebug::Print(_L("%S\r\n"), &TheBuf2);

		}

  	RDebug::Print(_L("\n"));

  	return 0;

	}

static TInt Compare(void*, TInt size1, const void* p1, TInt size2, const void* p2)

	{

	TPtrC8 ptr1((TUint8*)p1, size1);

	TPtrC8 ptr2((TUint8*)p2, size2);

	TInt res = ptr1.CompareF(ptr2);

	return res;		

	}

static TInt Compare2(void*, TInt size1, const void* p1, TInt size2, const void* p2)

	{

	TPtrC8 ptr1((TUint8*)p1, size1);

	TPtrC8 ptr2((TUint8*)p2, size2);

	TInt res = -ptr1.Compare(ptr2);

	return res;		

	}

static void Func(sqlite3_context* ctx, TInt nCnt, sqlite3_value** pp)

	{

	for(TInt i=0;i<nCnt;++i)

		{

		TInt v = sqlite3_value_int(*pp++);

		v = v;

		}

	sqlite3_result_int(ctx, 564);		

	}

///////////////////////////////////////////////////////////////////////////////////////

//Create database, table, view, INSERT, SELECT, DELETE sql statements, register collations, 

//user defined functions, CREATE INDEX, CREATE VIEW sql statements, GROUP BY, HAVING, EXCEPT,...

static void DoTests1()

	{

	char *zErrMsg = 0;

	TInt rc;

	rc = sqlite3_open(KDbName1, &TheDb1);

  	if(rc)

  		{

  		TPtrC p = ConvertToUtf16(sqlite3_errmsg(TheDb1));

  		RDebug::Print(_L("Can't open database, err %d, msg: %S\n"), rc, &p);

    	TEST(0);

  		}

	rc = sqlite3_create_collation(TheDb1, "Clt1", SQLITE_UTF8, NULL, &Compare);

  	if(rc)

  		{

  		TPtrC p = ConvertToUtf16(sqlite3_errmsg(TheDb1));

  		RDebug::Print(_L("Can't create collation, err %d, msg: %S\n"), rc, &p);

    	TEST(0);

  		}

	rc = sqlite3_create_collation(TheDb1, "Clt2", SQLITE_UTF8, NULL, &Compare2);

  	if(rc)

  		{

  		TPtrC p = ConvertToUtf16(sqlite3_errmsg(TheDb1));

  		RDebug::Print(_L("Can't create collation, err %d, msg: %S\n"), rc, &p);

    	TEST(0);

  		}

	rc = sqlite3_create_function(TheDb1, "Func", 2, SQLITE_UTF8, NULL, Func, NULL, NULL);

  	if(rc)

  		{

  		TPtrC p = ConvertToUtf16(sqlite3_errmsg(TheDb1));

  		RDebug::Print(_L("Can't create UDF, err %d, msg: %S\n"), rc, &p);

    	TEST(0);

  		}

	rc = sqlite3_exec(TheDb1, "\

	             BEGIN TRANSACTION;\

	             CREATE TABLE Pn(Id INTEGER PRIMARY KEY AUTOINCREMENT,\

	                             Name TEXT NOT NULL,\

	                             Surname TEXT NOT NULL COLLATE Clt1);\

	             CREATE INDEX PnSurname ON Pn(Surname);\

	             CREATE TABLE Addr(Id INTEGER PRIMARY KEY AUTOINCREMENT,\

	                               PnId INTEGER,\

	                               CityId INTEGER,\

	                               Location TEXT DEFAULT 'address');\

	             CREATE INDEX AddrPnId ON Addr(PnId);\

	             CREATE INDEX AddrCityId ON Addr(CityId);\

	             CREATE TABLE City(ID INTEGER PRIMARY KEY AUTOINCREMENT, Name TEXT Default 'None');\

	             CREATE TABLE Shop(Id INTEGER, Name TEXT);\

	             CREATE TABLE Sales(ShopId INTEGER, M MONEY, D TIMESTAMP, D1 DATE, T1 TIME);\

	             CREATE TRIGGER TrgPnDel BEFORE DELETE ON Pn \

	             BEGIN \

					SELECT Func(2,8);\

	                SELECT CASE WHEN ((SELECT COUNT(*) FROM Addr WHERE Addr.PnId = old.Id) > 0) \

	                            THEN RAISE (ABORT, 'Foreign key constraint violation!!!') \

	                END;\

	             END;\

	             CREATE VIEW V1 AS SELECT * FROM Pn;\

	             CREATE VIEW V2 AS SELECT * FROM Addr;\

	             CREATE VIEW V3 AS SELECT * FROM Sales;\

	             CREATE VIEW V4 AS SELECT * FROM Shop;\

	             COMMIT TRANSACTION;", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

    	TEST(0);

		}

	rc = sqlite3_exec(TheDb1, 

	"BEGIN TRANSACTION;\

	 INSERT INTO Pn(Name, Surname) VALUES('Aaaa1', 'Aaaa2');\

	 INSERT INTO Pn(Name, Surname) VALUES('Bbbb1', 'Bbbb2');\

	 INSERT INTO Pn(Name, Surname) VALUES('Cccc1', 'Cccc2');\

	 INSERT INTO Pn(Name, Surname) VALUES('Dddd1', 'Dddd2');\

	 INSERT INTO City(Name) VALUES('London');\

	 INSERT INTO City(Name) VALUES('Manchester');\

	 INSERT INTO Addr(PnId, CityId, Location) SELECT Pn.Id, City.Id, 'Here' FROM Pn, City WHERE Pn.Surname = 'Cccc2' AND City.Name = 'London';\

	 INSERT INTO Addr(PnId, CityId, Location) SELECT Pn.Id, City.Id, 'There' FROM Pn, City WHERE Pn.Surname = 'Bbbb2' AND City.Name = 'Manchester';\

	 INSERT INTO Shop(Id, Name) VALUES(1, 'Shop-1');\

	 INSERT INTO Shop(Id, Name) VALUES(2, 'Shop-2');\

	 INSERT INTO Shop(Id, Name) VALUES(3, 'Shop-3');\

	 INSERT INTO Sales(ShopId, M, D, D1, T1) VALUES(1, 123.0, '2005-01-01', CURRENT_DATE, '12:34:17');\

	 INSERT INTO Sales(ShopId, M, D, D1, T1) VALUES(2, 100.0, '2005-01-01', '2005-01-27', '12:34:18');\

	 INSERT INTO Sales(ShopId, M, D, D1, T1) VALUES(2, 200.0, '2005-01-02', '2005-01-28', '12:34:19');\

	 INSERT INTO Sales(ShopId, M, D, D1, T1) VALUES(3, 200.0, '2005-01-03', '2005-01-29', '12:34:23');\

	 COMMIT TRANSACTION;", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

    	TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT COUNT(*) FROM Shop", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

    	TEST(0);

		}

	for(TInt i=0;i<500;++i)

		{

		_LIT8(KSqlStmt, "INSERT INTO Shop(Id, Name) VALUES(%d, 'Shop-%d')\x0");

		TheBuf3.Format(KSqlStmt, i+1, i+1);

		rc = sqlite3_exec(TheDb1, (const char*)TheBuf3.Ptr(), callback, 0, &zErrMsg);

		if(rc != SQLITE_OK)

			{

	  		TPtrC p = ConvertToUtf16(zErrMsg);

	  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

	    	sqlite3_free(zErrMsg);

    		TEST(0);

			}

		}

	_LIT8(KSqlStmt2, "DELETE FROM Shop WHERE Id > 400\x0");

	rc = sqlite3_exec(TheDb1, (const char*)KSqlStmt2().Ptr(), callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	_LIT8(KSqlStmt3, "SELECT COUNT(*) FROM Shop\x0");

	rc = sqlite3_exec(TheDb1, (const char*)KSqlStmt3().Ptr(), callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT Pn.Surname, Addr.Location, City.Name FROM Pn \

	      INNER JOIN Addr ON Pn.Id = Addr.PnId \

	      INNER JOIN City ON Addr.CityId = City.Id \

	      ORDER BY Surname COLLATE Clt2", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT * FROM Addr", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT * FROM Sales WHERE D1 > '2005-08-27'", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT V1.* FROM V1 EXCEPT SELECT V1.* FROM V1 WHERE V1.Id = 2", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT V3.D, SUM(V3.M) FROM V3 GROUP BY V3.D HAVING SUM(V3.M) > 210.0", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT V4.Name, SUM(V3.M) FROM V4, V3 WHERE V4.Id = V3.ShopId GROUP BY V4.Id", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	sqlite3_close(TheDb1);

	TheFs.Delete(ConvertToUtf16(KDbName1));

	}

//Generic SQL tests.

static void DoTests2()

	{

	TheFs.Delete(ConvertToUtf16(KDbName1));

	char *zErrMsg = 0;

	TInt rc;

	rc = sqlite3_open(KDbName1, &TheDb1);

  	if(rc != SQLITE_OK)

  		{

  		TPtrC p = ConvertToUtf16(sqlite3_errmsg(TheDb1));

  		RDebug::Print(_L("Can't open database, err %d, msg: %S\n"), rc, &p);

    	TEST(0);

  		}

	rc = sqlite3_exec(TheDb1, "PRAGMA auto_vacuum = 1", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "\

	             BEGIN TRANSACTION;\

	             CREATE TABLE Shop(Id INTEGER, Name TEXT);\

	             CREATE INDEX ShopName ON Shop(Name);\

	             COMMIT TRANSACTION;", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "BEGIN TRANSACTION", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	for(TInt i=0;i<1000;++i)

		{

		_LIT8(KSqlStmt, "INSERT INTO Shop(Id, Name) VALUES(%d, 'Shop-%d')\x0");

		TheBuf3.Format(KSqlStmt, i, i);

		rc = sqlite3_exec(TheDb1, (const char*)TheBuf3.Ptr(), callback, 0, &zErrMsg);

		if(rc != SQLITE_OK)

			{

	  		TPtrC p = ConvertToUtf16(zErrMsg);

	  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

	    	sqlite3_free(zErrMsg);

	   		TEST(0);

			}

		}

	rc = sqlite3_exec(TheDb1, "COMMIT TRANSACTION", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT COUNT(*) FROM Shop", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "BEGIN TRANSACTION", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "DELETE FROM Shop WHERE Id > 100", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "COMMIT TRANSACTION", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	rc = sqlite3_exec(TheDb1, "SELECT COUNT(*) FROM Shop", callback, 0, &zErrMsg);

	if(rc != SQLITE_OK)

		{

  		TPtrC p = ConvertToUtf16(zErrMsg);

  		RDebug::Print(_L("SQL error %d, msg: %S\n"), rc, &p);

    	sqlite3_free(zErrMsg);

   		TEST(0);

		}

	sqlite3_close(TheDb1);

	TheFs.Delete(ConvertToUtf16(KDbName1));

	}