From bbf0878afc247917a82121328b01102fbd87a871 Mon Sep 17 00:00:00 2001
From: Johan Haals <johan.haals@gmail.com>
Date: Tue, 27 Apr 2021 18:28:55 +0200
Subject: [PATCH] WIP tests for DefaultProcessingDatabase

Signed-off-by: Johan Haals <johan.haals@gmail.com>
---
 .../DefaultProcessingDatabase.test.ts         | 62 +++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts b/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts
index f67b8603db..a5c1e56cff 100644
--- a/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts
+++ b/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts
@@ -429,4 +429,66 @@ describe('Default Processing Database', () => {
       ).toBeFalsy();
     });
   });
+
+  describe('updateProcessedEntity', () => {
+    it('should throw if the entity does not exist', async () => {
+      await processingDatabase.transaction(async tx => {
+        await expect(
+          processingDatabase.updateProcessedEntity(tx, {
+            id: '9',
+            processedEntity: {
+              apiVersion: '1.0.0',
+              metadata: {
+                name: 'new-root',
+              },
+              kind: 'Location',
+            } as Entity,
+            deferredEntities: [],
+            relations: [],
+          }),
+        ).rejects.toThrow('Processing state not found for 9');
+      });
+    });
+
+    it('should update a processed entity', async () => {
+      await db<DbRefreshStateRow>('refresh_state').insert({
+        entity_id: '123',
+        entity_ref: 'Component:default/wacka',
+        unprocessed_entity: '',
+        errors: '',
+        next_update_at: 'now()',
+        last_discovery_at: 'now()',
+      });
+
+      const deferredEntity = {
+        apiVersion: '1.0.0',
+        metadata: {
+          name: 'deferred',
+        },
+        kind: 'Location',
+      } as Entity;
+
+      await processingDatabase.transaction(async tx => {
+        await processingDatabase.updateProcessedEntity(tx, {
+          id: '123',
+          processedEntity: {
+            apiVersion: '1.0.0',
+            metadata: {
+              name: 'new-root',
+            },
+            kind: 'Location',
+          } as Entity,
+          deferredEntities: [deferredEntity],
+          relations: [],
+        });
+      });
+
+      console.log(
+        await db<DbRefreshStateRow>('refresh_state')
+          .where({ entity_ref: 'deferred' })
+          .select(),
+      );
+      expect(1).toBeDefined();
+    });
+  });
 });