--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/usbmgmt/usbmgrtest/t_usbmanager_suite/T_UsbManager/src/CUsbTestStepBase.cpp	Tue Feb 02 02:02:59 2010 +0200
@@ -0,0 +1,223 @@
+/*
+* 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;
+	}
+