Function TestConnectAndWait

cli-plugins/socket/socket_test.go:150–216

(t *testing.T)

}

func TestConnectAndWait(t *testing.T) {
	t.Run("calls cancel func on EOF", func(t *testing.T) {
		srv, err := NewPluginServer(nil)
		assert.NilError(t, err, "failed to setup server")
		defer srv.Close()

		done := make(chan struct{})
		t.Setenv(EnvKey, srv.Addr().String())
		cancelFunc := func() {
			done <- struct{}{}
		}
		ConnectAndWait(cancelFunc)

		select {
		case <-done:
			t.Fatal("unexpectedly done")
		default:
		}

		srv.Close()

		select {
		case <-done:
		case <-time.After(10 * time.Millisecond):
			t.Fatal("cancel function not closed after 10ms")
		}
	})

	// TODO: this test cannot be executed with `t.Parallel()`, due to
	// relying on goroutine numbers to ensure correct behaviour
	t.Run("connect goroutine exits after EOF", func(t *testing.T) {
		runtime.LockOSThread()
		defer runtime.UnlockOSThread()
		srv, err := NewPluginServer(nil)
		assert.NilError(t, err, "failed to setup server")

		defer srv.Close()

		t.Setenv(EnvKey, srv.Addr().String())

		runtime.Gosched()
		numGoroutines := runtime.NumGoroutine()

		ConnectAndWait(func() {})

		runtime.Gosched()
		poll.WaitOn(t, func(t poll.LogT) poll.Result {
			// +1 goroutine for the poll.WaitOn
			// +1 goroutine for the connect goroutine
			if runtime.NumGoroutine() < numGoroutines+1+1 {
				return poll.Continue("waiting for connect goroutine to spawn")
			}
			return poll.Success()
		}, poll.WithDelay(1*time.Millisecond), poll.WithTimeout(500*time.Millisecond))

		srv.Close()

		runtime.Gosched()