refactoring the test

Author: bartekwasielak

src/ServiceControl.AcceptanceTests/Recoverability/ExternalIntegration/When_a_failed_edit_is_resolved_by_retry.cs

@@ -40,13 +40,8 @@ public async Task Should_publish_notification()
                         return false;
                     }
 
-                    if (!ctx.OriginalMessageHasBeenHandled)
-                    {
-                        return false;
-                    }
-
                     // second message - edit & retry
-                    if (!ctx.EditedMessageHasBeenSent)
+                    if (ctx.MessageSentCount == 0 && ctx.MessageHandledCount == 1)
                     {
                         var failedMessagedId = await this.GetOnlyFailedUnresolvedMessageId();
                         if (failedMessagedId == null)
@@ -55,7 +50,7 @@ public async Task Should_publish_notification()
                         }
 
                         ctx.OriginalMessageFailureId = failedMessagedId;
-                        ctx.EditedMessageHasBeenSent = true;
+                        ctx.MessageSentCount = 1;
 
                         string editedMessage = JsonSerializer.Serialize(new EditResolutionMessage
                         {
@@ -74,13 +69,8 @@ public async Task Should_publish_notification()
                         return false;
                     }
 
-                    if (!ctx.EditedAndRetriedMessageHandled)
-                    {
-                        return false;
-                    }
-
                     // third message - retry
-                    if (!ctx.RetriedMessageHasBeenSent)
+                    if (ctx.MessageSentCount == 1 && ctx.MessageHandledCount == 2)
                     {
                         var failedMessageIdAfterId = await this.GetOnlyFailedUnresolvedMessageId();
                         if (failedMessageIdAfterId == null)
@@ -89,22 +79,21 @@ public async Task Should_publish_notification()
                         }
 
                         ctx.EditedMessageFailureId = failedMessageIdAfterId;
-                        ctx.RetriedMessageHasBeenSent = true;
-                        ctx.NeedToHandleRetriedMessage = true;
+                        ctx.MessageSentCount = 2;
 
                         await this.Post<object>($"/api/errors/{ctx.EditedMessageFailureId}/retry");
                         return false;
                     }
 
-                    if (!ctx.RetriedMessageHandled)
+                    if (ctx.MessageHandledCount != 3)
                     {
                         return false;
                     }
 
-                    if (ctx.EditedMessageEditOf == null)
-                    {
-                        return false;
-                    }
+                    // if (ctx.EditedMessageEditOf == null)
+                    // {
+                    //     return false;
+                    // }
 
                     if (!ctx.MessageResolved)
                     {
@@ -124,19 +113,17 @@ public async Task Should_publish_notification()
 
         public class EditMessageResolutionContext : ScenarioContext
         {
-            public bool OriginalMessageHasBeenHandled { get; set; }
             public string OriginalMessageFailureId { get; set; }
+            public int MessageSentCount { get; set; }
+            public int MessageHandledCount { get; set; }
 
             public string ResolvedMessageId { get; set; }
-            public bool MessageResolved { get; set; }
-            public bool EditedAndRetriedMessageHandled { get; set; }
-            public bool RetriedMessageHandled { get; set; }
+           
             public string EditedMessageFailureId { get; set; }
-            public bool EditedMessageHasBeenSent { get; set; }
-            public bool RetriedMessageHasBeenSent { get; set; }
+
             public string EditedMessageEditOf { get; set; }
             public bool ExternalProcessorSubscribed { get; set; }
-            public bool NeedToHandleRetriedMessage { get; set; }
+            public bool MessageResolved { get; set; }
         }
 
         public class MessageReceiver : EndpointConfigurationBuilder
@@ -150,22 +137,22 @@ public class EditMessageResolutionHandler(EditMessageResolutionContext testConte
                 public Task Handle(EditResolutionMessage message, IMessageHandlerContext context)
                 {
                     // First run - supposed to fail
-                    if (!message.ThisIsAnEditedMessage)
+                    if (testContext.MessageSentCount == 0)
                     {
-                        testContext.OriginalMessageHasBeenHandled = true;
+                        testContext.MessageHandledCount = 1;
                         throw new SimulatedException();
                     }
 
                     // Second run - edit retry - supposed to fail
-                    if (!testContext.NeedToHandleRetriedMessage)
+                    if (testContext.MessageSentCount == 1)
                     {
                         testContext.EditedMessageEditOf = context.MessageHeaders["ServiceControl.EditOf"];
-                        testContext.EditedAndRetriedMessageHandled = true;
+                        testContext.MessageHandledCount = 2;
                         throw new SimulatedException();
                     }
 
                     // Last run - normal retry - supposed to succeed
-                    testContext.RetriedMessageHandled = true;
+                    testContext.MessageHandledCount = 3;
                     return Task.CompletedTask;
                 }