Bill/AI-Trader
1
0
Fork 0
mirror of https://github.com/Xe138/AI-Trader.git synced 2026-04-01 17:17:24 -04:00
Code Issues Packages Projects Releases Wiki Activity

test: update integration test for new results endpoint behavior

Browse Source
This commit is contained in:
Bill Ballou
2025-11-07 19:41:01 -05:00
parent 58937774bf
commit dd99912ec7
1 changed files with 15 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
tests/integration/test_api_endpoints.py
View File

@@ -232,14 +232,13 @@ class TestSimulateStatusEndpoint:
class TestResultsEndpoint: class TestResultsEndpoint:
"""Test GET /results endpoint.""" """Test GET /results endpoint."""
def test_results_returns_all_results(self, api_client): def test_results_returns_404_when_no_data(self, api_client):
"""Should return all results without filters.""" """Should return 404 when no data exists for default date range."""
response = api_client.get("/results") response = api_client.get("/results")
assert response.status_code == 200 # With new endpoint, no data returns 404
data = response.json() assert response.status_code == 404
assert "results" in data assert "No trading data found" in response.json()["detail"]
assert isinstance(data["results"], list)
def test_results_filters_by_job_id(self, api_client): def test_results_filters_by_job_id(self, api_client):
"""Should filter results by job_id.""" """Should filter results by job_id."""
@@ -261,38 +260,32 @@ class TestResultsEndpoint:
def test_results_filters_by_date(self, api_client): def test_results_filters_by_date(self, api_client):
"""Should filter results by date.""" """Should filter results by date."""
response = api_client.get("/results?date=2025-01-16") response = api_client.get("/results?start_date=2025-01-16&end_date=2025-01-16")
assert response.status_code == 200 # No data exists, should return 404
data = response.json() assert response.status_code == 404
assert isinstance(data["results"], list)
def test_results_filters_by_model(self, api_client): def test_results_filters_by_model(self, api_client):
"""Should filter results by model.""" """Should filter results by model."""
response = api_client.get("/results?model=gpt-4") response = api_client.get("/results?model=gpt-4")
assert response.status_code == 200 # No data exists, should return 404
data = response.json() assert response.status_code == 404
assert isinstance(data["results"], list)
def test_results_combines_multiple_filters(self, api_client): def test_results_combines_multiple_filters(self, api_client):
"""Should support multiple filter parameters.""" """Should support multiple filter parameters."""
response = api_client.get("/results?date=2025-01-16&model=gpt-4") response = api_client.get("/results?start_date=2025-01-16&end_date=2025-01-16&model=gpt-4")
assert response.status_code == 200 # No data exists, should return 404
data = response.json() assert response.status_code == 404
assert isinstance(data["results"], list)
def test_results_includes_position_data(self, api_client): def test_results_includes_position_data(self, api_client):
"""Should include position and holdings data.""" """Should include position and holdings data."""
# This test will pass once we have actual data # This test will pass once we have actual data
response = api_client.get("/results") response = api_client.get("/results")
assert response.status_code == 200 # No data exists, should return 404
data = response.json() assert response.status_code == 404
# Each result should have expected structure
for result in data["results"]:
assert "job_id" in result or True # Pass if empty
@pytest.mark.integration @pytest.mark.integration