add schema checking testing

outerbase · Oct 13, 2024 · 31aa178 · 31aa178
1 parent 5af395d
commit 31aa178
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/tests/connections/connection.test.ts b/tests/connections/connection.test.ts
@@ -80,6 +80,45 @@ describe('Database Connection', () => {
             .query();
     });
 
+    test('Check the schema', async () => {
+        if (db.fetchDatabaseSchema) {
+            const schemas = await db.fetchDatabaseSchema();
+
+            // Column names are sorted for easier comparison
+            const expectedSchema = {
+                [DEFAULT_SCHEMA]: {
+                    persons: {
+                        columns: ['age', 'id', 'name'],
+                    },
+                },
+            };
+
+            // We only care about the columns for this test
+            const actualSchema = Object.entries(schemas).reduce(
+                (a, [schemaName, schemaTables]) => {
+                    a[schemaName] = Object.entries(schemaTables).reduce(
+                        (b, [tableName, table]) => {
+                            b[tableName] = {
+                                columns: table.columns
+                                    .map((column) => column.name)
+                                    .sort(),
+                            };
+                            return b;
+                        },
+                        {} as Record<string, { columns: string[] }>
+                    );
+
+                    return a;
+                },
+                {} as Record<string, Record<string, { columns: string[] }>>
+            );
+
+            expect(actualSchema).toEqual(expectedSchema);
+        }
+
+        expect(true).toBe(true);
+    });
+
     test('Insert data', async () => {
         // Insert data
         await qb