/*

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

*

*/

// ADD sentence "friend class UT_CAmrPayloadFormatRead" to AmrPayloadFormatRead.h

// 

//  CLASS HEADER

#include "UT_CAmrPayloadFormatRead.h"

#include "MccRtpDataSink.h"

#include "mccuids.hrh"

#include "Mccdldatapath.h"

#include "MccRtpDataSource.h"

#include "mccresourcepool.h"

#include "mccredpayloadread.h"

#include "amrpayloaddecoder.h"

//  EXTERNAL INCLUDES

#include <e32base.h>

#include <mmf/server/mmfvideoframebuffer.h>

#include <digia/eunit/eunitmacros.h>

#ifdef EUNIT_ALLOC_TEST_ON

    #include <digia/eunit/CEunitAllocTestCaseDecorator.h>   //for using EUNIT_ALLOC_TEST

#endif

//  INTERNAL INCLUDES

#include "amrpayloadformatread.h"

const TUid KRtpDataSource            =  { KImplUidRtpDataSource   }; 

const TUid KRtpDataSink              =  { KImplUidRtpDataSink     };

// CONSTRUCTION

UT_CAmrPayloadFormatRead* UT_CAmrPayloadFormatRead::NewL( )

    {

    UT_CAmrPayloadFormatRead* self = UT_CAmrPayloadFormatRead::NewLC();

    CleanupStack::Pop();

    return self;

    }

UT_CAmrPayloadFormatRead* UT_CAmrPayloadFormatRead::NewLC()

    {

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

    CleanupStack::PushL( self );

	self->ConstructL(); 

    return self;

    }

// Destructor (virtual by CBase)

UT_CAmrPayloadFormatRead::~UT_CAmrPayloadFormatRead()

    {

    }

// Default constructor

UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead()

    {

    }

// Second phase construct

void UT_CAmrPayloadFormatRead::ConstructL()

    {

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

    // It generates the test case table.

    CEUnitTestSuiteClass::ConstructL();

    }

//  METHODS

void UT_CAmrPayloadFormatRead::SetupAL(  )

    { 

    SetupL();

    iAlloc = ETrue; 

    }

