Added unittests to commands.

Author: srtab

tests/codebase/management/__init__.py

@@ -0,0 +1,100 @@
+from unittest.mock import Mock, call, patch
+
+from django.core.management import call_command
+
+import pytest
+
+from codebase.clients import RepoClient
+
+
+@pytest.fixture
+def mock_repo_client():
+    with patch.object(RepoClient, "create_instance") as mock:
+        client = Mock(spec=RepoClient)
+        mock.return_value = client
+        yield client
+
+
+def test_setup_webhooks_success(mock_repo_client):
+    """Test successful webhook setup for repositories."""
+    # Mock repositories
+    mock_repo1 = Mock(slug="repo1")
+    mock_repo2 = Mock(slug="repo2")
+    mock_repo_client.list_repositories.return_value = [mock_repo1, mock_repo2]
+    mock_repo_client.set_repository_webhooks.return_value = True
+
+    # Call the command
+    call_command("setup_webhooks", base_url="https://test.com")
+
+    # Verify list_repositories was called with load_all=True
+    mock_repo_client.list_repositories.assert_called_once_with(load_all=True)
+
+    # Verify set_repository_webhooks was called for each repo
+    expected_calls = [
+        call(
+            "repo1",
+            "https://test.com/api/codebase/callbacks/gitlab/",
+            ["push_events", "issues_events", "note_events", "pipeline_events"],
+            enable_ssl_verification=True,
+        ),
+        call(
+            "repo2",
+            "https://test.com/api/codebase/callbacks/gitlab/",
+            ["push_events", "issues_events", "note_events", "pipeline_events"],
+            enable_ssl_verification=True,
+        ),
+    ]
+    assert mock_repo_client.set_repository_webhooks.call_count == 2
+    mock_repo_client.set_repository_webhooks.assert_has_calls(expected_calls)
+
+
+def test_setup_webhooks_with_ssl_disabled(mock_repo_client):
+    """Test webhook setup with SSL verification disabled."""
+    # Mock single repository
+    mock_repo = Mock(slug="repo1")
+    mock_repo_client.list_repositories.return_value = [mock_repo]
+    mock_repo_client.set_repository_webhooks.return_value = True
+
+    # Call the command with SSL verification disabled
+    call_command("setup_webhooks", base_url="https://test.com", disable_ssl_verification=True)
+
+    # Verify set_repository_webhooks was called with SSL verification disabled
+    mock_repo_client.set_repository_webhooks.assert_called_once_with(
+        "repo1",
+        "https://test.com/api/codebase/callbacks/gitlab/",
+        ["push_events", "issues_events", "note_events", "pipeline_events"],
+        enable_ssl_verification=False,
+    )
+
+
+def test_setup_webhooks_update_existing(mock_repo_client):
+    """Test updating existing webhooks."""
+    # Mock repository
+    mock_repo = Mock(slug="repo1")
+    mock_repo_client.list_repositories.return_value = [mock_repo]
+    # Return False to simulate updating existing webhook
+    mock_repo_client.set_repository_webhooks.return_value = False
+
+    # Call the command
+    call_command("setup_webhooks", base_url="https://test.com")
+
+    # Verify webhook was updated
+    mock_repo_client.set_repository_webhooks.assert_called_once_with(
+        "repo1",
+        "https://test.com/api/codebase/callbacks/gitlab/",
+        ["push_events", "issues_events", "note_events", "pipeline_events"],
+        enable_ssl_verification=True,
+    )
+
+
+def test_setup_webhooks_no_repositories(mock_repo_client):
+    """Test behavior when no repositories are found."""
+    # Mock empty repository list
+    mock_repo_client.list_repositories.return_value = []
+
+    # Call the command
+    call_command("setup_webhooks", base_url="https://test.com")
+
+    # Verify list_repositories was called but set_repository_webhooks was not
+    mock_repo_client.list_repositories.assert_called_once_with(load_all=True)
+    mock_repo_client.set_repository_webhooks.assert_not_called()

tests/codebase/management/commands/__init__.py

