Extend CancellationToken (#653)

Author: BMurri

src/TriggerService.Tests/UpdateWorkflowStatusTests.cs

@@ -393,7 +393,7 @@ private static TesTask GetTesTask(string workflowId, int shard, int attempt, par
                 .Returns(Task.FromResult(new GetTimingResponse()));
 
             repository
-                .Setup(r => r.GetItemsAsync(It.IsAny<Expression<Func<TesTask, bool>>>()))
+                .Setup(r => r.GetItemsAsync(It.IsAny<Expression<Func<TesTask, bool>>>(), It.IsAny<System.Threading.CancellationToken>()))
                 .Returns(Task.FromResult(tesTasks));
 
             var logger = new Mock<ILogger<TriggerHostedService>>().Object;

src/TriggerService/TriggerHostedService.cs

@@ -546,7 +546,7 @@ private async Task<WorkflowFailureInfo> GetWorkflowFailureInfoAsync(Guid workflo
                 }
             }
 
-            var tesTasks = await tesTaskRepository.GetItemsAsync(t => t.WorkflowId == workflowId.ToString());
+            var tesTasks = await tesTaskRepository.GetItemsAsync(t => t.WorkflowId == workflowId.ToString(), CancellationToken.None);
 
             // Select the last attempt of each Cromwell task, and then select only the failed ones
             // If CromwellResultCode is > 0, point to Cromwell stderr/out. Otherwise, if batch exit code > 0, point to Batch stderr/out
@@ -589,7 +589,7 @@ private async Task<WorkflowFailureInfo> GetWorkflowFailureInfoAsync(Guid workflo
 
         private async Task<List<TaskWarning>> GetWorkflowTaskWarningsAsync(Guid workflowId)
         {
-            var tesTasks = await tesTaskRepository.GetItemsAsync(t => t.WorkflowId == workflowId.ToString());
+            var tesTasks = await tesTaskRepository.GetItemsAsync(t => t.WorkflowId == workflowId.ToString(), CancellationToken.None);
 
             // Select the last attempt of each Cromwell task, and then select only those that have a warning
             var taskWarnings = tesTasks