Spaces:
Running
Running
| import hashlib | |
| from pathlib import Path | |
| import subprocess | |
| import sys | |
| from types import SimpleNamespace | |
| import pytest | |
| from deploy.hf_space import DeploymentError, REQUIRED_FILES, stage_revision, upload_revision | |
| def run_git(root, *args): | |
| subprocess.run(["git", "-C", str(root), *args], check=True, capture_output=True) | |
| def source(tmp_path): | |
| root = tmp_path / "repository" | |
| root.mkdir() | |
| run_git(root, "init") | |
| for name in (*REQUIRED_FILES, "src/app.py", "console/package.json", "scripts/another_runtime.py"): | |
| file = root / name | |
| file.parent.mkdir(parents=True, exist_ok=True) | |
| file.write_text("committed content\n") | |
| run_git(root, "add", ".") | |
| run_git(root, "-c", "user.name=Deployment Test", "-c", "user.email=deployment-test@example.invalid", "commit", "-qm", "complete source") | |
| return root | |
| def test_stage_uses_complete_commit_and_excludes_local_changes(source, tmp_path): | |
| (source / "scripts/start_with_ollama.py").write_text("uncommitted replacement") | |
| (source / ".env").write_text("local-only configuration") | |
| stage = tmp_path / "stage" | |
| manifest = stage_revision(source, "HEAD", stage) | |
| assert (stage / "scripts/start_with_ollama.py").read_text() == "committed content\n" | |
| assert "scripts/another_runtime.py" in manifest["sha256"] | |
| assert not (stage / ".env").exists() | |
| assert manifest["sha256"]["scripts/start_with_ollama.py"] == hashlib.sha256(b"committed content\n").hexdigest() | |
| def test_missing_committed_wrapper_fails_even_if_local_file_exists(source, tmp_path): | |
| run_git(source, "rm", "--cached", "scripts/start_with_ollama.py") | |
| run_git(source, "-c", "user.name=Deployment Test", "-c", "user.email=deployment-test@example.invalid", "commit", "-qm", "partial source") | |
| assert (source / "scripts/start_with_ollama.py").exists() | |
| with pytest.raises(DeploymentError, match="scripts/start_with_ollama.py"): | |
| stage_revision(source, "HEAD", tmp_path / "stage") | |
| def test_upload_sends_every_staged_file_and_detects_remote_omission(source, tmp_path, monkeypatch): | |
| stage = tmp_path / "stage" | |
| manifest = stage_revision(source, "HEAD", stage) | |
| monkeypatch.setitem(sys.modules, "huggingface_hub", SimpleNamespace(CommitOperationAdd=lambda **kwargs: kwargs)) | |
| class API: | |
| def repo_info(self, *args, **kwargs): | |
| return SimpleNamespace(sha="previous-hf-commit") | |
| def create_commit(self, **kwargs): | |
| assert kwargs["parent_commit"] == "previous-hf-commit" | |
| assert {item["path_in_repo"] for item in kwargs["operations"]} == set(manifest["sha256"]) | |
| return SimpleNamespace(oid="new-hf-commit", commit_url="https://huggingface.co/spaces/test/commit/new-hf-commit") | |
| def list_repo_files(self, *args, **kwargs): | |
| assert kwargs["revision"] == "new-hf-commit" | |
| return [name for name in manifest["sha256"] if name != "scripts/start_with_ollama.py"] | |
| with pytest.raises(DeploymentError, match="Uploaded HF commit is missing source files: scripts/start_with_ollama.py"): | |
| upload_revision(API(), stage, manifest) | |