Function TestCopyNext

msgp/read_test.go:1024–1067

(t *testing.T)

}

func TestCopyNext(t *testing.T) {
	var buf bytes.Buffer
	en := NewWriter(&buf)

	en.WriteMapHeader(6)

	en.WriteString("thing_one")
	en.WriteString("value_one")

	en.WriteString("thing_two")
	en.WriteFloat64(3.14159)

	en.WriteString("some_bytes")
	en.WriteBytes([]byte("nkl4321rqw908vxzpojnlk2314rqew098-s09123rdscasd"))

	en.WriteString("the_time")
	en.WriteTime(time.Now())

	en.WriteString("what?")
	en.WriteBool(true)

	en.WriteString("ext")
	en.WriteExtension(&RawExtension{Type: 55, Data: []byte("raw data!!!")})

	en.Flush()

	// Read from a copy of the original buf.
	de := NewReader(bytes.NewReader(buf.Bytes()))

	w := new(bytes.Buffer)

	n, err := de.CopyNext(w)
	if err != nil {
		t.Fatal(err)
	}
	if n != int64(buf.Len()) {
		t.Fatalf("CopyNext returned the wrong value (%d != %d)",
			n, buf.Len())
	}

	if !bytes.Equal(buf.Bytes(), w.Bytes()) {
		t.Fatalf("not equal! %v, %v", buf.Bytes(), w.Bytes())
	}
}