From e0c39cd7e308a20f5e91b4cb04283589c5bc2251 Mon Sep 17 00:00:00 2001
From: Johan Haals <johan.haals@gmail.com>
Date: Mon, 26 Apr 2021 14:12:57 +0200
Subject: [PATCH] Add moar tests
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Fredrik Adelöw <freben@users.noreply.github.com>
Co-authored-by: Ben Lambert <ben@blam.sh>
Co-authored-by: Patrik Oldsberg <poldsberg@gmail.com>
Signed-off-by: Johan Haals <johan.haals@gmail.com>
---
 .../DefaultProcessingDatabase.test.ts         | 20 ++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts b/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts
index 14990a2a79..f67b8603db 100644
--- a/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts
+++ b/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts
@@ -359,16 +359,26 @@ describe('Default Processing Database', () => {
         });
       });
 
+      const currentRefreshState = await db<DbRefreshStateRow>(
+        'refresh_state',
+      ).select();
+
       const currentRefRowState = await db<DbRefreshStateReferencesRow>(
         'refresh_state_references',
       ).select();
 
-      expect(currentRefRowState).toEqual({
-        source_key: 'config-2',
-        target_entity_ref: 'location:default/root',
-      });
+      expect(currentRefRowState).toEqual([
+        expect.objectContaining({
+          source_key: 'config-2',
+          target_entity_ref: 'location:default/root',
+        }),
+      ]);
 
-      // TODO: assert that root wasn't removed
+      expect(currentRefreshState).toEqual([
+        expect.objectContaining({
+          entity_ref: 'location:default/root',
+        }),
+      ]);
     });
 
     it('should remove old locations using the delta options', async () => {