Skip to content

Commit f033e81

Browse files
update test_app.py file for init_openai method
1 parent 89b7912 commit f033e81

File tree

1 file changed

+15
-14
lines changed

1 file changed

+15
-14
lines changed

src/App/tests/test_app.py

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
delete_all_conversations,
99
generate_title,
1010
init_cosmosdb_client,
11-
init_openai_client,
11+
init_ai_projects_client,
1212
stream_chat_request,
1313
)
1414
from quart import Response
@@ -78,15 +78,16 @@ def test_create_app():
7878
assert "routes" in app.blueprints
7979

8080

81-
@patch("app.get_bearer_token_provider")
82-
@patch("app.AsyncAzureOpenAI")
83-
def test_init_openai_client(mock_async_openai, mock_token_provider):
84-
mock_token_provider.return_value = MagicMock()
85-
mock_async_openai.return_value = MagicMock()
81+
@patch("app.AIProjectClient")
82+
def test_init_ai_projects_client(mock_ai_projects_client):
83+
mock_project_instance = MagicMock()
84+
mock_openai_client = MagicMock()
85+
mock_project_instance.inference.get_azure_openai_client.return_value = mock_openai_client
86+
mock_ai_projects_client.return_value = mock_project_instance
8687

87-
client = init_openai_client()
88+
client = init_ai_projects_client()
8889
assert client is not None
89-
mock_async_openai.assert_called_once()
90+
mock_ai_projects_client.assert_called_once()
9091

9192

9293
@patch("app.CosmosConversationClient")
@@ -1179,27 +1180,27 @@ async def test_add_conversation_conversation_not_found(
11791180

11801181

11811182
@pytest.mark.asyncio
1182-
@patch("app.init_openai_client")
1183-
async def test_generate_title_success(mock_init_openai_client):
1183+
@patch("app.init_ai_projects_client")
1184+
async def test_generate_title_success(mock_init_ai_projects_client):
11841185
mock_openai_client = AsyncMock()
11851186
mock_openai_client.chat.completions.create.return_value = MagicMock(
11861187
choices=[
11871188
MagicMock(message=MagicMock(content=json.dumps({"title": "Test Title"})))
11881189
]
11891190
)
1190-
mock_init_openai_client.return_value = mock_openai_client
1191+
mock_init_ai_projects_client.return_value = mock_openai_client
11911192

11921193
conversation_messages = [{"role": "user", "content": "Hello"}]
11931194
title = await generate_title(conversation_messages)
11941195
assert title == "Test Title"
11951196

11961197

11971198
@pytest.mark.asyncio
1198-
@patch("app.init_openai_client")
1199-
async def test_generate_title_exception(mock_init_openai_client):
1199+
@patch("app.init_ai_projects_client")
1200+
async def test_generate_title_exception(mock_init_ai_projects_client):
12001201
mock_openai_client = AsyncMock()
12011202
mock_openai_client.chat.completions.create.side_effect = Exception("API error")
1202-
mock_init_openai_client.return_value = mock_openai_client
1203+
mock_init_ai_projects_client.return_value = mock_openai_client
12031204

12041205
conversation_messages = [{"role": "user", "content": "Hello"}]
12051206
title = await generate_title(conversation_messages)

0 commit comments

Comments
 (0)