Skip to content
Snippets Groups Projects
Commit da607c98 authored by maxyousif15's avatar maxyousif15
Browse files

tests: changing the tets to account for org id

parent 15f8af84
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment