run_tests.py

#!/usr/bin/env python
"""
Test runner script for Pokemon middleware.
Provides convenient ways to run different test suites.
"""
import os
import sys
import subprocess
import argparse


def run_command(command, description):
    """Run a command and handle errors."""
    print(f"\n{'='*60}")
    print(f"Running: {description}")
    print(f"Command: {command}")
    print(f"{'='*60}")
    
    result = subprocess.run(command, shell=True)
    if result.returncode != 0:
        print(f"❌ {description} failed with exit code {result.returncode}")
        return False
    else:
        print(f"✅ {description} completed successfully")
        return True


def main():
    """Main test runner function."""
    parser = argparse.ArgumentParser(description='Run Pokemon middleware tests')
    parser.add_argument('--all', action='store_true', help='Run all tests')
    parser.add_argument('--services', action='store_true', help='Run service tests')
    parser.add_argument('--factories', action='store_true', help='Run factory tests')
    parser.add_argument('--entities', action='store_true', help='Run entity tests')
    parser.add_argument('--repositories', action='store_true', help='Run repository tests')
    parser.add_argument('--coverage', action='store_true', help='Run with coverage report')
    parser.add_argument('--verbose', '-v', action='store_true', help='Verbose output')
    
    args = parser.parse_args()
    
    # Set Django settings
    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.local')
    
    # Base command
    base_cmd = "python manage.py test"
    if args.verbose:
        base_cmd += " --verbosity=2"
    
    success = True
    
    if args.coverage:
        # Install coverage if not available
        print("Installing coverage...")
        subprocess.run("pip install coverage", shell=True)
        base_cmd = f"coverage run --source='middleware' manage.py test"
    
    if args.all or (not any([args.services, args.factories, args.entities, args.repositories])):
        # Run all tests
        success &= run_command(f"{base_cmd} middleware.tests", "All tests")
    else:
        # Run specific test suites
        if args.services:
            success &= run_command(f"{base_cmd} middleware.tests.test_services", "Service tests")
        
        if args.factories:
            success &= run_command(f"{base_cmd} middleware.tests.test_factories", "Factory tests")
        
        if args.entities:
            success &= run_command(f"{base_cmd} middleware.tests.test_entities", "Entity tests")
        
        if args.repositories:
            success &= run_command(f"{base_cmd} middleware.tests.test_repositories", "Repository tests")
    
    if args.coverage:
        print("\n" + "="*60)
        print("Generating coverage report...")
        print("="*60)
        
        # Generate coverage report
        run_command("coverage report", "Coverage report")
        run_command("coverage html", "HTML coverage report")
        
        print("\n📊 Coverage report generated in htmlcov/index.html")
    
    if success:
        print("\n🎉 All tests completed successfully!")
        sys.exit(0)
    else:
        print("\n❌ Some tests failed!")
        sys.exit(1)


if __name__ == '__main__':
    main()