Refactor code structure for improved readability and maintainability

2026-07-06 21:37:20 +02:00
parent 6ca6a5687a
commit 5f35f56edc
48 changed files with 8128 additions and 0 deletions
@@ -0,0 +1,232 @@
+import json
+from typing import Any
+from unittest.mock import MagicMock
+from gitea.client import GiteaClient
+from gitea.models import PullRequestModel, CommentModel, RepositoryModel
+from gitea.tools.pr_tools import PRTools
+
+
+def test_get_pull_request_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    pr: PullRequestModel = PullRequestModel(number=1, title="Test PR", state="open")
+    mock_client.get_pull_request.return_value = pr
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.get_pull_request("owner", "repo", 1)
+
+    data: dict[str, Any] = json.loads(res)
+    assert data["number"] == 1
+    assert data["title"] == "Test PR"
+    mock_client.get_pull_request.assert_called_once_with("owner", "repo", 1)
+
+
+def test_get_pull_request_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.get_pull_request.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.get_pull_request("owner", "repo", 1)
+    assert "Error getting pull request: API Error" in res
+
+
+def test_close_pull_request_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.close_pull_request.return_value = PullRequestModel(number=1, state="closed")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.close_pull_request("owner", "repo", 1)
+    assert res == "Pull request #1 closed successfully."
+    mock_client.close_pull_request.assert_called_once_with("owner", "repo", 1)
+
+
+def test_close_pull_request_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.close_pull_request.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.close_pull_request("owner", "repo", 1)
+    assert "Error closing pull request: API Error" in res
+
+
+def test_get_pull_request_comments_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    comment: CommentModel = CommentModel(id=123, body="Comment body")
+    mock_client.get_pull_request_comments.return_value = [comment]
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.get_pull_request_comments("owner", "repo", 1)
+    data: list[dict[str, Any]] = json.loads(res)
+    assert len(data) == 1
+    assert data[0]["body"] == "Comment body"
+
+
+def test_get_pull_request_comments_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.get_pull_request_comments.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.get_pull_request_comments("owner", "repo", 1)
+    assert "Error getting PR comments: API Error" in res
+
+
+def test_list_assigned_pull_requests_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    repo: RepositoryModel = RepositoryModel(name="repo1", owner="owner1")
+    pr: PullRequestModel = PullRequestModel(number=1, title="Test PR")
+    mock_client.list_all_user_repos.return_value = [repo]
+    mock_client.list_assigned_pull_requests.return_value = [pr]
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: list[dict[str, Any]] = pr_tools.list_assigned_pull_requests()
+    assert len(res) == 1
+    assert res[0]["number"] == 1
+    mock_client.list_all_user_repos.assert_called_once()
+    mock_client.list_assigned_pull_requests.assert_called_once_with("owner1", "repo1")
+
+
+def test_list_assigned_pull_requests_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.list_all_user_repos.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: list[dict[str, Any]] = pr_tools.list_assigned_pull_requests()
+    assert res == []
+
+
+def test_list_pull_requests_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    pr: PullRequestModel = PullRequestModel(number=1, title="Test PR")
+    mock_client.list_repo_pull_requests.return_value = [pr]
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.list_pull_requests("owner", "repo")
+    assert "#1: Test PR" in res
+
+
+def test_list_pull_requests_empty() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.list_repo_pull_requests.return_value = []
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.list_pull_requests("owner", "repo")
+    assert res == "No PRs in owner/repo."
+
+
+def test_list_pull_requests_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.list_repo_pull_requests.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.list_pull_requests("owner", "repo")
+    assert "Error listing PRs: API Error" in res
+
+
+def test_create_pull_request_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    pr: PullRequestModel = PullRequestModel(number=2, title="Title")
+    mock_client.create_pr_via_tea.return_value = pr
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.create_pull_request("owner", "repo", "head", "base", "Title", "Desc")
+    data: dict[str, Any] = json.loads(res)
+    assert data["number"] == 2
+    mock_client.create_pr_via_tea.assert_called_once_with("owner", "repo", "Title", "Desc", "head", "base")
+
+
+def test_create_pull_request_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.create_pr_via_tea.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.create_pull_request("owner", "repo", "head", "base", "Title")
+    assert "Error creating PR: API Error" in res
+
+
+def test_add_label_to_pr_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.add_label_pr.return_value = {}
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.add_label_to_pr("owner", "repo", 1, "bug")
+    assert res == "Label 'bug' added to PR #1."
+
+
+def test_add_label_to_pr_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.add_label_pr.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.add_label_to_pr("owner", "repo", 1, "bug")
+    assert "Error adding label to PR #1: API Error" in res
+
+
+def test_get_pull_request_diff_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.get_pull_request_diff.return_value = "diff content"
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.get_pull_request_diff("owner", "repo", 1)
+    assert res == "diff content"
+
+
+def test_get_pull_request_diff_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.get_pull_request_diff.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.get_pull_request_diff("owner", "repo", 1)
+    assert "Error getting PR diff: API Error" in res
+
+
+def test_get_pull_request_patch_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.get_pull_request_patch.return_value = "patch content"
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.get_pull_request_patch("owner", "repo", 1)
+    assert res == "patch content"
+
+
+def test_get_pull_request_patch_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.get_pull_request_patch.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.get_pull_request_patch("owner", "repo", 1)
+    assert "Error getting PR patch: API Error" in res
+
+
+def test_approve_pull_request_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.approve_pr.return_value = {}
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.approve_pull_request("owner", "repo", 1, "good")
+    assert res == "Approved PR #1."
+
+
+def test_approve_pull_request_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.approve_pr.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.approve_pull_request("owner", "repo", 1, "good")
+    assert "Error approving PR: API Error" in res
+
+
+def test_request_changes_success() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.request_changes_pr.return_value = {}
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.request_changes("owner", "repo", 1, "bad")
+    assert res == "Requested changes on PR #1."
+
+
+def test_request_changes_failure() -> None:
+    mock_client: MagicMock = MagicMock(spec=GiteaClient)
+    mock_client.request_changes_pr.side_effect = Exception("API Error")
+
+    pr_tools: PRTools = PRTools(mock_client)
+    res: str = pr_tools.request_changes("owner", "repo", 1, "bad")
+    assert "Error requesting changes: API Error" in res