@@ -0,0 +1,134 @@
+from unittest.mock import Mock, patch
+
+from django.core.management import call_command
+
+import pytest
+from gitlab import GitlabGetError
+
+from codebase.clients import RepoClient
+from codebase.indexes import CodebaseIndex
+
+
+@pytest.fixture
+def mock_repo_client():
+    with patch.object(RepoClient, "create_instance") as mock:
+        client = Mock(spec=RepoClient)
+        mock.return_value = client
+        yield client
+
+
+@pytest.fixture
+def mock_indexer():
+    with patch("codebase.management.commands.update_index.CodebaseIndex") as mock:
+        indexer = Mock(spec=CodebaseIndex)
+        mock.return_value = indexer
+        yield indexer
+
+
+def test_update_index_all_repositories(mock_repo_client, mock_indexer):
+    """Test updating index for all repositories."""
+    # Mock repositories
+    mock_repo1 = Mock(slug="repo1")
+    mock_repo2 = Mock(slug="repo2")
+    mock_repo_client.list_repositories.return_value = [mock_repo1, mock_repo2]
+
+    # Call the command
+    call_command("update_index")
+
+    # Verify repositories were listed with correct parameters
+    mock_repo_client.list_repositories.assert_called_once_with(topics=None, load_all=True)
+
+    # Verify index was updated for each repository
+    assert mock_indexer.update.call_count == 2
+    mock_indexer.update.assert_any_call(repo_id="repo1", ref=None)
+    mock_indexer.update.assert_any_call(repo_id="repo2", ref=None)
+
+
+def test_update_index_specific_repository(mock_repo_client, mock_indexer):
+    """Test updating index for a specific repository."""
+    # Mock repository
+    mock_repo = Mock(slug="specific-repo")
+    mock_repo_client.get_repository.return_value = mock_repo
+
+    # Call the command
+    call_command("update_index", repo_id="specific-repo")
+
+    # Verify correct repository was fetched
+    mock_repo_client.get_repository.assert_called_once_with("specific-repo")
+
+    # Verify index was updated only for the specific repository
+    mock_indexer.update.assert_called_once_with(repo_id="specific-repo", ref=None)
+    mock_repo_client.list_repositories.assert_not_called()
+
+
+def test_update_index_with_topics(mock_repo_client, mock_indexer):
+    """Test updating index for repositories with specific topics."""
+    # Mock repositories
+    mock_repo1 = Mock(slug="repo1")
+    mock_repo2 = Mock(slug="repo2")
+    mock_repo_client.list_repositories.return_value = [mock_repo1, mock_repo2]
+
+    # Call the command with topics
+    call_command("update_index", topics=["python", "django"])
+
+    # Verify repositories were listed with correct topics
+    mock_repo_client.list_repositories.assert_called_once_with(topics=["python", "django"], load_all=True)
+
+    # Verify index was updated for each repository
+    assert mock_indexer.update.call_count == 2
+    mock_indexer.update.assert_any_call(repo_id="repo1", ref=None)
+    mock_indexer.update.assert_any_call(repo_id="repo2", ref=None)
+
+
+def test_update_index_with_ref(mock_repo_client, mock_indexer):
+    """Test updating index for a specific reference."""
+    # Mock repository
+    mock_repo = Mock(slug="repo1")
+    mock_repo_client.list_repositories.return_value = [mock_repo]
+
+    # Call the command with ref
+    call_command("update_index", ref="feature-branch")
+
+    # Verify index was updated with correct reference
+    mock_indexer.update.assert_called_once_with(repo_id="repo1", ref="feature-branch")
+
+
+def test_update_index_with_reset(mock_repo_client, mock_indexer):
+    """Test updating index with reset option."""
+    # Mock repository
+    mock_repo = Mock(slug="repo1")
+    mock_repo_client.list_repositories.return_value = [mock_repo]
+
+    # Call the command with reset
+    call_command("update_index", reset=True)
+
+    # Verify index was reset and then updated
+    mock_indexer.delete.assert_called_once_with(repo_id="repo1", ref=None)
+    mock_indexer.update.assert_called_once_with(repo_id="repo1", ref=None)
+
+
+def test_update_index_repository_not_found(mock_repo_client, mock_indexer):
+    """Test handling of non-existent repository."""
+    # Mock GitlabGetError for non-existent repository
+    mock_repo_client.get_repository.side_effect = GitlabGetError()
+
+    # Call the command with non-existent repository
+    call_command("update_index", repo_id="non-existent-repo")
+
+    # Verify error was handled gracefully
+    mock_repo_client.get_repository.assert_called_once_with("non-existent-repo")
+    mock_indexer.update.assert_not_called()
+    mock_indexer.delete.assert_not_called()
+
+
+def test_update_index_no_repositories(mock_repo_client, mock_indexer):
+    """Test behavior when no repositories are found."""
+    # Mock empty repository list
+    mock_repo_client.list_repositories.return_value = []
+
+    # Call the command
+    call_command("update_index")
+
+    # Verify no updates were performed
+    mock_indexer.update.assert_not_called()
+    mock_indexer.delete.assert_not_called()