Fix connect tests

hivemq · May 16, 2024 · 365db2a · 365db2a
1 parent 64e57aa
commit 365db2a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Tests/HiveMQtt.Test/HiveMQClient/ConnectTest.cs b/Tests/HiveMQtt.Test/HiveMQClient/ConnectTest.cs
@@ -68,6 +68,7 @@ public async Task TestConnectEventsAsync()
 
         // Wait for event handlers to finish
         await taskCompletionSource.Task.WaitAsync(TimeSpan.FromSeconds(5)).ConfigureAwait(false);
+        await Task.Delay(1000).ConfigureAwait(false);
 
         // Assert that all Events were called
         Assert.True(client.LocalStore.ContainsKey("BeforeConnectHandlerCalled"));
@@ -107,6 +108,7 @@ public async Task Test_AfterDisconnectEvent_Async()
 
         // Wait for event handlers to finish
         await taskCompletionSource.Task.WaitAsync(TimeSpan.FromSeconds(5)).ConfigureAwait(false);
+        await Task.Delay(1000).ConfigureAwait(false);
 
         // Assert that all Events were called
         Assert.True(client.LocalStore.ContainsKey("AfterDisconnectHandlerCalled"));