Ryan Malloy
8b43927493
- Update all dependencies to latest versions (fastmcp, httpx, packaging, etc.) - Downgrade click from yanked 8.2.2 to stable 8.1.7 - Fix code formatting and linting issues with ruff - Most tests passing (2 test failures in dependency resolver need investigation)
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""Direct test of fallback mechanisms."""
|
|
|
|
import asyncio
|
|
import os
|
|
import sys
|
|
|
|
sys.path.insert(0, os.path.abspath("."))
|
|
|
|
from pypi_query_mcp.core.stats_client import PyPIStatsClient
|
|
|
|
|
|
async def test_fallback():
|
|
"""Test fallback data generation directly."""
|
|
print("Testing fallback data generation...")
|
|
|
|
async with PyPIStatsClient() as client:
|
|
# Force API failure tracking to trigger fallback
|
|
client._api_health["consecutive_failures"] = 5 # Force fallback mode
|
|
|
|
# Test recent downloads fallback
|
|
fallback_recent = client._generate_fallback_recent_downloads(
|
|
"requests", "month"
|
|
)
|
|
print("✅ Fallback recent downloads generated for requests:")
|
|
print(f" Source: {fallback_recent.get('source')}")
|
|
print(f" Downloads: {fallback_recent['data']['last_month']:,}")
|
|
print(f" Note: {fallback_recent.get('note')}")
|
|
|
|
# Test overall downloads fallback
|
|
fallback_overall = client._generate_fallback_overall_downloads("numpy", False)
|
|
print("\n✅ Fallback time series generated for numpy:")
|
|
print(f" Source: {fallback_overall.get('source')}")
|
|
print(f" Data points: {len(fallback_overall['data'])}")
|
|
print(f" Note: {fallback_overall.get('note')}")
|
|
|
|
# Test the should_use_fallback logic
|
|
should_fallback = client._should_use_fallback()
|
|
print(f"\n✅ Fallback logic working: {should_fallback}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(test_fallback())
|