Emil Simeonov
790b5c3769
- Replace argparse/print/sys.exit with click commands and ClickException - Translate all user-facing messages via _() - Add friendly Oops! / Nice! prompts - Wrap HTTP errors in all scripts with user-friendly translated messages - Update all unit tests to use CliRunner and expect ClickException - Add 100% branch coverage for new HTTP error handling branches - Add missing translation keys to i18n.py - Fix pre-commit hook to use venv Python for validate_commit_msg.py
170 lines
6.6 KiB
Python
170 lines
6.6 KiB
Python
"""Unit tests for scripts/publish.py."""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import click
|
|
import pytest
|
|
import requests
|
|
from click.testing import CliRunner
|
|
|
|
from scripts.publish import (
|
|
GITEA_API,
|
|
build_package,
|
|
create_gitea_release,
|
|
main,
|
|
publish_to_pypi,
|
|
)
|
|
|
|
|
|
class TestBuildPackage:
|
|
@patch("scripts.publish.subprocess.run")
|
|
def test_success(self, mock_run: MagicMock) -> None:
|
|
mock_run.return_value = MagicMock(returncode=0, stderr="")
|
|
build_package()
|
|
args, _ = mock_run.call_args
|
|
assert args[0][1] == "-m"
|
|
assert args[0][2] == "build"
|
|
|
|
@patch("scripts.publish.subprocess.run")
|
|
def test_failure_raises(self, mock_run: MagicMock) -> None:
|
|
mock_run.return_value = MagicMock(returncode=1, stderr="build error")
|
|
with pytest.raises(click.ClickException) as exc:
|
|
build_package()
|
|
assert "build" in str(exc.value)
|
|
|
|
|
|
class TestPublishToPypi:
|
|
@patch("scripts.publish.subprocess.run")
|
|
def test_success(self, mock_run: MagicMock) -> None:
|
|
mock_run.return_value = MagicMock(returncode=0, stderr="")
|
|
publish_to_pypi("pypi-tok")
|
|
args, _ = mock_run.call_args
|
|
assert "twine" in args[0]
|
|
assert "pypi-tok" in args[0]
|
|
|
|
@patch("scripts.publish.subprocess.run")
|
|
def test_failure_raises(self, mock_run: MagicMock) -> None:
|
|
mock_run.return_value = MagicMock(returncode=1, stderr="upload failed")
|
|
with pytest.raises(click.ClickException) as exc:
|
|
publish_to_pypi("pypi-tok")
|
|
assert "PyPI" in str(exc.value)
|
|
|
|
|
|
class TestCreateGiteaRelease:
|
|
@patch("scripts.publish.requests.post")
|
|
def test_success(self, mock_post: MagicMock) -> None:
|
|
mock_response = MagicMock()
|
|
mock_post.return_value = mock_response
|
|
create_gitea_release("tok", "owner/repo", "v1.0.0")
|
|
args, kwargs = mock_post.call_args
|
|
assert GITEA_API in args[0]
|
|
assert kwargs["json"]["tag_name"] == "v1.0.0"
|
|
assert kwargs["json"]["draft"] is False
|
|
assert kwargs["json"]["prerelease"] is False
|
|
|
|
@patch("scripts.publish.requests.post")
|
|
def test_http_error_propagates(self, mock_post: MagicMock) -> None:
|
|
mock_response = MagicMock()
|
|
mock_response.raise_for_status.side_effect = requests.HTTPError("500")
|
|
mock_post.return_value = mock_response
|
|
with pytest.raises(requests.HTTPError):
|
|
create_gitea_release("tok", "owner/repo", "v1.0.0")
|
|
|
|
|
|
class TestMain:
|
|
@patch.dict("os.environ", {"GITEA_TOKEN": "gitea-tok", "PYPI_TOKEN": "pypi-tok"})
|
|
@patch("scripts.publish.create_gitea_release")
|
|
@patch("scripts.publish.publish_to_pypi")
|
|
@patch("scripts.publish.build_package")
|
|
def test_full_flow_with_pypi(
|
|
self,
|
|
mock_build: MagicMock,
|
|
mock_publish: MagicMock,
|
|
mock_release: MagicMock,
|
|
) -> None:
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["v1.0.0", "owner/repo"])
|
|
assert result.exit_code == 0
|
|
assert "Gitea release v1.0.0 created" in result.output
|
|
mock_build.assert_called_once()
|
|
mock_publish.assert_called_once_with("pypi-tok")
|
|
mock_release.assert_called_once_with("gitea-tok", "owner/repo", "v1.0.0")
|
|
|
|
@patch.dict("os.environ", {"GITEA_TOKEN": "gitea-tok"}, clear=True)
|
|
@patch("scripts.publish.create_gitea_release")
|
|
@patch("scripts.publish.build_package")
|
|
def test_without_pypi(
|
|
self,
|
|
mock_build: MagicMock,
|
|
mock_release: MagicMock,
|
|
) -> None:
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["v1.0.0", "owner/repo"])
|
|
assert result.exit_code == 0
|
|
mock_build.assert_called_once()
|
|
mock_release.assert_called_once_with("gitea-tok", "owner/repo", "v1.0.0")
|
|
assert "PYPI_TOKEN not set" in result.output
|
|
|
|
@patch.dict("os.environ", {"GITEA_TOKEN": ""}, clear=True)
|
|
def test_missing_gitea_token_exits(self) -> None:
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["v1.0.0", "owner/repo"])
|
|
assert result.exit_code == 1
|
|
assert "GITEA_TOKEN" in result.output
|
|
|
|
@patch.dict("os.environ", {"GITEA_TOKEN": "gitea-tok", "PYPI_TOKEN": "pypi-tok"})
|
|
@patch("scripts.publish.create_gitea_release")
|
|
@patch("scripts.publish.publish_to_pypi")
|
|
@patch("scripts.publish.build_package")
|
|
def test_build_failure_raises_click(
|
|
self, mock_build: MagicMock, mock_publish: MagicMock, mock_release: MagicMock
|
|
) -> None:
|
|
mock_build.side_effect = click.ClickException("build failed")
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["v1.0.0", "owner/repo"])
|
|
assert result.exit_code == 1
|
|
assert "build" in result.output
|
|
|
|
@patch.dict("os.environ", {"GITEA_TOKEN": "gitea-tok", "PYPI_TOKEN": "pypi-tok"})
|
|
@patch("scripts.publish.create_gitea_release")
|
|
@patch("scripts.publish.publish_to_pypi")
|
|
@patch("scripts.publish.build_package")
|
|
def test_publish_failure_raises_click(
|
|
self, mock_build: MagicMock, mock_publish: MagicMock, mock_release: MagicMock
|
|
) -> None:
|
|
mock_publish.side_effect = click.ClickException("publish failed")
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["v1.0.0", "owner/repo"])
|
|
assert result.exit_code == 1
|
|
assert "publish" in result.output
|
|
|
|
@patch.dict("os.environ", {"GITEA_TOKEN": "gitea-tok", "PYPI_TOKEN": "pypi-tok"})
|
|
@patch("scripts.publish.create_gitea_release")
|
|
@patch("scripts.publish.publish_to_pypi")
|
|
@patch("scripts.publish.build_package")
|
|
def test_release_failure_raises_click(
|
|
self, mock_build: MagicMock, mock_publish: MagicMock, mock_release: MagicMock
|
|
) -> None:
|
|
mock_release.side_effect = requests.HTTPError("500")
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["v1.0.0", "owner/repo"])
|
|
assert result.exit_code == 1
|
|
assert "HTTP" in result.output
|
|
|
|
@patch.dict("os.environ", {"GITEA_TOKEN": "gitea-tok", "PYPI_TOKEN": "pypi-tok"})
|
|
@patch("scripts.publish.create_gitea_release")
|
|
@patch("scripts.publish.publish_to_pypi")
|
|
@patch("scripts.publish.build_package")
|
|
def test_release_json_parse_failure(
|
|
self, mock_build: MagicMock, mock_publish: MagicMock, mock_release: MagicMock
|
|
) -> None:
|
|
mock_response = MagicMock()
|
|
mock_response.status_code = 502
|
|
mock_response.json.side_effect = ValueError("not json")
|
|
err = requests.HTTPError("502", response=mock_response)
|
|
mock_release.side_effect = err
|
|
runner = CliRunner()
|
|
result = runner.invoke(main, ["v1.0.0", "owner/repo"])
|
|
assert result.exit_code == 1
|
|
assert "502" in result.output
|