// Copyright (c) 2010 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: Wrapper for Testing CStateAwareSession2 class.

//

//

//

/**

 @file

 @test

 @internalComponent - Internal Symbian test code

*/

#ifndef __T_STATEAWARESESSIONWRAPPER2_H__

#define __T_STATEAWARESESSIONWRAPPER2_H__

#include <ssm/ssmstatemanager.h>

#include <ssm/ssmstateawaresession.h>

#include <test/testexecutestepbase.h>

#include <test/datawrapper.h>

// This structure is used to hold data read from the test ini file for CTestCStateAwareSession

struct TIniData3

    {

    TBool iAcknowledge;

    TInt iAsyncIndex;

    TIniData3();

    };

class CTestCStateAwareSession : public CDataWrapper, public MStateChangeNotificationSubscriber2

    {

public:

    CTestCStateAwareSession();

    ~CTestCStateAwareSession();

    TBool DoCommandL(const TTEFFunction& aCommand, const TTEFSectionName& aSection, const TInt /*aAsyncErrorIndex*/);

    TAny* GetObject();

    void  SetObjectL(TAny* aAny);

    void  DisownObjectL();

    void DoCmdNew();

    void DoCmdDestructor();

    void DoCmdRequestStateNotification();

    void DoCmdAcknowledgeStateNotification(const TDesC& aSection);

    void DoCmdAcknowledgeAndRequestStateNotification(const TDesC& aSection);

    void DoCmdRequestStateNotificationCancel();

    void DoCmdReturnCount(const TDesC& aSection);

    void DoCmdReadIniData(const TDesC& aSection);

    //helper

    void DoCleanup();

    //from MStateChangeNotificationSubscriber2

    void HandleTransition(TInt aError);

    TInt HandleDeferralError(TInt aError);

private:

    CSsmStateAwareSession2* iSsmStateAwareSession;

    TIniData3 iIniData;

    };

#endif // __T_STATEAWARESESSIONWRAPPER_H__