"""Unit tests for scripts/ci/post_merge.py.""" import http from unittest.mock import MagicMock, patch import click import pytest from click.testing import CliRunner from gitea_runner_manager.exceptions import APIError from scripts.ci.post_merge import ( build_comment, extract_conventional_msg, extract_task_id, main, resolve_task_id, ) class TestExtractTaskId: def test_extracts_from_first_line(self) -> None: assert extract_task_id("GRM-19: fix: resolve bug\n\nBody") == "GRM-19" def test_returns_empty_when_missing(self) -> None: assert extract_task_id("fix: resolve bug") == "" class TestExtractConventionalMsg: def test_strips_colon_prefix(self) -> None: """Legacy format: GRM-N: """ assert extract_conventional_msg("GRM-19: fix: resolve bug") == "fix: resolve bug" def test_strips_space_prefix(self) -> None: """Current format: GRM-N """ assert extract_conventional_msg("GRM-19 fix: resolve bug") == "fix: resolve bug" def test_returns_unchanged_without_prefix(self) -> None: assert extract_conventional_msg("fix: resolve bug") == "fix: resolve bug" class TestBuildComment: def test_html_format(self) -> None: html = build_comment("GRM-19", "fix: bug", "abc123") assert "GRM-19" in html assert "fix: bug" in html assert "abc123" in html class TestResolveTaskId: def test_found(self) -> None: mock_client = MagicMock() mock_client.list_project_tasks.return_value = [ {"id": 42, "identifier": "GRM-19"}, ] assert resolve_task_id(mock_client, "GRM-19") == 42 mock_client.list_project_tasks.assert_called_once() def test_not_found_raises(self) -> None: mock_client = MagicMock() mock_client.list_project_tasks.return_value = [] with pytest.raises(click.ClickException) as exc: resolve_task_id(mock_client, "GRM-99") assert "Could not find" in str(exc.value) def test_found_on_second_page(self) -> None: """Task is on page 2 when project has more than 50 tasks.""" mock_client = MagicMock() page1 = [{"id": i, "identifier": f"GRM-{i}"} for i in range(50)] page2 = [{"id": 100, "identifier": "GRM-99"}] mock_client.list_project_tasks.side_effect = [page1, page2] assert resolve_task_id(mock_client, "GRM-99") == 100 assert mock_client.list_project_tasks.call_count == 2 def test_stops_when_page_is_partial(self) -> None: """Stops paginating when a page has fewer than DEFAULT_PER_PAGE results.""" mock_client = MagicMock() page1 = [{"id": i, "identifier": f"GRM-{i}"} for i in range(10)] mock_client.list_project_tasks.return_value = page1 with pytest.raises(click.ClickException) as exc: resolve_task_id(mock_client, "GRM-99") assert "Could not find" in str(exc.value) assert mock_client.list_project_tasks.call_count == 1 def test_http_error_propagates(self) -> None: mock_client = MagicMock() mock_client.list_project_tasks.side_effect = APIError(http.HTTPStatus.INTERNAL_SERVER_ERROR, "server error") with pytest.raises(APIError): resolve_task_id(mock_client, "GRM-19") class TestMain: @patch.dict("os.environ", {"VIKUNJA_TOKEN": "tok"}) @patch("scripts.ci.post_merge.VikunjaClient") def test_full_flow(self, mock_client_cls: MagicMock) -> None: mock_client = MagicMock() mock_client.list_project_tasks.return_value = [ {"id": 267, "identifier": "GRM-20"}, ] mock_client_cls.return_value = mock_client runner = CliRunner() result = runner.invoke( main, ["GRM-20: fix: resolve bug\n\nBody", "--commit-sha", "abc123"], ) assert result.exit_code == 0 assert "updated and marked done" in result.output mock_client.post_comment.assert_called_once() mock_client.update_task.assert_called_once_with(267, done=True) @patch.dict("os.environ", {"VIKUNJA_TOKEN": "tok"}) @patch("scripts.ci.post_merge.VikunjaClient") def test_no_commit_sha(self, mock_client_cls: MagicMock) -> None: mock_client = MagicMock() mock_client.list_project_tasks.return_value = [ {"id": 267, "identifier": "GRM-20"}, ] mock_client_cls.return_value = mock_client runner = CliRunner() result = runner.invoke(main, ["GRM-20: fix: resolve bug"]) assert result.exit_code == 0 mock_client.post_comment.assert_called_once() args, _ = mock_client.post_comment.call_args assert "unknown" in args[1] @patch.dict("os.environ", {"VIKUNJA_TOKEN": ""}, clear=True) def test_missing_token_exits(self) -> None: runner = CliRunner() result = runner.invoke(main, ["GRM-20: fix: bug"]) assert result.exit_code == 1 assert "VIKUNJA_TOKEN" in result.output @patch.dict("os.environ", {"VIKUNJA_TOKEN": "tok"}) def test_no_task_id_non_release_fails(self) -> None: """Non-release commits without GRM-N prefix should fail.""" runner = CliRunner() result = runner.invoke(main, ["fix: resolve bug"]) assert result.exit_code == 1 assert "No task ID" in result.output @patch.dict("os.environ", {"VIKUNJA_TOKEN": "tok"}) def test_release_commit_without_task_id_skips(self) -> None: """Release commits without GRM-N prefix should skip gracefully.""" runner = CliRunner() result = runner.invoke(main, ["release: v0.3.2"]) assert result.exit_code == 0 assert "Release commit" in result.output assert "skipping" in result.output @patch.dict("os.environ", {"VIKUNJA_TOKEN": "tok"}) @patch("scripts.ci.post_merge.VikunjaClient") def test_resolve_failure_propagates(self, mock_client_cls: MagicMock) -> None: mock_client = MagicMock() mock_client.list_project_tasks.return_value = [] mock_client_cls.return_value = mock_client runner = CliRunner() result = runner.invoke(main, ["GRM-20: fix: bug"]) assert result.exit_code == 1 assert "Could not find" in result.output @patch.dict("os.environ", {"VIKUNJA_TOKEN": "tok"}) @patch("scripts.ci.post_merge.VikunjaClient") def test_post_comment_failure_raises_click(self, mock_client_cls: MagicMock) -> None: mock_client = MagicMock() mock_client.list_project_tasks.return_value = [ {"id": 267, "identifier": "GRM-20"}, ] mock_client.post_comment.side_effect = APIError(http.HTTPStatus.INTERNAL_SERVER_ERROR, "server error") mock_client_cls.return_value = mock_client runner = CliRunner() result = runner.invoke(main, ["GRM-20: fix: bug"]) assert result.exit_code == 1 assert "HTTP" in result.output @patch.dict("os.environ", {"VIKUNJA_TOKEN": "tok"}) @patch("scripts.ci.post_merge.VikunjaClient") def test_mark_done_failure_raises_click(self, mock_client_cls: MagicMock) -> None: mock_client = MagicMock() mock_client.list_project_tasks.return_value = [ {"id": 267, "identifier": "GRM-20"}, ] mock_client.post_comment.return_value = None mock_client.update_task.side_effect = APIError(http.HTTPStatus.INTERNAL_SERVER_ERROR, "server error") mock_client_cls.return_value = mock_client runner = CliRunner() result = runner.invoke(main, ["GRM-20: fix: bug"]) assert result.exit_code == 1 assert "HTTP" in result.output