118 lines
3.3 KiB
Go
118 lines
3.3 KiB
Go
package osd
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func testSnapshot() Snapshot {
|
|
var snapshot Snapshot
|
|
snapshot.Sequence = 0xbEEF
|
|
snapshot.TimestampMS = 123456
|
|
for i := range snapshot.Channels {
|
|
snapshot.Channels[i] = int16(-1024 + i*137)
|
|
}
|
|
snapshot.Armed = true
|
|
snapshot.Video = true
|
|
snapshot.OpConfig = OperatorConfig{Table: Table{Gen: 1}, MaxPower: 95}
|
|
snapshot.Policy = GSPolicy{Pending: true}
|
|
snapshot.Ack = VTXAck{Pending: true}
|
|
snapshot.Link = GSObservedLink{Known: false}
|
|
snapshot.PublishedMS = time.Now().UnixMilli()
|
|
return snapshot
|
|
}
|
|
|
|
func TestSnapshotSizeMatchesLayout(t *testing.T) {
|
|
// Header (magic, version, sequence, timestamp) + all 16 channels +
|
|
// flags + publish time + the four contract values + CRC must stay a
|
|
// small fixed record.
|
|
if Size != 50+1+12+17+14+8+2 {
|
|
t.Fatalf("Size=%d must match the documented layout", Size)
|
|
}
|
|
if encoded := Encode(testSnapshot()); len(encoded) != Size {
|
|
t.Fatalf("encoded=%d bytes want %d", len(encoded), Size)
|
|
}
|
|
}
|
|
|
|
func TestEncodeDecodeSnapshotRoundtrip(t *testing.T) {
|
|
want := testSnapshot()
|
|
|
|
got, err := Decode(Encode(want))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got != want {
|
|
t.Fatalf("decoded=%+v want %+v", got, want)
|
|
}
|
|
}
|
|
|
|
func TestDecodeRejectsDamagedSnapshots(t *testing.T) {
|
|
encoded := Encode(testSnapshot())
|
|
|
|
if _, err := Decode(encoded[:len(encoded)-1]); !errors.Is(err, ErrLength) {
|
|
t.Fatalf("truncated decode=%v want ErrLength", err)
|
|
}
|
|
magic := append([]byte(nil), encoded...)
|
|
magic[1] = 'X'
|
|
if _, err := Decode(magic); !errors.Is(err, ErrMagic) {
|
|
t.Fatalf("magic decode=%v want ErrMagic", err)
|
|
}
|
|
version := append([]byte(nil), encoded...)
|
|
version[2] = Version + 1
|
|
if _, err := Decode(version); !errors.Is(err, ErrVersion) {
|
|
t.Fatalf("version decode=%v want ErrVersion", err)
|
|
}
|
|
corrupt := append([]byte(nil), encoded...)
|
|
corrupt[10] ^= 0xff
|
|
if _, err := Decode(corrupt); !errors.Is(err, ErrCRC) {
|
|
t.Fatalf("corrupt payload decode=%v want ErrCRC", err)
|
|
}
|
|
corruptCRC := append([]byte(nil), encoded...)
|
|
corruptCRC[len(corruptCRC)-1] ^= 0xff
|
|
if _, err := Decode(corruptCRC); !errors.Is(err, ErrCRC) {
|
|
t.Fatalf("corrupt crc decode=%v want ErrCRC", err)
|
|
}
|
|
}
|
|
|
|
func TestFileSourceReportsFreshSnapshotAndHidesStaleOrBrokenOnes(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "osd-state.bin")
|
|
source := NewFileSource(path)
|
|
fixed := time.UnixMilli(10_000_000)
|
|
source.now = func() time.Time { return fixed }
|
|
|
|
if _, ok := source.Latest(); ok {
|
|
t.Fatal("a missing state file must report no snapshot")
|
|
}
|
|
if err := os.WriteFile(path, []byte("garbage that is not a snapshot"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, ok := source.Latest(); ok {
|
|
t.Fatal("a corrupt state file must report no snapshot")
|
|
}
|
|
|
|
want := testSnapshot()
|
|
want.PublishedMS = fixed.UnixMilli()
|
|
if err := os.WriteFile(path, Encode(want), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
got, ok := source.Latest()
|
|
if !ok {
|
|
t.Fatal("a fresh state file must report its snapshot")
|
|
}
|
|
if got != want {
|
|
t.Fatalf("got=%+v want %+v", got, want)
|
|
}
|
|
|
|
stale := want
|
|
stale.PublishedMS = fixed.Add(-Freshness - time.Second).UnixMilli()
|
|
if err := os.WriteFile(path, Encode(stale), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, ok := source.Latest(); ok {
|
|
t.Fatal("a snapshot older than the freshness window must not render")
|
|
}
|
|
}
|