sl@0: // Copyright (c) 1997-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: // Overload of the CTransition test that provides
sl@0: // Heap and Handle leak testing upon a test method.
sl@0: // 
sl@0: //
sl@0: 
sl@0: #include <ecom/test_bed/datalogger.h>
sl@0: #include "LeakTestTransition.h"
sl@0: 
sl@0: EXPORT_C CLeakTestTransition::CLeakTestTransition(const TDesC&			aTransitionId, 
sl@0: 												  CUnitTestContext&		aUTContext,
sl@0: 												  TTransitionValidator&	aValidator) 
sl@0: :	CTransition(aTransitionId, aUTContext, aValidator), 
sl@0: 	iFailStep(KMemoryLeakTestFailInit), 
sl@0: 	iBreakStep(KMemoryLeakTestBreakNone)
sl@0: 	{
sl@0: 	// Do nothing here
sl@0: 	}
sl@0: 
sl@0: // Define the overloaded RunL behaviour here
sl@0: EXPORT_C void CLeakTestTransition::RunL()
sl@0: 	{
sl@0: 	// Setup leak check and call the base RunL
sl@0: 	iThread.HandleCount(iStartProcessHandleCount, iStartThreadHandleCount);
sl@0: 	__UHEAP_SETFAIL(RHeap::EDeterministic,iFailStep);
sl@0: 	__UHEAP_MARK;
sl@0: 	if(iBreakStep == iFailStep)
sl@0: 		{
sl@0: 		// Drop into the debugger because an unexpected leave occured
sl@0: 		// on the last run of the RunL. (This is therefore a repeat run...)
sl@0: 		__DEBUGGER();
sl@0: 		iBreakStep = KMemoryLeakTestBreakNone;
sl@0: 		}
sl@0: 
sl@0:     if(iStatus == KErrNoMemory) 
sl@0:         { 
sl@0:         // Special case of Async Process signalling 
sl@0:         // a failure during the notification 
sl@0:         _LIT(KAsyncProblem, "CLeakTestTransition::RunL() async completion with error %d."); 
sl@0:         iUTContext.DataLogger().LogInformationWithParameters(KAsyncProblem, iStatus.Int()); 
sl@0:         User::Leave(iStatus.Int()); 
sl@0:         }
sl@0: 
sl@0: 	CTransition::RunL();
sl@0: 	if(iTransitionFinished)
sl@0: 		{
sl@0: 		// Successful completion.
sl@0: 		if(iStatus == KErrNone)
sl@0: 			{
sl@0: 			_LIT(KTransitionRunSuccess, "CLeakTestTransition::TransitMethodL() successful completion on iteration %d.");
sl@0: 			iUTContext.DataLogger().LogInformationWithParameters(KTransitionRunSuccess, iFailStep);
sl@0: 			}
sl@0: 		else if(iStatus == KRequestPending)
sl@0: 			{
sl@0: 			//This transition was to set up an asynchronous request
sl@0: 			_LIT(KTransitionRunAsync, "CLeakTestTransition::TransitMethodL() successful called async method on iteration %d.");
sl@0: 			iUTContext.DataLogger().LogInformationWithParameters(KTransitionRunAsync, iFailStep);
sl@0: 			}
sl@0: 		else
sl@0: 			{
sl@0: 			_LIT(KTransitionRunError, "CLeakTestTransition::TransitMethodL() error %d completion on iteration %d.");
sl@0: 			iUTContext.DataLogger().LogInformationWithParameters(KTransitionRunError, iStatus.Int(), iFailStep);
sl@0: 			}
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		__DEBUGGER();		// An impossible case ????
sl@0: 		}
sl@0: 	}
sl@0: 
sl@0: EXPORT_C TInt CLeakTestTransition::RunError(TInt aErrorCode)
sl@0: 	{
sl@0: 	if(aErrorCode != KErrNoMemory)
sl@0: 		{
sl@0: 		iLeaveError = aErrorCode;
sl@0: 		// Check if the leave is associated with a repeat request
sl@0: 		// I.e. it was an execution path test from a stub.
sl@0: 		if(iLeaveError == KTestBedRepeatTest && iRepeatThis)
sl@0: 			{
sl@0: 			_LIT(KTransitionRunRepeat, "CLeakTestTransition::TransitMethodL() leaving on iteration %d for repeat test.");
sl@0: 			iUTContext.DataLogger().LogInformationWithParameters(KTransitionRunRepeat, iFailStep);
sl@0: 			User::RequestComplete(iUnitTestStatus, KTestBedRepeatTest);
sl@0: 			}
sl@0: 		else
sl@0: 			{
sl@0: 			iBreakStep = iFailStep;				// Unexpected so
sl@0: 			// Record the leave and signal completed with a leave code
sl@0: 			_LIT(KTransitionRunError, "CLeakTestTransition::TransitMethodL() leaving error %d on iteration %d.");
sl@0: 			iUTContext.DataLogger().LogInformationWithParameters(KTransitionRunError, 
sl@0: 																 aErrorCode, 
sl@0: 																 iFailStep);
sl@0: 			User::RequestComplete(iUnitTestStatus, KTestBedTestLeft);
sl@0: 			}
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		++iFailStep;					// Caused by this test harness
sl@0: 		// Safety check in case we are testing a function which always leaves with KErrNoMemory
sl@0: 		if(iFailStep > KProbablyInfinitelyLooping)
sl@0: 			User::RequestComplete(iUnitTestStatus, KTestBedLeakTestLoopDetected);
sl@0: 		else
sl@0: 			{
sl@0: 			// Re-Schedule
sl@0: 			// Only Reset if its not a stub repeat 
sl@0: 			// request via a leave
sl@0: 			if(aErrorCode != KTestBedRepeatTest)
sl@0: 				iTransitionInfo.iIteration = 0;	
sl@0: 			SetActive();
sl@0: 			TRequestStatus* status = &iStatus;
sl@0: 			User::RequestComplete(status, KErrNone);
sl@0: 			}
sl@0: 		}
sl@0: 	// Check leak cleanup
sl@0: 	iThread.HandleCount(iEndProcessHandleCount, iEndThreadHandleCount);
sl@0: 	if(iStartThreadHandleCount != iEndThreadHandleCount)
sl@0: 		{
sl@0: 		__DEBUGGER();										// Oops leaked some handles
sl@0: 		}
sl@0: 	__UHEAP_MARKEND;
sl@0: 	__UHEAP_SETFAIL(RHeap::ENone, KMemoryLeakTestFailInit);	// No more fails
sl@0: 	return KErrNone;
sl@0: 	}
sl@0: 
sl@0: EXPORT_C void CLeakTestTransition::PostTransitionCleanup()
sl@0: 	{
sl@0: 	__UHEAP_SETFAIL(RHeap::ENone, KMemoryLeakTestFailInit);	// No more fails
sl@0: 	}