txqueue work

tcp/txqueue_test.go

@@ -0,0 +1,35 @@
+package tcp
+
+import (
+	"bytes"
+	"testing"
+)
+
+func TestTxQueueWrite(t *testing.T) {
+	const (
+		bufsize = 1024
+		maxPkt  = 3
+		msg     = "hello world"
+	)
+	buf := make([]byte, bufsize)
+	rtx := newRingTx(buf, maxPkt)
+
+	bufs := bytes.SplitAfter([]byte(msg), []byte("e"))
+	var data [bufsize]byte
+	for i, buf := range bufs {
+		n, err := rtx.Write(buf)
+		if err != nil {
+			t.Fatalf("writing packet %d: %s", i, err)
+		} else if n != len(buf) {
+			t.Fatalf("want %d written, got %d", len(buf), n)
+		}
+		n, err = rtx.MakePacket(data[:])
+		if err != nil {
+			t.Fatalf("making packet %d: %s", i, err)
+		} else if n != len(buf) {
+			t.Fatalf("want %d packet read, got %d", len(buf), n)
+		} else if !bytes.Equal(buf, data[:n]) {
+			t.Fatalf("want data %q, got data read %q", buf, data[:n])
+		}
+	}
+}