added event type to tell if insert, modify or remove

Author: AlexStansfield

src/Table.d.ts

@@ -11,6 +11,7 @@ export type EntityGroup = {
 
 export type StreamEntityGroup = {
     [key: string]: {
+        type: 'INSERT' | 'MODIFY' | 'REMOVE',
         new?: AnyEntity,
         old?: AnyEntity
     }[]

src/Table.js

@@ -1066,7 +1066,9 @@ export class Table {
                 continue
             }
 
-            const model = {}
+            const model = {
+                type: record.eventName
+            }
             let typeNew
             let typeOld
 

test/stream.ts

@@ -116,18 +116,19 @@ const event = {
 
 test('Stream has New Images', async () => {
     const streamModels = table.stream(event.Records as DynamoDBRecord[]);
-    const newModels = streamModels.User.filter((streamModel) => !!streamModel.new)
-      .map((streamModel) => streamModel.new)
+    const models = streamModels.User.filter((streamModel) => !!streamModel.new)
 
-    expect(newModels).toHaveLength(2);
-    expect(newModels[0]).toEqual(
+    expect(models).toHaveLength(2)
+    expect(models[0].type).toEqual('INSERT')
+    expect(models[0].new).toEqual(
         expect.objectContaining({
             id: '1234',
             name: 'alice',
             registered: new Date('2022-01-01Z'),
         })
     )
-    expect(newModels[1]).toEqual(
+    expect(models[1].type).toEqual('MODIFY')
+    expect(models[1].new).toEqual(
       expect.objectContaining({
           id: '1235',
           name: 'bob',
@@ -138,11 +139,11 @@ test('Stream has New Images', async () => {
 
 test('Stream has Old Images', async () => {
     const streamModels = table.stream(event.Records as DynamoDBRecord[]);
-    const oldModels = streamModels.User.filter((streamModel) => !!streamModel.old)
-      .map((streamModel) => streamModel.old)
+    const models = streamModels.User.filter((streamModel) => !!streamModel.old)
 
-    expect(oldModels).toHaveLength(1);
-    expect(oldModels[0]).toEqual(
+    expect(models).toHaveLength(1)
+    expect(models[0].type).toEqual('MODIFY')
+    expect(models[0].old).toEqual(
       expect.objectContaining({
           name: 'rob',
           registered: new Date('2022-01-02Z'),