From da607c983c83b7c3e454405071c694a766056f77 Mon Sep 17 00:00:00 2001
From: maxyousif15 <max.yousif@15gifts.com>
Date: Mon, 19 Feb 2024 18:42:25 +0000
Subject: [PATCH] tests: changing the tets to account for org id
---
tests/unit/encoders/test_openai.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/tests/unit/encoders/test_openai.py b/tests/unit/encoders/test_openai.py
index 4679ee93..508e9e9e 100644
--- a/tests/unit/encoders/test_openai.py
+++ b/tests/unit/encoders/test_openai.py
@@ -14,7 +14,9 @@ def openai_encoder(mocker):
class TestOpenAIEncoder:
def test_openai_encoder_init_success(self, mocker):
- mocker.patch("os.getenv", return_value="fake-api-key")
+ # -- Mock the return value of os.getenv 3 times: model name, api key and org ID
+ side_effect = ["fake-model-name", "fake-api-key", "fake-org-id"]
+ mocker.patch("os.getenv", side_effect=side_effect)
encoder = OpenAIEncoder()
assert encoder.client is not None
--
GitLab