chore: use task.delay

SpecterOps · Aug 15, 2024 · 65499f0 · 65499f0
1 parent 56557f8
commit 65499f0
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/test/unit/ComputerSessionProcessorTest.cs b/test/unit/ComputerSessionProcessorTest.cs
@@ -235,11 +235,8 @@ public async Task ComputerSessionProcessor_ReadUserSessionsPrivileged_FilteringW
         public async Task ComputerSessionProcessor_TestTimeout() {
             var nativeMethods = new Mock<NativeMethods>();
             nativeMethods.Setup(x => x.NetSessionEnum(It.IsAny<string>())).Callback(() => {
-                Thread.Sleep(200);
+                Task.Delay(1000).Wait();
             }).Returns(Array.Empty<NetSessionEnumResults>());
-            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";
@@ -258,7 +255,7 @@ public async Task ComputerSessionProcessor_TestTimeout() {
         public async Task ComputerSessionProcessor_TestTimeoutPrivileged() {
             var nativeMethods = new Mock<NativeMethods>();
             nativeMethods.Setup(x => x.NetWkstaUserEnum(It.IsAny<string>())).Callback(() => {
-                Thread.Sleep(200);
+                Task.Delay(1000).Wait();
             }).Returns(Array.Empty<NetWkstaUserEnumResults>());
             var processor = new ComputerSessionProcessor(new MockLdapUtils(),"", nativeMethods.Object);
             var receivedStatus = new List<CSVComputerStatus>();