test_runner.py

#!/usr/bin/env python3
"""
Simple test runner script to demonstrate the endpoint tester
"""

import asyncio
import sys
from pathlib import Path

# Add current directory to path for imports
sys.path.append(str(Path(__file__).parent))

from main import EndpointTesterApp


async def run_demo():
    """Run a demonstration of the endpoint tester"""
    print("=== Endpoint Performance Tester Demo ===")
    print("This demo will test some public APIs to show the functionality")
    print()
    
    # Create app instance
    app = EndpointTesterApp()
    
    # Create a demo test configuration
    demo_config = {
        "name": "Demo API Test",
        "base_url": "https://httpbin.org",
        "concurrent_requests": 2,
        "test_duration_seconds": 10,
        "warmup_requests": 1,
        "description": "Demo test using httpbin.org",
        "endpoints": [
            {
                "name": "Get Request",
                "url": "/get",
                "method": "GET",
                "expected_status_code": 200,
                "description": "Test GET request",
                "timeout": 5
            },
            {
                "name": "Post Request",
                "url": "/post",
                "method": "POST",
                "headers": {
                    "Content-Type": "application/json"
                },
                "json_data": {
                    "test": "data",
                    "timestamp": "2024-01-01"
                },
                "expected_status_code": 200,
                "description": "Test POST request",
                "timeout": 5
            },
            {
                "name": "Delay Request",
                "url": "/delay/1",
                "method": "GET",
                "expected_status_code": 200,
                "description": "Test request with 1 second delay",
                "timeout": 5
            }
        ]
    }
    
    # Create test suite from config
    test_suite = app.test_manager.create_test_suite_from_dict(demo_config)
    app.test_manager.add_test_suite(test_suite)
    
    print(f"Running demo test: {test_suite.name}")
    print(f"Duration: {test_suite.test_duration_seconds} seconds")
    print(f"Concurrent requests: {test_suite.concurrent_requests}")
    print(f"Endpoints: {len(test_suite.endpoints)}")
    print()
    
    # Run the test
    result = await app.test_manager.run_test_suite(test_suite.name)
    
    if result:
        print("\n" + "="*60)
        print("DEMO COMPLETED SUCCESSFULLY!")
        print("="*60)
        
        # Display results
        app.result_display.display_test_suite_result(result, detailed=True)
        
        # Export results
        app._export_results(result)
        
        print("\nDemo completed! Check the generated files for detailed results.")
    else:
        print("Demo failed!")


if __name__ == "__main__":
    asyncio.run(run_demo())