/*

* Copyright (c) 2007 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_CMusAvaCapabilityQuery.h"

#include "musavasettingsimp.h"

#include "musavaobserverimp.h"

#include "musavaavailabilityobserver.h"

#include "musavaavailabilityobserverimp.h"

#include "muslogger.h"

#include "mussesseioninformationapi.h"

//  EXTERNAL INCLUDES

#include <EscapeUtils.h>

#include <digia/eunit/eunitmacros.h>

#include <sipstrings.h>

#include <sipstrconsts.h>

#include <stringpool.h>

#include <sipservertransaction.h>

#include <siptransactionbase.h>

//  INTERNAL INCLUDES

#include "musavaoptionhandler.h"

#include "musavacapabilitytesthelper.h"

#include "CSipSseTestTls.h"

#include "musavaCapabilityExchange.h"

#include "musavacapabilitysipagent.h"

#include "MusAvaCapabilityQuery.h"

#include "musavaterminal.h"

#include "sipprofile.h"

#include "musavasharedobject.h"

#include "mussettingskeys.h"

#include "mussettings.h"

#include "musavasip.h"

#include "musavacapabilityqueryobserverimp.h"

// CONSTRUCTION

UT_CMusAvaCapabilityQuery* UT_CMusAvaCapabilityQuery::NewL()

    {

    UT_CMusAvaCapabilityQuery* self = UT_CMusAvaCapabilityQuery::NewLC(); 

    CleanupStack::Pop();

    return self;

    }

UT_CMusAvaCapabilityQuery* UT_CMusAvaCapabilityQuery::NewLC()

    {

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

    CleanupStack::PushL( self );

	self->ConstructL(); 

    return self;

    }

// Destructor (virtual by CBase)

UT_CMusAvaCapabilityQuery::~UT_CMusAvaCapabilityQuery()

    {

    }

// Default constructor

UT_CMusAvaCapabilityQuery::UT_CMusAvaCapabilityQuery()

    {

    }

// Second phase construct

void UT_CMusAvaCapabilityQuery::ConstructL()

    {

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

    // It generates the test case table.

    CEUnitTestSuiteClass::ConstructL();

    }

//  METHODS

void UT_CMusAvaCapabilityQuery::SetupL(  )

    {

    iSharedObject = CMusAvaSharedObject::GetSingletonL();

    iAvailabilityObserver = CMusAvaAvailabilityObserverImp::NewL();

    iConcreteSettings = CMusAvaSettingsImp::NewL();

    iOptionHandler = CMusAvaOptionHandler::NewL( *iAvailabilityObserver,

                                                 *iConcreteSettings );

    iExchange = iOptionHandler->iCapabilityExchange;

    iSipAgent = CMusAvaCapabilitySipAgent::NewL( *iExchange, 

                                                *iOptionHandler->iSharedObj,

                                                iOptionHandler->iObserver );

    iCapability = iOptionHandler->iSwisCapability;

    MSIPConnectionObserver& connectionObserver = 

                            iSharedObject->MusAvaSip().ConnectionObserver();

    CSIP& sip = iSharedObject->MusAvaSip().Sip();

    iSharedObject->MusAvaSip().CreateProfileL();    

    iProfile = iSharedObject->MusAvaSip().Profile();

    if ( !iProfile->iArray )

        {

        iProfile->iArray = new ( ELeave ) CDesC8ArrayFlat( 1 );

        iProfile->iArray->AppendL( _L8("sip:user@domain.com") );

        }

    if ( !iProfile->iRegisteredContact )

        {

        iProfile->iRegisteredContact = KRegisteredContact().AllocL();

        }

    TUint32 iapId;

    TInt err = iProfile->GetParameter( KSIPAccessPointId, iapId );

    iSIPConnection = CSIPConnection::NewL( sip, 

                                           iapId,

                                           connectionObserver );

    iQuery = CMusAvaCapabilityQuery::NewL( *iCapability,

                                            *iSIPConnection,

                                            *iProfile,

                                            KCapabilityTestSIPAddressUri_B );

    CSipSseTestTls::OpenL();

    iStorage = CSipSseTestTls::Storage();

    iOpVariantSetting = MultimediaSharingSettings::OperatorVariantSettingL();

    } 

void UT_CMusAvaCapabilityQuery::Teardown(  )

    {

    delete HelpFlags::iTestSipAddress;

    HelpFlags::iTestSipAddress = NULL;

    delete iQuery;

    CMusAvaSharedObject::DeleteSingleton();

    if ( iStorage )

        {

        iStorage->Clear();

        CSipSseTestTls::Close();

        iStorage = NULL;

        }

    delete iSipAgent;

    delete iOptionHandler;

    delete iConcreteSettings;

    delete iAvailabilityObserver;

    delete iSIPConnection;

    PropertyHelper::Close();

    }

void UT_CMusAvaCapabilityQuery::UT_CMusAvaCapabilityQuery_NewLL(  )

    {    

    // 1.basic test and iQuery is constructed already. Check it

    EUNIT_ASSERT( iQuery );

    // 2.test for sipaddress is empty,should leave

    delete iQuery;

    iQuery = NULL; // clean the previous one

    TRAPD( error, iQuery = CMusAvaCapabilityQuery::NewL( *iCapability,

                                              *iSIPConnection,

                                              *iProfile,

                                              KNullDesC ) );

    EUNIT_ASSERT( error == KErrNoMemory || error == KErrArgument );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT( !iQuery );   // iQuery Should be empty.

    // 3.Contact user part tests, getting KSIPRegisteredContact leaves

    delete iProfile->iRegisteredContact;

    iProfile->iRegisteredContact = NULL;    

    TRAP( error, iQuery = CMusAvaCapabilityQuery::NewL( *iCapability,

                                            *iSIPConnection,

                                            *iProfile,

                                            KCapabilityTestSIPAddressUri_B ) );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT( error == KErrNotFound );    

    EUNIT_ASSERT( !iQuery );   // iQuery Should be empty. 

    // 4.Contact user part tests, CSIPContactHeader::DecodeL has less than 1.

    iProfile->iRegisteredContact = KRegisteredContact().AllocL();

    HelpFlags::iEmptyContactHeader = ETrue;

    TRAP( error, iQuery = CMusAvaCapabilityQuery::NewL( *iCapability,

                                            *iSIPConnection,

                                            *iProfile,

                                            KCapabilityTestSIPAddressUri_B ) );

    HelpFlags::iEmptyContactHeader = EFalse;

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT( error == KErrNotFound );    

    EUNIT_ASSERT( !iQuery );   // iQuery Should be empty. 

    // 5.Contact user part tests ,CSIPContactHeader::DecodeL has 1 contact

    delete iQuery;

    iQuery = NULL; // clean the previous one    

    delete iProfile->iRegisteredContact;

    iProfile->iRegisteredContact = NULL;

    iProfile->iRegisteredContact = KRegisteredContact().AllocL();    

    HelpFlags::iTestSipAddress = CSIPAddress::DecodeL( KRegisteredContact ); 

    HelpFlags::iUseTestSipAddress = ETrue;

    TRAP( error, iQuery = CMusAvaCapabilityQuery::NewL( *iCapability,

                                            *iSIPConnection,

                                            *iProfile,

                                            KCapabilityTestSIPAddressUri_B ) );             

    HelpFlags::iUseTestSipAddress = EFalse;

    if ( error == KErrNoMemory ) User::Leave( error );

    delete HelpFlags::iTestSipAddress;

    HelpFlags::iTestSipAddress = NULL;

    EUNIT_ASSERT( error == KErrNone );

    EUNIT_ASSERT( iQuery );

    // 6.Contact user part tests ,CSIPContactHeader::DecodeL has 1 contact but

    // sip address empty

    delete iQuery;

    iQuery = NULL; // clean the previous one

    delete iProfile->iRegisteredContact;

    iProfile->iRegisteredContact = NULL;

    iProfile->iRegisteredContact = KRegisteredContact().AllocL();

    HelpFlags::iUseTestSipAddress = ETrue;

    TRAP(error,iQuery = CMusAvaCapabilityQuery::NewL( *iCapability,

                                            *iSIPConnection,

                                            *iProfile,

                                            KCapabilityTestSIPAddressUri_B ));   

    HelpFlags::iUseTestSipAddress = EFalse;

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT( error == KErrNotFound );

    EUNIT_ASSERT( !iQuery ); // iQuery Should be empty. 

    }

void UT_CMusAvaCapabilityQuery::UT_CMusAvaCapabilityQuery_ExecuteLL(  )

    {

    /* includes testing of functions:

     * Prepare(),

     * DoCompletedL() case KMUSAVASIP200

     * DoCompleted200OKL() 

     * ValidateCodecL()

     * ValidateAttributesL()

     * ValidateContactL(),

     * Retrying()

     * Capability()

     * SetState()

     * Terminal()

     * ExecuteL()

     * CompletedL()

     * State()

     * Result()

     * AddAcceptContactHeaderL()

     * AddContactHeaderL()

     * ResetAndDestroyTrx()

     * LocalAddress()

     */

    EUNIT_ASSERT( iExchange->Terminals().Count() == 1 );

    CMusAvaTerminal& terminal = iQuery->Terminal();

    EUNIT_ASSERT( terminal.iQueries == 1 );

    EUNIT_ASSERT( terminal.iQuery == NULL );

    EUNIT_ASSERT( iQuery->Terminal().Uri().Compare( KCapabilityTestSIPAddressUri8_B ) == 0 );

    iQuery->SetState( CMusAvaCapabilityQueryBase::ECapabilityQueryPrepared );

    TRAPD( error, iQuery->ExecuteL() );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT( error != KErrNone );

    iQuery->SetState( CMusAvaCapabilityQueryBase::ECapabilityQueryCreated );

    iQuery->ExecuteL( );

    EUNIT_ASSERT( iQuery->State() ==  

                    CMusAvaCapabilityQueryBase::ECapabilityQueryExecuting );

    EUNIT_ASSERT( !iQuery->iTimer );

    EUNIT_ASSERT( iQuery->iTrx );

    EUNIT_ASSERT( iStorage->iClientRequest->Method() == 

                            SIPStrings::StringF( SipStrConsts::EOptions ) );

    EUNIT_ASSERT( iStorage->iClientRequest->MessageElements().Content().Length() 

                != 0 );

    _LIT8(contentType,"application/sdp");            

    EUNIT_ASSERT(

        iStorage->iClientRequest->MessageElements().ContentType() );

    EUNIT_ASSERT( 

            iStorage->iClientRequest->MessageElements().UserHeaders().Count() 

                == 6 );

    MUS_HEADER_ASSERT( iStorage->iClientRequest, 0, 

                    KCapabilityTestAcceptContactHeader );

    MUS_HEADER_ASSERT( iStorage->iClientRequest, 5, 

                    KCapabilityTestContactHeader_A1Updated );

    MUS_HEADER_ASSERT( iStorage->iClientRequest, 3, 

                    KCapabilityTestAcceptHeader );

    EUNIT_ASSERT( iOptionHandler->State() == 

                            MMusAvaObserver::EMusAvaStatusNotExecuted );

    // set response 200 OK FT

    CapabilityTestHelper::SetResponseL( *iQuery->iTrx, KMUSAVASIP200, 

                                        SipStrConsts::EPhraseOk,

                                        KCapabilityTestAddress_B,

                                        KCapabilitySwisFeature,

                                        KCapabilityTestTerminalID_B,

                                        KPAssertedIdentity_B,

                                        KCapabilityTestSDP );

    iQuery->CompletedL( *iQuery->iTrx );

    MusSettingsKeys::TOperatorVariant op = MultimediaSharingSettings::OperatorVariantSettingL();

    EUNIT_ASSERT( op == MusSettingsKeys::EStandard );

    EUNIT_ASSERT( iQuery->iState == 

                CMusAvaCapabilityQueryBase::ECapabilityQueryCompleted );

    EUNIT_ASSERT( !iQuery->Retrying() );

    EUNIT_ASSERT( iQuery->iTimer == NULL );

    EUNIT_ASSERT( iStorage->iClientRequest->Method() == 

                            SIPStrings::StringF( SipStrConsts::EOptions ) );

    EUNIT_ASSERT( iStorage->iClientRequest->MessageElements().Content().Length()

             > 0 );

    EUNIT_ASSERT( iStorage->iClientRequest->MessageElements().ContentType() );

    EUNIT_ASSERT( iStorage->iClientRequest->RemoteUri().Uri().UriDes().Compare( 

            KCapabilityTestSIPAddressUri8_B ) == 0);

    EUNIT_ASSERT( iStorage->iClientRequest->MessageElements().UserHeaders().

            Count() == 6 );

    MUS_HEADER_ASSERT( iStorage->iClientRequest, 

                    0, 

                    KCapabilityTestAcceptContactHeader );

    MUS_HEADER_ASSERT( iStorage->iClientRequest, 

                    5, 

                    KCapabilityTestContactHeader_A1Updated );

    MUS_HEADER_ASSERT( iStorage->iClientRequest, 3, 

                    KCapabilityTestAcceptHeader );

    EUNIT_ASSERT( iOptionHandler->State() == 

                            MMusAvaObserver::EMusAvaStatusAvailable );

    iQuery->SetState( CMusAvaCapabilityQueryBase::ECapabilityQueryCompleted );                                

    TRAP( error, iQuery->ExecuteL() );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT ( error == KErrNone );

    iQuery->DoRetryL();

    TRAP( error, iQuery->ExecuteL() );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT ( error == KErrNone );

    iQuery->DoRetryL();

    // set TerminalId for CMusAvaCapabilityExchange, terminal Id will be added to headers

    CMusAvaCapabilityQueryObserverImp* capabQueryObserver = 

                        new ( ELeave ) CMusAvaCapabilityQueryObserverImp();

    CleanupStack::PushL( capabQueryObserver );                        

    CMusAvaCapabilityExchange* exchange =

                        CMusAvaCapabilityExchange::NewL( 

                                            KCapabilityTestTerminalID_A, 

                                            *capabQueryObserver );

    CleanupStack::PushL( exchange );                        

    CMusAvaCapability* capability = CMusAvaCapability::NewL( *exchange );

    CleanupStack::PushL( capability );                        

    CMusAvaCapabilityQuery* query = CMusAvaCapabilityQuery::NewL( *capability,

                                            *iSIPConnection,

                                            *iProfile,

                                            KCapabilityTestSIPAddressUri_B );

    CleanupStack::PushL( query ); 

    query->SetState( CMusAvaCapabilityQueryBase::ECapabilityQueryCreated );

    query->ExecuteL();

    EUNIT_ASSERT_EQUALS( iStorage->iClientRequest->MessageElements().UserHeaders().Count(), 7 ); 

    MUS_HEADER_ASSERT( iStorage->iClientRequest, 6, 

                    KCapabilityTestUserAgent_A );

    CleanupStack::PopAndDestroy( query );

    CleanupStack::PopAndDestroy( capability );

    CleanupStack::PopAndDestroy( exchange );

    CleanupStack::PopAndDestroy( capabQueryObserver );

    }

