// Copyright (c) 2001-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_WSPTRHNDSTUB_H__
#define __T_WSPTRHNDSTUB_H__


// System includes
//
#include <wsp/mwspcapabilityprovider.h>
#include <wsp/mwspproxyinfoprovider.h>
#include <wsp/mwspsessionheadersprovider.h>
#include <wsp/mwspcosessioncallback.h>
#include <wsp/mwspcomethodcallback.h>

// Local includes
//
#include "HttpTestCore.h"

// Forward class declarations
//
class CWspTransportHandler;
class CWspTrHndTestDriver;

// Definition of class 'CWspTrHndStubTest'.  This test instantiates a WSP Transport Handler,
// via ECOM, to obtain the stub version.
class CWspTrHndStubTest : public CHttpTestBase, public MWspCapabilityProvider,
						  public MWspProxyInfoProvider, public MWspSessionHeadersProvider
	{
public: // methods

	static CWspTrHndStubTest* NewLC();
	virtual ~CWspTrHndStubTest();
	void DoTestsL();
	virtual void DoRunL();
	virtual const TDesC& TestName();
	virtual void DoCancel();
	void Done();

public: // methods inherited from MWspCapabilityProvider

	virtual const MWspCapabilityViewer& ClientCapabilities() const;
	virtual MWspCapabilitySetter& ServerCapabilities() const;

public: // methods inherited from MWspProxyInfoProvider

	virtual const TDesC8& ProxyAddress() const;
	virtual TUint16 LocalPort() const;
	virtual TUint16 RemotePort() const;
	virtual TWspBearer Bearer() const;
	virtual const TDesC8& ServiceCentreNumber() const;
	virtual TWspSession WspSessionService() const;
	virtual TBool SecureConnection() const;

public: // methods inherited from MWspSessionHeadersProvider

	virtual const TDesC8& ClientHeaders() const;
	virtual void SetServerHeadersL(const TDesC8& aBuffer);

private: // methods

	CWspTrHndStubTest();
	void ConstructL();

private: // attributes

	RStringPool iStringPool;
	CWspTransportHandler* iTrHnd;
	CWspTrHndTestDriver* iDriver;
	};


class CWspTrHndTestDriver : public CActive, public MWspCOSessionCallback, public MWspCOMethodCallback
	{
public: // methods

	static CWspTrHndTestDriver* NewL(CWspTrHndStubTest& aOwner, RStringPool aStringPool);
	virtual ~CWspTrHndTestDriver();
	void Start();
	void SetTransportHandler(CWspTransportHandler& aTrHnd);

public: // methods from CActive

	virtual void RunL();
	virtual void DoCancel();
	virtual TInt RunError(TInt aError);

public: // methods from MWspCOSessionCallback

	virtual void ConnectCnf();
	virtual void ResumeCnf();
	virtual void DisconnectInd(
							  TWspReason				aReasonCode, 
							  TBool						aRedirectSecurity, 
							  TWspRedirectedAddress&	aRedirectAddresses, 
							  const TDesC8&				aErrorHeader, 
							  const TDesC8&				aErrorBody
							  );

	virtual void SuspendInd(TWspReason aReason);
	virtual void ExceptionInd(const TDesC8& aExceptionData);

public: // methods from MWspCOSessionCallback

	virtual void MethodInvokeCnf();
	virtual void MethodInvokeDataCnf();
	virtual void MethodAbortInd(TWspReason aReason);
	virtual void MethodResultInd(
								TInt					aStatus, 
								const TDesC8&			aResponseHeaders, 
								MHTTPDataSupplier&		aResponseBody,
								TBool					aMoreData
								);


	virtual void MethodResultDataInd(const TDesC8& aTrailerHeaders, TBool aMoreData);

private: // embedded classes

/*
	class TBody : public MHTTPDataSupplier
		{
	public:
		TBody();
		virtual TBool GetNextDataPart(TPtrC8& aDataPart);
		virtual void ReleaseData();
		virtual TInt OverallDataSize();
		virtual TInt Reset();
	private:
		TInt iPartCount;
		};

	class THeader : public MHTTPDataSupplier
		{
	public:
		THeader();
		virtual TBool GetNextDataPart(TPtrC8& aDataPart);
		virtual void ReleaseData();
		virtual TInt OverallDataSize();
		virtual TInt Reset();
	private:
		TInt iPartCount;
		};
*/
private: // methods

	CWspTrHndTestDriver(CWspTrHndStubTest& aOwner, RStringPool aStringPool);
	void ConstructL();
	void CompleteSelf();
	void DoConnectReq();
	void DoMethodInvokeReq();
	void DoMethodInvokeDataReq();
	void DoDisconnectReq();

private: // type definitions

	enum TState
		{
		EInit,
		EConnecting,
		EConnected,
		EInvoking,
		EInvokingMore,
		EReplying,
		EReplyingMore,
		EReplyComplete,
		EDisconnecting,
		EDisconnected
		};

private: // attributes

	TState					iState;
	CWspTrHndStubTest&		iOwner;
	CWspTransportHandler*	iTrHnd;
	RStringPool				iStringPool;
	HBufC8*					iHeader;
	MHTTPDataSupplier*		iResponseBody;
	};

#endif // __T_WSPTRHNDSTUB_H__