Mock PyGithub in unit tests, rename API token env var

Author: smkent

cookie_python/manage/github.py

@@ -3,14 +3,14 @@
 from functools import cached_property, lru_cache
 from typing import Optional
 
-from github import Github
+import github
 from github.PullRequest import PullRequest
 from github.Repository import Repository
 
 
 class GithubRepo:
     def __init__(self) -> None:
-        self._gh = Github(os.environ["GITHUB_ACCESS_TOKEN"])
+        self._gh = github.Github(os.environ["GITHUB_API_TOKEN"])
 
     @cached_property
     def username(self) -> str:

tests/test_manage_cookie.py

@@ -6,8 +6,9 @@
 from pathlib import Path
 from types import SimpleNamespace
 from typing import Any, Dict, Iterable, Iterator, Tuple
-from unittest.mock import patch
+from unittest.mock import MagicMock, patch
 
+import github
 import pytest
 
 from cookie_python.manage.main import main as manage_cookie_main
@@ -27,6 +28,7 @@ def environ() -> Iterator[None]:
             GIT_AUTHOR_EMAIL=AUTHOR_EMAIL,
             GIT_COMMITTER_NAME=AUTHOR_NAME,
             GIT_COMMITTER_EMAIL=AUTHOR_EMAIL,
+            GITHUB_API_TOKEN="unittest_token",
         )
     )
     with patch.dict(os.environ, env):
@@ -62,6 +64,25 @@ def new_cookie(
     yield Path(temp_dir) / PROJECT_NAME
 
 
+@pytest.fixture(autouse=True)
+def mock_pygithub(new_cookie: Path) -> Iterator[MagicMock]:
+    with patch.object(github, "Github") as obj:
+        gh = obj.return_value
+        gh.get_repo.return_value = SimpleNamespace(
+            name=PROJECT_NAME,
+            full_name=f"{AUTHOR_NAME}/{PROJECT_NAME}",
+            ssh_url=str(new_cookie),
+            get_pulls=MagicMock(
+                return_value=[
+                    SimpleNamespace(
+                        url="https://unittest.example.com/repo/pulls/1138"
+                    )
+                ],
+            ),
+        )
+        yield obj
+
+
 @pytest.fixture
 def new_cookie_with_lock(new_cookie: Path, temp_dir: str) -> Iterator[Path]:
     for cmd in (