Update tests

aws · Oct 30, 2024 · 662930d · 662930d
1 parent 299265b
commit 662930d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/sdk/src/Services/DynamoDBv2/Custom/DocumentModel/_async/DocumentBatchWrite.Async.cs b/sdk/src/Services/DynamoDBv2/Custom/DocumentModel/_async/DocumentBatchWrite.Async.cs
@@ -14,6 +14,7 @@
  */
 #pragma warning disable 1574
 
+using Amazon.Runtime.Telemetry.Tracing;
 using System.Threading;
 using System.Threading.Tasks;
 

diff --git a/sdk/test/Services/DynamoDBv2/UnitTests/Custom/DynamoDBTests.cs b/sdk/test/Services/DynamoDBv2/UnitTests/Custom/DynamoDBTests.cs
@@ -489,6 +489,9 @@ public void DisableFetchingTableMetadata_KeyWithConverter_DateTimeToString()
                     request.Key.ContainsKey("CreationDate") && 
                     request.Key["CreationDate"].S == "0001-01-01T00:00:00.000Z")));
 
+
+            mock.Verify(x => x.Config, Times.AtLeastOnce());
+
             mock.VerifyNoOtherCalls();
         }
 
@@ -514,6 +517,9 @@ public void DisableFetchingTableMetadata_KeyWithConverter_DateTimeToNumber()
                     request.Key.ContainsKey("CreationDate") &&
                     request.Key["CreationDate"].N == "1024")));
 
+
+            mock.Verify(x => x.Config, Times.AtLeastOnce());
+
             mock.VerifyNoOtherCalls();
         }
 
@@ -539,6 +545,8 @@ public void DisableFetchingTableMetadata_KeyWithConverter_DateTimeToBinary()
                     request.Key.ContainsKey("CreationDate") &&
                     BitConverter.ToString(request.Key["CreationDate"].B.ToArray()) == "00-04-00-00-00-00-00-40")));
 
+            mock.Verify(x => x.Config, Times.AtLeastOnce());
+
             mock.VerifyNoOtherCalls();
         }
 
@@ -556,6 +564,8 @@ public void DisableFetchingTableMetadata_KeyWithConverter_DateTimeToBool_ThrowsE
             Assert.ThrowsException<InvalidOperationException>(() => 
                 context.Load<HashKeyConverter_DateTimeToBool>(new DateTime(1024, DateTimeKind.Utc)));
 
+            mock.Verify(x => x.Config, Times.AtLeastOnce());
+
             mock.VerifyNoOtherCalls();
         }