gouring/util_ptr_arith_test.go

package gouring

import (
	"encoding/binary"
	"fmt"
	"testing"
	"unsafe"

	"github.com/stretchr/testify/assert"
)

func TestUserdata(t *testing.T) {
	type test struct {
		v   any
		exp uint64
	}
	ts := []test{
		{uint64(0), 0},
		{uint64(0xff), 0xff},
		{uint64(0xfffefd), 0xfffefd},
		{uintptr(0xcafeba), 0xcafeba},
		{unsafe.Pointer(nil), 0},
	}
	bo := binary.LittleEndian
	for _, tc := range ts {
		var u UserData
		switch v := tc.v.(type) {
		case uint64:
			u.SetUint64(v)
		case uintptr:
			u.SetUintptr(v)
		case unsafe.Pointer:
			u.SetUnsafe(v)
		default:
			panic(fmt.Sprintf("unhandled type: %T", v))
		}

		assert.Equal(t, tc.exp, u.GetUint64())

		var exp [8]byte
		bo.PutUint64(exp[:], tc.exp)
		// assert.Equal(t, exp[:], u[:])
		assert.Equal(t, tc.exp, u.GetUint64())
	}
}