void UT_CMusAvaCapabilityQuery::UT_CMusAvaCapabilityQuery_CanceledL(  )

    {

    iQuery->ExecuteL( );

    EUNIT_ASSERT( iQuery->State() ==  

                    CMusAvaCapabilityQueryBase::ECapabilityQueryExecuting );

    // set response 200 OK FT

    CapabilityTestHelper::SetResponseL( *iQuery->iTrx, 

                                        KMUSAVASIP200,

                                        SipStrConsts::EPhraseOk,

                                        KCapabilityTestAddress_B,

                                        KCapabilitySwisFeature,

                                        KCapabilityTestTerminalID_B,

                                        KPAssertedIdentity_B,

                                        KCapabilityTestSDP );

    iQuery->Canceled( *iQuery->iTrx );

    EUNIT_ASSERT( iQuery->State() == 

                CMusAvaCapabilityQueryBase::ECapabilityQueryCompleted );

    EUNIT_ASSERT( iQuery->Result() == KCapabilityCapabilitesNotFound );

    EUNIT_ASSERT( iOptionHandler->State() == 

                            MMusAvaObserver::EMusAvaOptionNotAvailable );

    // give different response than iTrx                                        

    iQuery->SetState( CMusAvaCapabilityQueryBase::ECapabilityQueryCreated );

    iQuery->ExecuteL();

    CSIPClientTransaction* cliTransaction = NULL;

    cliTransaction = CSIPClientTransaction::NewL( iQuery->iTrx->Type() );

    CleanupStack::PushL( cliTransaction );

    CapabilityTestHelper::SetResponseL( *cliTransaction, 

                                        KMUSAVASIPForbidden, 

                                        SipStrConsts::EPhraseOk,

                                        KCapabilityTestAddress_B,

                                        KCapabilitySwisFeature,

                                        KCapabilityTestTerminalID_B,

                                        KPAssertedIdentity_B,

                                        KCapabilityTestSDP );

    iQuery->Canceled( *cliTransaction );                              

    EUNIT_ASSERT_NOT_EQUALS( iQuery->State(), CMusAvaCapabilityQueryBase::ECapabilityQueryCompleted );              

    iQuery->ResetAndDestroyTrx();

    iQuery->Canceled( *cliTransaction );

    EUNIT_ASSERT_NOT_EQUALS( iQuery->State(), CMusAvaCapabilityQueryBase::ECapabilityQueryCompleted );                  

    CleanupStack::PopAndDestroy( cliTransaction );                            

    }

