mirror of
https://github.com/Azure-Samples/graphrag-accelerator.git
synced 2025-10-31 02:29:05 +00:00
34 lines
839 B
Python
34 lines
839 B
Python
|
# Copyright (c) Microsoft Corporation.
|
||
|
|
# Licensed under the MIT License.
|
||
|
|
"""
|
||
|
|
Integration tests for the /index/config API endpoints.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from unittest.mock import AsyncMock, patch
|
||
|
|
|
||
|
|
import pytest_asyncio
|
||
|
|
|
||
|
|
|
||
|
|
@pytest_asyncio.fixture
|
||
|
|
def mock_generate_indexing_prompts():
|
||
|
|
with patch(
|
||
|
|
"graphrag.api.generate_indexing_prompts", new_callable=AsyncMock
|
||
|
|
) as mock:
|
||
|
|
mock.return_value = (
|
||
|
|
"synthetic-prompt1",
|
||
|
|
"synthetic-prompt2",
|
||
|
|
"synthetic-prompt3",
|
||
|
|
)
|
||
|
|
yield mock
|
||
|
|
|
||
|
|
|
||
|
|
def test_generate_prompts(
|
||
|
|
blob_with_data_container_name, mock_generate_indexing_prompts, client
|
||
|
|
):
|
||
|
|
"""Test generating prompts."""
|
||
|
|
response = client.get(
|
||
|
|
"/index/config/prompts",
|
||
|
|
params={"storage_name": blob_with_data_container_name},
|
||
|
|
)
|
||
|
|
assert response.status_code == 200
|