utdf_test.go

package utdfgo

import (
	"bytes"
	"crypto/rand"
	"encoding/binary"
	"io/ioutil"
	"testing"
)

type UDTFPacket struct {
	b1      byte
	b2      byte
	b3      byte
	router1 uint8
	router2 uint8
	YY      uint8
	sic     uint16
	vid     uint16
	scY     uint32
	mcS     uint32
}

var packet UDTFPacket

const pl = 75
const filename = "utdf_test_file"

// build fake utdf packet for testing
func TestBuildUTDF(t *testing.T) {
	packet = UDTFPacket{
		b1:      1,
		b2:      2,
		b3:      3,
		router1: 'D',
		router2: 'D',
		YY:      19,
		sic:     3675,
		vid:     1,
		scY:     3772500,
		mcS:     0,
	}

	buf := new(bytes.Buffer)

	err := binary.Write(buf, binary.BigEndian, packet)
	if err != nil {
		t.Fatal("Cannot package utdf packet")
	}

	data := make([]byte, 57)
	rand.Read(data)

	err = binary.Write(buf, binary.BigEndian, data)
	if err != nil {
		t.Fatal("Cannot package utdf packet")
	}

	err = ioutil.WriteFile(filename, buf.Bytes(), 0644)
	if err != nil {
		t.Fatal("Cannot write packet to file")
	}

}

// test if application can successfully ID it gets passed a non utdf file
func TestUTDFValid(t *testing.T) {
	t.Run("given unknown file", func(t *testing.T) {
		if _, err := Run("utdf.go"); err == nil {
			t.Error("expected error: got nil")
		}
	})
}

// ensure the first 13 bytes are correct
func TestUTDFHeader(t *testing.T) {

	// test suite for header
	t.Run("given utdf file", func(t *testing.T) {
		p, _ := Run(filename)
		op := p[0]

		// make sure utdf packet is 75 bytes
		t.Run("checking utdf length", func(t *testing.T) {
			if len(op) != pl {
				t.Error("Packet in file not equal to", pl)
			}
		})

		// packet year
		t.Run("checking year", func(t *testing.T) {
			if op.GetYear() != 2000+int(+packet.YY) {
				t.Error("Unexpected Year:", op.GetYear())
			}
		})

		// packet router
		t.Run("checking router", func(t *testing.T) {
			if op[3] != packet.router1 && op[4] != packet.router2 {
				t.Error("Unexpecting router:", op[3])
			}
		})

		// packet SIC
		t.Run("checking SIC", func(t *testing.T) {
			if op.GetSIC() != uint64(packet.sic) {
				t.Error("Unexpected SIC:", op.GetSIC())
			}
		})

		// packet VID
		t.Run("checking VID", func(t *testing.T) {
			if op.GetVID() != uint64(packet.vid) {
				t.Error("Unexpected VID:", op.GetVID())
			}
		})

		// packet seconds
		t.Run("checking seconds of year", func(t *testing.T) {
			if op.GetSeconds() != int(packet.scY) {
				t.Error("Unexpected seconds of year:", op.GetSeconds())
			}
		})

		// packet microseconds
		t.Run("checking microseconds", func(t *testing.T) {
			if op.GetMicroseconds() != int(packet.mcS) {
				t.Error("Unexpected microseconds:", op.GetMicroseconds())
			}
		})
	})
}