chore: fix tests

SpecterOps · Aug 15, 2024 · 56557f8 · 56557f8
1 parent 68664a6
commit 56557f8
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/test/unit/ComputerSessionProcessorTest.cs b/test/unit/ComputerSessionProcessorTest.cs
@@ -252,14 +252,26 @@ public async Task ComputerSessionProcessor_TestTimeout() {
             Assert.Single(receivedStatus);
             var status = receivedStatus[0];
             Assert.Equal("Timeout", status.Status);
+        }
+
+        [Fact]
+        public async Task ComputerSessionProcessor_TestTimeoutPrivileged() {
+            var nativeMethods = new Mock<NativeMethods>();
+            nativeMethods.Setup(x => x.NetWkstaUserEnum(It.IsAny<string>())).Callback(() => {
+                Thread.Sleep(200);
+            }).Returns(Array.Empty<NetWkstaUserEnumResults>());
+            var processor = new ComputerSessionProcessor(new MockLdapUtils(),"", nativeMethods.Object);
+            var receivedStatus = new List<CSVComputerStatus>();
+            var machineDomainSid = $"{Consts.MockDomainSid}-1000";
+            processor.ComputerStatusEvent += async status =>  {
+                receivedStatus.Add(status);
+            };
 
-            receivedStatus.Clear();
-
-            results = await processor.ReadUserSessionsPrivileged("primary.testlab.local", machineDomainSid, "testlab.local",
+            var results = await processor.ReadUserSessionsPrivileged("primary.testlab.local", machineDomainSid, "testlab.local",
                 TimeSpan.FromMilliseconds(1));
             Assert.Empty(results.Results);
             Assert.Single(receivedStatus);
-            status = receivedStatus[0];
+            var status = receivedStatus[0];
             Assert.Equal("Timeout", status.Status);
         }
     }