Function TestMultiPoolWithFuncGenericReleaseContext

ants_test.go:1699–1749

(t *testing.T)

}

func TestMultiPoolWithFuncGenericReleaseContext(t *testing.T) {
	ch := make(chan struct{})
	mp, err := ants.NewMultiPoolWithFuncGeneric(10, 5, longRunningPoolFuncCh, ants.RoundRobin)
	require.NoError(t, err)

	for i := 0; i < 50; i++ {
		err = mp.Invoke(ch)
		require.NoError(t, err)
	}
	require.EqualValues(t, 50, mp.Running())

	close(ch)
	err = mp.ReleaseContext(context.Background())
	require.NoError(t, err)
	require.Zero(t, mp.Running())
	require.True(t, mp.IsClosed())

	// Calling ReleaseContext on a closed pool should return ErrPoolClosed.
	require.ErrorIs(t, mp.ReleaseContext(context.Background()), ants.ErrPoolClosed)

	// Test with LeastTasks strategy.
	ch = make(chan struct{})
	mp, err = ants.NewMultiPoolWithFuncGeneric(10, 5, longRunningPoolFuncCh, ants.LeastTasks)
	require.NoError(t, err)
	for i := 0; i < 50; i++ {
		err = mp.Invoke(ch)
		require.NoError(t, err)
	}
	close(ch)
	err = mp.ReleaseContext(context.Background())
	require.NoError(t, err)
	require.Zero(t, mp.Running())
	require.True(t, mp.IsClosed())

	// Test that a cancelled context returns an error.
	ch = make(chan struct{})
	mp, err = ants.NewMultiPoolWithFuncGeneric(10, 5, longRunningPoolFuncCh, ants.RoundRobin)
	require.NoError(t, err)
	for i := 0; i < 50; i++ {
		err = mp.Invoke(ch)
		require.NoError(t, err)
	}
	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	err = mp.ReleaseContext(ctx)
	require.Error(t, err)
	close(ch)
	require.Eventually(t, func() bool {
		return mp.Running() == 0
	}, 3*time.Second, 100*time.Millisecond)
}

func TestRebootNewPoolCalc(t *testing.T) {
	atomic.StoreInt32(&sum, 0)