fix: dynamodb tests

vighnesh153 · Nov 9, 2024 · 8165425 · 8165425
1 parent a835640
commit 8165425
Showing 1 changed file with 21 additions and 7 deletions.
diff --git a/tools-deno/tools-server/src/aws_dynamodb/dynamodb_table_impl.test.ts b/tools-deno/tools-server/src/aws_dynamodb/dynamodb_table_impl.test.ts
@@ -1,7 +1,7 @@
 import { assertEquals } from "@std/assert";
 import { beforeEach, describe, it } from "@std/testing/bdd";
 import {
-  PutCommand,
+  BatchWriteCommand,
   QueryCommand,
   type ServiceOutputTypes,
 } from "@aws-sdk/lib-dynamodb";
@@ -129,9 +129,16 @@ describe("createOne tests", () => {
 
     assertEquals(
       fakeDocumentClient.sendCalledWithArgs?.[0]?.input,
-      new PutCommand({
-        Item: item,
-        TableName: tableName,
+      new BatchWriteCommand({
+        RequestItems: {
+          [tableName]: [
+            {
+              PutRequest: {
+                Item: item,
+              },
+            },
+          ],
+        },
       }).input,
     );
     assertEquals(fakeDocumentClient.sendCalledWithArgs?.[1], undefined);
@@ -153,9 +160,16 @@ describe("createOne tests", () => {
 
     assertEquals(
       fakeDocumentClient.sendCalledWithArgs?.[0]?.input,
-      new PutCommand({
-        Item: item,
-        TableName: tableName,
+      new BatchWriteCommand({
+        RequestItems: {
+          [tableName]: [
+            {
+              PutRequest: {
+                Item: item,
+              },
+            },
+          ],
+        },
       }).input,
     );
     assertEquals(fakeDocumentClient.sendCalledWithArgs?.[1], undefined);