void UT_CAmrPayloadFormatRead::SetupL(  )

    {

    iFourCC.Set( TFourCC( ' ','A','M','R' ) );

    iAlloc = EFalse; 

    iRtpDataSourceStub = MDataSource::NewSourceL( KRtpDataSource, KNullDesC8 );

    iMediaId = KUidMediaTypeAudio;

    MAsyncEventHandler* eventHandler = NULL;

    CMccResourcePool* mccresource = NULL;

    iDLDataPathStub = CMccDlDataPath::NewL( eventHandler, mccresource, iMediaId ); 

    iRead = CAmrPayloadFormatRead::NewL( iRtpDataSourceStub );

    iBuf = CMMFDescriptorBuffer::NewL( 300 );

    iBuffi = CMMFDescriptorBuffer::NewL( 300 );

    iBuffi->Data().Append( _L("1234512345") ); 

    iConsumer = MDataSink::NewSinkL( KRtpDataSink, KNullDesC8 );

    iCodec.iFourCC.Set(TFourCC(' ','A','M','R') );

    iCodec.iBitrate = 12200;

    iCodec.iCodecMode = KAmrCodecModeOctetAlign;

    iCodec.iPtime = 160;

    iCodec.iMaxPtime = 200;

    iCodec.iPayloadType = 106;

    iCodec.iRedundancyCount = 0;

    iCodec.iEnableDTX = EFalse;

    //iCodec.iFwStream = ETrue;

    iCodec.iRedundantPayload = 0;

    iCodec.iHwFrameTime = 20;

    iCodec.iFrameSize = 32;

    iCodec.iAlgoUsed = 0;

    iCodec.iPayloadFormatEncoder = KImplUidAmrPayloadFormatEncode;

    iCodec.iPayloadFormatDecoder = KImplUidAmrPayloadFormatDecode;

    TMccCodecInfoBuffer buffer( iCodec );

    RDebug::Print( _L("- Algo used       = %d"), iCodec.iAlgoUsed ); // 0 jos redundanssi ei päällä, 2 jos haluat asettaa sen päälle

    RDebug::Print( _L("- Bitrate         = %d"), iCodec.iBitrate ); // 64000

    RDebug::Print( _L("- CNFrameSize     = %d"), iCodec.iCNFrameSize ); // 0

    RDebug::Print( _L("- CodecMode       = %d"), iCodec.iCodecMode ); // 3

    RDebug::Print( _L("- EnableDTX       = %d"), iCodec.iEnableDTX ); // 1

    RDebug::Print( _L("- FourCC          = %x"), iCodec.iFourCC ); // 31313747

    RDebug::Print( _L("- FrameSize       = %d"), iCodec.iFrameSize ); // 0

    RDebug::Print( _L("- HWFrameTime     = %d"), iCodec.iHwFrameTime ); // 20

    RDebug::Print( _L("- MaxPTime        = %d"), iCodec.iMaxPtime ); // 200

    RDebug::Print( _L("- PayloadType     = %d"), iCodec.iPayloadType ); // 106

    RDebug::Print( _L("- PTime           = %d"), iCodec.iPtime ); // 160

    RDebug::Print( _L("- RedundancyCount = %d"), iCodec.iRedundancyCount ); // 0 jos redundanssi ei päällä, 1 jos redundanssi päällä

    RDebug::Print( _L("- RedPayload      = %d"), iCodec.iRedundantPayload ); // 0 jos redundanssi ei päällä, 97-127 jos redundanssi päällä

    iRead->SetSourceDataTypeCode( iFourCC, KUidMediaTypeAudio );

    iRead->SourceThreadLogon( *this );

    iRead->ConfigurePayloadFormatL( buffer );

    iRead->NegotiateSourceL( *iDLDataPathStub );

    iRead->SourcePrimeL();

    iRead->SourcePlayL();

    } 

void UT_CAmrPayloadFormatRead::SetupWbAL(  )

    { 

    SetupWbL();

    iAlloc = ETrue; 

    }

