Skip to content

Fix tests #21

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
brylie wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Fix tests #21

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 8 additions & 4 deletions tests/test_chat_gpt_client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,9 @@ def load_env_variables(monkeypatch):
@pytest.mark.asyncio
@patch("app.chat_gpt_client.client")
async def test_get_chat_response_with_history_success(
mock_client, chat_history, load_env_variables
mock_client, chat_history, load_env_variables, monkeypatch
):
monkeypatch.setenv("OPENAI_API_KEY", "test_api_key")
mock_client.chat.completions.create = AsyncMock(return_value=MockResponse())
response = await get_chat_response_with_history(chat_history)
assert response == "Mocked response content"
Expand All @@ -31,8 +32,9 @@ async def test_get_chat_response_with_history_success(
@pytest.mark.asyncio
@patch("app.chat_gpt_client.client")
async def test_get_chat_response_with_history_api_error(
mock_client, chat_history, load_env_variables
mock_client, chat_history, load_env_variables, monkeypatch
):
monkeypatch.setenv("OPENAI_API_KEY", "test_api_key")
mock_client.chat.completions.create = AsyncMock(side_effect=Exception("API error"))
response = await get_chat_response_with_history(chat_history)
assert "I'm sorry, but I encountered an error: API error" in response
Expand All @@ -41,8 +43,9 @@ async def test_get_chat_response_with_history_api_error(
@pytest.mark.asyncio
@patch("app.chat_gpt_client.client")
async def test_get_chat_response_with_history_empty_message(
mock_client, load_env_variables
mock_client, load_env_variables, monkeypatch
):
monkeypatch.setenv("OPENAI_API_KEY", "test_api_key")
# Using AsyncMock to simulate async API call
mock_client.chat.completions.create = AsyncMock(return_value=MockResponse())
response = await get_chat_response_with_history([])
Expand All @@ -62,8 +65,9 @@ async def test_get_chat_response_with_history_empty_message(
@pytest.mark.asyncio
@patch("app.chat_gpt_client.client")
async def test_get_chat_response_with_history_parameterized(
mock_client, message_content, expected, load_env_variables
mock_client, message_content, expected, load_env_variables, monkeypatch
):
monkeypatch.setenv("OPENAI_API_KEY", "test_api_key")
mock_client.chat.completions.create = AsyncMock(return_value=MockResponse())
messages = [
Message(role=MessageRole.user, content=content) for content in message_content
Expand Down
3 changes: 3 additions & 0 deletions tests/test_main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,9 @@ def mock_services(monkeypatch):
"app.main.get_chat_response_with_history", mock_get_chat_response_with_history
)

# Set the OPENAI_API_KEY environment variable
monkeypatch.setenv("OPENAI_API_KEY", "test_api_key")


def test_sources_toggle_independent(mock_services):
messages = [
Expand Down