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: #include <s32file.h>
sl@0: #include <e32test.h>
sl@0: 
sl@0: #define UNUSED_VAR(a) a = a
sl@0: 
sl@0: const TInt KTestCleanupStack=0x20;
sl@0: const TPtrC KTestDir=_L("\\STOR-TST\\T_OOM\\");
sl@0: 
sl@0: #ifdef _DEBUG
sl@0: const TPtrC desOriginalReverted2(_S("original/reverted A"),19);
sl@0: const TPtrC desOriginalReverted3(_S("original/reverted B"),19);
sl@0: const TPtrC desNewOverwritten2(_S("new/overwritten X"),17);
sl@0: const TPtrC desNewOverwritten3(_S("new/overwritten Y"),17);
sl@0: const TPtrC alphabet(_S("abcdefghijklmnopqrstuvwxyz"),26);
sl@0: LOCAL_D CFileStore* store;
sl@0: RStoreWriteStream out;
sl@0: RStoreReadStream in;
sl@0: TInt KMemoryAllocsInTestFunction=1;
sl@0: #endif
sl@0: 
sl@0: LOCAL_D CTrapCleanup* TheTrapCleanup;
sl@0: LOCAL_D RTest test(_L("t_storoom"));
sl@0: LOCAL_D RFs TheFs;
sl@0: 
sl@0: LOCAL_C void setupTestDirectory()
sl@0:     {// Prepare the test directory.
sl@0: 	TInt r=TheFs.Connect();
sl@0: 	test(r==KErrNone);
sl@0: //
sl@0: 	r=TheFs.MkDirAll(KTestDir);
sl@0: 	test(r==KErrNone||r==KErrAlreadyExists);
sl@0: 	r=TheFs.SetSessionPath(KTestDir);
sl@0: 	test(r==KErrNone);
sl@0: 	}
sl@0: 
sl@0: LOCAL_C void setupCleanup()
sl@0:     {// Initialise the cleanup stack
sl@0: 	TheTrapCleanup=CTrapCleanup::New();
sl@0: 	test(TheTrapCleanup!=NULL);
sl@0: 	TRAPD(r,\
sl@0: 		{\
sl@0: 		for (TInt i=KTestCleanupStack;i>0;i--)\
sl@0: 			CleanupStack::PushL((TAny*)0);\
sl@0: 		CleanupStack::Pop(KTestCleanupStack);\
sl@0: 		});
sl@0: 	test(r==KErrNone);
sl@0: 	}
sl@0: 
sl@0: #ifdef _DEBUG
sl@0: LOCAL_D void CreateStoreSetRootAndDestroyStoreL()
sl@0: 	{
sl@0: 	TheFs.Delete(_L("pfs"));
sl@0: 	store=CPermanentFileStore::CreateLC(TheFs,_L("pfs"),EFileWrite|EFileRead);
sl@0: 	store->SetTypeL(KPermanentFileStoreLayoutUid);
sl@0: 	TStreamId rootId = store->ExtendL();
sl@0: 	store->SetRootL(rootId);
sl@0: 	store->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	}
sl@0: 
sl@0: LOCAL_D void AlterStoreL()
sl@0: 	{
sl@0: 	RStoreWriteStream out2;
sl@0: 	RStoreWriteStream out3;
sl@0: 	RStoreWriteStream out4;
sl@0: 	RStoreReadStream in;
sl@0: 
sl@0: 	TStreamId id2 = out.CreateLC(*store);
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	TStreamId id3 = out.CreateLC(*store);
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	TStreamId id4 = out.CreateLC(*store);
sl@0: 	out << _L("mum");
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	out.ReplaceLC(*store,store->Root());
sl@0: 	out << id2;
sl@0: 	out << id3;
sl@0: 	out << id4;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	in.OpenLC(*store,store->Root());// use the root for in and out streams
sl@0: 	out.ReplaceLC(*store,store->Root());
sl@0: 	out.WriteL(in);
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: 
sl@0: 	out.ReplaceLC(*store,store->Root());// swap the order
sl@0: 	in.OpenLC(*store,store->Root());
sl@0: 	out.WriteL(in);
sl@0: 	out << _L("fromage");
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: 
sl@0: 	store->CommitL();
sl@0: 
sl@0: 	in.OpenLC(*store,store->Root());
sl@0: 	TStreamId idX,idZ;
sl@0: 	in >> idX;
sl@0: 	in >> idX;
sl@0: 	in >> idZ;// id4 "mum"
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	out.OpenLC(*store,idZ);
sl@0: 	in.OpenLC(*store,idZ);
sl@0: 	out2.OpenLC(*store,idZ);
sl@0: 	out3.OpenLC(*store,idZ);
sl@0: 	out4.OpenLC(*store,idZ);
sl@0: 	out4.WriteL(in);
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy(5);
sl@0: 	}
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1170
sl@0: @SYMTestCaseDesc	    Allocation failure in store test
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Tests for any memory errors during allocation of store
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_D void AllocFailInSampleStoreCodeL()
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1170 "));
sl@0: 	test.Console()->Printf(_L("AllocFailInSampleStoreCodeL()\n"));
sl@0: 	TRAPD(r,CreateStoreSetRootAndDestroyStoreL())
sl@0:     UNUSED_VAR(r);
sl@0: 	const TInt KAllocFail=15;
sl@0: 	for (TInt ii=1;ii<=20;++ii)
sl@0: 		{
sl@0: 		store=CPermanentFileStore::OpenLC(TheFs,_L("pfs"),EFileWrite|EFileRead);
sl@0: 		__UHEAP_FAILNEXT(ii);
sl@0: 		TRAPD(r,AlterStoreL());
sl@0: 		if (ii<KAllocFail)
sl@0: 			test(r==KErrNoMemory);
sl@0: 		if (ii>=KAllocFail)
sl@0: 			test(r==KErrNone);
sl@0: 		__UHEAP_RESET;
sl@0: 		CleanupStack::PopAndDestroy();
sl@0: 		}
sl@0: 	TheFs.Delete(_L("pfs"));
sl@0: 	}
sl@0: 
sl@0: LOCAL_D void InitialseStoreWithDataL()
sl@0: 	{
sl@0: 	TheFs.Delete(_L("pope"));
sl@0: 	store=CPermanentFileStore::CreateLC(TheFs,_L("pope"),EFileWrite|EFileRead);
sl@0: 	store->SetTypeL(KPermanentFileStoreLayoutUid);
sl@0: 	TStreamId rootId = store->ExtendL();
sl@0: 	store->SetRootL(rootId);
sl@0: 	store->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	store=CPermanentFileStore::OpenLC(TheFs,_L("pope"),EFileWrite|EFileRead);
sl@0: 	TStreamId id2 = out.CreateLC(*store);
sl@0: 	out << desOriginalReverted2;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	TStreamId id3 = out.CreateLC(*store);
sl@0: 	out << desOriginalReverted3;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	out.ReplaceLC(*store,store->Root());
sl@0: 	out << id2;
sl@0: 	out << id3;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();// out
sl@0: 
sl@0: 	store->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();// store
sl@0: 	}
sl@0: 
sl@0: LOCAL_D void AlterStoreDuringOutOfMemoryL(TInt aFail)
sl@0: 	{
sl@0: 	store=CPermanentFileStore::OpenLC(TheFs,_L("pope"),EFileWrite|EFileRead);
sl@0: 	in.OpenLC(*store,store->Root());
sl@0: 	TStreamId id2;
sl@0: 	TStreamId id3;
sl@0: 	in >> id2;
sl@0: 	in >> id3;
sl@0: 	CleanupStack::PopAndDestroy();// in
sl@0: 
sl@0: 	out.ReplaceLC(*store,id2);
sl@0: 	out << desNewOverwritten2;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();// out
sl@0: 
sl@0: 	store->CommitL();
sl@0: 	__UHEAP_FAILNEXT(aFail);// Out of memory
sl@0: 
sl@0: 	out.ReplaceLC(*store,id3);
sl@0: 	out << desNewOverwritten3;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();// out
sl@0: 
sl@0: 	store->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();// store
sl@0: 
sl@0: 	__UHEAP_RESET;
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1346
sl@0: @SYMTestCaseDesc	    Streaming of data test
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Tests for RStoreReadStream::>> operator
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_D void TestStreamDataL(TInt aFail)
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1346 "));
sl@0: 	store=CPermanentFileStore::OpenLC(TheFs,_L("pope"),EFileWrite|EFileRead);
sl@0: 	in.OpenLC(*store,store->Root());
sl@0: 	TStreamId id2;
sl@0: 	TStreamId id3;
sl@0: 	in >> id2;
sl@0: 	in >> id3;
sl@0: 	CleanupStack::PopAndDestroy();// in
sl@0: 
sl@0: 	TBuf<32> buf;
sl@0: 
sl@0: 	in.OpenLC(*store,id2);
sl@0: 	in >> buf;
sl@0: 	test(buf==desNewOverwritten2);
sl@0: 
sl@0: 	CleanupStack::PopAndDestroy();// in
sl@0: 
sl@0: 	in.OpenLC(*store,id3);
sl@0: 	in >> buf;
sl@0: 	if (aFail > KMemoryAllocsInTestFunction)
sl@0: 		test(buf==desNewOverwritten3);
sl@0: 	else if (aFail<=KMemoryAllocsInTestFunction)
sl@0: 		test(buf==desOriginalReverted3);
sl@0: 
sl@0: 	CleanupStack::PopAndDestroy();// in
sl@0: 
sl@0: 	CleanupStack::PopAndDestroy();// store
sl@0: 	}
sl@0: 
sl@0: LOCAL_D void ResetStreamDataL()
sl@0: 	{
sl@0: 	store=CPermanentFileStore::OpenLC(TheFs,_L("pope"),EFileWrite|EFileRead);
sl@0: 	in.OpenLC(*store,store->Root());
sl@0: 	TStreamId id2;
sl@0: 	TStreamId id3;
sl@0: 	in >> id2;
sl@0: 	in >> id3;
sl@0: 	CleanupStack::PopAndDestroy();// in
sl@0: 
sl@0: 	out.ReplaceLC(*store,id2);
sl@0: 	out << desOriginalReverted2;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();// out
sl@0: 
sl@0: 	out.ReplaceLC(*store,id3);
sl@0: 	out << desOriginalReverted3;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();// out
sl@0: 
sl@0: 	store->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();// store
sl@0: 	}
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1171
sl@0: @SYMTestCaseDesc	    Out of memory errors test
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Tests for out of memory conditions before commiting to the store
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_D void OutOfMemoryBeforeStoreCommitL()
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1171 "));
sl@0: 	test.Console()->Printf(_L("OutOfMemoryBeforeStoreCommitL()\n"));
sl@0: 	InitialseStoreWithDataL();
sl@0: 	for (TInt fail=1; fail<=5; ++ fail)
sl@0: 		{
sl@0: 		TRAPD(r,AlterStoreDuringOutOfMemoryL(fail));
sl@0: 		if (fail<=KMemoryAllocsInTestFunction)
sl@0: 			test(r==KErrNoMemory);// store saved when r!=KErrNone
sl@0: 		else
sl@0: 			test(r==KErrNone);
sl@0: 		TestStreamDataL(fail);
sl@0: 		ResetStreamDataL();
sl@0: 		}
sl@0: 	TheFs.Delete(_L("pope"));
sl@0: 	}
sl@0: 
sl@0: 
sl@0: LOCAL_D void OpenCloseStoreL(TInt aFail)
sl@0: 	{
sl@0: 	__UHEAP_FAILNEXT(aFail);
sl@0: 	TheFs.Delete(_L("pope"));
sl@0: 	store=CPermanentFileStore::CreateLC(TheFs,_L("pope"),EFileWrite|EFileRead);
sl@0: 	store->SetTypeL(KPermanentFileStoreLayoutUid);
sl@0: 	TStreamId rootId = store->ExtendL();
sl@0: 	store->SetRootL(rootId);
sl@0: 	store->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	store=CPermanentFileStore::OpenLC(TheFs,_L("pope"),EFileWrite|EFileRead);
sl@0: 	TStreamId id2 = out.CreateLC(*store);
sl@0: 	out << desOriginalReverted2;
sl@0: 	out << id2;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: 	}
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1172
sl@0: @SYMTestCaseDesc	    Out of memory test
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Test for memory errors during opening and closing of store operation.
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: 
sl@0: LOCAL_D void OutOfMemoryWhenOpeningClosingStoreL()
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1172 "));
sl@0: 	test.Console()->Printf(_L("OutOfMemoryWhenOpeningClosingStoreL()\n"));
sl@0: 	const TInt KAllocs=12;
sl@0: 	for (TInt fail=1; fail<=20; ++ fail)
sl@0: 		{
sl@0: 		TRAPD(r,OpenCloseStoreL(fail))
sl@0: 		if (fail<KAllocs)
sl@0: 			test(r==KErrNoMemory);
sl@0: 		else
sl@0: 			test(r==KErrNone);
sl@0: 		}
sl@0: 	TheFs.Delete(_L("pope"));
sl@0: 	__UHEAP_RESET;
sl@0: 	}
sl@0: #endif
sl@0: 
sl@0: GLDEF_C TInt E32Main()
sl@0:     {// Test permanent file store
sl@0: 	test.Title();
sl@0: 	setupTestDirectory();
sl@0: 	setupCleanup();
sl@0: #ifdef _DEBUG
sl@0: 	__UHEAP_MARK;
sl@0: //
sl@0: 	test.Start(_L("Begin tests"));
sl@0: 	TRAPD(r,AllocFailInSampleStoreCodeL());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,OutOfMemoryBeforeStoreCommitL());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,OutOfMemoryWhenOpeningClosingStoreL());
sl@0: 	test(r==KErrNone);
sl@0: 	test.End();
sl@0: 
sl@0: 	TheFs.Delete(_L("pope"));
sl@0: 	TheFs.Delete(_L("pfs"));
sl@0: //
sl@0: 	__UHEAP_MARKEND;
sl@0: #endif
sl@0: 
sl@0: #ifndef _DEBUG
sl@0: 	test.Start(_L("The tests are not valid in release mode"));
sl@0: 	test.End();
sl@0: #endif
sl@0: 	delete TheTrapCleanup;
sl@0: 	TheFs.Close();
sl@0: 	test.Close();
sl@0: 	return 0;
sl@0:     }
sl@0: