fixing cloudflare test

outerbase · Oct 22, 2024 · 7e79982 · 7e79982
1 parent ced0ee1
commit 7e79982
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/connections/sqlite/cloudflare.ts b/src/connections/sqlite/cloudflare.ts
@@ -227,12 +227,15 @@ export class CloudflareD1Connection extends SqliteBaseConnection {
                     schema: table.schema,
                     tableName: table.name,
                     type: 'FOREIGN KEY',
+                    referenceSchema: table.schema,
+                    referenceTableName: fkConstraintData[0].table,
                     columns: [],
                 };
 
                 fkConstraintData.forEach((fkConstraint) => {
                     const currentConstraint: ConstraintColumn = {
                         columnName: fkConstraint.from,
+                        referenceColumnName: fkConstraint.to,
                     };
                     fkConstraints.columns.push(currentConstraint);
                 });

diff --git a/tests/connections/connection.test.ts b/tests/connections/connection.test.ts
@@ -166,7 +166,11 @@ describe('Database Connection', () => {
         expect(actualSchema).toEqual(expectedSchema);
 
         // Check teams and persons table reference
-        if (!['bigquery', 'mongodb'].includes(process.env.CONNECTION_TYPE!)) {
+        if (
+            !['bigquery', 'mongodb', 'motherduck'].includes(
+                process.env.CONNECTION_TYPE!
+            )
+        ) {
             expect(
                 schemas[DEFAULT_SCHEMA].persons!.columns!.find(
                     (c) => c.name === 'team_id'