fix(tests): Fixes tests.

Author: ryasmi

src/tests/getEntity/test.ts

@@ -14,7 +14,7 @@ export default (facade: Facade<TestId, TestEntity>) => {
 
     it('should not error when identifier does exist', async () => {
       await facade.createEntity({ id: testId, entity: testEntity });
-      const retrievedEntity = await facade.getEntity({ id: testId });
+      const { entity: retrievedEntity } = await facade.getEntity({ id: testId });
       assert.deepEqual(retrievedEntity, testEntity);
     });
   });

src/tests/overwriteEntity/test.ts

@@ -20,8 +20,11 @@ export default (facade: Facade<TestId, TestEntity>) => {
         stringProp: 'test_string_prop_overwrite',
       };
       await facade.createEntity({ id: testId, entity: testEntity });
-      const overwrittenEntity = await facade.overwriteEntity({ id: testId, entity: testOverwrite });
-      const retrievedEntity = await facade.getEntity({ id: testId });
+      const { entity: overwrittenEntity } = await facade.overwriteEntity({
+        entity: testOverwrite,
+        id: testId,
+      });
+      const { entity: retrievedEntity } = await facade.getEntity({ id: testId });
       assert.deepEqual(overwrittenEntity, testOverwrite);
       assert.deepEqual(retrievedEntity, testOverwrite);
     });

src/tests/patchEntity/test.ts

@@ -8,8 +8,11 @@ import { TestEntity, testEntity, testId, TestId } from '../utils/testEntity';
 export default (facade: Facade<TestId, TestEntity>) => {
   describe('patchEntity', () => {
     const assertPatch = async (testPatch: Partial<TestEntity>) => {
-      const patchedEntity = await facade.patchEntity({ id: testId, patch: testPatch });
-      const retrievedEntity = await facade.getEntity({ id: testId });
+      const { entity: patchedEntity } = await facade.patchEntity({
+        id: testId,
+        patch: testPatch,
+      });
+      const { entity: retrievedEntity } = await facade.getEntity({ id: testId });
       const expectedEntity = { ...testEntity, ...testPatch };
       assert.deepEqual(patchedEntity, expectedEntity);
       assert.deepEqual(retrievedEntity, expectedEntity);

src/tests/upsertEntity/test.ts

@@ -6,8 +6,11 @@ import { TestEntity, testEntity, testId, TestId } from '../utils/testEntity';
 export default (facade: Facade<TestId, TestEntity>) => {
   describe('upsertEntity', () => {
     it('should create when identifier does not exist', async () => {
-      const createdEntity = await facade.upsertEntity({ id: testId, entity: testEntity });
-      const retrievedEntity = await facade.getEntity({ id: testId });
+      const { entity: createdEntity } = await facade.upsertEntity({
+        entity: testEntity,
+        id: testId,
+      });
+      const { entity: retrievedEntity } = await facade.getEntity({ id: testId });
       assert.deepEqual(createdEntity, testEntity);
       assert.deepEqual(retrievedEntity, testEntity);
     });
@@ -20,8 +23,11 @@ export default (facade: Facade<TestId, TestEntity>) => {
         stringProp: 'test_string_prop_overwrite',
       };
       await facade.createEntity({ id: testId, entity: testEntity });
-      const overwrittenEntity = await facade.upsertEntity({ id: testId, entity: testOverwrite });
-      const retrievedEntity = await facade.getEntity({ id: testId });
+      const { entity: overwrittenEntity } = await facade.upsertEntity({
+        entity: testOverwrite,
+        id: testId,
+      });
+      const { entity: retrievedEntity } = await facade.getEntity({ id: testId });
       assert.deepEqual(overwrittenEntity, testOverwrite);
       assert.deepEqual(retrievedEntity, testOverwrite);
     });