#!/usr/bin/env python3
"""
Simple validation script to check if the new test modules can be imported correctly.
"""
def validate_imports():
"""Validate that all test modules can be imported."""
print("🔍 Validating test imports...")
try:
# Test retry module imports
from mcp_vultr.retry import (
NetworkError,
RateLimitError,
RetryableError,
create_retry_decorator,
retry_api_call,
retry_async,
retry_network,
retry_rate_limit,
)
print("✅ Retry module imports successful")
# Test cache module imports
from mcp_vultr.cache import (
CacheManager,
cached_request,
clear_cache,
get_cache_manager,
get_cache_stats,
)
print("✅ Cache module imports successful")
# Test basic functionality
cache_manager = CacheManager()
print(f"✅ CacheManager instantiated with {len(cache_manager.get_stats())} stats")
retry_decorator = create_retry_decorator(max_attempts=2)
print("✅ Retry decorator created successfully")
return True
except ImportError as e:
print(f"❌ Import error: {e}")
return False
except Exception as e:
print(f"❌ Unexpected error: {e}")
return False
def count_test_functions():
"""Count the number of test functions in our new test files."""
import ast
from pathlib import Path
test_counts = {}
for test_file in ['tests/test_retry.py', 'tests/test_cache.py']:
path = Path(test_file)
if path.exists():
with open(path, 'r') as f:
tree = ast.parse(f.read())
test_count = 0
for node in ast.walk(tree):
if isinstance(node, ast.FunctionDef) and node.name.startswith('test_'):
test_count += 1
test_counts[test_file] = test_count
print(f"📊 {test_file}: {test_count} test functions")
total_tests = sum(test_counts.values())
print(f"📊 Total new test functions: {total_tests}")
return test_counts
if __name__ == "__main__":
print("🧪 Validating new test modules for retry.py and cache.py")
print("=" * 60)
# Validate imports
imports_ok = validate_imports()
print("\n" + "=" * 60)
# Count tests
test_counts = count_test_functions()
print("\n" + "=" * 60)
if imports_ok:
print("✅ All validations passed! Tests are ready to run.")
print("\n🚀 Next steps:")
print(" 1. Run: uv run pytest tests/test_retry.py tests/test_cache.py -v")
print(" 2. Run with coverage: uv run pytest tests/test_retry.py tests/test_cache.py --cov=mcp_vultr.retry --cov=mcp_vultr.cache --cov-report=term-missing")
else:
print("❌ Validation failed. Please check the import errors above.")
print("\n📈 Expected coverage improvements:")
print(" • retry.py: from 61% to 90%+ (comprehensive function and edge case testing)")
print(" • cache.py: from 53% to 90%+ (TTL, eviction, decorator, and integration testing)")
