/*

* 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:  

*

*/

//  CLASS HEADER

#include "ut_cnatfwsockethandler.h"

//  EXTERNAL INCLUDES

#include <digia/eunit/eunitmacros.h>

#include "natfwconnectivityapidefs.h"

//  INTERNAL INCLUDES

#include "natfwsockethandler.h"

#include "socketmediaconnwrappertestdefs.h"

#include "natfwunittestmacros.h"

_LIT8(KData,"qwerty");

const TUint KTestPortUdp = 5000;

const TUint KTestPortTcp = 5001;

const TUint KTestPort2 = 5010;

// CONSTRUCTION

ut_cnatfwsockethandler* ut_cnatfwsockethandler::NewL()

    {

    ut_cnatfwsockethandler* self = ut_cnatfwsockethandler::NewLC();

    CleanupStack::Pop();

    return self;

    }

ut_cnatfwsockethandler* ut_cnatfwsockethandler::NewLC()

    {

    ut_cnatfwsockethandler* self = new( ELeave ) ut_cnatfwsockethandler();

    CleanupStack::PushL( self );

    self->ConstructL();

    return self;

    }

// Destructor (virtual by CBase)

ut_cnatfwsockethandler::~ut_cnatfwsockethandler()

    {

    }

// Default constructor

ut_cnatfwsockethandler::ut_cnatfwsockethandler()

    {

    }

// Second phase construct

void ut_cnatfwsockethandler::ConstructL()

    {

    // The ConstructL from the base class CEUnitTestSuiteClass must be called.

    // It generates the test case table.

    CEUnitTestSuiteClass::ConstructL();

    }

// From observer interface

void ut_cnatfwsockethandler::GetNewFreePort( TUint& aPort )

    {

    aPort = KTestPort2;

    }

//  METHODS

void ut_cnatfwsockethandler::SetupL(  )

    {

    iHandler = CNATFWSocketHandler::NewL( *this, *this, KProtocolInetUdp, KTestPortUdp );

    User::LeaveIfError( iSocketServer.Connect() );

    }

void ut_cnatfwsockethandler::Setup2L(  )

    {

    iHandler = CNATFWSocketHandler::NewL( *this, *this, KProtocolInetTcp, KTestPortTcp );

    User::LeaveIfError( iSocketServer.Connect() );

    }    

