How to Mock a GitHub Actions Workflow in Python

import json
from dataclasses import dataclass, asdict
from typing import List, Dict, Any


@dataclass
class Step:
    name: str
    run: str


@dataclass
class Job:
    name: str
    steps: List[Step]
    runs_on: str = "ubuntu-latest"


@dataclass
class Workflow:
    name: str
    jobs: List[Job]

    def to_github_actions_yaml(self) -> Dict[str, Any]:
        """Generate a GitHub Actions workflow dictionary (simulated)."""
        workflow_dict = {
            "name": self.name,
            "on": ["push"],
            "jobs": {},
        }
        for job in self.jobs:
            job_dict = {
                "runs-on": job.runs_on,
                "steps": [
                    {"name": step.name, "run": step.run}
                    for step in job.steps
                ],
            }
            workflow_dict["jobs"][job.name] = job_dict
        return workflow_dict

    def execute(self) -> Dict[str, List[str]]:
        """Mock-execute each step, collecting printed outputs."""
        results: Dict[str, List[str]] = {}
        for job in self.jobs:
            outputs = []
            for step in job.steps:
                outputs.append(f"Running: {step.name}")
                outputs.append(f"Command: {step.run}")
                outputs.append(f"Result: SUCCESS")
            results[job.name] = outputs
        return results


if __name__ == "__main__":
    workflow = Workflow(
        name="CI Pipeline",
        jobs=[
            Job(
                name="test",
                steps=[
                    Step(name="Install deps", run="pip install -r requirements.txt"),
                    Step(name="Run tests", run="pytest"),
                ],
            ),
            Job(
                name="build",
                steps=[
                    Step(name="Compile", run="python setup.py build"),
                    Step(name="Package", run="python setup.py sdist"),
                ],
            ),
        ],
    )

    generated_yaml = workflow.to_github_actions_yaml()
    print("Generated workflow (JSON):")
    print(json.dumps(generated_yaml, indent=2))

    print("\nExecuting workflow simulation:")
    print(json.dumps(workflow.execute(), indent=2, default=str))