Function TestRuntimeOwnerCheckHooksAndStrictThreadAffinity

runtime_test.go:1560–1609

(t *testing.T)

}

func TestRuntimeOwnerCheckHooksAndStrictThreadAffinity(t *testing.T) {
	oldGIDHook := ownerCheckCurrentGoroutineID
	oldThreadHook := ownerCheckCurrentThreadID
	defer func() {
		ownerCheckCurrentGoroutineID = oldGIDHook
		ownerCheckCurrentThreadID = oldThreadHook
	}()

	var gid atomic.Uint64
	var tid atomic.Uint64
	gid.Store(11)
	tid.Store(101)

	ownerCheckCurrentGoroutineID = func() uint64 { return gid.Load() }
	ownerCheckCurrentThreadID = func() uint64 { return tid.Load() }

	var nilRuntime *Runtime
	require.False(t, nilRuntime.ensureOwnerAccess())

	rt := NewRuntime(WithStrictOSThread(true))
	require.NotNil(t, rt)
	defer rt.Close()

	require.True(t, rt.ensureOwnerAccess())
	require.Equal(t, uint64(11), rt.ownerGoroutineID.Load())
	require.Equal(t, uint64(101), rt.ownerThreadID.Load())
	require.NotZero(t, currentGoroutineID())
	require.NotZero(t, currentThreadID())

	gid.Store(0)
	require.False(t, rt.ensureOwnerAccess())
	gid.Store(11)
	tid.Store(0)
	require.False(t, rt.ensureOwnerAccess())
	tid.Store(101)

	gid.Store(12)
	require.False(t, rt.ensureOwnerAccess())

	gid.Store(11)
	tid.Store(102)
	require.False(t, rt.ensureOwnerAccess())

	tid.Store(101)
	require.True(t, rt.ensureOwnerAccess())

	rtNoCheck := NewRuntime()
	require.True(t, rtNoCheck.ensureOwnerAccess())
	rtNoCheck.Close()
}

func TestRuntimeOwnerGoroutineCheckOption(t *testing.T) {
	oldGIDHook := ownerCheckCurrentGoroutineID