@ngocsangairvds/vsaf 3.1.27 → 3.2.2

package/tools/vds-scripts/bitbucket_orchestrator/tests/test_cli.py

@@ -0,0 +1,115 @@
+from __future__ import annotations
+
+from pathlib import Path
+from unittest.mock import MagicMock, Mock, patch
+
+import pytest
+from typer.testing import CliRunner
+from vds_bitbucket_orchestrator.cli import app
+from vds_bitbucket_orchestrator.config import BitbucketSettings
+
+
+@pytest.fixture
+def runner() -> CliRunner:
+    """Create a CLI runner for testing."""
+    return CliRunner()
+
+
+@pytest.fixture
+def mock_settings() -> BitbucketSettings:
+    """Create mock settings for testing."""
+    return BitbucketSettings(
+        VDS_USERNAME="test_user",
+        VDS_PASSWORD="test_pass",
+        BITBUCKET_TOKEN="test_token",
+        BITBUCKET_BASE_URL="https://bitbucket.test.com",
+        BITBUCKET_MODE="server",
+    )
+
+
+class TestBitbucketCLI:
+    """Regression coverage for top-level Bitbucket CLI commands."""
+
+    def test_version_option(self, runner: CliRunner) -> None:
+        """`--version` should print the CLI version and exit cleanly."""
+        result = runner.invoke(app, ["--version"])
+        assert result.exit_code == 0
+        assert "vds-bitbucket-orchestrator" in result.stdout
+
+    @patch("vds_bitbucket_orchestrator.cli.BitbucketClient")
+    @patch("vds_bitbucket_orchestrator.cli.load_settings")
+    def test_projects_command_success(
+        self,
+        mock_load: Mock,
+        mock_client_class: Mock,
+        runner: CliRunner,
+        mock_settings: BitbucketSettings,
+        sample_project_response: dict,
+    ) -> None:
+        """`projects` should call list_projects and emit JSON when --no-markdown is provided."""
+        mock_load.return_value = mock_settings
+
+        client_manager = MagicMock()
+        client_instance = client_manager.__enter__.return_value
+        client_instance.list_projects.return_value = [sample_project_response]
+        mock_client_class.return_value = client_manager
+
+        with runner.isolated_filesystem():
+            result = runner.invoke(app, ["--report-dir", ".", "projects", "--limit", "5"])
+        assert result.exit_code == 0
+        assert "Test Project" in result.stdout
+        client_instance.list_projects.assert_called_once_with(limit=5)
+        assert Path("projects.md").exists()
+
+    @patch("vds_bitbucket_orchestrator.cli.BitbucketClient")
+    @patch("vds_bitbucket_orchestrator.cli.load_settings")
+    def test_projects_command_handles_error(
+        self,
+        mock_load: Mock,
+        mock_client_class: Mock,
+        runner: CliRunner,
+        mock_settings: BitbucketSettings,
+    ) -> None:
+        """`projects` should exit non-zero when the client raises."""
+        mock_load.return_value = mock_settings
+
+        client_manager = MagicMock()
+        client_instance = client_manager.__enter__.return_value
+        client_instance.list_projects.side_effect = RuntimeError("boom")
+        mock_client_class.return_value = client_manager
+
+        with runner.isolated_filesystem():
+            result = runner.invoke(app, ["projects"])
+
+        assert result.exit_code != 0
+
+    @patch("vds_bitbucket_orchestrator.cli.BitbucketClient")
+    @patch("vds_bitbucket_orchestrator.cli.load_settings")
+    def test_repositories_command_success(
+        self,
+        mock_load: Mock,
+        mock_client_class: Mock,
+        runner: CliRunner,
+        mock_settings: BitbucketSettings,
+        sample_repository_response: dict,
+    ) -> None:
+        """`repositories` should call list_repositories with the provided project key."""
+        mock_load.return_value = mock_settings
+
+        client_manager = MagicMock()
+        client_instance = client_manager.__enter__.return_value
+        client_instance.list_repositories.return_value = [sample_repository_response]
+        mock_client_class.return_value = client_manager
+
+        with runner.isolated_filesystem():
+            result = runner.invoke(app, ["--report-dir", ".", "repositories", "TEST", "--limit", "7"])
+        assert result.exit_code == 0
+        assert "test-repo" in result.stdout
+        client_instance.list_repositories.assert_called_once_with("TEST", limit=7)
+        assert Path("TEST_repositories.md").exists()
+
+    def test_repositories_missing_project_argument(self, runner: CliRunner) -> None:
+        """`repositories` without the required project key should exit with usage error."""
+        result = runner.invoke(app, ["repositories"])
+        assert result.exit_code == 2
+        assert "Missing argument 'PROJECT_KEY'" in (result.stdout or result.stderr)

