Function TestModuleService_InterruptedSlowStartup

modules/modules_test.go:395–428

(t *testing.T)

}

func TestModuleService_InterruptedSlowStartup(t *testing.T) {
	subserviceStarted := make(chan struct{})
	subserviceStopped := false

	subService := services.NewBasicService(func(serviceContext context.Context) error {
		close(subserviceStarted)
		// Prolong the startup until we have to stop.
		<-serviceContext.Done()
		return nil
	}, func(context.Context) error {
		require.Fail(t, "did not expect to enter running state; service should have been canceled while in starting Fn and go directly to stoppingFn")
		return nil
	}, func(error) error {
		subserviceStopped = true
		return nil
	})

	noDepsFunc := func(string) map[string]services.Service { return nil }
	moduleSvc := NewModuleService("A", log.NewNopLogger(), subService, noDepsFunc, noDepsFunc)

	// Don't wait for startup because it will be very slow.
	require.NoError(t, moduleSvc.StartAsync(context.Background()))

	<-subserviceStarted

	// We get context.Canceled from moduleService.start.
	err := services.StopAndAwaitTerminated(context.Background(), moduleSvc)
	require.ErrorIs(t, err, context.Canceled)
	require.ErrorContains(t, err, "starting module A")

	assert.True(t, subserviceStopped)
	require.Equal(t, services.Failed, moduleSvc.State())
	require.Equal(t, services.Terminated, subService.State())
}

func getStopDependenciesForModule(module string, services map[string]services.Service) []string {
	var deps []string