Fix #365 (#366)

* Fix #365 * Fixing the unit tests. Co-authored-by: James O'Neill <[email protected]> Co-authored-by: Donovan Brown <[email protected]>
MethodsAndPractices · Oct 15, 2020 · f95ce17 · f95ce17
1 parent 7daec7c
commit f95ce17
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/Source/Classes/Cache/QueryCache.cs b/Source/Classes/Cache/QueryCache.cs
@@ -36,11 +36,10 @@ public static void Update(IEnumerable<string> list)
                }
 
                // This will return just the names
-               list = Cache.Shell.AddArgument(queries)
-                                 .AddCommand("Select-Object")
+               list = Cache.Shell.AddCommand("Select-Object")
                                  .AddParameter("ExpandProperty", "Name")
                                  .AddCommand("Sort-Object")
-                                 .Invoke<string>();
+                                 .Invoke<string>(queries);
             }
          }
 

diff --git a/Tests/library/Cache/QueryCacheTests.cs b/Tests/library/Cache/QueryCacheTests.cs
@@ -45,6 +45,7 @@ public void QueryCache_Update_With_Null_List()
          var expected = 2;
          var ps = BaseTests.PrepPowerShell();
          ps.Invoke().Returns(this._queries);
+         ps.Invoke<string>(this._queries).Returns(this._queryNames);
          ps.Invoke<string>().Returns(this._defaultProject, this._queryNames);
          QueryCache.Cache.Shell = ps;
 
@@ -62,6 +63,8 @@ public void QueryCache_GetCurrent()
          var expected = 2;
          var ps = BaseTests.PrepPowerShell();
          ps.Invoke().Returns(this._queries);
+         ps.Invoke<string>(this._queries).Returns(this._queryNames);
+
          ps.Invoke<string>().Returns(this._defaultProject, this._queryNames);
          QueryCache.Invalidate();
          QueryCache.Cache.Shell = ps;