chore: fixed imports and tests configuration
This commit is contained in:
@@ -10,7 +10,7 @@ class TestHealthEndpoint:
|
||||
|
||||
def test_health_check(self, memory):
|
||||
"""Should return healthy status."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -25,7 +25,7 @@ class TestModelsEndpoint:
|
||||
|
||||
def test_list_models(self, memory):
|
||||
"""Should return model list."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -43,7 +43,7 @@ class TestMemoryEndpoints:
|
||||
|
||||
def test_get_memory_state(self, memory):
|
||||
"""Should return full memory state."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -57,7 +57,7 @@ class TestMemoryEndpoints:
|
||||
|
||||
def test_get_search_results_empty(self, memory):
|
||||
"""Should return empty when no search results."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -69,7 +69,7 @@ class TestMemoryEndpoints:
|
||||
|
||||
def test_get_search_results_with_data(self, memory_with_search_results):
|
||||
"""Should return search results when available."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -83,7 +83,7 @@ class TestMemoryEndpoints:
|
||||
|
||||
def test_clear_session(self, memory_with_search_results):
|
||||
"""Should clear session memories."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -102,10 +102,10 @@ class TestChatCompletionsEndpoint:
|
||||
|
||||
def test_chat_completion_success(self, memory):
|
||||
"""Should return chat completion."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
# Patch the agent's step method directly
|
||||
with patch("app.agent.step", return_value="Hello! How can I help?"):
|
||||
with patch("alfred.app.agent.step", return_value="Hello! How can I help?"):
|
||||
client = TestClient(app)
|
||||
|
||||
response = client.post(
|
||||
@@ -123,7 +123,7 @@ class TestChatCompletionsEndpoint:
|
||||
|
||||
def test_chat_completion_no_user_message(self, memory):
|
||||
"""Should return error if no user message."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -146,7 +146,7 @@ class TestChatCompletionsEndpoint:
|
||||
|
||||
def test_chat_completion_empty_messages(self, memory):
|
||||
"""Should return error for empty messages."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -162,7 +162,7 @@ class TestChatCompletionsEndpoint:
|
||||
|
||||
def test_chat_completion_invalid_json(self, memory):
|
||||
"""Should return error for invalid JSON."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
@@ -176,9 +176,9 @@ class TestChatCompletionsEndpoint:
|
||||
|
||||
def test_chat_completion_streaming(self, memory):
|
||||
"""Should support streaming mode."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
with patch("app.agent.step", return_value="Streaming response"):
|
||||
with patch("alfred.app.agent.step", return_value="Streaming response"):
|
||||
client = TestClient(app)
|
||||
|
||||
response = client.post(
|
||||
@@ -195,9 +195,9 @@ class TestChatCompletionsEndpoint:
|
||||
|
||||
def test_chat_completion_extracts_last_user_message(self, memory):
|
||||
"""Should use last user message."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
with patch("app.agent.step", return_value="Response") as mock_step:
|
||||
with patch("alfred.app.agent.step", return_value="Response") as mock_step:
|
||||
client = TestClient(app)
|
||||
|
||||
response = client.post(
|
||||
@@ -218,9 +218,9 @@ class TestChatCompletionsEndpoint:
|
||||
|
||||
def test_chat_completion_response_format(self, memory):
|
||||
"""Should return OpenAI-compatible format."""
|
||||
from app import app
|
||||
from alfred.app import app
|
||||
|
||||
with patch("app.agent.step", return_value="Test response"):
|
||||
with patch("alfred.app.agent.step", return_value="Test response"):
|
||||
client = TestClient(app)
|
||||
|
||||
response = client.post(
|
||||
|
||||
Reference in New Issue
Block a user