reflect2/tests/map_test.go

package tests

import (
	"testing"
	"github.com/modern-go/reflect2"
	"github.com/modern-go/test/must"

	"github.com/modern-go/test"
	"unsafe"
	"reflect"
	"context"
)

func Test_map(t *testing.T) {
	var pInt = func(val int) *int {
		return &val
	}
	t.Run("New", testOp(func(api reflect2.API) interface{} {
		valType := api.TypeOf(map[int]int{})
		m := valType.New().(*map[int]int)
		return m
	}))
	t.Run("IsNil", testOp(func(api reflect2.API) interface{} {
		valType := api.TypeOf(map[int]int{})
		var nilMap map[int]int
		m := map[int]int{}
		return []interface{}{
			valType.IsNil(&nilMap),
			valType.IsNil(&m),
		}
	}))
	t.Run("MakeMap", testOp(func(api reflect2.API) interface{} {
		valType := api.TypeOf(map[int]int{}).(reflect2.MapType)
		m := *(valType.MakeMap(0).(*map[int]int))
		m[2] = 4
		m[3] = 9
		return m
	}))
	t.Run("UnsafeMakeMap", test.Case(func(ctx context.Context) {
		valType := reflect2.TypeOf(map[int]int{}).(reflect2.MapType)
		m := *(*map[int]int)(valType.UnsafeMakeMap(0))
		m[2] = 4
		m[3] = 9
	}))
	t.Run("PackEFace", test.Case(func(ctx context.Context) {
		valType := reflect2.TypeOf(map[int]int{}).(reflect2.MapType)
		m := valType.UnsafeMakeMap(0)
		must.Equal(&map[int]int{}, valType.PackEFace(unsafe.Pointer(m)))
	}))
	t.Run("Indirect", testOp(func(api reflect2.API) interface{} {
		valType := reflect2.TypeOf(map[int]int{})
		return valType.Indirect(&map[int]int{})
	}))
	t.Run("SetIndex", testOp(func(api reflect2.API) interface{} {
		obj := map[int]int{}
		valType := api.TypeOf(obj).(reflect2.MapType)
		valType.SetIndex(&obj, pInt(2), pInt(4))
		valType.SetIndex(&obj, pInt(3), pInt(9))
		must.Equal(4, obj[2])
		return obj
	}))
	t.Run("UnsafeSetIndex", test.Case(func(ctx context.Context) {
		obj := map[int]int{}
		valType := reflect2.TypeOf(obj).(reflect2.MapType)
		valType.UnsafeSetIndex(unsafe.Pointer(&obj), reflect2.PtrOf(2), reflect2.PtrOf(4))
		must.Equal(map[int]int{2: 4}, obj)
	}))
	t.Run("GetIndex", testOp(func(api reflect2.API) interface{} {
		obj := map[int]int{3: 9, 2: 4}
		valType := api.TypeOf(obj).(reflect2.MapType)
		return []interface{}{
			*valType.GetIndex(&obj, pInt(3)).(*int),
			valType.GetIndex(&obj, pInt(0)).(*int),
		}
	}))
	t.Run("UnsafeGetIndex", test.Case(func(ctx context.Context) {
		obj := map[int]int{3: 9, 2: 4}
		valType := reflect2.TypeOf(obj).(reflect2.MapType)
		elem := valType.UnsafeGetIndex(unsafe.Pointer(&obj), reflect2.PtrOf(3))
		must.Equal(9, *(*int)(elem))
	}))
	t.Run("Iterate", testOp(func(api reflect2.API) interface{} {
		obj := map[int]int{2: 4}
		valType := api.TypeOf(obj).(reflect2.MapType)
		iter := valType.Iterate(&obj)
		must.Pass(iter.HasNext(), "api", api)
		key1, elem1 := iter.Next()
		must.Pass(!iter.HasNext(), "api", api)
		return []interface{}{key1, elem1}
	}))
	t.Run("UnsafeIterate", test.Case(func(ctx context.Context) {
		obj := map[int]int{2: 4}
		valType := reflect2.TypeOf(obj).(reflect2.MapType)
		iter := valType.UnsafeIterate(unsafe.Pointer(&obj))
		must.Pass(iter.HasNext())
		key, elem := iter.UnsafeNext()
		must.Equal(2, *(*int)(key))
		must.Equal(4, *(*int)(elem))
	}))
}

func Benchmark_map_unsafe(b *testing.B) {
	obj := map[int]int{}
	valType := reflect2.TypeOf(obj).(*reflect2.UnsafeMapType)
	m := unsafe.Pointer(&obj)
	b.ReportAllocs()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		valType.UnsafeSetIndex(m, reflect2.PtrOf(2), reflect2.PtrOf(4))
	}
}

func Benchmark_map_safe(b *testing.B) {
	obj := map[int]int{}
	val := reflect.ValueOf(obj)
	b.ReportAllocs()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		val.SetMapIndex(reflect.ValueOf(2), reflect.ValueOf(4))
	}
}