Fix bug where error in prompt function crashed REPL

src/PowerShellEditorServices/Services/PowerShell/Host/PsesInternalHost.cs

@@ -31,7 +31,7 @@ namespace Microsoft.PowerShell.EditorServices.Services.PowerShell.Host
 
     internal class PsesInternalHost : PSHost, IHostSupportsInteractiveSession, IRunspaceContext, IInternalPowerShellExecutionService
     {
-        private const string DefaultPrompt = "> ";
+        internal const string DefaultPrompt = "> ";
 
         private static readonly PropertyInfo s_scriptDebuggerTriggerObjectProperty;
 
@@ -865,21 +865,25 @@ private void DoOneRepl(CancellationToken cancellationToken)
             }
         }
 
-        private string GetPrompt(CancellationToken cancellationToken)
+        internal string GetPrompt(CancellationToken cancellationToken)
         {
             Runspace.ThrowCancelledIfUnusable();
             string prompt = DefaultPrompt;
             try
             {
                 // TODO: Should we cache PSCommands like this as static members?
                 PSCommand command = new PSCommand().AddCommand("prompt");
-                IReadOnlyList<string> results = InvokePSCommand<string>(command, executionOptions: null, cancellationToken);
+                IReadOnlyList<string> results = InvokePSCommand<string>(
+                    command,
+                    executionOptions: new PowerShellExecutionOptions { ThrowOnError = false },
+                    cancellationToken);
+
                 if (results?.Count > 0)
                 {
                     prompt = results[0];
                 }
             }
-            catch (CommandNotFoundException) { } // Use default prompt
+            catch (RuntimeException) { } // Use default prompt
 
             if (CurrentRunspace.RunspaceOrigin != RunspaceOrigin.Local)
             {

test/PowerShellEditorServices.Test/Session/PsesInternalHostTests.cs

@@ -162,12 +162,30 @@ await psesHost.ExecuteDelegateAsync(
                 CancellationToken.None).ConfigureAwait(true);
         }
 
+        // NOTE: Tests where we call functions that use PowerShell runspaces are slightly more
+        // complicated than one would expect because we explicitly need the methods to run on the
+        // pipeline thread, otherwise Windows complains about the the thread's apartment state not
+        // matching. Hence we use a delegate where it looks like we could just call the method.
+
+        [Fact]
+        public async Task CanHandleBrokenPrompt()
+        {
+            await psesHost.ExecutePSCommandAsync(
+                new PSCommand().AddScript("function prompt { throw }"),
+                CancellationToken.None).ConfigureAwait(true);
+
+            string prompt = await psesHost.ExecuteDelegateAsync(
+                nameof(psesHost.GetPrompt),
+                executionOptions: null,
+                (_, _) => psesHost.GetPrompt(CancellationToken.None),
+                CancellationToken.None).ConfigureAwait(true);
+
+            Assert.Equal(PsesInternalHost.DefaultPrompt, prompt);
+        }
+
         [Fact]
         public async Task CanLoadPSReadLine()
         {
-            // NOTE: This is slightly more complicated than one would expect because we explicitly
-            // need it to run on the pipeline thread otherwise Windows complains about the the
-            // thread's apartment state not matching.
             Assert.True(await psesHost.ExecuteDelegateAsync(
                 nameof(psesHost.TryLoadPSReadLine),
                 executionOptions: null,