| from __future__ import annotations |
|
|
| from hughub.config import RepoConfig |
| from hughub.process import Result |
| from hughub.workflows import doctor, run_workflow |
|
|
| from .conftest import FakeRunner |
|
|
|
|
| def write_workflow(tmp_path, content): |
| directory = tmp_path / ".github" / "workflows" |
| directory.mkdir(parents=True) |
| path = directory / "test.yml" |
| path.write_text(content) |
| return path |
|
|
|
|
| def test_doctor_accepts_shell_workflow(tmp_path, capsys): |
| write_workflow( |
| tmp_path, |
| """ |
| name: test |
| on: [push, workflow_dispatch] |
| jobs: |
| tests: |
| runs-on: ubuntu-latest |
| steps: |
| - uses: actions/checkout@v4 |
| - run: pytest |
| """, |
| ) |
|
|
| result = doctor("test.yml", tmp_path) |
|
|
| assert result == 0 |
| assert "compatible" in capsys.readouterr().out |
|
|
|
|
| def test_doctor_rejects_unsupported_action(tmp_path, capsys): |
| write_workflow( |
| tmp_path, |
| """ |
| on: push |
| jobs: |
| tests: |
| runs-on: ubuntu-latest |
| steps: |
| - uses: vendor/unknown@v1 |
| """, |
| ) |
|
|
| assert doctor(None, tmp_path) == 1 |
| assert "not supported" in capsys.readouterr().out |
|
|
|
|
| def test_workflow_launches_hf_job_with_repo_labels(tmp_path, capsys): |
| write_workflow( |
| tmp_path, |
| """ |
| on: workflow_dispatch |
| jobs: |
| tests: |
| runs-on: ubuntu-latest |
| steps: |
| - uses: actions/checkout@v4 |
| - run: python -m pytest |
| """, |
| ) |
| git_dir = tmp_path / ".git" |
| git_dir.mkdir() |
| runner = FakeRunner(git_dir) |
| runner.scripted[("git", "branch", "--show-current")] = Result(("git",), 0, "feature\n", "") |
| config = RepoConfig("acme/widget", "hf-acme/widget", mode="overlay") |
|
|
| result = run_workflow(config, runner, name="test.yml", ref=None, cwd=tmp_path) |
|
|
| assert result == 0 |
| jobs_calls = [call for call in runner.calls if call[:3] == ("hf", "jobs", "run")] |
| assert len(jobs_calls) == 1 |
| assert "hughub-repo=hf-acme--widget" in jobs_calls[0] |
| assert "Launched 1" in capsys.readouterr().out |
|
|
