parser_test.go

package ulogo

import (
	"bytes"
	"os"
	"testing"
)

func TestLoadPyulogSample(t *testing.T) {
	log, err := Load("testdata/pyulog/sample.ulg")
	if err != nil {
		t.Fatal(err)
	}
	if log.Version != 0 {
		t.Fatalf("version = %d, want 0", log.Version)
	}
	if got := log.Info["sys_name"].String(); got != "PX4" {
		t.Fatalf("sys_name = %q, want PX4", got)
	}
	if len(log.Dropouts) != 4 {
		t.Fatalf("dropouts = %d, want 4", len(log.Dropouts))
	}
	if len(log.LoggedMessages) != 4 {
		t.Fatalf("logged messages = %d, want 4", len(log.LoggedMessages))
	}
	if len(log.Data) != 15 {
		t.Fatalf("datasets = %d, want 15", len(log.Data))
	}
	dataset, ok := log.Dataset("sensor_combined", 0)
	if !ok {
		t.Fatal("sensor_combined dataset not found")
	}
	if got := len(dataset.Rows); got != 17070 {
		t.Fatalf("sensor_combined rows = %d, want 17070", got)
	}
	if got := dataset.MessageSize(); got != 72 {
		t.Fatalf("sensor_combined message size = %d, want 72", got)
	}
}

func TestReadAcceptsPlainReader(t *testing.T) {
	raw, err := os.ReadFile("testdata/pyulog/sample_appended.ulg")
	if err != nil {
		t.Fatal(err)
	}

	log, err := Read(bytes.NewBuffer(raw))
	if err != nil {
		t.Fatal(err)
	}
	if !log.HasDataAppended() {
		t.Fatal("expected appended data flag")
	}
	if len(log.AppendedOffsets) != 1 {
		t.Fatalf("appended offsets = %d, want 1", len(log.AppendedOffsets))
	}
	if got := len(log.Data); got != 20 {
		t.Fatalf("datasets = %d, want 20", got)
	}
}

func TestReadFileMatchesLoad(t *testing.T) {
	fromReadFile, err := ReadFile("testdata/pyulog/sample.ulg")
	if err != nil {
		t.Fatal(err)
	}
	fromLoad, err := Load("testdata/pyulog/sample.ulg")
	if err != nil {
		t.Fatal(err)
	}

	if len(fromReadFile.Data) != len(fromLoad.Data) {
		t.Fatalf("datasets = %d, want %d", len(fromReadFile.Data), len(fromLoad.Data))
	}
	if fromReadFile.StartTimestamp != fromLoad.StartTimestamp {
		t.Fatalf("start timestamp = %d, want %d", fromReadFile.StartTimestamp, fromLoad.StartTimestamp)
	}
}

func TestWithMessageFilterNilLoadsAllData(t *testing.T) {
	log, err := ReadFile("testdata/pyulog/sample.ulg", WithMessageFilter(nil))
	if err != nil {
		t.Fatal(err)
	}
	if len(log.Data) != 15 {
		t.Fatalf("datasets = %d, want 15", len(log.Data))
	}
}

func TestWithMessageFilterEmptyLoadsDefinitionsOnly(t *testing.T) {
	log, err := ReadFile("testdata/pyulog/sample.ulg", WithMessageFilter([]string{}))
	if err != nil {
		t.Fatal(err)
	}
	if len(log.Data) != 0 {
		t.Fatalf("datasets = %d, want 0", len(log.Data))
	}
	if len(log.MessageFormats) == 0 {
		t.Fatal("expected message formats")
	}
}

func TestHeaderOnlyLoadsDefinitionsWithoutData(t *testing.T) {
	log, err := ReadFile("testdata/pyulog/sample.ulg", HeaderOnly())
	if err != nil {
		t.Fatal(err)
	}
	if len(log.Data) != 0 {
		t.Fatalf("datasets = %d, want 0", len(log.Data))
	}
	if len(log.MessageFormats) == 0 {
		t.Fatal("expected message formats")
	}
	if len(log.InitialParameters) == 0 {
		t.Fatal("expected initial parameters")
	}
}

func TestReadRejectsInvalidInput(t *testing.T) {
	tests := []struct {
		name  string
		input []byte
	}{
		{name: "empty", input: nil},
		{name: "short header", input: []byte{0x55, 0x4c}},
		{name: "bad magic", input: []byte("not-a-valid-ulog-header")},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if _, err := Read(bytes.NewReader(tt.input)); err == nil {
				t.Fatal("Read() error = nil, want error")
			}
		})
	}
}

func TestReadRejectsNilReader(t *testing.T) {
	if _, err := Read(nil); err == nil {
		t.Fatal("Read(nil) error = nil, want error")
	}
}

func TestPacketLooksCorrupt(t *testing.T) {
	tests := []struct {
		header messageHeader
		want   bool
	}{
		{header: messageHeader{typ: 0, size: 4}, want: true},
		{header: messageHeader{typ: msgTypeData, size: 0}, want: true},
		{header: messageHeader{typ: msgTypeData, size: 10001}, want: true},
		{header: messageHeader{typ: msgTypeData, size: 32}, want: false},
	}

	for _, tt := range tests {
		if got := packetLooksCorrupt(tt.header); got != tt.want {
			t.Fatalf("packetLooksCorrupt(%+v) = %v, want %v", tt.header, got, tt.want)
		}
	}
}