Method TestBuffer_Slice

mem/buffers_test.go:302–347

(t *testing.T)

}

func (s) TestBuffer_Slice(t *testing.T) {
	freed := make(chan struct{})
	data := []byte{1, 2, 3, 4}
	buf := mem.NewBuffer(&data, poolFunc(func(*[]byte) {
		close(freed)
	}))

	// Slice the buffer and verify the data.
	slice1 := buf.Slice(1, 3)
	if !bytes.Equal(slice1.ReadOnlyData(), data[1:3]) {
		t.Fatalf("Buffer did not contain expected data. got %v, want %v", slice1.ReadOnlyData(), data[1:3])
	}

	// Verify the original buffer is not modified.
	if !bytes.Equal(buf.ReadOnlyData(), data) {
		t.Fatalf("Buffer did not contain expected data. got %v, want %v", buf.ReadOnlyData(), data)
	}

	// Slice the slice.
	slice2 := slice1.Slice(0, 1)
	if !bytes.Equal(slice2.ReadOnlyData(), data[1:2]) {
		t.Fatalf("Buffer did not contain expected data. got %v, want %v", slice2.ReadOnlyData(), data[1:2])
	}

	// Free original and first slice — root should not be freed yet.
	buf.Free()
	slice1.Free()
	select {
	case <-freed:
		t.Fatalf("Freed too early")
	default:
	}

	// The last slice keeps the root alive.
	if !bytes.Equal(slice2.ReadOnlyData(), data[1:2]) {
		t.Fatalf("Buffer did not contain expected data. got %v, want %v", slice2.ReadOnlyData(), data[1:2])
	}

	slice2.Free()

	select {
	case <-freed:
	default:
		t.Fatalf("Still not freed")
	}
}

func (s) TestBuffer_SliceAfterFree(t *testing.T) {
	buf := newBuffer([]byte("abcd"), mem.NopBufferPool{})