Function TestServiceFailureWatcherClose

services/failure_watch_test.go:58–113

(t *testing.T)

}

func TestServiceFailureWatcherClose(t *testing.T) {
	s1 := serviceThatDoesntDoAnything()
	s2 := serviceThatDoesntDoAnything()
	m, err := NewManager(s1, s2)
	require.NoError(t, err)

	s3, errorsS3 := errorReturningService()
	s4, errorsS4 := errorReturningService()

	require.NoError(t, StartManagerAndAwaitHealthy(context.Background(), m))
	require.NoError(t, StartAndAwaitRunning(context.Background(), s3))
	require.NoError(t, StartAndAwaitRunning(context.Background(), s4))

	// All goroutines created until now are unrelated to failure watcher.
	defer goleak.VerifyNone(t, goleak.IgnoreCurrent(), goleak.Cleanup(func(_ int) {
		require.NoError(t, StopManagerAndAwaitStopped(context.Background(), m))
		// These services are expected to errors.
		require.Error(t, StopAndAwaitTerminated(context.Background(), s3))
		require.Error(t, StopAndAwaitTerminated(context.Background(), s4))

		goleak.VerifyNone(t)
	}))

	// Watch manager and both services
	w := NewFailureWatcher()
	w.WatchManager(m)
	w.WatchService(s3)
	w.WatchService(s4)

	ch := w.Chan()

	// Verify that we receive error from s3.
	err = fmt.Errorf("service3 error")
	errorsS3 <- err

	require.ErrorIs(t, <-ch, err)

	// After closing failure watcher, we don't receive any more errors.
	w.Close()
	require.Nil(t, <-ch)

	// Even sending error from service 4 doesn't make it to failure watcher.
	errorsS4 <- fmt.Errorf("service4 error")
	require.Nil(t, <-ch)

	// Since watcher is now closed, it cannot be used to watch services or managers.
	require.Panics(t, func() {
		w.WatchService(s3)
	})
	require.Panics(t, func() {
		w.WatchManager(m)
	})

	// Repeated call to Close doesn't panic.
	w.Close()
}

// Creates service which will return first error passed to the channel.