From 38fc65c5419082dd106c8fab764f9f6a5de61842 Mon Sep 17 00:00:00 2001
From: Martin McKeaveney <martinmckeaveney@gmail.com>
Date: Tue, 10 Sep 2024 22:25:47 +0100
Subject: [PATCH] fix tests

---
 .../src/api/controllers/global/tests/configs.spec.ts     | 9 +++++----
 packages/worker/src/tests/structures/configs.ts          | 2 +-
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/packages/worker/src/api/controllers/global/tests/configs.spec.ts b/packages/worker/src/api/controllers/global/tests/configs.spec.ts
index 06d1ad0b2f..ad9efc4398 100644
--- a/packages/worker/src/api/controllers/global/tests/configs.spec.ts
+++ b/packages/worker/src/api/controllers/global/tests/configs.spec.ts
@@ -49,6 +49,7 @@ describe("Global configs controller", () => {
         active: true,
         isDefault: true,
         name: "Budibase AI",
+        defaultModel: "",
       },
       ai: {
         active: true,
@@ -86,20 +87,20 @@ describe("Global configs controller", () => {
   it("Should not update existing secrets when updating an existing AI Config", async () => {
     const data = structures.configs.ai()
     await config.api.configs.saveConfig(data)
-    const existingConfig = await config.api.configs.getAIConfig()
 
     const newConfig: AIInnerConfig = {
-      aiconfig: {
+      ai: {
         provider: "OpenAI",
         isDefault: true,
+        apiKey: "--secret-value--",
         name: "MyConfig",
         active: true,
         defaultModel: "gpt4",
       },
     }
 
-    await verifyAIConfig(newConfig, existingConfig.body)
+    await verifyAIConfig(newConfig, data)
     // should be unchanged
-    expect(newConfig.aiconfig.apiKey).toEqual("myapikey")
+    expect(newConfig.ai.apiKey).toEqual("myapikey")
   })
 })
diff --git a/packages/worker/src/tests/structures/configs.ts b/packages/worker/src/tests/structures/configs.ts
index 9257e813cb..5aa073408c 100644
--- a/packages/worker/src/tests/structures/configs.ts
+++ b/packages/worker/src/tests/structures/configs.ts
@@ -93,7 +93,7 @@ export function ai(): AIConfig {
         name: "Test",
         active: true,
         defaultModel: "gpt4",
-        apiKey: "abc123APIKey",
+        apiKey: "myapikey",
         baseUrl: "https://api.example.com",
       },
     },