we are speed

Energinet-DataHub · Dec 5, 2024 · 52fdb57 · 52fdb57
1 parent 11e96b3
commit 52fdb57
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/source/IntegrationTests/Behaviours/BehavioursTestBase.cs b/source/IntegrationTests/Behaviours/BehavioursTestBase.cs
@@ -362,16 +362,25 @@ protected async Task<List<PeekResultDto>> WhenActorPeeksAllMessages(ActorNumber
         var timeoutAt = DateTime.UtcNow.AddMinutes(1);
         while (DateTime.UtcNow < timeoutAt)
         {
+            var thereWasNothingToPeek = true;
             foreach (var messageCategory in EnumerationType.GetAll<MessageCategory>())
             {
                 var peekResult = await WhenActorPeeksMessage(actorNumber, actorRole, documentFormat, messageCategory);
 
                 if (peekResult is null)
+                {
                     break;
+                }
 
+                thereWasNothingToPeek = false;
                 peekResults.Add(peekResult);
                 await WhenActorDequeuesMessage(peekResult.MessageId.Value, actorNumber, actorRole);
             }
+
+            if (thereWasNothingToPeek)
+            {
+                break;
+            }
         }
 
         return peekResults;