fix(testing): add support for sdk v3 types

typedorm · Aug 21, 2022 · 6027aaa · 6027aaa
1 parent 706414b
commit 6027aaa
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/packages/testing/index.ts b/packages/testing/index.ts
@@ -5,17 +5,18 @@ import {
   Container,
   ConnectionManager,
 } from '@typedorm/core';
+import {DocumentClient} from '@typedorm/document-client';
 
-import {DynamoDB} from 'aws-sdk';
-
-export function createTestConnection(
+export function createTestConnection<T = any>(
   connectionOptions: Replace<
     ConnectionOptions,
     'documentClient',
     {
-      documentClient?: {
-        [key in keyof DynamoDB.DocumentClient]?: jest.SpyInstance;
-      };
+      documentClient?:
+        | {
+            [key in keyof DocumentClient]?: jest.SpyInstance;
+          }
+        | T;
     }
   >
 ) {