Function TestReadWriteExtension

msgp/extension_test.go:18–51

(t *testing.T)

}

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

	t.Run("interface", func(t *testing.T) {
		for range 25 {
			buf.Reset()
			e := randomExt()
			en.WriteExtension(&e)
			en.Flush()
			err := dc.ReadExtension(&e)
			if err != nil {
				t.Errorf("error with extension (length %d): %s", len(buf.Bytes()), err)
			}
		}
	})

	t.Run("raw", func(t *testing.T) {
		for range 25 {
			buf.Reset()
			e := randomExt()
			en.WriteExtensionRaw(e.Type, e.Data)
			en.Flush()
			typ, payload, err := dc.ReadExtensionRaw()
			if err != nil {
				t.Errorf("error with extension (length %d): %s", len(buf.Bytes()), err)
			}
			if typ != e.Type || !bytes.Equal(payload, e.Data) {
				t.Errorf("extension mismatch: %d %x != %d %x", typ, payload, e.Type, e.Data)
			}
		}
	})
}

func TestReadWriteLargeExtensionRaw(t *testing.T) {
	var buf bytes.Buffer