diff --git a/internal/processor/test/artnet-packet-decode_test.go b/internal/processor/test/artnet-packet-decode_test.go index 63f7075..3d43fa2 100644 --- a/internal/processor/test/artnet-packet-decode_test.go +++ b/internal/processor/test/artnet-packet-decode_test.go @@ -1,13 +1,15 @@ package processor_test import ( + "reflect" "testing" + "github.com/jwetzell/artnet-go" "github.com/jwetzell/showbridge-go/internal/config" "github.com/jwetzell/showbridge-go/internal/processor" ) -func TestArtnetPacketCreateFromRegistry(t *testing.T) { +func TestArtnetPacketDecodeFromRegistry(t *testing.T) { registration, ok := processor.ProcessorRegistry["artnet.packet.decode"] if !ok { t.Fatalf("artnet.packet.decode processor not registered") @@ -25,3 +27,81 @@ func TestArtnetPacketCreateFromRegistry(t *testing.T) { t.Fatalf("artnet.packet.decode processor has wrong type: %s", processorInstance.Type()) } } + +func TestGoodArtnetPacketDecode(t *testing.T) { + packetDecoder := processor.ArtNetPacketDecode{} + + tests := []struct { + name string + payload any + expected artnet.ArtNetPacket + }{ + { + name: "number", + payload: []byte{65, 114, 116, 45, 78, 101, 116, 0, 0, 80, 0, 14, 237, 0, 1, 0, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, + expected: &artnet.ArtDmx{ + ID: []byte{'A', 'r', 't', '-', 'N', 'e', 't', 0x00}, + OpCode: artnet.OpDmx, + ProtVerHi: 0, + ProtVerLo: 14, + Sequence: 237, + Physical: 0, + SubUni: 1, + Net: 0, + Length: 512, + Data: make([]uint8, 512), + }, + }, + } + + for _, test := range tests { + t.Run(test.name, func(t *testing.T) { + + got, err := packetDecoder.Process(t.Context(), test.payload) + + if err != nil { + t.Fatalf("artnet.packet.decode failed: %s", err) + } + + //TODO(jwetzell): work out better way to compare the any/any + if !reflect.DeepEqual(got, test.expected) { + t.Fatalf("artnet.packet.decode got %+v (%T), expected %+v (%T)", got, got, test.expected, test.expected) + } + }) + } +} + +func TestBadArtnetPacketDecode(t *testing.T) { + packetDecoder := processor.ArtNetPacketDecode{} + tests := []struct { + name string + payload any + errorString string + }{ + { + name: "not a byte slice", + payload: "not a byte slice", + errorString: "artnet.packet.decode processor only accepts a []byte", + }, + { + name: "not enough bytes", + payload: []byte{1, 2, 3}, + errorString: "ArtNet packet must be at least 12 bytes", + }, + } + + for _, test := range tests { + t.Run(test.name, func(t *testing.T) { + + got, err := packetDecoder.Process(t.Context(), test.payload) + + if err == nil { + t.Fatalf("artnet.packet.decode expected to fail but succeeded, got: %v", got) + + } + if err.Error() != test.errorString { + t.Fatalf("artnet.packet.decode got error '%s', expected '%s'", err.Error(), test.errorString) + } + }) + } +} diff --git a/internal/processor/test/artnet-packet-encode_test.go b/internal/processor/test/artnet-packet-encode_test.go index d3f3db0..03e1b07 100644 --- a/internal/processor/test/artnet-packet-encode_test.go +++ b/internal/processor/test/artnet-packet-encode_test.go @@ -1,8 +1,10 @@ package processor_test import ( + "reflect" "testing" + "github.com/jwetzell/artnet-go" "github.com/jwetzell/showbridge-go/internal/config" "github.com/jwetzell/showbridge-go/internal/processor" ) @@ -25,3 +27,76 @@ func TestArtnetPacketEncodeFromRegistry(t *testing.T) { t.Fatalf("artnet.packet.encode processor has wrong type: %s", processorInstance.Type()) } } + +func TestGoodArtnetPacketEncode(t *testing.T) { + packetEncoder := processor.ArtNetPacketEncode{} + + tests := []struct { + name string + expected []byte + payload any + }{ + { + name: "number", + expected: []byte{65, 114, 116, 45, 78, 101, 116, 0, 0, 80, 0, 14, 237, 0, 1, 0, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, + payload: &artnet.ArtDmx{ + ID: []byte{'A', 'r', 't', '-', 'N', 'e', 't', 0x00}, + OpCode: artnet.OpDmx, + ProtVerHi: 0, + ProtVerLo: 14, + Sequence: 237, + Physical: 0, + SubUni: 1, + Net: 0, + Length: 512, + Data: make([]uint8, 512), + }, + }, + } + + for _, test := range tests { + t.Run(test.name, func(t *testing.T) { + + got, err := packetEncoder.Process(t.Context(), test.payload) + + if err != nil { + t.Fatalf("artnet.packet.encode failed: %s", err) + } + + //TODO(jwetzell): work out better way to compare the any/any + if !reflect.DeepEqual(got, test.expected) { + t.Fatalf("artnet.packet.encode got %+v (%T), expected %+v (%T)", got, got, test.expected, test.expected) + } + }) + } +} + +func TestBadArtnetPacketEncode(t *testing.T) { + packetEncoder := processor.ArtNetPacketEncode{} + tests := []struct { + name string + payload any + errorString string + }{ + { + name: "not an ArtNet packet", + payload: "test", + errorString: "artnet.packet.encode processor only accepts an ArtNetPacket", + }, + } + + for _, test := range tests { + t.Run(test.name, func(t *testing.T) { + + got, err := packetEncoder.Process(t.Context(), test.payload) + + if err == nil { + t.Fatalf("artnet.packet.encode expected to fail but succeeded, got: %v", got) + + } + if err.Error() != test.errorString { + t.Fatalf("artnet.packet.encode got error '%s', expected '%s'", err.Error(), test.errorString) + } + }) + } +}