Co-operatively cancel RPC Calls

OctopusDeploy · Dec 6, 2023 · a2e09aa · a2e09aa
1 parent 9b19f36
commit a2e09aa
Show file tree

Hide file tree

Showing 30 changed files with 625 additions and 1,016 deletions.
diff --git a/source/Octopus.Tentacle.Client.Tests/RpcCallExecutorFixture.cs b/source/Octopus.Tentacle.Client.Tests/RpcCallExecutorFixture.cs
@@ -164,13 +164,13 @@ await AssertionExtensions.Should(
                             throw exception;
                         },
                         RetryDuration, clientOperationMetricsBuilder, cancellationToken))
-                .ThrowAsync<TaskCanceledException>();
+                .ThrowAsync<HalibutClientException>();
 
             // Assert
             var metric = rpcCallObserver.RpcCallMetrics.Should().ContainSingle().Subject;
 
             metric.Succeeded.Should().BeFalse();
-            metric.Exception?.GetType().Should().Be(typeof(TaskCanceledException));
+            metric.Exception.Should().BeEquivalentTo(exception);
             metric.HasException.Should().BeTrue();
             metric.WasCancelled.Should().BeTrue();
 
@@ -302,7 +302,6 @@ await sut.ExecuteWithRetries(
                 new RpcCall(RpcService, RpcCallName),
                 action,
                 Substitute.For<ILog>(),
-                false,
                 clientOperationMetricsBuilder,
                 cancellationToken);
         }
@@ -320,7 +319,6 @@ private static async Task<Guid> ExecuteResult(
                 new RpcCall(RpcService, RpcCallName),
                 action,
                 Substitute.For<ILog>(),
-                abandonActionOnCancellation: false,
                 clientOperationMetricsBuilder,
                 cancellationToken);
         }
@@ -342,7 +340,6 @@ await sut.ExecuteWithNoRetries(
                     return true;
                 },
                 Substitute.For<ILog>(),
-                abandonActionOnCancellation: false,
                 clientOperationMetricsBuilder,
                 cancellationToken);
         }