--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/natplugins/natptraversalcontroller/tsrc/ut_NatTraversalController/inc/ut_cnattraversalcontroller.h	Tue Feb 02 01:04:58 2010 +0200
@@ -0,0 +1,129 @@
+/*
+* 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