Function TestManagerAwaitWithContextCancellation

services/manager_test.go:232–260

(t *testing.T)

}

func TestManagerAwaitWithContextCancellation(t *testing.T) {
	t.Parallel()

	s1 := serviceThatStopsOnItsOwnAfterTimeout(200 * time.Millisecond)
	s2 := serviceThatStopsOnItsOwnAfterTimeout(300 * time.Millisecond)
	gl := newGatheringManagerListener(t)

	m, err := NewManager(s1, s2)
	require.NoError(t, err)
	m.AddListener(gl)

	// test context cancellation
	stoppedContext, cancel := context.WithCancel(context.Background())
	cancel()

	require.Equal(t, context.Canceled, m.AwaitHealthy(stoppedContext)) // since no services have started yet, and context is stopped, must return error quickly

	// start all services
	require.NoError(t, m.StartAsync(context.Background()))

	shortContext, cancel := context.WithTimeout(context.Background(), 10*time.Millisecond)
	defer cancel()

	require.Equal(t, context.DeadlineExceeded, m.AwaitStopped(shortContext))
	require.NoError(t, m.AwaitStopped(context.Background()))

	require.NoError(t, gl.AwaitTerminated(context.Background()))
	require.Equal(t, []string{"healthy", "stopped"}, gl.log)
}

func mergeStates(m map[State][]Service, states ...State) []Service {
	result := []Service(nil)