// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the License "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:

// @file testpolicy.cpp

// @internalComponent

// 

//

#include "TestPolicy.h"

#include "TestCaseFactory.h"

#include "BaseTestCase.h"

#include "TestCaseFactory.h"

#include "testdebug.h"

namespace NUnitTesting_USBDI

	{

CBasicTestPolicy* CBasicTestPolicy::NewL()

	{

	CBasicTestPolicy* self = new (ELeave) CBasicTestPolicy;

	CleanupStack::PushL(self);

	self->ConstructL();

	CleanupStack::Pop(self);

	return self;

	}

CBasicTestPolicy::CBasicTestPolicy()

:	CActive(EPriorityStandard)

	{

	CActiveScheduler::Add(this);

	}

CBasicTestPolicy::~CBasicTestPolicy()

	{

	Cancel();

	}

void CBasicTestPolicy::ConstructL()

	{

	}

void CBasicTestPolicy::RunTestCaseL(const TDesC& aTestCaseId,TRequestStatus& aNotifierStatus)

	{

	iNotifierStatus = &aNotifierStatus;

	// Create the specified test case

	iTestCase = RTestFactory::CreateTestCaseL(aTestCaseId,EFalse);

	iTestCase->SetTestPolicy(this);

	iTestCase->PerformTestL();

	// Set the request of the notifier to pending

	*iNotifierStatus = iStatus = KRequestPending;

	SetActive();

	}

void CBasicTestPolicy::DoCancel()

	{

	// Cancel running the test cases

	iTestCase->Cancel();

	// Notify the test case controller that test case execution was cancelled

	User::RequestComplete(iNotifierStatus,KErrCancel);

	}

void CBasicTestPolicy::SignalTestComplete(TInt aCompletionCode)

	{

	// Complete the test policy request with the test case completion code

	// (Basically self completion)

	TRequestStatus* s = &iStatus;	

	User::RequestComplete(s,aCompletionCode);

	}

void CBasicTestPolicy::RunL()

	{

	// Complete the request of the notifier with the test case 

	// completion code

	User::RequestComplete(iNotifierStatus,iStatus.Int());

	// Destroy the test case

	delete iTestCase;

	}

TInt CBasicTestPolicy::RunError(TInt aError)

	{

	aError = KErrNone;

	return aError;

	}

CThreadTestPolicy* CThreadTestPolicy::NewL()

	{

	CThreadTestPolicy* self = new (ELeave) CThreadTestPolicy;

	CleanupStack::PushL(self);

	self->ConstructL();

	CleanupStack::Pop(self);

	return self;

	}

CThreadTestPolicy::CThreadTestPolicy()

	{

	}

CThreadTestPolicy::~CThreadTestPolicy()

	{

	iTestThread.Close();

	if(iTestCaseId)

		{

		delete iTestCaseId;

		iTestCaseId = NULL;

		}

	}

void CThreadTestPolicy::ConstructL()

	{

	}

void CThreadTestPolicy::RunTestCaseL(const TDesC& aTestCaseId,TRequestStatus& aNotifierStatus)

	{

	iNotifierStatus = &aNotifierStatus;

	if(iTestCaseId)

		{

		delete iTestCaseId;

		iTestCaseId = NULL;

		}

	iTestCaseId = HBufC::NewL(aTestCaseId.Length()); 

	*iTestCaseId = aTestCaseId;

	// Create the thread to run the test case in

	TInt err(iTestThread.Create(aTestCaseId,CThreadTestPolicy::ThreadFunction,

			KDefaultStackSize,NULL,reinterpret_cast<TAny*>(iTestCaseId)));

	if(err != KErrNone)

		{

		User::Leave(err);

		}

	// Start the test case in the thread

	iTestThread.Logon(iStatus);

	SetActive();

	iTestThread.Resume();

	*iNotifierStatus = KRequestPending;

	}

void CThreadTestPolicy::SignalTestComplete(TInt aCompletionCode)

	{

	if(aCompletionCode == KErrNone)

		{

		CActiveScheduler::Stop();

		}

	else

		{

		iTestThread.Kill(aCompletionCode);

		}

	}

void CThreadTestPolicy::DoCancel()

	{

	iTestCase->Cancel();

	TInt err(iTestThread.LogonCancel(iStatus));

	if(err != KErrNone)

		{

		}

	TRequestStatus cancelStatus;

	iTestThread.Logon(cancelStatus);

	iTestThread.Kill(KErrCancel);		

	User::RequestComplete(iNotifierStatus,cancelStatus.Int());

	}

TInt CThreadTestPolicy::ThreadFunction(TAny* aThreadParameter)

	{

	TInt err(KErrNone);

	TInt leaveCode(KErrNone);

	HBufC* testCaseId = reinterpret_cast<HBufC*>(aThreadParameter);

	// Create the cleanup stack for this thread

	CTrapCleanup* cleanup = CTrapCleanup::New();

	if(cleanup == NULL)

		{

		return KErrNoMemory;

		}

	TRAP(leaveCode,err = CThreadTestPolicy::DoTestL(*testCaseId));

	if(leaveCode != KErrNone)

		{

		err = leaveCode;

		}

	delete cleanup;

	return err;

	}

TInt CThreadTestPolicy::DoTestL(const TDesC& aTestCaseId)

	{

	TInt err(KErrNone);

	// Create a new active scheduler for this thread

	CActiveScheduler* sched = new (ELeave) CActiveScheduler;

	CleanupStack::PushL(sched);

	CActiveScheduler::Install(sched);

	CBaseTestCase* testCase = RTestFactory::CreateTestCaseL(aTestCaseId,ETrue);

	CleanupStack::PushL(testCase);

	// Do the test

	testCase->PerformTestL();

	if(!testCase->IsHostOnly())

		{

		// Loop for active objects

		CActiveScheduler::Start();

		}	

	// Get the test case execution result 

	err = testCase->TestResult();

	// Destroy test case

	CleanupStack::PopAndDestroy(testCase);

	// Destroy the active scheduler	

	CleanupStack::PopAndDestroy(sched);

	return err;

	}

void CThreadTestPolicy::RunL()

	{

	TInt completionCode(iStatus.Int());

	TExitType exitType(iTestThread.ExitType());

	TExitCategoryName exitName(iTestThread.ExitCategory());

	TInt exitReason(iTestThread.ExitReason());

	switch(exitType)

		{

		// The test thread has panicked

		// This will occur if test API panics or RTest expression is false (i.e. test case fails)

		case EExitPanic:

			{

			// May require to stop and start host/client USB depending on what panic category it is

			// can no longer trust RUsbHubDriver/RDevUsbcClient to be in good state

			completionCode = KErrAbort;

			}

			break;

		// The thread has been terminated

		case EExitTerminate:

			{

			}

			break;

		// The thread has been killed

		// This will occur when the test thread executes normally or is cancelled

		case EExitKill:

			{

			}

			break;

		// These cases should not occur at this stage

		case EExitPending: //follow through

		default:

			gtest(EFalse); // Panic main thread

			break;

		}

	// Close the just executed test thread

	iTestThread.Close();

	// Complete the notifier's request status

	User::RequestComplete(iNotifierStatus,completionCode);

	}

TInt CThreadTestPolicy::RunError(TInt aError)

	{

	return KErrNone;

	}

	}