payload almost done

payload/copy_test.go

@@ -0,0 +1,29 @@
+package payload
+
+import (
+	"bytes"
+	"io"
+	"testing"
+)
+
+type Buffer struct {
+	bytes.Buffer
+	io.ReaderFrom // conflicts with and hides bytes.Buffer's ReaderFrom.
+	io.WriterTo   // conflicts with and hides bytes.Buffer's WriterTo.
+}
+
+// Simple tests, primarily to verify the ReadFrom and WriteTo callouts inside Copy and CopyN.
+
+func TestCopy(t *testing.T) {
+	rb := new(Buffer)
+	wb1 := new(Buffer)
+	wb2 := new(Buffer)
+	rb.WriteString("hello, world.")
+	CopyToTwoWriters(wb1, wb2, rb)
+	if wb1.String() != "hello, world." {
+		t.Errorf("Copy did not work properly")
+	}
+	if wb2.String() != "hello, world." {
+		t.Errorf("Copy did not work properly")
+	}
+}