Use async delay

inputfalken · Jul 15, 2024 · 4e12b2e · 4e12b2e
1 parent b4dced2
commit 4e12b2e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/src/Dynatello/Handlers/TableAccessExtensions.cs b/src/Dynatello/Handlers/TableAccessExtensions.cs
@@ -1,4 +1,3 @@
-using Amazon.DynamoDBv2;
 using Amazon.DynamoDBv2.Model;
 using DynamoDBGenerator;
 using Dynatello.Builders;

diff --git a/tests/Dynatello.Tests/HandlerTests/GetRequestHandlerTests.cs b/tests/Dynatello.Tests/HandlerTests/GetRequestHandlerTests.cs
@@ -15,13 +15,14 @@ private class TestPipeLine : IRequestPipeLine
     {
         public DateTime? TimeStamp = null;
 
-        public Task<AmazonWebServiceResponse> Invoke(
+        public async Task<AmazonWebServiceResponse> Invoke(
             RequestContext requestContext,
             Func<RequestContext, Task<AmazonWebServiceResponse>> continuation
         )
         {
+            await Task.Delay(50);
             TimeStamp = DateTime.Now;
-            return continuation(requestContext);
+            return await continuation(requestContext);
         }
     }
 
@@ -39,14 +40,16 @@ public async Task Send_SuccessMockWithMockedRequestPipeline_ShouldReturnItemAndE
 
         amazonDynamoDB
             .GetItemAsync(Arg.Any<GetItemRequest>())
-            .Returns(x =>
+            .Returns(async x =>
             {
-                Thread.Sleep(200);
+                await Task.Delay(200);
                 return response;
             });
 
         var pipelines = new TestPipeLine[]
         {
+            new TestPipeLine(),
+            new TestPipeLine(),
             new TestPipeLine(),
             new TestPipeLine(),
             new TestPipeLine()
@@ -62,18 +65,21 @@ public async Task Send_SuccessMockWithMockedRequestPipeline_ShouldReturnItemAndE
                     x.RequestsPipelines.Add(pipelines[0]);
                     x.RequestsPipelines.Add(pipelines[1]);
                     x.RequestsPipelines.Add(pipelines[2]);
+                    x.RequestsPipelines.Add(pipelines[3]);
+                    x.RequestsPipelines.Add(pipelines[4]);
                     x.AmazonDynamoDB = amazonDynamoDB;
                 }
             )
             .Send(expected.Id, default);
         Assert.Equal(expected, actual);
         Assert.All(pipelines, x => Assert.NotNull(x.TimeStamp));
         var pipeLineTimestamps = pipelines.Select(x => x.TimeStamp!.Value).ToArray();
+
         Assert.DoesNotContain(
             false,
             pipeLineTimestamps.Zip(
                 pipeLineTimestamps.Skip(1),
-                (x, y) => (x < y) && (y - x) < TimeSpan.FromMilliseconds(20) // ugly hack to verify that request comes last.
+                (x, y) => (x < y) && (y - x) < TimeSpan.FromMilliseconds(100) // ugly hack to verify that request comes last.
             )
         );
     }