Fix module path during testing with runtime hook

Author: andyleejordan

src/PowerShellEditorServices/Services/PowerShellContext/Session/PSReadLinePromptContext.cs

@@ -22,13 +22,18 @@ internal class PSReadLinePromptContext : IPromptContext
             "..",
             "..",
             "..",
-#if TEST
-            // When using xUnit (dotnet test) the assemblies are deployed to the
-            // test project folder, invalidating our relative path assumption.
+            "PSReadLine");
+
+        // When using xUnit (dotnet test) the assemblies are deployed to the
+        // test project folder, invalidating our relative path assumption.
+        private static readonly string _psReadLineTestModulePath = Path.Combine(
+            Path.GetDirectoryName(typeof(PSReadLinePromptContext).Assembly.Location),
+            "..",
+            "..",
+            "..",
             "..",
             "..",
             "module",
-#endif
             "PSReadLine");
 
         private static readonly Lazy<CmdletInfo> s_lazyInvokeReadLineForEditorServicesCmdletInfo = new Lazy<CmdletInfo>(() =>
@@ -79,15 +84,16 @@ internal PSReadLinePromptContext(
         internal static bool TryGetPSReadLineProxy(
             ILogger logger,
             Runspace runspace,
-            out PSReadLineProxy readLineProxy)
+            out PSReadLineProxy readLineProxy,
+            bool testing = false)
         {
             readLineProxy = null;
             logger.LogTrace("Attempting to load PSReadLine");
             using (var pwsh = PowerShell.Create())
             {
                 pwsh.Runspace = runspace;
                 pwsh.AddCommand("Microsoft.PowerShell.Core\\Import-Module")
-                    .AddParameter("Name", _psReadLineModulePath)
+                    .AddParameter("Name", testing ? _psReadLineTestModulePath : _psReadLineModulePath)
                     .Invoke();
 
                 var psReadLineType = Type.GetType("Microsoft.PowerShell.PSConsoleReadLine, Microsoft.PowerShell.PSReadLine2");

test/PowerShellEditorServices.Test/Session/PowerShellContextTests.cs

@@ -154,7 +154,8 @@ public void CanGetPSReadLineProxy()
             Assert.True(PSReadLinePromptContext.TryGetPSReadLineProxy(
                 NullLogger.Instance,
                 PowerShellContextFactory.initialRunspace,
-                out PSReadLineProxy proxy));
+                out PSReadLineProxy proxy,
+                true));
         }
 
         #region Helper Methods