package/tools/vds-scripts/bitbucket_orchestrator/tests/test_client.py

@@ -0,0 +1,157 @@
+from __future__ import annotations
+
+import os
+from unittest.mock import Mock, patch
+
+import pytest
+from vds_bitbucket_orchestrator.client import BitbucketClient
+from vds_bitbucket_orchestrator.config import BitbucketSettings
+
+
+@pytest.fixture
+def mock_settings() -> BitbucketSettings:
+    """Create server-mode settings for testing."""
+    return BitbucketSettings(
+        VDS_USERNAME="test_user",
+        VDS_PASSWORD="test_pass",
+        BITBUCKET_TOKEN="test_token",
+        BITBUCKET_BASE_URL="https://bitbucket.test.com",
+        BITBUCKET_MODE="server",
+    )
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_client_initialization_server_mode(mock_bitbucket_class: Mock, mock_settings: BitbucketSettings) -> None:
+    """Client should initialize correctly in server mode."""
+    client = BitbucketClient(mock_settings)
+    assert client._settings == mock_settings
+    assert client._settings.is_server_mode
+    assert not client._settings.is_cloud_mode
+    mock_bitbucket_class.assert_called_once()
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_client_initialization_cloud_mode(mock_bitbucket_class: Mock) -> None:
+    """Client should initialize correctly in cloud mode."""
+    settings = BitbucketSettings(
+        BITBUCKET_TOKEN="test_token",
+        BITBUCKET_MODE="cloud",
+        BITBUCKET_CLOUD_WORKSPACE="test_workspace",
+    )
+    client = BitbucketClient(settings)
+    assert client._settings == settings
+    assert client._settings.is_cloud_mode
+    assert not client._settings.is_server_mode
+    mock_bitbucket_class.assert_called_once()
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_list_projects_server_mode(mock_bitbucket_class: Mock, mock_settings: BitbucketSettings) -> None:
+    client = BitbucketClient(mock_settings)
+    mock_bitbucket = Mock()
+    mock_bitbucket.project_list.return_value = [{"key": "TEST", "name": "Test Project"}]
+    client._client = mock_bitbucket
+
+    result = client.list_projects()
+
+    assert result == [{"key": "TEST", "name": "Test Project"}]
+    mock_bitbucket.project_list.assert_called_once_with(limit=25)
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_list_repositories_server_mode(mock_bitbucket_class: Mock, mock_settings: BitbucketSettings) -> None:
+    client = BitbucketClient(mock_settings)
+    mock_bitbucket = Mock()
+    mock_bitbucket.repo_list.return_value = [{"slug": "test-repo", "name": "Test Repository"}]
+    client._client = mock_bitbucket
+
+    result = client.list_repositories("TEST")
+
+    assert result == [{"slug": "test-repo", "name": "Test Repository"}]
+    mock_bitbucket.repo_list.assert_called_once_with("TEST", limit=25)
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_get_project_server_mode(mock_bitbucket_class: Mock, mock_settings: BitbucketSettings) -> None:
+    client = BitbucketClient(mock_settings)
+    mock_bitbucket = Mock()
+    mock_bitbucket.project.return_value = {"key": "TEST", "name": "Test Project"}
+    client._client = mock_bitbucket
+
+    result = client.get_project("TEST")
+
+    assert result == {"key": "TEST", "name": "Test Project"}
+    mock_bitbucket.project.assert_called_once_with("TEST")
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_get_repository_server_mode(mock_bitbucket_class: Mock, mock_settings: BitbucketSettings) -> None:
+    client = BitbucketClient(mock_settings)
+    mock_bitbucket = Mock()
+    mock_bitbucket.get_repo.return_value = {"slug": "test-repo", "name": "Test Repository"}
+    client._client = mock_bitbucket
+
+    result = client.get_repository("TEST", "test-repo")
+
+    assert result == {"slug": "test-repo", "name": "Test Repository"}
+    mock_bitbucket.get_repo.assert_called_once_with("TEST", "test-repo")
+
+
+def test_get_url_server_mode(mock_settings: BitbucketSettings) -> None:
+    url = mock_settings.get_url()
+    assert url == "https://bitbucket.test.com"
+
+
+def test_get_url_cloud_mode() -> None:
+    settings = BitbucketSettings(
+        BITBUCKET_TOKEN="test_token",
+        BITBUCKET_MODE="cloud",
+        BITBUCKET_CLOUD_WORKSPACE="test_workspace",
+    )
+    url = settings.get_url()
+    assert url == "https://api.bitbucket.org/workspace/test_workspace"
+
+
+def test_get_url_cloud_mode_missing_workspace() -> None:
+    settings = BitbucketSettings(BITBUCKET_TOKEN="test_token", BITBUCKET_MODE="cloud")
+    with pytest.raises(ValueError, match="BITBUCKET_CLOUD_WORKSPACE is required"):
+        settings.get_url()
+
+
+def test_get_auth_kwargs_with_token() -> None:
+    settings = BitbucketSettings(BITBUCKET_TOKEN="test_token")
+    auth_kwargs = settings.get_auth_kwargs()
+    assert auth_kwargs == {"token": "test_token"}
+
+
+def test_get_auth_kwargs_with_username_password() -> None:
+    settings = BitbucketSettings(VDS_USERNAME="test_user", VDS_PASSWORD="test_pass")
+    auth_kwargs = settings.get_auth_kwargs()
+    assert auth_kwargs == {"username": "test_user", "password": "test_pass"}
+
+
+@patch.dict(os.environ, {}, clear=True)
+def test_get_auth_kwargs_no_credentials() -> None:
+    settings = BitbucketSettings(
+        VDS_USERNAME=None,
+        VDS_PASSWORD=None,
+        BITBUCKET_TOKEN=None,
+    )
+    with pytest.raises(ValueError, match="No valid authentication credentials"):
+        settings.get_auth_kwargs()
+
+
+def test_require_basic_or_token_success() -> None:
+    settings = BitbucketSettings(BITBUCKET_TOKEN="test_token")
+    settings.require_basic_or_token()
+
+
+@patch.dict(os.environ, {}, clear=True)
+def test_require_basic_or_token_failure() -> None:
+    settings = BitbucketSettings(
+        VDS_USERNAME=None,
+        VDS_PASSWORD=None,
+        BITBUCKET_TOKEN=None,
+    )
+    with pytest.raises(ValueError, match="Missing credentials"):
+        settings.require_basic_or_token()

