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 index ecc4800b..c4c55f58 100644 --- 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(), delegates = new List(), }) }; - - _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();