Transcriptarr/test_backend.py

#!/usr/bin/env python3
"""Test script for TranscriptorIO backend components."""
import sys
import logging

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


def test_config():
    """Test configuration loading."""
    logger.info("Testing configuration...")
    try:
        from backend.config import settings
        logger.info(f"✓ Config loaded successfully")
        logger.info(f"  - Mode: {settings.transcriptarr_mode}")
        logger.info(f"  - Database: {settings.database_type.value}")
        logger.info(f"  - Whisper Model: {settings.whisper_model}")
        logger.info(f"  - Device: {settings.transcribe_device}")
        return True
    except Exception as e:
        logger.error(f"✗ Config test failed: {e}")
        return False


def test_database():
    """Test database connection and table creation."""
    logger.info("\nTesting database...")
    try:
        from backend.core.database import database
        from backend.core.models import Base

        # Clean database for fresh test
        try:
            database.drop_tables()
            logger.info(f"  - Dropped existing tables for clean test")
        except:
            pass

        database.create_tables()
        logger.info(f"✓ Database initialized with fresh tables")

        # Test connection with health check
        if database.health_check():
            logger.info(f"✓ Database connection OK")
        else:
            logger.error("✗ Database health check failed (but tables were created)")
            # Don't fail the test if health check fails but tables exist
            return True

        # Get stats
        stats = database.get_stats()
        logger.info(f"  - Type: {stats['type']}")
        logger.info(f"  - URL: {stats['url']}")

        return True
    except Exception as e:
        logger.error(f"✗ Database test failed: {e}")
        import traceback
        traceback.print_exc()
        return False


def test_queue_manager():
    """Test queue manager operations."""
    logger.info("\nTesting queue manager...")
    try:
        from backend.core.queue_manager import queue_manager
        from backend.core.models import QualityPreset

        # Add a test job
        job = queue_manager.add_job(
            file_path="/test/anime.mkv",
            file_name="anime.mkv",
            source_lang="ja",
            target_lang="es",
            quality_preset=QualityPreset.FAST,
            priority=5
        )

        if job:
            logger.info(f"✓ Job created: {job.id}")
            logger.info(f"  - File: {job.file_name}")
            logger.info(f"  - Status: {job.status.value}")
            logger.info(f"  - Priority: {job.priority}")
        else:
            logger.error("✗ Failed to create job")
            return False

        # Get queue stats
        stats = queue_manager.get_queue_stats()
        logger.info(f"✓ Queue stats:")
        logger.info(f"  - Total: {stats['total']}")
        logger.info(f"  - Queued: {stats['queued']}")
        logger.info(f"  - Processing: {stats['processing']}")
        logger.info(f"  - Completed: {stats['completed']}")

        # Try to add duplicate
        duplicate = queue_manager.add_job(
            file_path="/test/anime.mkv",
            file_name="anime.mkv",
            source_lang="ja",
            target_lang="es",
            quality_preset=QualityPreset.FAST
        )

        if duplicate is None:
            logger.info(f"✓ Duplicate detection working")
        else:
            logger.warning(f"⚠ Duplicate job was created (should have been rejected)")

        # Get next job
        next_job = queue_manager.get_next_job("test-worker-1")
        if next_job:
            logger.info(f"✓ Got next job: {next_job.id} (assigned to test-worker-1)")
            logger.info(f"  - Status: {next_job.status.value}")
        else:
            logger.error("✗ Failed to get next job")
            return False

        return True
    except Exception as e:
        logger.error(f"✗ Queue manager test failed: {e}")
        import traceback
        traceback.print_exc()
        return False


def main():
    """Run all tests."""
    logger.info("=" * 60)
    logger.info("TranscriptorIO Backend Test Suite")
    logger.info("=" * 60)

    results = {
        "Config": test_config(),
        "Database": test_database(),
        "Queue Manager": test_queue_manager(),
    }

    logger.info("\n" + "=" * 60)
    logger.info("Test Results:")
    logger.info("=" * 60)

    all_passed = True
    for test_name, passed in results.items():
        status = "✓ PASSED" if passed else "✗ FAILED"
        logger.info(f"{test_name}: {status}")
        if not passed:
            all_passed = False

    logger.info("=" * 60)

    if all_passed:
        logger.info("🎉 All tests passed!")
        return 0
    else:
        logger.error("❌ Some tests failed")
        return 1


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