Function TestModuleService_InterruptedFastStartup

modules/modules_test.go:354–393

(t *testing.T)

}

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

	subService := services.NewBasicService(func(_ context.Context) error {
		// We want to control the execution of this function via the test code,
		// so ignore the passed context because it will be cancelled shortly after entering this function.
		close(subserviceStarted)
		<-finishStarting
		return nil
	}, func(serviceContext context.Context) error {
		<-serviceContext.Done()
		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)

	ctx, cancel := context.WithCancel(context.Background())
	go func() {
		<-subserviceStarted
		cancel()
		close(finishStarting)
	}()

	// Start service using context that's going to be cancelled
	require.NoError(t, moduleSvc.StartAsync(ctx))
	// We get context cancelled error from service context cancellation.
	err := moduleSvc.AwaitRunning(context.Background())
	require.ErrorIs(t, err, context.Canceled)
	require.ErrorContains(t, err, "starting module A")

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

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