void UT_CMusAvaCapabilityQuery::UT_CMusAvaCapabilityQuery_CompletedLL(  )

    {

    TRAPD( error, RProperty::Define(

                            NMusSessionInformationApi::KCategoryUid,

                            NMusSessionInformationApi::KMUSForbidden,

                            RProperty::EInt,

                            0 ) );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT ( error == KErrNone );

    TRAP( error, RProperty::Set(

    	NMusSessionInformationApi::KCategoryUid,

    	NMusSessionInformationApi::KMUSForbidden,

    	( TInt ) NMusSessionInformationApi::EMUSAllowed ) );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT ( error == KErrNone );

    // Test for forbidden response (403)

    //Cingular specific case && 403 Forbidden

    MultimediaSharingSettings::SetOperatorVariantSettingL( 

                                        MusSettingsKeys::EOperatorSpecific );

    iQuery->ExecuteL( );

    EUNIT_ASSERT( iQuery->State() ==  

                    CMusAvaCapabilityQueryBase::ECapabilityQueryExecuting );

    // set response 200 OK FT

    CapabilityTestHelper::SetResponseL( *iQuery->iTrx, 

                                        KMUSAVASIPForbidden, 

                                        SipStrConsts::EPhraseOk,

                                        KCapabilityTestAddress_B,

                                        KCapabilitySwisFeature,

                                        KCapabilityTestTerminalID_B,

                                        KPAssertedIdentity_B,

                                        KCapabilityTestSDP );

    iQuery->CompletedL( *iQuery->iTrx );

    EUNIT_ASSERT( iQuery->State() == 

                CMusAvaCapabilityQueryBase::ECapabilityQueryCompleted );

    EUNIT_ASSERT( iQuery->Result() == KCapabilityCapabilitiesForbidden );

    EUNIT_ASSERT( iOptionHandler->State() == 

                            MMusAvaObserver::EMusAvaFailureCode );

    // check if it is "forbidden"

    TInt forbidden;

    TRAP( error, RProperty::Get( NMusSessionInformationApi::KCategoryUid,

	                    NMusSessionInformationApi::KMUSForbidden,

	                    forbidden ) );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT ( error == KErrNone );

	EUNIT_ASSERT_EQUALS( NMusSessionInformationApi::EMUSForbidden, forbidden );                            

    // set back the forbidden flag to allowed	

    TRAP( error, RProperty::Set( NMusSessionInformationApi::KCategoryUid,

                        NMusSessionInformationApi::KMUSForbidden,

                        ( TInt ) NMusSessionInformationApi::EMUSAllowed ) );

    if ( error == KErrNoMemory ) User::Leave( error );

    EUNIT_ASSERT ( error == KErrNone );

    MultimediaSharingSettings::SetOperatorVariantSettingL( 

                                        MusSettingsKeys::EStandard );

    // give different response than iTrx                                        

    iQuery->SetState( CMusAvaCapabilityQueryBase::ECapabilityQueryCreated );

    iQuery->ExecuteL( );

    CSIPClientTransaction* cliTransaction = NULL;

    cliTransaction = CSIPClientTransaction::NewL( iQuery->iTrx->Type() );

    CleanupStack::PushL( cliTransaction );

    CapabilityTestHelper::SetResponseL( *cliTransaction, 

                                        KMUSAVASIPForbidden, 

                                        SipStrConsts::EPhraseOk,

                                        KCapabilityTestAddress_B,

                                        KCapabilitySwisFeature,

                                        KCapabilityTestTerminalID_B,

                                        KPAssertedIdentity_B,

                                        KCapabilityTestSDP );

    iQuery->CompletedL( *cliTransaction );

    EUNIT_ASSERT_NOT_EQUALS( iQuery->State(), CMusAvaCapabilityQueryBase::ECapabilityQueryCompleted );              

    iQuery->ResetAndDestroyTrx();

    iQuery->CompletedL( *cliTransaction );

    EUNIT_ASSERT_NOT_EQUALS( iQuery->State(), CMusAvaCapabilityQueryBase::ECapabilityQueryCompleted );                  

    CleanupStack::PopAndDestroy( cliTransaction );                                        

    // set response 199

    iQuery->SetState( CMusAvaCapabilityQueryBase::ECapabilityQueryCreated );

    iQuery->ExecuteL( ); 

    CapabilityTestHelper::SetResponseL( *iQuery->iTrx, 

                                        KMUSAVASIP200-1, 

                                        SipStrConsts::EPhraseOk,

                                        KCapabilityTestAddress_B,

                                        KCapabilitySwisFeature,

                                        KCapabilityTestTerminalID_B,

                                        KPAssertedIdentity_B,