void UT_CAmrPayloadFormatRead::SetupWbL(  )

    {

    iFourCC.Set( TFourCC( ' ','A','W','B' ) );

    iAlloc = EFalse; 

    iRtpDataSourceStub = MDataSource::NewSourceL( KRtpDataSource, KNullDesC8 );

    iMediaId = KUidMediaTypeAudio;

    MAsyncEventHandler* eventHandler = NULL;

    CMccResourcePool* mccresource = NULL;

    iDLDataPathStub = CMccDlDataPath::NewL( eventHandler, mccresource, iMediaId ); 

    iRead = CAmrPayloadFormatRead::NewL( iRtpDataSourceStub );

    iBuf = CMMFDescriptorBuffer::NewL( 300 );

    iBuffi = CMMFDescriptorBuffer::NewL( 300 );

    iBuffi->Data().Append( _L("1234512345") ); 

    iConsumer = MDataSink::NewSinkL( KRtpDataSink, KNullDesC8 );

    iCodec.iFourCC.Set(TFourCC(' ','A','W','B') );

    iCodec.iBitrate = 23850;

    iCodec.iCodecMode = KAmrCodecModeOctetAlign;

    iCodec.iPtime = 160;

    iCodec.iMaxPtime = 200;

    iCodec.iPayloadType = 106;

    iCodec.iRedundancyCount = 0;

    iCodec.iEnableDTX = EFalse;

    //iCodec.iFwStream = ETrue;

    iCodec.iRedundantPayload = 0;

    iCodec.iHwFrameTime = 20;

    iCodec.iFrameSize = 64;

    iCodec.iAlgoUsed = 0;

    iCodec.iPayloadFormatEncoder = KImplUidAmrPayloadFormatEncode;

    iCodec.iPayloadFormatDecoder = KImplUidAmrPayloadFormatDecode;

    TMccCodecInfoBuffer buffer( iCodec );

    RDebug::Print( _L("- Algo used       = %d"), iCodec.iAlgoUsed ); // 0 jos redundanssi ei päällä, 2 jos haluat asettaa sen päälle

    RDebug::Print( _L("- Bitrate         = %d"), iCodec.iBitrate ); // 64000

    RDebug::Print( _L("- CNFrameSize     = %d"), iCodec.iCNFrameSize ); // 0

    RDebug::Print( _L("- CodecMode       = %d"), iCodec.iCodecMode ); // 3

    RDebug::Print( _L("- EnableDTX       = %d"), iCodec.iEnableDTX ); // 1

    RDebug::Print( _L("- FourCC          = %x"), iCodec.iFourCC ); // 31313747

    RDebug::Print( _L("- FrameSize       = %d"), iCodec.iFrameSize ); // 0

    RDebug::Print( _L("- HWFrameTime     = %d"), iCodec.iHwFrameTime ); // 20

    RDebug::Print( _L("- MaxPTime        = %d"), iCodec.iMaxPtime ); // 200

    RDebug::Print( _L("- PayloadType     = %d"), iCodec.iPayloadType ); // 106

    RDebug::Print( _L("- PTime           = %d"), iCodec.iPtime ); // 160

    RDebug::Print( _L("- RedundancyCount = %d"), iCodec.iRedundancyCount ); // 0 jos redundanssi ei päällä, 1 jos redundanssi päällä

    RDebug::Print( _L("- RedPayload      = %d"), iCodec.iRedundantPayload ); // 0 jos redundanssi ei päällä, 97-127 jos redundanssi päällä

    iRead->SetSourceDataTypeCode( iFourCC, KUidMediaTypeAudio );

    iRead->SourceThreadLogon( *this );

    iRead->ConfigurePayloadFormatL( buffer );

    iRead->NegotiateSourceL( *iDLDataPathStub );

    iRead->SourcePrimeL();

    iRead->SourcePlayL();

    } 

void UT_CAmrPayloadFormatRead::Teardown(  )

    {

    delete iBuffi;

    delete iDLDataPathStub;   

    delete iRtpDataSourceStub;

    delete iRead; 

    delete iBuf;  

    delete iConsumer;

    delete iFrameData;

    iFrameData = NULL;

    REComSession::FinalClose();

    }

// HELPERS

void UT_CAmrPayloadFormatRead::CreateFramesL( TInt aNumFrames, TBool aOkData )

    {

    iRead->iNumOfFrames = aNumFrames;

    iRead->iCurrentFrame = 0;

    __ASSERT_ALWAYS( iRead->iPayloadDecoder, User::Leave( KErrNotReady ) );

    for ( TInt i = 0; i < aNumFrames; i++ )

        {

        // Add some damaged frames

        TAmrFrame frame;

        if ( aOkData )

            {

            if ( !iFrameData )

                {

                iFrameData = HBufC8::NewL( 300 );

                iFrameData->Des().Append( _L8("ð¼¼¼¼¼¼¼<‘(˜Èž\x01�À‰å/áPv\x14€") );

                iFrameData->Des().SetLength( iFrameData->Des().MaxLength() );

                }

            frame.iFrameType = EAmrFrame12_2;

            frame.iFrameQualityInd = 1;

            frame.iFrameData = const_cast<TUint8*>( iFrameData->Des().Ptr() );

            }

        else

            {

            frame.iFrameType = EAmrFrameNoData;

            frame.iFrameQualityInd = 0;

            }

        iRead->iPayloadDecoder->iFrames[ i ] = frame;

        }

    }

