root/experimental/components/Packetizer/trunk/Packetizer-metadata/tests/CreateExtract_testsuite.h @ 5394

#ifndef __CREATE_EXTRACT_TEST_H__
#define __CREATE_EXTRACT_TEST_H__

#include <cxxtest/TestSuite.h>
#include "src/PacketizerDSP.h"

//
// A simple test suite: Just inherit CxxTest::TestSuite and write tests!
//
// NOTE: " : public CxxTest::TestSuite" must be on the same line as the class
//       definition, otherwise the python script does not recognize it
//       as a test class
class CreateExtract_testsuite1 : public CxxTest::TestSuite,
    public PacketizerDSP
{
public:

    //
    // Basic test for creating and extracting a packet of type PACKET_RAW_400
    // 
    void test_CreateExtract_PACKET_RAW_400_01() {
        // generate input buffer
        char * input;
        input = (char*) calloc( 400, sizeof(char) );
        input[0] = 1;
        input[1] = 0;
        input[2] = 1;

        // generate packet buffer
        char * packet;
        packet = (char*) calloc( 1024, sizeof(char) );
        memset( packet, 0x01, 1024*sizeof(char) );
        unsigned int packet_length = 1024;

        // generate output buffer
        char * output;
        output = (char*) calloc( 1024, sizeof(char) );
        memset( output, 0x01, 1024*sizeof(char) );
        unsigned int output_length = 1024;

        // generate processing buffers
        char * b1 = (char*) calloc( MAX_PACKET_SIZE, sizeof(char) );
        char * b2 = (char*) calloc( MAX_PACKET_SIZE, sizeof(char) );
        SetBuffer(b1, b2, MAX_PACKET_SIZE);

        // Initialize properties
        packet_id = 6;
        CreatePacket(PACKET_RAW_400, input, 400, &packet[0], 1024, packet_length );

        packet_id = 0;

        bool packet_extracted;
        packet_extracted = ExtractPacket(packet, packet_length, output, 1024, output_length);

        TS_ASSERT_EQUALS( packet_extracted, true );
        TS_ASSERT_EQUALS( packet_id, 6 );
        TS_ASSERT_EQUALS( output_length, 400 );
        TS_ASSERT_SAME_DATA( input, output, 400 );

        // Test again, but set output to offset position in buffer and break data into pieces
        packet_id = 6;
        CreatePacket(PACKET_RAW_400, input, 400, &packet[55], 1024, packet_length );

        packet_id = 0;

        for (unsigned int i=0; i<1024; i+=16) {
            packet_extracted = ExtractPacket(packet+i, 16, output, 1024, output_length);
            if (packet_extracted)
                break;
        }

        TS_ASSERT_EQUALS( packet_extracted, true );
        TS_ASSERT_EQUALS( packet_id, 6 );
        TS_ASSERT_EQUALS( output_length, 400 );
        TS_ASSERT_SAME_DATA( input, output, 400 );


        
        free(input);
        free(packet);
        free(output);
        free(b1);
        free(b2);
    }

};


#endif