test_pipeline.py

#!/usr/bin/env python3
"""Test script for pipeline runner"""

import subprocess
import sys


def test_successful_pipeline():
    """Test pipeline with all successful jobs"""
    print("\n" + "=" * 60)
    print("TEST 1: Successful Pipeline")
    print("=" * 60)

    result = subprocess.run(
        ["python3", "pipeline_runner.py", "pipeline_config.yaml"],
        text=True
    )
    return result.returncode == 0


def test_failed_pipeline():
    """Test pipeline with job failure"""
    print("\n" + "=" * 60)
    print("TEST 2: Pipeline with Failure")
    print("=" * 60)

    result = subprocess.run(
        ["python3", "pipeline_runner.py", "pipeline_config_fail.yaml"],
        text=True
    )
    return result.returncode == 0


if __name__ == "__main__":
    success_1 = test_successful_pipeline()
    success_2 = test_failed_pipeline()

    if success_1 and success_2:
        sys.exit(0)
    sys.exit(1)