fix tests

Author: hjgraca

libraries/tests/AWS.Lambda.Powertools.Idempotency.Tests/IdempotencyTest.cs

@@ -14,6 +14,7 @@
  */
 
 using System;
+using System.Collections.Generic;
 using System.IO;
 using System.Text.Json;
 using System.Threading.Tasks;
@@ -68,6 +69,13 @@ public async Task EndToEndTest()
             TableName = _tableName
         });
         scanResponse.Count.Should().Be(1);
+        
+        // delete row dynamo
+        var key = new Dictionary<string, AttributeValue>
+        {
+            ["id"] = new AttributeValue { S = "testFunction.GetPageContents#ff323c6f0c5ceb97eed49121babcec0f" }
+        };
+        await _client.DeleteItemAsync(new DeleteItemRequest{TableName = _tableName, Key = key});
     }
 
     [Fact]
@@ -96,13 +104,22 @@ public async Task EndToEndTestMethod()
         var response2 = await function.Handle(request, context);
         function.MethodCalled.Should().BeFalse();
 
-        JsonSerializer.Serialize(response).Should().Be(JsonSerializer.Serialize(response));
+        // Assert
+        JsonSerializer.Serialize(response).Should().Be(JsonSerializer.Serialize(response2));
+        response.Body.Should().Contain("hello world");
         response2.Body.Should().Contain("hello world");
 
         var scanResponse = await _client.ScanAsync(new ScanRequest
         {
             TableName = _tableName
         });
         scanResponse.Count.Should().Be(1);
+        
+        // delete row dynamo
+        var key = new Dictionary<string, AttributeValue>
+        {
+            ["id"] = new AttributeValue { S = "testFunction.GetPageContents#ff323c6f0c5ceb97eed49121babcec0f" }
+        };
+        await _client.DeleteItemAsync(new DeleteItemRequest{TableName = _tableName, Key = key});
     }
 }

libraries/tests/AWS.Lambda.Powertools.Idempotency.Tests/Internal/IdempotentAspectTests.cs

@@ -310,8 +310,8 @@ public void Handle_WhenIdempotencyOnSubMethodAnnotated_AndSecondCall_AndNotExpir
     {
         // Arrange
         var store = Substitute.For<BasePersistenceStore>();
-        store.SaveInProgress(Arg.Any<JsonDocument>(), Arg.Any<DateTimeOffset>(), null)
-            .Throws(new IdempotencyItemAlreadyExistsException());
+        store.SaveInProgress(Arg.Any<JsonDocument>(), Arg.Any<DateTimeOffset>(), Arg.Any<double>())
+            .Returns(_ => throw new IdempotencyItemAlreadyExistsException());
 
         Idempotency.Configure(builder => builder.WithPersistenceStore(store));
 
@@ -327,7 +327,7 @@ public void Handle_WhenIdempotencyOnSubMethodAnnotated_AndSecondCall_AndNotExpir
             .Returns(record);
 
         // Act
-        var function = new IdempotencyInternalFunction(false);
+        var function = new IdempotencyInternalFunction(true);
         Basket resultBasket = function.HandleRequest(product, new TestLambdaContext());
 
         // assert