mirror of https://github.com/ceph/go-ceph
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package cutil
|
|
|
|
import (
|
|
"testing"
|
|
"unsafe"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIovec(t *testing.T) {
|
|
strs := []string{
|
|
"foo",
|
|
"barbar",
|
|
"bazbazbaz",
|
|
}
|
|
var data [][]byte
|
|
for _, s := range strs {
|
|
data = append(data, []byte(s))
|
|
}
|
|
iovec := ByteSlicesToIovec(data)
|
|
p := iovec.Pointer()
|
|
assert.NotNil(t, p)
|
|
assert.Equal(t, iovec.Len(), len(data))
|
|
assert.Equal(t, p, unsafe.Pointer(&iovec.iovec[0]))
|
|
for i, iov := range iovec.iovec {
|
|
require.NotNil(t, iov.iov_base)
|
|
assert.Equal(t, int(iov.iov_len), len(data[i]))
|
|
assert.Equal(t, data[i], (*[MaxIdx]byte)(iov.iov_base)[:iov.iov_len:iov.iov_len])
|
|
}
|
|
// data didn't change
|
|
for i, b := range data {
|
|
assert.Equal(t, string(b), strs[i])
|
|
}
|
|
// clear iovec buffers
|
|
for _, iov := range iovec.iovec {
|
|
b := (*[MaxIdx]byte)(iov.iov_base)[:iov.iov_len:iov.iov_len]
|
|
for i := range b {
|
|
b[i] = 0
|
|
}
|
|
}
|
|
iovec.Sync()
|
|
// data must be cleared
|
|
for _, b := range data {
|
|
for i := range b {
|
|
assert.Zero(t, b[i])
|
|
}
|
|
}
|
|
iovec.Free()
|
|
for _, iov := range iovec.iovec {
|
|
assert.Equal(t, iov.iov_base, unsafe.Pointer(nil))
|
|
assert.Zero(t, iov.iov_len)
|
|
}
|
|
iovec.Free()
|
|
iovec.Sync()
|
|
iovec.Sync()
|
|
iovec.Free()
|
|
}
|