From d1e1ad4930ae91ab595485d4f4a3696aee6567aa Mon Sep 17 00:00:00 2001
From: Adria Navarro <adria@budibase.com>
Date: Wed, 13 Sep 2023 16:33:04 +0200
Subject: [PATCH] Revert TestConfiguration changes

---
 .../src/tests/utilities/TestConfiguration.ts   | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/packages/server/src/tests/utilities/TestConfiguration.ts b/packages/server/src/tests/utilities/TestConfiguration.ts
index 94896932ad..da7af8acd7 100644
--- a/packages/server/src/tests/utilities/TestConfiguration.ts
+++ b/packages/server/src/tests/utilities/TestConfiguration.ts
@@ -535,7 +535,7 @@ class TestConfiguration {
     { skipReassigning } = { skipReassigning: false }
   ): Promise<Table> {
     config = config || basicTable()
-    const response = await this.api.table.create(config)
+    const response = await this._req(config, null, controllers.table.save)
     if (!skipReassigning) {
       this.table = response
     }
@@ -589,7 +589,7 @@ class TestConfiguration {
       }
     }
 
-    const linkedTable = await this.api.table.create(tableConfig)
+    const linkedTable = await this.createTable(tableConfig)
     return linkedTable
   }
 
@@ -609,7 +609,7 @@ class TestConfiguration {
     }
     const tableId = (config && config.tableId) || this.table._id
     config = config || basicRow(tableId!)
-    return this.api.row.save(tableId!, config)
+    return this._req(config, { tableId }, controllers.row.save)
   }
 
   async getRow(tableId: string, rowId: string): Promise<Row> {
@@ -715,14 +715,18 @@ class TestConfiguration {
     datasource: Datasource
   }): Promise<Datasource> {
     config = config || basicDatasource()
-    const response = await this.api.datasource.create({ ...config.datasource })
-    this.datasource = response
+    const response = await this._req(config, null, controllers.datasource.save)
+    this.datasource = response.datasource
     return this.datasource!
   }
 
   async updateDatasource(datasource: Datasource): Promise<Datasource> {
-    const response = await this.api.datasource.update(datasource)
-    this.datasource = response
+    const response = await this._req(
+      datasource,
+      { datasourceId: datasource._id },
+      controllers.datasource.update
+    )
+    this.datasource = response.datasource
     return this.datasource!
   }