Function TestLinkBufferReadCopy

nocopy_linkbuffer_test.go:543–697

(t *testing.T)

}

func TestLinkBufferReadCopy(t *testing.T) {
	t.Run("SingleNode", func(t *testing.T) {
		LinkBufferCap = 128
		buf := NewLinkBuffer(128)
		p, _ := buf.Malloc(16)
		for i := range p {
			p[i] = byte(i)
		}
		buf.Flush()

		dst := make([]byte, 10)
		n := buf.readCopy(dst)
		Equal(t, n, 10)
		for i := 0; i < 10; i++ {
			Equal(t, dst[i], byte(i))
		}
		Equal(t, buf.Len(), 6)
		// readCopy must not set readExposed
		MustTrue(t, !buf.read.readExposed())
	})

	t.Run("MultiNode", func(t *testing.T) {
		LinkBufferCap = 8
		buf := NewLinkBuffer(8)
		p, _ := buf.Malloc(8)
		for i := range p {
			p[i] = byte(i)
		}
		buf.Flush()
		p, _ = buf.Malloc(8)
		for i := range p {
			p[i] = byte(i + 8)
		}
		buf.Flush()

		dst := make([]byte, 16)
		n := buf.readCopy(dst)
		Equal(t, n, 16)
		for i := 0; i < 16; i++ {
			Equal(t, dst[i], byte(i))
		}
		Equal(t, buf.Len(), 0)
	})

	t.Run("PartialRead", func(t *testing.T) {
		LinkBufferCap = 128
		buf := NewLinkBuffer(128)
		p, _ := buf.Malloc(4)
		for i := range p {
			p[i] = byte(i + 1)
		}
		buf.Flush()

		// read more than available
		dst := make([]byte, 16)
		n := buf.readCopy(dst)
		Equal(t, n, 4)
		Equal(t, dst[0], byte(1))