diff --git a/agent/pkg/agent/stream/payload_test.go b/agent/pkg/agent/stream/payload_test.go index 09424bcf..6083a371 100644 --- a/agent/pkg/agent/stream/payload_test.go +++ b/agent/pkg/agent/stream/payload_test.go @@ -1,7 +1,6 @@ package stream import ( - "bytes" "testing" ) @@ -25,29 +24,3 @@ func TestStreamRequestHeader_RoundTripAndChecksum(t *testing.T) { t.Fatalf("unexpected host/port: %q:%q", host, port) } } - -func TestStreamRequestPayload_WriteTo_WritesFullHeader(t *testing.T) { - h, err := NewStreamRequestHeader("127.0.0.1", "53") - if err != nil { - t.Fatalf("NewStreamRequestHeader: %v", err) - } - - p := &StreamRequestPayload{StreamRequestHeader: *h, Data: []byte("hello")} - - var out bytes.Buffer - n, err := p.WriteTo(&out) - if err != nil { - t.Fatalf("WriteTo: %v", err) - } - if int(n) != headerSize+len(p.Data) { - t.Fatalf("unexpected bytes written: got %d want %d", n, headerSize+len(p.Data)) - } - - written := out.Bytes() - if len(written) != headerSize+len(p.Data) { - t.Fatalf("unexpected output size: got %d", len(written)) - } - if !bytes.Equal(written[:headerSize], h.Bytes()) { - t.Fatalf("expected full header (including checksum) to be written") - } -}