@ngocsangairvds/vsaf 3.1.27 → 3.2.2

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

@@ -0,0 +1,156 @@
+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 TestBitbucketClientRepositoryVariables:
+    """Tests for BitbucketClient repository variable operations."""
+
+    @pytest.fixture
+    def cloud_settings(self) -> BitbucketSettings:
+        return BitbucketSettings(
+            BITBUCKET_TOKEN="token",
+            BITBUCKET_MODE="cloud",
+            BITBUCKET_CLOUD_WORKSPACE="workspace",
+        )
+
+    @pytest.fixture
+    def server_settings(self) -> BitbucketSettings:
+        return BitbucketSettings(
+            VDS_USERNAME="user",
+            VDS_PASSWORD="pass",
+            BITBUCKET_BASE_URL="https://example.com",
+            BITBUCKET_MODE="server",
+        )
+
+    def _mock_repository(self) -> Mock:
+        repository = Mock()
+        repository.repository_variables = Mock()
+        variable = Mock()
+        variable.data = {"uuid": "{uuid}", "key": "API_TOKEN", "secured": False, "value": "secret"}
+        repository.repository_variables.each.return_value = [variable]
+        repository.repository_variables.get.return_value = variable
+        repository.repository_variables.create.return_value = variable
+        variable.update.return_value = variable
+        return repository
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_list_repository_variables_success(
+        self, mock_bitbucket_class: Mock, cloud_settings: BitbucketSettings
+    ) -> None:
+        client = BitbucketClient(cloud_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.cloud = Mock()
+        repository = self._mock_repository()
+        mock_bitbucket.cloud.repositories.get.return_value = repository
+        client._client = mock_bitbucket
+
+        variables = client.list_repository_variables("workspace", "repo")
+
+        mock_bitbucket.cloud.repositories.get.assert_called_once_with("workspace", "repo")
+        assert len(variables) == 1
+        assert variables[0]["key"] == "API_TOKEN"
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_get_repository_variable_success(
+        self, mock_bitbucket_class: Mock, cloud_settings: BitbucketSettings
+    ) -> None:
+        client = BitbucketClient(cloud_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.cloud = Mock()
+        repository = self._mock_repository()
+        mock_bitbucket.cloud.repositories.get.return_value = repository
+        client._client = mock_bitbucket
+
+        variable = client.get_repository_variable("workspace", "repo", "{uuid}")
+
+        repository.repository_variables.get.assert_called_once_with("{uuid}")
+        assert variable["key"] == "API_TOKEN"
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_create_repository_variable_success(
+        self, mock_bitbucket_class: Mock, cloud_settings: BitbucketSettings
+    ) -> None:
+        client = BitbucketClient(cloud_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.cloud = Mock()
+        repository = self._mock_repository()
+        mock_bitbucket.cloud.repositories.get.return_value = repository
+        client._client = mock_bitbucket
+
+        result = client.create_repository_variable("workspace", "repo", "API_TOKEN", "secret", secured=True)
+
+        repository.repository_variables.create.assert_called_once_with("API_TOKEN", "secret", True)
+        assert result["key"] == "API_TOKEN"
+        assert result["secured"] is False  # mocked data
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_update_repository_variable_success(
+        self, mock_bitbucket_class: Mock, cloud_settings: BitbucketSettings
+    ) -> None:
+        client = BitbucketClient(cloud_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.cloud = Mock()
+        repository = self._mock_repository()
+        mock_bitbucket.cloud.repositories.get.return_value = repository
+        client._client = mock_bitbucket
+
+        result = client.update_repository_variable(
+            "workspace",
+            "repo",
+            "{uuid}",
+            key="NEW_TOKEN",
+            value="new-secret",
+            secured=False,
+        )
+
+        repository.repository_variables.get.assert_called_once_with("{uuid}")
+        repository.repository_variables.get.return_value.update.assert_called_once_with(
+            key="NEW_TOKEN", value="new-secret", secured=False
+        )
+        assert result["key"] == "API_TOKEN"
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_update_repository_variable_requires_fields(
+        self, mock_bitbucket_class: Mock, cloud_settings: BitbucketSettings
+    ) -> None:
+        client = BitbucketClient(cloud_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.cloud = Mock()
+        repository = self._mock_repository()
+        mock_bitbucket.cloud.repositories.get.return_value = repository
+        client._client = mock_bitbucket
+
+        with pytest.raises(BitbucketClientError):
+            client.update_repository_variable("workspace", "repo", "{uuid}")
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_delete_repository_variable_success(
+        self, mock_bitbucket_class: Mock, cloud_settings: BitbucketSettings
+    ) -> None:
+        client = BitbucketClient(cloud_settings)
+        mock_bitbucket = Mock()
+        mock_bitbucket.cloud = Mock()
+        repository = self._mock_repository()
+        mock_bitbucket.cloud.repositories.get.return_value = repository
+        client._client = mock_bitbucket
+
+        client.delete_repository_variable("workspace", "repo", "{uuid}")
+
+        repository.repository_variables.get.assert_called_once_with("{uuid}")
+        repository.repository_variables.get.return_value.delete.assert_called_once_with()
+
+    @patch("vds_bitbucket_orchestrator.client.AtlassianBitbucket")
+    def test_repository_variable_operations_require_cloud(
+        self, mock_bitbucket_class: Mock, server_settings: BitbucketSettings
+    ) -> None:
+        client = BitbucketClient(server_settings)
+
+        with pytest.raises(BitbucketClientError):
+            client.list_repository_variables("workspace", "repo")
+

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

@@ -0,0 +1,98 @@
+from __future__ import annotations
+
+from unittest.mock import Mock, patch
+
+from typer.testing import CliRunner
+from vds_bitbucket_orchestrator.cli import app
+from vds_bitbucket_orchestrator.errors import BitbucketClientError
+
+
+def _runner() -> CliRunner:
+    return CliRunner()
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_file_requires_file(mock_load: Mock, mock_build: Mock) -> None:
+    runner = _runner()
+    result = runner.invoke(app, ["code", "file", "PRJ", "repo"])
+    assert result.exit_code != 0
+    assert "--file required for file action" in result.stdout
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_commits(mock_load: Mock, mock_build: Mock) -> None:
+    runner = _runner()
+    client = Mock()
+    client.get_commits.return_value = []
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(app, ["code", "commits", "PRJ", "repo", "--limit", "5"])
+    assert result.exit_code == 0
+    client.get_commits.assert_called_once_with("PRJ", "repo", None, None, limit=5)
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_commits_error(mock_load: Mock, mock_build: Mock) -> None:
+    runner = _runner()
+    client = Mock()
+    client.get_commits.side_effect = BitbucketClientError("nope")
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(app, ["code", "commits", "PRJ", "repo"])
+    assert result.exit_code != 0
+    assert "bitbucket_error" in result.stdout
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_diff_requires_arguments(mock_load: Mock, mock_build: Mock) -> None:
+    runner = _runner()
+    result = runner.invoke(app, ["code", "diff", "PRJ", "repo"])
+    assert result.exit_code != 0
+    assert "--file, --hash-oldest, and --hash-newest required" in result.stdout
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_diff(mock_load: Mock, mock_build: Mock) -> None:
+    runner = _runner()
+    client = Mock()
+    client.get_diff.return_value = {"diff": "..."}
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(
+        app,
+        [
+            "code",
+            "diff",
+            "PRJ",
+            "repo",
+            "--file",
+            "path/to/file.py",
+            "--hash-oldest",
+            "aaa",
+            "--hash-newest",
+            "bbb",
+        ],
+    )
+    assert result.exit_code == 0
+    client.get_diff.assert_called_once_with("PRJ", "repo", "path/to/file.py", "aaa", "bbb")
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_changelog(mock_load: Mock, mock_build: Mock) -> None:
+    runner = _runner()
+    client = Mock()
+    client.get_changelog.return_value = []
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(
+        app,
+        ["code", "changelog", "PRJ", "repo", "--ref-from", "main", "--ref-to", "develop", "--limit", "10"],
+    )
+    assert result.exit_code == 0
+    client.get_changelog.assert_called_once_with("PRJ", "repo", "main", "develop", limit=10)

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

@@ -0,0 +1,282 @@
+from __future__ import annotations
+
+from unittest.mock import Mock, patch
+
+from typer.testing import CliRunner
+from vds_bitbucket_orchestrator.cli import app
+from vds_bitbucket_orchestrator.errors import BitbucketClientError
+
+
+def _runner() -> CliRunner:
+    return CliRunner()
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_file_list_requires_project(mock_load: Mock, mock_build: Mock) -> None:
+    """Test that file-list requires --project-key."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    result = runner.invoke(app, ["code-advanced", "file-list", "--repository-slug", "repo"])
+    assert result.exit_code != 0
+    assert "--project-key required" in result.stderr
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_file_list_success(mock_load: Mock, mock_build: Mock) -> None:
+    """Test successful file-list."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    client = Mock()
+    client.get_file_list.return_value = [{"path": "file1.py"}, {"path": "file2.py"}]
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(
+        app,
+        [
+            "code-advanced",
+            "file-list",
+            "--project-key",
+            "PRJ",
+            "--repository-slug",
+            "repo",
+            "--sub-folder",
+            "src",
+            "--query",
+            "*.py",
+        ],
+    )
+    assert result.exit_code == 0
+    client.get_file_list.assert_called_once_with("PRJ", "repo", sub_folder="src", query="*.py", start=0, limit=None)
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_commit_info_requires_commit(mock_load: Mock, mock_build: Mock) -> None:
+    """Test that commit-info requires --commit."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    result = runner.invoke(
+        app, ["code-advanced", "commit-info", "--project-key", "PRJ", "--repository-slug", "repo"]
+    )
+    assert result.exit_code != 0
+    assert "--commit required" in result.stderr
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_commit_info_success(mock_load: Mock, mock_build: Mock) -> None:
+    """Test successful commit-info."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    client = Mock()
+    client.get_commit_info.return_value = {"id": "abc123", "message": "Test commit"}
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(
+        app,
+        [
+            "code-advanced",
+            "commit-info",
+            "--project-key",
+            "PRJ",
+            "--repository-slug",
+            "repo",
+            "--commit",
+            "abc123",
+            "--path",
+            "file.py",
+        ],
+    )
+    assert result.exit_code == 0
+    client.get_commit_info.assert_called_once_with("PRJ", "repo", "abc123", path="file.py")
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_commit_changes_requires_commit(mock_load: Mock, mock_build: Mock) -> None:
+    """Test that commit-changes requires --commit-id or --hash-newest."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    result = runner.invoke(
+        app, ["code-advanced", "commit-changes", "--project-key", "PRJ", "--repository-slug", "repo"]
+    )
+    assert result.exit_code != 0
+    assert "--commit-id or --hash-newest required" in result.stderr
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_commit_changes_with_commit_id(mock_load: Mock, mock_build: Mock) -> None:
+    """Test successful commit-changes with commit-id."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    client = Mock()
+    client.get_commit_changes.return_value = {"changes": []}
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(
+        app,
+        [
+            "code-advanced",
+            "commit-changes",
+            "--project-key",
+            "PRJ",
+            "--repository-slug",
+            "repo",
+            "--commit-id",
+            "abc123",
+            "--merges",
+            "exclude",
+        ],
+    )
+    assert result.exit_code == 0
+    client.get_commit_changes.assert_called_once_with("PRJ", "repo", commit_id="abc123", hash_newest=None, merges="exclude")
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_commit_changes_with_hash_newest(mock_load: Mock, mock_build: Mock) -> None:
+    """Test successful commit-changes with hash-newest."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    client = Mock()
+    client.get_commit_changes.return_value = {"changes": []}
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(
+        app,
+        [
+            "code-advanced",
+            "commit-changes",
+            "--project-key",
+            "PRJ",
+            "--repository-slug",
+            "repo",
+            "--hash-newest",
+            "def456",
+        ],
+    )
+    assert result.exit_code == 0
+    client.get_commit_changes.assert_called_once_with("PRJ", "repo", commit_id=None, hash_newest="def456", merges="include")
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_search_requires_cloud_mode(mock_load: Mock, mock_build: Mock) -> None:
+    """Test that search requires Cloud mode."""
+    runner = _runner()
+    settings = Mock()
+    settings.is_cloud_mode = False
+    mock_load.return_value = settings
+
+    result = runner.invoke(
+        app, ["code-advanced", "search", "--team", "workspace", "--search-query", "test"]
+    )
+    assert result.exit_code != 0
+    # Exit code 1 indicates Cloud mode requirement failed
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_search_requires_team(mock_load: Mock, mock_build: Mock) -> None:
+    """Test that search requires --team."""
+    runner = _runner()
+    settings = Mock(is_cloud_mode=True)
+    mock_load.return_value = settings
+
+    result = runner.invoke(app, ["code-advanced", "search", "--search-query", "test"])
+    assert result.exit_code != 0
+    assert "--team required" in result.stderr
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_search_requires_query(mock_load: Mock, mock_build: Mock) -> None:
+    """Test that search requires --search-query."""
+    runner = _runner()
+    settings = Mock(is_cloud_mode=True)
+    mock_load.return_value = settings
+
+    result = runner.invoke(app, ["code-advanced", "search", "--team", "workspace"])
+    assert result.exit_code != 0
+    assert "--search-query required" in result.stderr
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_search_success(mock_load: Mock, mock_build: Mock) -> None:
+    """Test successful code search."""
+    runner = _runner()
+    settings = Mock(is_cloud_mode=True)
+    mock_load.return_value = settings
+
+    client = Mock()
+    client.search_code.return_value = {"values": [{"file": {"path": "file.py"}}]}
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(
+        app,
+        [
+            "code-advanced",
+            "search",
+            "--team",
+            "workspace",
+            "--search-query",
+            "def test",
+            "--page",
+            "2",
+            "--limit",
+            "20",
+        ],
+    )
+    assert result.exit_code == 0
+    client.search_code.assert_called_once_with("workspace", "def test", page=2, limit=20)
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_unknown_action(mock_load: Mock, mock_build: Mock) -> None:
+    """Test that unknown action raises error."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    result = runner.invoke(app, ["code-advanced", "unknown"])
+    assert result.exit_code != 0
+    assert "Unknown action" in result.stderr
+
+
+@patch("vds_bitbucket_orchestrator.cli._build_client")
+@patch("vds_bitbucket_orchestrator.cli.load_settings")
+def test_code_advanced_client_error(mock_load: Mock, mock_build: Mock) -> None:
+    """Test that client errors are handled."""
+    runner = _runner()
+    settings = Mock()
+    mock_load.return_value = settings
+
+    client = Mock()
+    client.get_file_list.side_effect = BitbucketClientError("Client error", context={})
+    mock_build.return_value.__enter__.return_value = client
+
+    result = runner.invoke(
+        app, ["code-advanced", "file-list", "--project-key", "PRJ", "--repository-slug", "repo"]
+    )
+    assert result.exit_code != 0
+    assert "Error" in result.stderr
+