1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/os/persistentdata/persistentstorage/sql/TEST/t_sqloomutil.cpp Fri Jun 15 03:10:57 2012 +0200
1.3 @@ -0,0 +1,227 @@
1.4 +// Copyright (c) 2005-2009 Nokia Corporation and/or its subsidiary(-ies).
1.5 +// All rights reserved.
1.6 +// This component and the accompanying materials are made available
1.7 +// under the terms of "Eclipse Public License v1.0"
1.8 +// which accompanies this distribution, and is available
1.9 +// at the URL "http://www.eclipse.org/legal/epl-v10.html".
1.10 +//
1.11 +// Initial Contributors:
1.12 +// Nokia Corporation - initial contribution.
1.13 +//
1.14 +// Contributors:
1.15 +//
1.16 +// Description:
1.17 +//
1.18 +
1.19 +#include "t_sqloom.h"
1.20 +#include "SqlResourceTester.h" //TSqlResourceTester
1.21 +
1.22 +
1.23 +TInt TheHandleCount1B = 0;
1.24 +TInt TheHandleCount2B = 0;
1.25 +TInt TheAllocatedCellsCountB = 0;
1.26 +TOomTestType TheOomTestType[2] = {EClientSideTest, EServerSideTest};
1.27 +
1.28 +///////////////////////////////////////////////////////////////////////////////////////
1.29 +/////////////// OOM tests - utility functions //////////////////////////
1.30 +///////////////////////////////////////////////////////////////////////////////////////
1.31 +
1.32 +void DeleteTestFiles()
1.33 + {
1.34 + (void)RSqlDatabase::Delete(KSecureAttachDb);
1.35 + (void)RSqlDatabase::Delete(KAttachDb);
1.36 + (void)RSqlDatabase::Delete(KSecureTestDb);
1.37 + (void)RSqlDatabase::Delete(KTestDb);
1.38 + (void)RSqlDatabase::Delete(KSecureDb2);
1.39 + (void)RSqlDatabase::Delete(KTestDb2);
1.40 + (void)RSqlDatabase::Delete(KPrivateTestDb);
1.41 + }
1.42 +
1.43 +void Check(TInt64 aValue, TInt aLine)
1.44 + {
1.45 + if(!aValue)
1.46 + {
1.47 + DeleteTestFiles();
1.48 + TheTest(EFalse, aLine);
1.49 + }
1.50 + }
1.51 +
1.52 +void Check(TInt64 aValue, TInt64 aExpected, TInt aLine)
1.53 + {
1.54 + if(aValue != aExpected)
1.55 + {
1.56 + DeleteTestFiles();
1.57 + RDebug::Print(_L("*** Expected error: %ld, got: %ld\r\n"), aExpected, aValue);
1.58 + TheTest(EFalse, aLine);
1.59 + }
1.60 + }
1.61 +
1.62 +void CreateTestDir()
1.63 + {
1.64 + RFs fs;
1.65 + TInt err = fs.Connect();
1.66 + TEST2(err, KErrNone);
1.67 +
1.68 + err = fs.MkDir(KTestDir);
1.69 + TEST(err == KErrNone || err == KErrAlreadyExists);
1.70 +
1.71 + err = fs.CreatePrivatePath(EDriveC);
1.72 + TEST(err == KErrNone || err == KErrAlreadyExists);
1.73 +
1.74 + fs.Close();
1.75 + }
1.76 +
1.77 +void PrepareAttachFromHandle()
1.78 + {
1.79 + (void)RSqlDatabase::Delete(KAttachDb);
1.80 + (void)RSqlDatabase::Delete(KPrivateTestDb);
1.81 +
1.82 + RSqlDatabase db;
1.83 + TInt err = db.Create(KAttachDb);
1.84 + db.Close();
1.85 + TEST2(err, KErrNone);
1.86 +
1.87 + RFs fs;
1.88 + err = fs.Connect();
1.89 + TEST2(err, KErrNone);
1.90 +
1.91 + err = BaflUtils::CopyFile(fs, KAttachDb, KPrivateTestDb);
1.92 + TEST2(err, KErrNone);
1.93 + }
1.94 +
1.95 +void MarkHandles()
1.96 + {
1.97 + RThread().HandleCount(TheHandleCount1B, TheHandleCount2B);
1.98 + }
1.99 +
1.100 +void CheckHandles()
1.101 + {
1.102 + TInt endHandleCount1E;
1.103 + TInt endHandleCount2E;
1.104 +
1.105 + RThread().HandleCount(endHandleCount1E, endHandleCount2E);
1.106 +
1.107 + TEST(TheHandleCount1B == endHandleCount1E);
1.108 + TEST(TheHandleCount2B == endHandleCount2E);
1.109 + }
1.110 +
1.111 +void MarkAllocatedCells()
1.112 + {
1.113 + TheAllocatedCellsCountB = User::CountAllocCells();
1.114 + }
1.115 +
1.116 +void CheckAllocatedCells()
1.117 + {
1.118 + TInt allocatedCellsCountE = User::CountAllocCells();
1.119 + TEST(allocatedCellsCountE == TheAllocatedCellsCountB);
1.120 + }
1.121 +
1.122 +void PrintEndOfOomTest(TOomTestType aOomTestType, TInt aFailingAllocationNo)
1.123 + {
1.124 + _LIT(KClientSide, "Client side");
1.125 + _LIT(KServerSide, "Server side");
1.126 + RDebug::Print(_L("=== %S OOM Test succeeded at heap failure rate of %d ===\r\n"),
1.127 + aOomTestType == EClientSideTest ? &KClientSide() : &KServerSide(),
1.128 + aFailingAllocationNo);
1.129 + }
1.130 +
1.131 +//If aDelayed is true, then the SQL server will delay the heap failure simulation until the database is opened.
1.132 +void SetDbHeapFailure(TOomTestType aOomTestType, TInt aFailingAllocationNo, TBool aDelayed)
1.133 + {
1.134 + const TInt KDelayedDbHeapFailureMask = 0x1000;
1.135 + if(aOomTestType == EClientSideTest)
1.136 + {
1.137 + __UHEAP_SETFAIL(RHeap::EDeterministic, aFailingAllocationNo);
1.138 + }
1.139 + else
1.140 + {
1.141 + TSqlResourceTester::SetDbHeapFailure(RHeap::EDeterministic | (aDelayed ? KDelayedDbHeapFailureMask : 0), aFailingAllocationNo);
1.142 + }
1.143 + }
1.144 +
1.145 +void ResetDbHeapFailure(TOomTestType aOomTestType)
1.146 + {
1.147 + if(aOomTestType == EClientSideTest)
1.148 + {
1.149 + __UHEAP_SETFAIL(RHeap::ENone, 0);
1.150 + }
1.151 + else
1.152 + {
1.153 + TSqlResourceTester::SetDbHeapFailure(RHeap::ENone, 0);
1.154 + }
1.155 + }
1.156 +
1.157 +void SetHeapFailure(TOomTestType aOomTestType, TInt aFailingAllocationNo)
1.158 + {
1.159 + if(aOomTestType == EClientSideTest)
1.160 + {
1.161 + __UHEAP_SETFAIL(RHeap::EDeterministic, aFailingAllocationNo);
1.162 + }
1.163 + else
1.164 + {
1.165 + TSqlResourceTester::Mark();
1.166 + TSqlResourceTester::SetHeapFailure(RHeap::EDeterministic, aFailingAllocationNo);
1.167 + }
1.168 + }
1.169 +
1.170 +void ResetHeapFailure(TOomTestType aOomTestType)
1.171 + {
1.172 + if(aOomTestType == EClientSideTest)
1.173 + {
1.174 + __UHEAP_SETFAIL(RHeap::ENone, 0);
1.175 + }
1.176 + else
1.177 + {
1.178 + TSqlResourceTester::SetHeapFailure(RHeap::ENone, 0);
1.179 + TSqlResourceTester::Check();
1.180 + }
1.181 + }
1.182 +
1.183 +void CreateTestSecurityPolicy(RSqlSecurityPolicy& aSecurityPolicy)
1.184 + {
1.185 + TInt err = aSecurityPolicy.Create(KDefaultPolicy);
1.186 + TEST2(err, KErrNone);
1.187 +
1.188 + err = aSecurityPolicy.SetDbPolicy(RSqlSecurityPolicy::ESchemaPolicy, KPolicy1);
1.189 + TEST2(err, KErrNone);
1.190 + err = aSecurityPolicy.SetDbPolicy(RSqlSecurityPolicy::EWritePolicy, KPolicy3);
1.191 + TEST2(err, KErrNone);
1.192 +
1.193 + err = aSecurityPolicy.SetPolicy(RSqlSecurityPolicy::ETable, KTableName1, RSqlSecurityPolicy::EReadPolicy, KPolicy1);
1.194 + TEST2(err, KErrNone);
1.195 + err = aSecurityPolicy.SetPolicy(RSqlSecurityPolicy::ETable, KTableName1, RSqlSecurityPolicy::EWritePolicy, KPolicy2);
1.196 + TEST2(err, KErrNone);
1.197 +
1.198 + err = aSecurityPolicy.SetPolicy(RSqlSecurityPolicy::ETable, KTableName2, RSqlSecurityPolicy::EReadPolicy, KPolicy1);
1.199 + TEST2(err, KErrNone);
1.200 + err = aSecurityPolicy.SetPolicy(RSqlSecurityPolicy::ETable, KTableName3, RSqlSecurityPolicy::EReadPolicy, KPolicy2);
1.201 + TEST2(err, KErrNone);
1.202 +
1.203 + err = aSecurityPolicy.SetPolicy(RSqlSecurityPolicy::ETable, KTableName2, RSqlSecurityPolicy::EWritePolicy, KPolicy1);
1.204 + TEST2(err, KErrNone);
1.205 +
1.206 + err = aSecurityPolicy.SetPolicy(RSqlSecurityPolicy::ETable, KTableName3, RSqlSecurityPolicy::EWritePolicy, KPolicy3);
1.207 + TEST2(err, KErrNone);
1.208 + }
1.209 +
1.210 +void CreateAttachDb()
1.211 + {
1.212 + RSqlDatabase db;
1.213 +
1.214 + TInt err = db.Create(KAttachDb);
1.215 + TEST2(err, KErrNone);
1.216 + err = db.Exec(_L("CREATE TABLE Aab(F1 INTEGER, F2 TEXT)"));
1.217 + TEST(err >= 0);
1.218 + db.Close();
1.219 +
1.220 + RSqlSecurityPolicy securityPolicy;
1.221 + CreateTestSecurityPolicy(securityPolicy);
1.222 +
1.223 + err = db.Create(KSecureAttachDb, securityPolicy);
1.224 + TEST2(err, KErrNone);
1.225 + err = db.Exec(_L("CREATE TABLE Aab(F1 INTEGER, F2 TEXT)"));
1.226 + TEST(err >= 0);
1.227 + db.Close();
1.228 +
1.229 + securityPolicy.Close();
1.230 + }