Function createTestInvoker

internal/codespaces/rpc/invoker_test.go:59–86

createTestInvoker is the main test setup function. It returns an Invoker using the provided mockServer, as well as a shutdown function. The Invoker does not need to be closed directly, that will be handled by the shutdown function.

(t *testing.T, server *mockServer)

// createTestInvoker is the main test setup function. It returns an Invoker using the provided mockServer, as well as a shutdown function.
// The Invoker does not need to be closed directly, that will be handled by the shutdown function.
func createTestInvoker(t *testing.T, server *mockServer) (Invoker, func(), error) {
	listener, err := net.Listen("tcp", "127.0.0.1:16634")
	if err != nil {
		return nil, nil, fmt.Errorf("failed to listen: %w", err)
	}

	ctx, cancel := context.WithCancel(context.Background())
	ch := make(chan error)
	go func() { ch <- runTestGrpcServer(ctx, listener, server) }()

	close := func() {
		cancel()
		<-ch
		listener.Close()
	}

	// Create a new invoker with a mock port forwarder
	invoker, err := CreateInvoker(context.Background(), rpctest.PortForwarder{})
	if err != nil {
		close()
		return nil, nil, fmt.Errorf("error connecting to internal server: %w", err)
	}

	return invoker, func() {
		invoker.Close()
		close()
	}, nil
}

// Test that the RPC invoker notifies the codespace of client activity on connection
func verifyNotifyCodespaceOfClientActivity(t *testing.T, server *mockServer) {