av/nal/nal_test.go

/*
NAME
  MpegTs.go - provides a data structure intended to encapsulate the properties
  of an MpegTs packet.

DESCRIPTION
  See Readme.md

AUTHOR
  Saxon Nelson-Milton <saxon.milton@gmail.com>

LICENSE
  MpegTs.go is Copyright (C) 2017 the Australian Ocean Lab (AusOcean)

  It is free software: you can redistribute it and/or modify them
  under the terms of the GNU General Public License as published by the
  Free Software Foundation, either version 3 of the License, or (at your
  option) any later version.

  It is distributed in the hope that it will be useful, but WITHOUT
  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 for more details.

  You should have received a copy of the GNU General Public License
  along with revid in gpl.txt.  If not, see [GNU licenses](http://www.gnu.org/licenses).
*/

package nal

import (
  "testing"
)

var parseInput = []byte{
	0x6C, // 3NalUnitBits = 101(5), Fragment type = 1100 (type = 12 )
	0x94, // starbit = 1, endbit = 0, Reservedbit = 0, 5NalUnitBits = 10100 (20)
	0x8E, // 10001110 random frame byte
	0x26, // 00100110 random frame byte
	0xD0, // 11010000 random frame byte
}

var expectedParsing = []interface{}{
	byte(3),
	byte(12),
	bool(true),
	bool(false),
	bool(false),
	byte(20),
	[]byte{0x8E, 0x26, 0xD0},
}

const (
	nalTestType = 12
)

func TestNalFragmentParsing(t *testing.T) {
	nalUnit := ParseNALFragment(parseInput)
	value := reflect.ValueOf(*nalUnit)
	length := value.NumField()
	fields := make([]interface{}, length)
	for ii := 0; ii < length; ii++ {
		fields[ii] = value.Field(ii).Interface()
	}
	for ii := range fields {
		if !reflect.DeepEqual(fields[ii], expectedParsing[ii]) {
			t.Errorf("Bad Parsing! Field: %v wanted: %v got: %v\n", ii, expectedParsing[ii],
				fields[ii])
		}
	}
}

func TestNalFragmentToByteSlice(t *testing.T) {
	nalUnit := ParseNALFragment(parseInput)
	output := nalUnit.ToByteSlice()
	for ii := range output {
		if output[ii] != parseInput[ii] {
			t.Errorf("Bad conversion to byte slice at %vth byte! wanted: %v got: %v",
				parseInput[ii], output[ii])
		}
	}
}

func TestNalFragmentType(t *testing.T) {
	nalUnit := ParseNALFragment(parseInput)
	nalType := nalUnit.GetType()
	if nalType != nalTestType {
		t.Errorf("Returned wrong type!")
	}
}

func TestNalSpsPpsParsing(t *testing.T) {
	nalSpsPps := ParseNALSpsPps(parseInput)
	for ii := range parseInput {
		if nalSpsPps.Data[ii] != parseInput[ii] {
			t.Errorf("Bad Parsing! Byte: %v wanted: %v got: %v\n", ii, parseInput[ii],
				nalSpsPps.Data[ii])
		}
	}
}

func TestNalSpsPpsToByteSlice(t *testing.T) {
	nalSpsPps := ParseNALSpsPps(parseInput)
	nalSpsPpsByteSlice := nalSpsPps.ToByteSlice()
	for ii := range parseInput {
		if nalSpsPpsByteSlice[ii] != parseInput[ii] {
			t.Errorf("Bad conversion to byte slice! Byte: %v wanted: %v got: %v\n", ii,
				parseInput[ii], nalSpsPpsByteSlice[ii])
		}
	}
}

func TestNalSpsPpsType(t *testing.T) {
	nalSpsPps := ParseNALSpsPps(parseInput)
	if nalSpsPps.GetType() != nalTestType {
		t.Errorf("Returned wrong type!")
	}
}