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

/**
 @file
*/
#include <test/testexecutelog.h>
#include "CUsbTestStepBase.h"

CUsbTestStepBase::CUsbTestStepBase():
	iDoNotConnect(EFalse),
	iConnected(EFalse),
	iStopService(ETrue)
	{
	}

TVerdict CUsbTestStepBase::doTestStepPreambleL()
/**
 * @return - TVerdict code
 * Override of base class virtual
 */
	{
	__UHEAP_MARK;
	SetTestStepResult(EPass);
	
	if (!iDoNotConnect)
		{
		TInt res = iUsb.Connect();
		if (res != KErrNone)
			{
			INFO_PRINTF2(_L("Error connecting to USB service: %d"), res);
			SetTestStepResult(EFail);
			return TestStepResult();
			}
		iConnected = ETrue;
		}
	return TestStepResult();
	}

TVerdict CUsbTestStepBase::doTestStepPostambleL()
/**
 * @return - TVerdict code
 * Override of base class virtual
 */
	{
	if (iStopService && iConnected)
		{
		TRequestStatus status;
		iUsb.Stop(status);
		User::WaitForRequest(status);
		}
	if (iConnected)
		{
		iUsb.Close();
		}
	__UHEAP_MARKEND;
	return TestStepResult();
	}

TInt CUsbTestStepBase::CheckState(RUsb sess, TUsbServiceState aState)
	{
	TUsbServiceState currState;
	TInt err;
	err = sess.GetServiceState(currState);
	if (err != KErrNone)
		{
		INFO_PRINTF2(_L("GetServiceState returned error code %d"), err);
		return err;
		}
	if (currState != aState)
		{
		INFO_PRINTF3(_L("GetServiceState returned unexpected state %d, expected was: %d"), currState, aState);
		return KErrGeneral;
		}
	return KErrNone;
	}

TInt CUsbTestStepBase::SetIdle(RUsb sess)
	{
	TRequestStatus status;
	TUsbServiceState currState;
	TInt err;
	err = sess.GetServiceState(currState);
	if (err != KErrNone)
		{
		INFO_PRINTF2(_L("GetServiceState returned error code %d"), err);
		return err;
		}
	if (currState == EUsbServiceIdle)
		{
		return KErrNone;
		}
	
	sess.Stop(status);
	User::WaitForRequest(status);
	if (status != KErrNone)
		{
		INFO_PRINTF2(_L("Unable to stop service: %d"), status.Int());
		return status.Int();
		}
	return KErrNone;
	}

TInt CUsbTestStepBase::SetStarted(RUsb sess)
	{
	TRequestStatus status;
	TUsbServiceState currState;
	TInt err;
	err = sess.GetServiceState(currState);
	if (err != KErrNone)
		{
		INFO_PRINTF2(_L("GetServiceState returned error code %d"), err);
		return err;
		}
	if (currState == EUsbServiceStarted)
		{
		return KErrNone;
		}
	
	sess.Start(status);
	User::WaitForRequest(status);
	if (status != KErrNone)
		{
		INFO_PRINTF2(_L("Unable to start service: %d"), status.Int());
		return status.Int();
		}
	return KErrNone;
	}

TInt CUsbTestStepBase::SetStarting(RUsb sess)
	{
	TRequestStatus status;
	TUsbServiceState currState;
	TInt err;
	err = sess.GetServiceState(currState);
	if (err != KErrNone)
		{
		INFO_PRINTF2(_L("GetServiceState returned error code %d"), err);
		return err;
		}
	if (currState == EUsbServiceStarting)
		{
		return KErrNone;
		}
	
	if (currState == EUsbServiceStarted)
		{
		sess.Stop(status);
		User::WaitForRequest(status);
		if (status != KErrNone)
			{
			INFO_PRINTF2(_L("Unable to stop service: %d"), status.Int());
			return status.Int();
			}
		if (CheckState(sess, EUsbServiceIdle) != KErrNone)
			{
				INFO_PRINTF1(_L("Can't set state to EUsbServiceIdle"));
				return KErrGeneral;
			}
		}
	
	//Start and don't wait for request, otherwise the state will be "started"
	sess.Start(status);
	sess.CancelInterest(RUsb::EStart);
	User::WaitForRequest(status);
	
	return KErrNone;
	}
	
TInt CUsbTestStepBase::SetStopping(RUsb sess)
	{
	TRequestStatus status;
	TUsbServiceState currState;
	TInt err;
	err = sess.GetServiceState(currState);
	if (err != KErrNone)
		{
		INFO_PRINTF2(_L("GetServiceState returned error code %d"), err);
		return err;
		}
	if (currState == EUsbServiceStopping)
		{
		return KErrNone;
		}
	
	if (currState == EUsbServiceIdle)
		{
		sess.Start(status);
		User::WaitForRequest(status);
		if (status != KErrNone)
			{
			INFO_PRINTF2(_L("Unable to start service: %d"), status.Int());
			return status.Int();
			}
		if (CheckState(sess, EUsbServiceStarted) != KErrNone)
			{
			INFO_PRINTF1(_L("Can't set state to EUsbServiceStarted"));
			return KErrGeneral;
			}
		}
	
	//Stop and don't wait for request, otherwise the state will be "idle"
	sess.Stop(status);
	sess.CancelInterest(RUsb::EStop);
	User::WaitForRequest(status);
	
	return KErrNone;
	}