package/tools/vds-scripts/bitbucket_orchestrator/tests/test_client_branch_conditions.py

@@ -0,0 +1,79 @@
+from __future__ import annotations
+
+from unittest.mock import MagicMock, patch
+
+import pytest
+from vds_bitbucket_orchestrator.client import BitbucketClient, BitbucketClientError
+from vds_bitbucket_orchestrator.config import BitbucketSettings
+
+
+@pytest.fixture
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def client(mock_bb: MagicMock) -> BitbucketClient:
+    mock_instance = MagicMock()
+    mock_bb.return_value = mock_instance
+    settings = BitbucketSettings(BITBUCKET_TOKEN="token")
+    return BitbucketClient(settings)
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_set_branches_permissions_defaults(mock_bb: MagicMock) -> None:
+    stub = MagicMock()
+    stub.set_branches_permissions.return_value = {"ok": True}
+    mock_bb.return_value = stub
+    client = BitbucketClient(BitbucketSettings(BITBUCKET_TOKEN="token"))
+
+    result = client.set_branches_permissions(
+        "PRJ",
+        matcher_type="BRANCH",
+        matcher_value="refs/heads/main",
+        permission_type="READ_ONLY",
+    )
+
+    assert result == {"ok": True}
+    stub.set_branches_permissions.assert_called_once_with(
+        "PRJ",
+        multiple_permissions=False,
+        matcher_type="BRANCH",
+        matcher_value="refs/heads/main",
+        permission_type="READ_ONLY",
+        repository_slug=None,
+        except_users=[],
+        except_groups=[],
+        except_access_keys=[],
+        start=0,
+        limit=25,
+    )
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_set_branches_permissions_error(mock_bb: MagicMock) -> None:
+    stub = MagicMock()
+    stub.set_branches_permissions.side_effect = RuntimeError("boom")
+    mock_bb.return_value = stub
+    client = BitbucketClient(BitbucketSettings(BITBUCKET_TOKEN="token"))
+
+    with pytest.raises(BitbucketClientError):
+        client.set_branches_permissions("PRJ")
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_get_project_conditions_coerces_list(mock_bb: MagicMock) -> None:
+    stub = MagicMock()
+    stub.get_project_conditions.return_value = {"id": 1}
+    mock_bb.return_value = stub
+    client = BitbucketClient(BitbucketSettings(BITBUCKET_TOKEN="token"))
+
+    result = client.get_project_conditions("PRJ")
+    assert result == [{"id": 1}]
+
+
+@patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+def test_get_project_conditions_handles_none(mock_bb: MagicMock) -> None:
+    stub = MagicMock()
+    stub.get_project_conditions.return_value = None
+    mock_bb.return_value = stub
+    client = BitbucketClient(BitbucketSettings(BITBUCKET_TOKEN="token"))
+
+    result = client.get_project_conditions("PRJ")
+    assert result == []

