Function TestSupervisor_StopConcurrent

pkg/tools/lifecycle/supervisor_test.go:802–830

TestSupervisor_StopConcurrent exercises the s.stopping guard: several goroutines call Stop concurrently while the watcher is live in sess.Wait(). All calls must return without error and observe a fully-shut-down supervisor.

(t *testing.T)

// sess.Wait(). All calls must return without error and observe a
// fully-shut-down supervisor.
func TestSupervisor_StopConcurrent(t *testing.T) {
	t.Parallel()

	sess := newFakeSession()
	c := newScriptedConnector(scriptStep{session: sess})
	s := lifecycle.New("test", c, lifecycle.Policy{})

	assert.NilError(t, s.Start(t.Context()))
	sess.waitParked(t)

	const n = 4
	errs := make(chan error, n)
	var wg sync.WaitGroup
	wg.Add(n)
	for range n {
		go func() {
			defer wg.Done()
			errs <- s.Stop(t.Context())
		}()
	}
	wg.Wait()
	close(errs)

	for err := range errs {
		assert.NilError(t, err)
	}
	assert.Check(t, is.Equal(s.State().State, lifecycle.StateStopped))
	assert.Check(t, sess.waitDone.Load(), "a Stop returned before watcher's Wait() completed")
}