.

OctopusDeploy · Dec 5, 2023 · 80a8ea9 · 80a8ea9
1 parent c8fdca0
commit 80a8ea9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...ce/Octopus.Tentacle.Tests.Integration/ClientScriptExecutionCanRecoverFromNetworkIssues.cs b/...ce/Octopus.Tentacle.Tests.Integration/ClientScriptExecutionCanRecoverFromNetworkIssues.cs
@@ -239,7 +239,11 @@ await clientTentacle.TentacleClient.ExecuteScript(startScriptCommand,
                 actualException = ex;
             }
 
-            actualException.Should().NotBeNull().And.BeOfType<OperationCanceledException>().And.Match<Exception>(x => x.Message == "Script execution was cancelled");
+            var expectedException = new ExceptionContractAssertionBuilder(FailureScenario.ScriptExecutionCancelled, tentacleConfigurationTestCase, clientTentacle)
+                .ForScriptService(ScriptServiceOperation.CancelScript).Build();
+
+            actualException.Should().NotBeNull();
+            actualException!.ShouldMatchExceptionContract(expectedException);
 
             var allLogs = logs.JoinLogs();
             allLogs.Should().Contain("hello");