// Copyright (c) 1998-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:
//

#ifndef T_DBFAIL_H
#define T_DBFAIL_H

GLREF_D RTest test;
GLREF_D RDbs TheDbs;
GLREF_D RDbNamedDatabase TheDatabase;

GLREF_C void Connect();
GLREF_C void Disconnect();
GLREF_C void PrepareDbFmtString();
GLREF_C void TestOpen2();
GLREF_C void Origins2();

class THeapFail
	{
public:
	virtual void Mark() const =0;
	virtual void Fail(TInt aCount) const =0;
	virtual void Reset() const =0;
	virtual void Check() const =0;
	};

class TContext
	{
public:
	virtual void OpenDbL() const =0;
	};

class TFail
	{
private:
	void Test(const THeapFail& aHeap,const TContext* aContext);
	virtual void RunL() 
		{
		}
	virtual void End() 
		{
		}

public:
	inline void Test(const THeapFail& aHeap,const TContext& aContext)
		{
		Test(aHeap,&aContext);
		}
	inline void Test(const THeapFail& aHeap)
		{
		Test(aHeap,0);
		}
	};

class TClientHeap : public THeapFail
	{
public:
	TClientHeap() 
		{
		}
		
private:
	void Mark() const
		{
		User::__DbgMarkStart(RHeap::EUser);
		}
	void Fail(TInt aCount) const
		{
		User::__DbgSetAllocFail(RHeap::EUser,RHeap::EFailNext,aCount);
		}
	void Reset() const
		{
		User::__DbgSetAllocFail(RHeap::EUser,RHeap::ENone,1);
		}
	void Check() const
		{
		User::__DbgMarkEnd(RHeap::EUser,0);
		}
		
	};

class TServerHeap : public THeapFail
	{
public:
	TServerHeap() 
		{
		}
		
private:
	void Mark() const
		{
		}
	void Fail(TInt aCount) const
		{
		TheDbs.SetHeapFailure(RHeap::EFailNext,aCount);
		}
	void Reset() const
		{
		TheDbs.SetHeapFailure(RHeap::ENone,0);
		}
	void Check() const
		{
		}
		
	};

GLREF_D TClientHeap KClientHeap;
GLREF_D TServerHeap KServerHeap;

#endif//T_DBFAIL_H