feat: add tests for github_update_external_repositories

Author: rudransh-shrivastava

backend/tests/apps/github/management/commands/github_update_external_repositories_test.py

@@ -0,0 +1,122 @@
+from unittest import mock
+
+import pytest
+
+from apps.github.management.commands.github_update_external_repositories import (
+    GITHUB_ITEMS_PER_PAGE,
+    Command,
+    Organization,
+    Project
+)
+
+@pytest.fixture
+def command():
+    return Command()
+
+@pytest.fixture
+def mock_gh_repository():
+    repo = mock.Mock()
+    repo.name = "test-repo"
+    repo.html_url = "https://github.com/TestOrg/test-repo"
+    return repo
+
+@pytest.mark.parametrize(
+    "num_orgs, num_repos_per_org, expected_sync_calls, expected_bulk_save_calls",
+    [
+        (1, [2], 2, 1),    # 1 org with 2 repos
+        (2, [1, 3], 4, 2),  # 2 orgs with 1 and 3 repos
+        (2, [0, 2], 2, 1),  # 1 org with repos, 1 without
+    ]
+)
+@mock.patch.dict("os.environ", {"GITHUB_TOKEN": "valid-token"})
+@mock.patch("apps.github.management.commands.github_update_external_repositories.github.Github")
+@mock.patch("apps.github.management.commands.github_update_external_repositories.sync_repository")
+@mock.patch("apps.github.management.commands.github_update_external_repositories.Project.bulk_save")
+@mock.patch("apps.github.management.commands.github_update_external_repositories.Organization.objects.filter")
+def test_handle_success(
+    mock_org_filter,
+    mock_bulk_save,
+    mock_sync_repository,
+    mock_github,
+    command,
+    mock_gh_repository,
+    num_orgs,
+    num_repos_per_org,
+    expected_sync_calls,
+    expected_bulk_save_calls
+):
+    """Test command execution with varying organizations and repositories."""
+    mock_gh = mock_github.return_value
+
+    orgs = []
+    gh_orgs = []
+    for i in range(num_orgs):
+        org = mock.Mock(spec=Organization)
+        org.login = f"TestOrg{i+1}"
+        org.is_owasp_related_organization = True
+        orgs.append(org)
+
+        gh_org = mock.Mock()
+        gh_repos = mock.MagicMock()
+        gh_repos.totalCount = num_repos_per_org[i]
+        gh_repos.__iter__.return_value = [mock_gh_repository] * num_repos_per_org[i]
+        gh_org.get_repos.return_value = gh_repos
+        gh_orgs.append(gh_org)
+
+    qs_mock = mock.MagicMock()
+    qs_mock.count.return_value = num_orgs
+    qs_mock.__iter__.return_value = orgs
+    mock_org_filter.return_value.exclude.return_value = qs_mock
+
+    mock_gh.get_organization.side_effect = gh_orgs
+
+    mock_project = mock.Mock(spec=Project)
+    mock_sync_repository.side_effect = lambda gh_repo: (orgs[0], mock.Mock(project=mock_project))
+
+    with mock.patch("builtins.print"):
+        command.handle()
+
+    mock_github.assert_called_once_with("valid-token", per_page=GITHUB_ITEMS_PER_PAGE)
+
+    assert mock_gh.get_organization.call_count == num_orgs
+    for i, org in enumerate(orgs):
+        mock_gh.get_organization.assert_any_call(org.login)
+
+    assert mock_sync_repository.call_count == expected_sync_calls
+
+    assert mock_bulk_save.call_count == expected_bulk_save_calls
+
+@pytest.mark.parametrize(
+    "num_repos, expected_project_count",
+    [
+        (1, 1),  # 1 repo with  1 project
+        (3, 3),  # 3 repos with 3 projects
+        (2, 0),  # 2 repos without 0 projects
+    ]
+)
+@mock.patch("apps.github.management.commands.github_update_external_repositories.sync_repository")
+def test_sync_organization_repositories(
+    mock_sync_repository,
+    command,
+    mock_gh_repository,
+    num_repos,
+    expected_project_count
+):
+    """Test repository synchronization with varying repository counts."""
+    mock_organization = mock.Mock(spec=Organization)
+    mock_organization.login = "TestOrg"
+
+    mock_repositories = mock.MagicMock()
+    mock_repositories.totalCount = num_repos
+    mock_repositories.__iter__.return_value = [mock_gh_repository] * num_repos
+
+    mock_project = mock.Mock(spec=Project)
+    if expected_project_count > 0:
+        mock_sync_repository.return_value = (mock_organization, mock.Mock(project=mock_project))
+    else:
+        mock_sync_repository.return_value = (mock_organization, mock.Mock(project=None))
+
+    projects = command.sync_organization_repositories(mock_organization, mock_repositories)
+
+    assert len(projects) == expected_project_count
+    assert mock_sync_repository.call_count == num_repos