void ut_cnatfwsockethandler::Teardown(  )

    {

    delete iHandler;

    iSocketServer.Close();

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_NewLL(  )

    {

    TUint port = 0;

    CNATFWSocketHandler* handler = NULL;

    NATFW_EUNIT_ASSERT_SPECIFIC_LEAVE( 

        handler = CNATFWSocketHandler::NewL( *this, *this, 0, port ), KErrArgument );        

    delete handler;

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_LocalAddressL(  )

    {

    iHandler->LocalAddress( iAddr );

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_SetAddrLL(  )

    {

    iHandler->SetLocalAddress( iSocketServer, iAddr );

    NATFW_EUNIT_ASSERT_NO_LEAVE( iHandler->SetAddrL( iAddr ) );

    NATFW_EUNIT_ASSERT_SPECIFIC_LEAVE( iHandler->SetAddrL( iAddr ), KErrArgument );

    delete iHandler;

    iHandler = NULL;  

    iHandler = CNATFWSocketHandler::NewL( *this, *this, KProtocolInetTcp, KTestPortTcp );

    iHandler->SetLocalAddress( iSocketServer, iAddr );    

    NATFW_EUNIT_ASSERT_NO_LEAVE( iHandler->SetAddrL( iAddr ) );

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_SetAddrL2L(  )

    {

    TSockAddr addr;

    RSocket socket;

    User::LeaveIfError( socket.Open( iSocketServer, KAfInet, KSockDatagram, KProtocolInetUdp ) );

    socket.SetLocalPort( KTestPortUdp );

    // RSocket is stubbed and currently the used ports are not stored

    // so we can make the next localportcall fail with this

    TBool shouldItFail = ETrue;

	socket.NextSetLocalPortShouldFail(ETrue, shouldItFail); // fails

    iHandler->SetMediaObserverL( this );

    iHandler->SetLocalAddress( iSocketServer, iAddr );

    // Will try to use same port as KTestPortUdp, but

    // since it's reserved, should switch to KTestPort2

    NATFW_EUNIT_ASSERT_NO_LEAVE( iHandler->SetAddrL( iAddr ) );

    iHandler->LocalAddress( addr );

    EUNIT_ASSERT_EQUALS( KTestPort2, addr.Port() );

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_ConnectLL(  )

    {

    iHandler->iSendState = iHandler->EConnected;

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_StartListeningLL(  )

    {

    iHandler->SetLocalAddress( iSocketServer, iAddr );

    NATFW_EUNIT_ASSERT_NO_LEAVE( iHandler->SetAddrL( iAddr ) );    

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_SendLL(  )

    {

    iHandler->SetLocalAddress( iSocketServer, iAddr );

    NATFW_EUNIT_ASSERT_NO_LEAVE( iHandler->SetAddrL( iAddr ) ); 

    iHandler->iSendState = iHandler->EConnected;

    iHandler->SetSendingStateL();           

    NATFW_EUNIT_ASSERT_NO_LEAVE( iHandler->SendL( KData ) );

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_SetLocalAddressL(  )

    {

    iHandler->SetLocalAddress( iSocketServer, iAddr );

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_DeactivateSendingL(  )

    {

    iHandler->DeactivateSending();

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_DeactivateReceivingL(  )

    {

    iHandler->DeactivateReceiving();

    }

void ut_cnatfwsockethandler::ut_cnatfwsockethandler_SetupTcpConnectionL(  )

    {

    iHandler->SetLocalAddress( iSocketServer, iAddr );

    NATFW_EUNIT_ASSERT_NO_LEAVE( iHandler->SetAddrL( iAddr ) );

    iHandler->OpenTcpConnectionL( ETcpSetupActive, *this );

    delete iHandler;

    iHandler = NULL;  

    iHandler = CNATFWSocketHandler::NewL( *this, *this, KProtocolInetTcp, KTestPortTcp );

    iHandler->SetLocalAddress( iSocketServer, iAddr );

    NATFW_EUNIT_ASSERT_NO_LEAVE( iHandler->SetAddrL( iAddr ) );

    iHandler->OpenTcpConnectionL( ETcpSetupPassive, *this );

    iHandler->iSendState = iHandler->EConnected;

    NATFW_EUNIT_ASSERT_SPECIFIC_LEAVE( iHandler->OpenTcpConnectionL( ETcpSetupPassive, *this ), KErrInUse );

    }      

//  TEST TABLE

EUNIT_BEGIN_TEST_TABLE(

    ut_cnatfwsockethandler,

    "Add test suite description here.",

    "UNIT" )

EUNIT_TEST(

    "NewL - test ",

    "CNATFWSocketHandler",

    "NewL",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_NewLL, Teardown)

EUNIT_TEST(

    "LocalAddress - test ",

    "CNATFWSocketHandler",

    "LocalAddress",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_LocalAddressL, Teardown)

EUNIT_TEST(

    "SetAddrL - test ",

    "CNATFWSocketHandler",

    "SetAddrL",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_SetAddrLL, Teardown)

EUNIT_TEST(

    "SetAddrL2 - port in use test ",

    "CNATFWSocketHandler",

    "SetAddrL",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_SetAddrL2L, Teardown)

EUNIT_TEST(

    "ConnectL - test ",

    "CNATFWSocketHandler",

    "ConnectL",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_ConnectLL, Teardown)

EUNIT_TEST(

    "StartListeningL - test ",

    "CNATFWSocketHandler",

    "StartListeningL",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_StartListeningLL, Teardown)

EUNIT_TEST(

    "SendL - test ",

    "CNATFWSocketHandler",

    "SendL",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_SendLL, Teardown)

EUNIT_TEST(

    "SetLocalAddress - test ",

    "CNATFWSocketHandler",

    "SetLocalAddress",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_SetLocalAddressL, Teardown)

EUNIT_TEST(

    "DeactivateSending - test ",

    "CNATFWSocketHandler",

    "DeactivateSending",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_DeactivateSendingL, Teardown)

EUNIT_TEST(

    "DeactivateReceiving - test ",

    "CNATFWSocketHandler",

    "DeactivateReceiving",

    "FUNCTIONALITY",

    SetupL, ut_cnatfwsockethandler_DeactivateReceivingL, Teardown) 

EUNIT_TEST(

    "SetupTcpConnection - test ",

    "CNATFWSocketHandler",

    "SetupTcpConnection",

    "FUNCTIONALITY",

    Setup2L, ut_cnatfwsockethandler_SetupTcpConnectionL, Teardown)    

EUNIT_END_TEST_TABLE

//  END OF FILE