/*
* 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_CNATTRAVERSALCONTROLLER_H
#define UT_CNATTRAVERSALCONTROLLER_H

#include <es_sock.h>
#include <in_sock.h>

#include <digia/eunit/ceunittestsuiteclass.h>

#include "sipnattraversalrequestobserver.h"
#include "sipnatbindingobserver.h"
#include "nattraversalsocketuser.h"

#include <digia/eunit/ceunittestsuiteclass.h>

#include <e32def.h>
#ifndef NONSHARABLE_CLASS
    #define NONSHARABLE_CLASS(x) class x
#endif

class CNATTraversalController;


NONSHARABLE_CLASS(UT_CNATTraversalController) :
    public CEUnitTestSuiteClass,
    public MSIPNATTraversalRequestObserver,
    public MSIPNATBindingObserver,
    public MNATTraversalSocketUser
    {
    public:     // Constructors and destructors

        /**
         * Two phase construction
         */
        static UT_CNATTraversalController* NewL();
        static UT_CNATTraversalController* NewLC();

        /**
         * Destructor
         */
        ~UT_CNATTraversalController();

    private:    // Constructors and destructors

        UT_CNATTraversalController();
        void ConstructL();

    public: // MSIPNATTraversalRequestObserver

        void RequestComplete(
            TUint32 aRequestId,
            TInt aCompletionCode );

        void PublicAddrResolved(
            TUint32 aRequestId,
            const TInetAddr& aPublicAddr );

    public: // MSIPNATBindingObserver

        void FlowFailure( TInt aError );

    public: // MNATTraversalSocketUser

        const RSocket& Socket() const;
        void SendingAllowed();
        TBool IsSending() const;
        void CancelSending();

    private: // New Methods

        void SetupL();
        void Teardown();

        void TestHandleNextFromSocketUserQueueL();
        void TestNextRequestIdL();
        void TestCreateBindingL();
        void TestFindSessionIndexL();
        void TestHandleRequestL();
        void TestCancelL();
        void TestRefreshNATBindingL();
        void TestRefreshTlsNATBindingL();
        void TestUpdateNextHopL();
        void TestSocketIdleL();
        void TestTlsSocketIdleL();
        void TestDataReceivedL();
        void TestFreeResourcesL();
        void TestRequestCompletedL();
        void TestSTUNClientInitCompletedL();
        void TestSTUNBindingEventOccurredL();
        void TestSTUNBindingErrorOccurredL();
        void TestSTUNBindingErrorOccurred2L();
        void TestAddToSendingQueueL();
        void TestSendingCompletedL();

    private: // Data

        EUNIT_DECLARE_TEST_TABLE;

        RSocketServ iSocketServ;
        RConnection iConnection;
        RSocket iSocket;
        TInetAddr iLocalAddr;
        TInetAddr iNextHopAddr;
        CNATTraversalController* iController;
        TUint32 iRequestId;
        TBool iIsSending;
        TInt iRequestCompletionCode;
    };

#endif // UT_CNATTRAVERSALCONTROLLER_H

// End of File