Function TestAllFunctionality

pkg/util/services/basic_service_test.go:101–186

(t *testing.T)

}

func TestAllFunctionality(t *testing.T) {
	errStartFailed := errors.New("start failed")
	errRunFailed := errors.New("runFn failed")
	errStopFailed := errors.New("stop failed")

	testCases := map[string]testCase{
		"normal flow": {
			listenerLog: []string{"starting", "running", "stopping: Running", "terminated: Stopping"},
		},

		"start returns error": {
			startRetVal:          errStartFailed,
			awaitRunningError:    []error{invalidServiceStateWithFailureError(Failed, Running, errStartFailed)},
			awaitTerminatedError: invalidServiceStateWithFailureError(Failed, Terminated, errStartFailed), // Failed in start
			failureCase:          errStartFailed,
			listenerLog:          []string{"starting", "failed: Starting: start failed"},
		},

		"start is canceled via context and returns cancellation error": {
			cancelAfterStartAsync: true,
			startReturnContextErr: true,
			awaitRunningError:     []error{invalidServiceStateWithFailureError(Failed, Running, context.Canceled)},
			awaitTerminatedError:  invalidServiceStateWithFailureError(Failed, Terminated, context.Canceled),
			failureCase:           context.Canceled,
			listenerLog:           []string{"starting", "failed: Starting: context canceled"},
		},

		"start is canceled via context, doesn't return error. Run shouldn't runFn, since context is canceled now.": {
			cancelAfterStartAsync: true,
			startReturnContextErr: false,
			// we can observe Stopping or Terminated
			awaitRunningError:    []error{invalidServiceStateError(Stopping, Running), invalidServiceStateError(Terminated, Running)},
			awaitTerminatedError: nil, // but still terminates correctly, since Start or RunningFn didn't return error
			failureCase:          nil, // start didn't return error, service stopped without calling run
			listenerLog:          []string{"starting", "stopping: Starting", "terminated: Stopping"},
		},

		"start is canceled via StopAsync, but start doesn't return error": {
			startReturnContextErr: false, // don't return error on cancellation, just stop early
			stopAfterStartAsync:   true,
			// we can observe Stopping or Terminated
			awaitRunningError:    []error{invalidServiceStateError(Stopping, Running), invalidServiceStateError(Terminated, Running)},
			awaitTerminatedError: nil, // stopped while starting, but no error. Should be in Terminated state.
			failureCase:          nil, // start didn't return error, service stopped without calling run
			listenerLog:          []string{"starting", "stopping: Starting", "terminated: Stopping"},
		},

		"runFn returns error": {
			runRetVal:            errRunFailed,
			awaitTerminatedError: invalidServiceStateWithFailureError(Failed, Terminated, errRunFailed), // service will get into Failed state, since run failed
			failureCase:          errRunFailed,
			listenerLog:          []string{"starting", "running", "stopping: Running", "failed: Stopping: runFn failed"},
		},

		"runFn returns error from context cancellation": {
			runReturnContextErr:     true,
			cancelAfterAwaitRunning: true,
			awaitTerminatedError:    invalidServiceStateWithFailureError(Failed, Terminated, context.Canceled), // service will get into Failed state, since run failed