Add a unit test

ScottLogic · Aug 4, 2023 · 3fd63c9 · 3fd63c9
1 parent 5e60a7e
commit 3fd63c9
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/backend/test/unit/openai.test.js b/backend/test/unit/openai.test.js
@@ -0,0 +1,9 @@
+const { OpenAIApi } = require("openai");
+const { setGptModel } = require("../../src/openai");
+
+test("GIVEN a new model WHEN setting the GPT model THEN the model is set", () => {
+ const session = { gptModel: "gpt-4" };
+ const model = "gpt-3.5-turbo";
+ setGptModel(session, model);
+ expect(session.gptModel).toBe(model);
+});