run_tests_clean.py

#!/usr/bin/env python3
"""
Test runner script that ensures all tests use temporary directories and clean up properly.

This script runs the test suite and ensures that no temporary files are left behind.
"""

import os
import sys
import tempfile
import shutil
import subprocess
from pathlib import Path
import argparse


def cleanup_temp_files():
    """Clean up any temporary files that might have been left behind."""
    temp_dir = Path(tempfile.gettempdir())
    
    # Look for test-related temporary files
    patterns = [
        "graphflow_test_*",
        "tmp*",
        "test_*",
        "pytest_*"
    ]
    
    cleaned_count = 0
    for pattern in patterns:
        for temp_file in temp_dir.glob(pattern):
            try:
                if temp_file.is_dir():
                    shutil.rmtree(temp_file, ignore_errors=True)
                    cleaned_count += 1
                else:
                    temp_file.unlink(missing_ok=True)
                    cleaned_count += 1
            except (OSError, PermissionError):
                # Ignore errors - some files might be in use
                pass
    
    if cleaned_count > 0:
        print(f"Cleaned up {cleaned_count} temporary files/directories")


def run_tests(test_args=None):
    """Run the test suite with proper cleanup."""
    if test_args is None:
        test_args = []
    
    # Clean up before running tests
    print("Cleaning up any existing temporary files...")
    cleanup_temp_files()
    
    # Run the tests
    cmd = [sys.executable, "-m", "pytest"] + test_args
    print(f"Running tests with command: {' '.join(cmd)}")
    
    try:
        result = subprocess.run(cmd, cwd=Path(__file__).parent)
        return result.returncode
    finally:
        # Clean up after running tests
        print("Cleaning up temporary files after tests...")
        cleanup_temp_files()


def main():
    """Main entry point."""
    parser = argparse.ArgumentParser(description="Run GraphFlow tests with proper cleanup")
    parser.add_argument("test_args", nargs="*", help="Additional arguments to pass to pytest")
    parser.add_argument("--clean-only", action="store_true", help="Only clean up temporary files")
    parser.add_argument("--verbose", "-v", action="store_true", help="Verbose output")
    
    args = parser.parse_args()
    
    if args.clean_only:
        cleanup_temp_files()
        return 0
    
    # Add verbose flag if requested
    test_args = args.test_args
    if args.verbose:
        test_args = ["-v"] + test_args
    
    # Run tests
    return run_tests(test_args)


if __name__ == "__main__":
    sys.exit(main())