diff --git a/tests/unit/encoders/test_openai.py b/tests/unit/encoders/test_openai.py index 4679ee939f7d4b494150a8a52f4b4c33a0e6c8db..508e9e9e197a8a0b87fb179554d43c18686c1b44 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