Fix tests

blazejkustra · Apr 14, 2024 · a85ea6c · a85ea6c
1 parent 023b54a
commit a85ea6c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/tests/unit/entity/helpers/converters.test.ts b/tests/unit/entity/helpers/converters.test.ts
@@ -88,6 +88,10 @@ const mockEntityAttributes = {
     propertyName: 'boolean',
     type: Boolean,
   },
+  binary: {
+    propertyName: 'binary',
+    type: Uint8Array,
+  },
 } as any as AttributesMetadata;
 
 const dynamoObject = {
@@ -102,6 +106,7 @@ const dynamoObject = {
   set: { SS: ['1', '2', '3'] },
   array: { L: [{ S: '1' }, { S: '2' }] },
   boolean: { BOOL: true },
+  binary: { B: new Uint8Array([1, 2, 3]) },
 };
 
 describe('Converters entity helpers', () => {
@@ -136,15 +141,15 @@ describe('Converters entity helpers', () => {
       getEntityMetadataSpy.mockReturnValue(metadata as any);
 
       expect(convertAttributeValuesToEntity(MockEntity, dynamoObject)).toEqual(mockInstance);
-      expect(truncateValueSpy).toBeCalledTimes(15);
+      expect(truncateValueSpy).toBeCalledTimes(16);
     });
 
     test('Should return object in dynamode format', async () => {
       getEntityAttributesSpy.mockReturnValue(mockEntityAttributes);
       getEntityMetadataSpy.mockReturnValue(metadata as any);
 
       expect(convertAttributeValuesToEntity(MockEntity, dynamoObject)).toEqual(mockInstance);
-      expect(truncateValueSpy).toBeCalledTimes(15);
+      expect(truncateValueSpy).toBeCalledTimes(16);
     });
   });
 
@@ -162,7 +167,7 @@ describe('Converters entity helpers', () => {
       getEntityAttributesSpy.mockReturnValue(mockEntityAttributes);
 
       expect(convertEntityToAttributeValues(MockEntity, mockInstance)).toEqual(dynamoObject);
-      expect(transformValueSpy).toBeCalledTimes(15);
+      expect(transformValueSpy).toBeCalledTimes(16);
     });
   });
 

diff --git a/tests/unit/stream/index.test.ts b/tests/unit/stream/index.test.ts
@@ -21,6 +21,7 @@ const validMockEntityImage = {
   boolean: { BOOL: true },
   strDate: { S: '2001-09-09T01:46:40.000Z' },
   numDate: { N: '1000000000000' },
+  binary: { B: new Uint8Array([1, 2, 3]) },
 };
 
 const validNewImageStream: DynamoDBRecord = {