// TEST FUNCTIONS

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_CancelDlRequestL(  )

    {

    EUNIT_ASSERT_NO_LEAVE( iRead->CancelDlRequest() );

    delete iRead->iFrameBufferOne;

    iRead->iFrameBufferOne = NULL;

    EUNIT_ASSERT_NO_LEAVE( iRead->CancelDlRequest() );

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_SendDataToSinkLL(  )

    {

    if ( !iAlloc )

        {       

        EUNIT_ASSERT_NO_LEAVE( iRead->FillBufferL( iBuffi, iDLDataPathStub, iMediaId ) );

        EUNIT_ASSERT_NO_LEAVE( iRead->SendDataToSinkL() );

        EUNIT_ASSERT_NO_LEAVE( iRead->FillBufferL( iBuffi, iDLDataPathStub, iMediaId ) );

        EUNIT_ASSERT_NO_LEAVE( iRead->SendDataToSinkL() );

        iRead->iBufferToReadExists = ETrue;

        iRead->iFrameBufferOne->SetLastBuffer( ETrue );

        EUNIT_ASSERT_NO_LEAVE( iRead->SendDataToSinkL() );

        iRead->iBufferToReadExists = ETrue;

        iRead->iFrameBufferOne->SetLastBuffer( ETrue );

        EUNIT_ASSERT_NO_LEAVE( iRead->SendDataToSinkL() );

        }        

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->FillBufferL( iBuffi, iDLDataPathStub, iMediaId ) );

        EUNIT_ASSERT_NO_LEAVE( iRead->SendDataToSinkL() );       

        } 

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_FillSinkBufferL(  )

    {

    if ( !iAlloc )

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->FillSinkBufferL() );

        iRead->iBufferToReadExists = 1;

        EUNIT_ASSERT_NO_LEAVE( iRead->FillSinkBufferL() );  

        iRead->iCurrentBuffer = CAmrPayloadFormatRead::EBufferTwo;

        // No data

        iRead->iFrameBufferTwo->Data().SetLength( 0 );  

        EUNIT_ASSERT_NO_LEAVE( iRead->FillSinkBufferL() );  

        EUNIT_ASSERT( iRead->iStateMachine->CurrentState() == EDecodeIdle );

        // Some data to forward

        iRead->iBufferToRead = iBuf;

        iRead->iFrameBufferTwo->Data().SetLength( 1 );  

        CreateFramesL( 1, ETrue );

        EUNIT_ASSERT_NO_LEAVE( iRead->FillSinkBufferL() );

        }        

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->FillSinkBufferL() );

        } 

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_FillSourceBufferLL(  )

    {

    EUNIT_ASSERT_NO_LEAVE( iRead->FillSourceBufferL() );

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_StreamsL(  )

    {

    TUid uid = {0x101F7D07};

    if ( !iAlloc )

        {

        EUNIT_ASSERT_EQUALS( iRead->Streams( KUidMediaTypeAudio ), 1 );

        EUNIT_ASSERT_EQUALS( iRead->Streams( KUidMediaTypeVideo ), 0 );

        EUNIT_ASSERT_EQUALS( iRead->Streams( KUidMediaTypeMidi ), 0 );

        EUNIT_ASSERT_EQUALS( iRead->Streams( uid ), 0 );

        }

    else

        {

        EUNIT_ASSERT_EQUALS( iRead->Streams( KUidMediaTypeAudio ), 1 );

        EUNIT_ASSERT_EQUALS( iRead->Streams( KUidMediaTypeVideo ), 0 );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_FrameTimeIntervalL(  )

    {

    if ( !iAlloc )

        {

        RDebug::Print( _L("**FrameInterval = %d"), iRead->FrameTimeInterval( KUidMediaTypeAudio ).Int64() );

        EUNIT_ASSERT_EQUALS( iRead->FrameTimeInterval( KUidMediaTypeAudio ).Int64(), 20 );

        EUNIT_ASSERT_EQUALS( iRead->FrameTimeInterval( KUidMediaTypeVideo ).Int64(), 0 );

        }

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->FrameTimeInterval( KUidMediaTypeAudio ) );

        EUNIT_ASSERT_NO_LEAVE( iRead->FrameTimeInterval( KUidMediaTypeVideo ) );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_DurationL(  )

    {

    if ( !iAlloc )

        {

        /*

        EUNIT_ASSERT_EQUALS( iRead->Duration( KUidMediaTypeAudio ).Int64(), 1000000000 );

        EUNIT_ASSERT_EQUALS( iRead->Duration( KUidMediaTypeMidi ).Int64(), 1000000000 );

        */

        EUNIT_ASSERT_EQUALS( iRead->Duration( KUidMediaTypeMidi ).Int64(), 0 );

        }

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->Duration( KUidMediaTypeAudio ) );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_NumChannelsL(  )

    {

    if ( !iAlloc )

        {

        EUNIT_ASSERT_EQUALS( iRead->NumChannels( ), KMono );

        }

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->NumChannels( ) );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_SampleRateL(  )

    {

    if ( !iAlloc )

        {

        EUNIT_ASSERT_EQUALS( iRead->SampleRate(), iRead->iSamplingRate );

        }

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->SampleRate() );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_SetSampleRateL(  )

    {

    TUint sample(0);

    if ( !iAlloc )

        {

        sample = iRead->SampleRate(); // 8000

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( sample ), KErrNone );

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( sample + 1 ), KErrNotSupported );

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( sample - 8000 ), KErrNotSupported );

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( 8000 ), KErrNone );

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( 16000 ), KErrNotSupported );

        }

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->SetSampleRate( sample ) );

        EUNIT_ASSERT_NO_LEAVE( iRead->SetSampleRate( sample + 1 ) );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_SetSampleRateWbL(  )

    {

    TUint sample(0);

    if ( !iAlloc )

        {

        sample = iRead->SampleRate(); // 16000

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( sample ), KErrNone );

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( sample + 1 ), KErrNotSupported );

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( sample - 8000 ), KErrNotSupported );

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( 8000 ), KErrNotSupported );

        EUNIT_ASSERT_EQUALS( iRead->SetSampleRate( 16000 ), KErrNone );

        }

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->SetSampleRate( sample ) );

        EUNIT_ASSERT_NO_LEAVE( iRead->SetSampleRate( sample + 1 ) );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_FillBufferLL(  )

    {

    if ( !iAlloc )

        {

        iRead->iCurrentBuffer = iRead->EBufferOne;

        EUNIT_ASSERT_NO_LEAVE( iRead->FillBufferL( iBuffi, iDLDataPathStub, KUidMediaTypeAudio ) );

        EUNIT_ASSERT_LEAVE( iRead->FillBufferL( iBuffi, iDLDataPathStub, KUidMediaTypeVideo ) ); // KErrNotSupported

        iRead->iCurrentBuffer = iRead->EBufferTwo;

        EUNIT_ASSERT_NO_LEAVE( iRead->FillBufferL( iBuffi, iDLDataPathStub, KUidMediaTypeAudio ) );

        EUNIT_ASSERT_LEAVE( iRead->FillBufferL( NULL, iDLDataPathStub, KUidMediaTypeAudio ) );

        }        

    else

        {

        EUNIT_ASSERT_LEAVE( iRead->FillBufferL( iBuffi, iDLDataPathStub, KUidMediaTypeMidi ) );    

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_SourcePrimeLL(  ) 

    {

    if ( !iAlloc )

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->SourcePrimeL() );

        }

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->SourcePrimeL() );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_SourcePlayLL(  )

    {

    if ( !iAlloc )

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->SourcePlayL() );

        }

    else

        {

        EUNIT_ASSERT_NO_LEAVE( iRead->SourcePlayL() );

        }

    }

void UT_CAmrPayloadFormatRead::UT_CAmrPayloadFormatRead_SourcePauseLL(  )

    {