python-github-backup/tests/test_all_starred.py at 1d6d474408968f728b11aa50c55ec9bb7ddf068e · Iamrodos/python-github-backup · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
"""Tests for --all-starred flag behavior (issue #225)."""

import pytest
from unittest.mock import Mock, patch

from github_backup import github_backup


class TestAllStarredCloning:
    """Test suite for --all-starred repository cloning behavior.

    Issue #225: --all-starred should clone starred repos without requiring --repositories.
    """

    def _create_mock_args(self, **overrides):
        """Create a mock args object with sensible defaults."""
        args = Mock()
        args.user = "testuser"
        args.output_directory = "/tmp/backup"
        args.include_repository = False
        args.include_everything = False
        args.include_gists = False
        args.include_starred_gists = False
        args.all_starred = False
        args.skip_existing = False
        args.bare_clone = False
        args.lfs_clone = False
        args.no_prune = False
        args.include_wiki = False
        args.include_issues = False
        args.include_issue_comments = False
        args.include_issue_events = False
        args.include_pulls = False
        args.include_pull_comments = False
        args.include_pull_commits = False
        args.include_pull_details = False
        args.include_labels = False
        args.include_hooks = False
        args.include_milestones = False
        args.include_releases = False
        args.include_assets = False
        args.include_attachments = False
        args.incremental = False
        args.incremental_by_files = False
        args.github_host = None
        args.prefer_ssh = False
        args.token_classic = None
        args.token_fine = None
        args.username = None
        args.password = None
        args.as_app = False
        args.osx_keychain_item_name = None
        args.osx_keychain_item_account = None

        for key, value in overrides.items():
            setattr(args, key, value)

        return args

    @patch('github_backup.github_backup.fetch_repository')
    @patch('github_backup.github_backup.get_github_repo_url')
    def test_all_starred_clones_without_repositories_flag(self, mock_get_url, mock_fetch):
        """--all-starred should clone starred repos without --repositories flag.

        This is the core fix for issue #225.
        """
        args = self._create_mock_args(all_starred=True)
        mock_get_url.return_value = "https://github.com/otheruser/awesome-project.git"

        # A starred repository (is_starred flag set by retrieve_repositories)
        starred_repo = {
            "name": "awesome-project",
            "full_name": "otheruser/awesome-project",
            "owner": {"login": "otheruser"},
            "private": False,
            "fork": False,
            "has_wiki": False,
            "is_starred": True,  # This flag is set for starred repos
        }

        with patch('github_backup.github_backup.mkdir_p'):
            github_backup.backup_repositories(args, "/tmp/backup", [starred_repo])

        # fetch_repository should be called for the starred repo
        assert mock_fetch.called, "--all-starred should trigger repository cloning"
        mock_fetch.assert_called_once()
        call_args = mock_fetch.call_args
        assert call_args[0][0] == "awesome-project"  # repo name

    @patch('github_backup.github_backup.fetch_repository')
    @patch('github_backup.github_backup.get_github_repo_url')
    def test_starred_repo_not_cloned_without_all_starred_flag(self, mock_get_url, mock_fetch):
        """Starred repos should NOT be cloned if --all-starred is not set."""
        args = self._create_mock_args(all_starred=False)
        mock_get_url.return_value = "https://github.com/otheruser/awesome-project.git"

        starred_repo = {
            "name": "awesome-project",
            "full_name": "otheruser/awesome-project",
            "owner": {"login": "otheruser"},
            "private": False,
            "fork": False,
            "has_wiki": False,
            "is_starred": True,
        }

        with patch('github_backup.github_backup.mkdir_p'):
            github_backup.backup_repositories(args, "/tmp/backup", [starred_repo])

        # fetch_repository should NOT be called
        assert not mock_fetch.called, "Starred repos should not be cloned without --all-starred"

    @patch('github_backup.github_backup.fetch_repository')
    @patch('github_backup.github_backup.get_github_repo_url')
    def test_non_starred_repo_not_cloned_with_only_all_starred(self, mock_get_url, mock_fetch):
        """Non-starred repos should NOT be cloned when only --all-starred is set."""
        args = self._create_mock_args(all_starred=True)
        mock_get_url.return_value = "https://github.com/testuser/my-project.git"

        # A regular (non-starred) repository
        regular_repo = {
            "name": "my-project",
            "full_name": "testuser/my-project",
            "owner": {"login": "testuser"},
            "private": False,
            "fork": False,
            "has_wiki": False,
            # No is_starred flag
        }

        with patch('github_backup.github_backup.mkdir_p'):
            github_backup.backup_repositories(args, "/tmp/backup", [regular_repo])

        # fetch_repository should NOT be called for non-starred repos
        assert not mock_fetch.called, "Non-starred repos should not be cloned with only --all-starred"

    @patch('github_backup.github_backup.fetch_repository')
    @patch('github_backup.github_backup.get_github_repo_url')
    def test_repositories_flag_still_works(self, mock_get_url, mock_fetch):
        """--repositories flag should still clone repos as before."""
        args = self._create_mock_args(include_repository=True)
        mock_get_url.return_value = "https://github.com/testuser/my-project.git"

        regular_repo = {
            "name": "my-project",
            "full_name": "testuser/my-project",
            "owner": {"login": "testuser"},
            "private": False,
            "fork": False,
            "has_wiki": False,
        }

        with patch('github_backup.github_backup.mkdir_p'):
            github_backup.backup_repositories(args, "/tmp/backup", [regular_repo])

        # fetch_repository should be called
        assert mock_fetch.called, "--repositories should trigger repository cloning"


if __name__ == "__main__":
    pytest.main([__file__, "-v"])