/*

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

#define __UT_CSRTPSESSION_H__

//  EXTERNAL INCLUDES

#include <digia/eunit/ceunittestsuiteclass.h>

//  INTERNAL INCLUDES

#include <in_sock.h>

#include "srtpmastersalt.h"

#include "srtpmasterkey.h"

#include "msrtprekeyingobserver.h"

#include "srtpcryptocontext.h"

#include "srtpsession.h"

//#include "SRTPCryptoHandlerSRTP.h"

//#include "SRTPCryptoHandlerSRTCP.h"

#include "srtputils.h"

//  FORWARD DECLARATIONS

class CSRTPCryptoHandlerSRTP;

class CSRTPSession;

//  CLASS DEFINITION

/**

 *

 * EUnitWizard generated test class. 

 *

 */

class UT_CSRTPSession

     : public CEUnitTestSuiteClass,

     public MSRTPReKeyingObserver

    {

    public:     // Constructors and destructors

        /**

         * Two phase construction

         */

        static UT_CSRTPSession* NewL();

        static UT_CSRTPSession* NewLC();

        /**

         * Destructor

         */

        ~UT_CSRTPSession();

    private:    // Constructors and destructors

        UT_CSRTPSession();

        void ConstructL();

	public: 	// From observer interface

		void SRTPMasterKeyStaleEvent(const CSRTPStream& aStream);

		void SRTPMasterKeyStaleEvent(const CSRTPSession& aSession);

    private:    // New methods

         void SetupL();

         void Teardown();

         void UT_CSRTPSession_RemoveStreamL();

         void UT_CSRTPSession_GetCryptoContextL();

         void Hex(HBufC8& aString);

         void UT_CSRTPSession_StreamLL();

         void UT_CSRTPSession_ProtectRTPLL();

         void UT_CSRTPSession_UnprotectRTPLL();	

         void UT_CSRTPSession_UnprotectRTPL2L();

      	 void UT_CSRTPSession_ProtectRTCPLL();

         void UT_CSRTPSession_UnprotectRTCPLL();

         void UT_CSRTPSession_UnprotectRTCPL2L();	         

         void UT_CSRTPSession_FindStreamL();

         void UT_CSRTPSession_ConstructLL(  );

         void FindLateBindingStreamAndUnprotectRTPLL();

         void FindLateBindingStreamAndUnprotectRTCPLL();

         void UT_CSRTPSession_UpdateCryptoContextLL();

         void UT_CSRTPSession_RemoteAddrL();

         void UT_CSRTPSession_RemoveAllStreamL();

    private:    // Data

		TInetAddr iDestination;

        CSRTPMasterSalt* iMasterSalt; 

        CSRTPMasterKey*  iMasterKey;         

        CSRTPCryptoContext* iContext;

        HBufC8* iTestPayload160Bits;

        HBufC8* iTestMKI128Bits;

        HBufC8* iTestAuthTag80Bits;

        CSRTPSession* iSRTPSession;

        CSRTPCryptoHandlerSRTP* iCryptoHandlerSRTP;

		EUNIT_DECLARE_TEST_TABLE; 

    };

#endif      //  __UT_CSRTPSESSION_H__

// End of file