update ls test for parent parsing

MythicAgents · Feb 16, 2024 · fbeb505 · fbeb505
1 parent 28b5595
commit fbeb505
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Payload_Type/athena/athena/agent_code/Tests/Agent.Tests/PluginTests/LsTests.cs b/Payload_Type/athena/athena/agent_code/Tests/Agent.Tests/PluginTests/LsTests.cs
@@ -36,14 +36,17 @@ public LsTests()
         [TestMethod]
         public void TestValidParentPath()
         {
-            string path = string.Empty;
+            string path;
+            string parent;
             if (OperatingSystem.IsWindows())
             {
                 path = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.System), "drivers", "etc");
+                parent = "drivers";
             }
             else
             {
                 path = Path.Combine("/", "etc");
+                parent = "";
             }
             Dictionary<string, string> parameters = new Dictionary<string, string>
             {
@@ -56,7 +59,7 @@ public void TestValidParentPath()
             ((TestMessageManager)_messageManager).hasResponse.WaitOne();
             string response = ((TestMessageManager)_messageManager).GetRecentOutput().Result;
             FileBrowserTaskResponse fb = JsonSerializer.Deserialize<FileBrowserTaskResponse>(response);
-            Assert.AreEqual(fb.file_browser.parent_path, Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.System), "drivers"));
+            Assert.AreEqual(fb.file_browser.parent_path, Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.System), parent));
         }
         [TestMethod]
         public void TestGetFileListingLocal()