Function TestSampleService

rivershared/startstop/start_stop_test.go:220–282

BaseStartStop tests that need specific internal implementation (like ones we can add to sampleService) to be able to verify.

(t *testing.T)

// BaseStartStop tests that need specific internal implementation (like ones we
// can add to sampleService) to be able to verify.
func TestSampleService(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	type testBundle struct{}

	setup := func(t *testing.T) (*sampleService, *testBundle) { //nolint:unparam
		t.Helper()

		return &sampleService{}, &testBundle{}
	}

	t.Run("StartedChannel", func(t *testing.T) {
		t.Parallel()

		service, _ := setup(t)

		require.NoError(t, service.Start(ctx))
		t.Cleanup(service.Stop)

		riversharedtest.WaitOrTimeout(t, service.Started())
		require.True(t, service.state)
	})

	t.Run("StartError", func(t *testing.T) {
		t.Parallel()

		service, _ := setup(t)
		service.startErr = errors.New("error on start")

		require.ErrorIs(t, service.Start(ctx), service.startErr)

		riversharedtest.WaitOrTimeout(t, service.Started()) // start channel also closed on erroneous start
		riversharedtest.WaitOrTimeout(t, service.Stopped())
	})

	t.Run("StartErrorThenSuccessfulRestart", func(t *testing.T) {
		t.Parallel()

		service, _ := setup(t)
		service.startErr = errors.New("error on start")

		// First start fails with our simulated error.
		require.ErrorIs(t, service.Start(ctx), service.startErr)

		riversharedtest.WaitOrTimeout(t, service.Started())
		riversharedtest.WaitOrTimeout(t, service.Stopped())

		// Clear error so the next start succeeds.
		service.startErr = nil

		// Second start should succeed despite the prior failure. Without the
		// reset-on-failed-start logic in StartInit, isRunning would still be
		// true and StartInit would return shouldStart=false, causing Start to
		// return nil without actually starting the service.
		require.NoError(t, service.Start(ctx))
		t.Cleanup(service.Stop)