package/tools/vds-scripts/bitbucket_orchestrator/tests/test_client_code_advanced.py

@@ -0,0 +1,163 @@
+from __future__ import annotations
+
+from unittest.mock import Mock, patch
+
+import pytest
+from vds_bitbucket_orchestrator.client import BitbucketClient
+from vds_bitbucket_orchestrator.config import BitbucketSettings
+from vds_bitbucket_orchestrator.errors import BitbucketClientError
+
+
+class TestBitbucketClientCodeAdvanced:
+    """Test suite for BitbucketClient advanced code operations."""
+
+    @pytest.fixture
+    def server_settings(self) -> BitbucketSettings:
+        """Create server mode settings for testing."""
+        return BitbucketSettings(
+            VDS_USERNAME="test_user",
+            VDS_PASSWORD="test_pass",
+            BITBUCKET_BASE_URL="https://bitbucket.test.com",
+            BITBUCKET_MODE="server",
+        )
+
+    @pytest.fixture
+    def cloud_settings(self) -> BitbucketSettings:
+        """Create cloud mode settings for testing."""
+        return BitbucketSettings(
+            BITBUCKET_TOKEN="test_token",
+            BITBUCKET_MODE="cloud",
+            BITBUCKET_CLOUD_WORKSPACE="test_workspace",
+        )
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_file_list_success(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test successful file list retrieval."""
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_file_list.return_value = [{"path": "file1.py"}, {"path": "file2.py"}]
+        client._client = mock_bitbucket
+
+        result = client.get_file_list("PRJ", "repo", sub_folder="src", query="*.py", start=0, limit=100)
+
+        mock_bitbucket.get_file_list.assert_called_once_with("PRJ", "repo", sub_folder="src", query="*.py", start=0, limit=100)
+        assert len(result) == 2
+        assert result[0]["path"] == "file1.py"
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_file_list_returns_empty_list(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test that get_file_list returns empty list when client returns None."""
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_file_list.return_value = None
+        client._client = mock_bitbucket
+
+        result = client.get_file_list("PRJ", "repo")
+
+        assert result == []
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_file_list_handles_error(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test that get_file_list handles errors."""
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_file_list.side_effect = Exception("API error")
+        client._client = mock_bitbucket
+
+        with pytest.raises(BitbucketClientError):
+            client.get_file_list("PRJ", "repo")
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_commit_info_success(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test successful commit info retrieval."""
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_commit_info.return_value = {"id": "abc123", "message": "Test commit"}
+        client._client = mock_bitbucket
+
+        result = client.get_commit_info("PRJ", "repo", "abc123", path="file.py")
+
+        mock_bitbucket.get_commit_info.assert_called_once_with("PRJ", "repo", "abc123", path="file.py")
+        assert result["id"] == "abc123"
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_commit_info_handles_error(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test that get_commit_info handles errors."""
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_commit_info.side_effect = Exception("API error")
+        client._client = mock_bitbucket
+
+        with pytest.raises(BitbucketClientError):
+            client.get_commit_info("PRJ", "repo", "abc123")
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_commit_changes_with_commit_id(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test successful commit changes retrieval with commit_id."""
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_commit_changes.return_value = {"changes": []}
+        client._client = mock_bitbucket
+
+        result = client.get_commit_changes("PRJ", "repo", commit_id="abc123", merges="exclude")
+
+        mock_bitbucket.get_commit_changes.assert_called_once_with("PRJ", "repo", hash_newest=None, merges="exclude", commit_id="abc123")
+        assert result == {"changes": []}
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_commit_changes_with_hash_newest(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test successful commit changes retrieval with hash_newest."""
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_commit_changes.return_value = {"changes": []}
+        client._client = mock_bitbucket
+
+        result = client.get_commit_changes("PRJ", "repo", hash_newest="def456", merges="include")
+
+        mock_bitbucket.get_commit_changes.assert_called_once_with("PRJ", "repo", hash_newest="def456", merges="include", commit_id=None)
+        assert result == {"changes": []}
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_commit_changes_handles_error(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test that get_commit_changes handles errors."""
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_commit_changes.side_effect = Exception("API error")
+        client._client = mock_bitbucket
+
+        with pytest.raises(BitbucketClientError):
+            client.get_commit_changes("PRJ", "repo", commit_id="abc123")
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_search_code_requires_cloud_mode(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        """Test that search_code requires Cloud mode."""
+        client = BitbucketClient(server_settings)
+
+        with pytest.raises(BitbucketClientError, match="Code search is only available in Bitbucket Cloud mode"):
+            client.search_code("workspace", "test query")
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_search_code_success(self, mock_bitbucket_class: Mock, cloud_settings: BitbucketSettings) -> None:
+        """Test successful code search in Cloud mode."""
+        client = BitbucketClient(cloud_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.search_code.return_value = {"values": [{"file": {"path": "file.py"}}]}
+        client._client = mock_bitbucket
+
+        result = client.search_code("workspace", "def test", page=1, limit=20)
+
+        mock_bitbucket.search_code.assert_called_once_with("workspace", "def test", page=1, limit=20)
+        assert "values" in result
+        assert len(result["values"]) == 1
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_search_code_handles_error(self, mock_bitbucket_class: Mock, cloud_settings: BitbucketSettings) -> None:
+        """Test that search_code handles errors."""
+        client = BitbucketClient(cloud_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.search_code.side_effect = Exception("API error")
+        client._client = mock_bitbucket
+
+        with pytest.raises(BitbucketClientError):
+            client.search_code("workspace", "test query")
+

package/tools/vds-scripts/bitbucket_orchestrator/tests/test_client_code_file.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+from unittest.mock import Mock, patch
+
+import pytest
+from vds_bitbucket_orchestrator.client import BitbucketClient
+from vds_bitbucket_orchestrator.config import BitbucketSettings
+
+
+class TestBitbucketClientCodeFile:
+    """Tests for file content retrieval."""
+
+    @pytest.fixture
+    def server_settings(self) -> BitbucketSettings:
+        return BitbucketSettings(
+            VDS_USERNAME="test_user",
+            VDS_PASSWORD="test_pass",
+            BITBUCKET_BASE_URL="https://bitbucket.test.com",
+            BITBUCKET_MODE="server",
+        )
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_content_of_file_decodes_bytes(self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings) -> None:
+        client = BitbucketClient(server_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.get_content_of_file.return_value = b"hello world"
+        client._client = mock_bitbucket
+
+        result = client.get_content_of_file("PRJ", "repo", "README.md")
+
+        assert result == "hello world"
+