fix taskingtests

MythicAgents · Feb 27, 2024 · bf9b92d · bf9b92d
1 parent 00f6f52
commit bf9b92d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Payload_Type/athena/athena/agent_code/Tests/Agent.Tests/AgentTests/TaskingTests.cs b/Payload_Type/athena/athena/agent_code/Tests/Agent.Tests/AgentTests/TaskingTests.cs
@@ -75,12 +75,12 @@ public async Task TestGetTaskingMultiple()
                 rpfwd = new List<ServerDatagram>(),
                 delegates = new List<DelegateMessage>(),
             }) };
-
-            _profile.First().SetTaskingReceived += (sender, args) => taskingReceived.Set();
             Agent _agent = new Agent(_profile, _taskManager, _logger, _config, _tokenManager);
             TestProfile prof = (TestProfile)_profile.First();
 
             Task.Run(_agent.Start);
+            _profile.First().SetTaskingReceived += (sender, args) => taskingReceived.Set();
+
             ((TestTaskManager)_taskManager).WaitForNumberOfJobs(3);
             //prof.taskingSent.WaitOne(1000);
             _profile.First().StopBeacon();