test: add codebase.create_pr tests (#422)

Author: christinewangcw

src/codegen/git/repo_operator/local_repo_operator.py

@@ -15,7 +15,6 @@
 from codegen.git.schemas.repo_config import RepoConfig
 from codegen.git.utils.clone_url import url_to_github
 from codegen.git.utils.file_utils import create_files
-from codegen.shared.configs.config import config
 
 logger = logging.getLogger(__name__)
 
@@ -41,7 +40,7 @@ def __init__(
         github_api_key: str | None = None,
         bot_commit: bool = False,
     ) -> None:
-        self._github_api_key = github_api_key or config.secrets.github_token
+        self._github_api_key = github_api_key
         self._remote_git_repo = None
         super().__init__(repo_config, bot_commit)
         os.makedirs(self.repo_path, exist_ok=True)

tests/integration/codegen/git/codebase/conftest.py

@@ -0,0 +1,37 @@
+import os
+
+import pytest
+from git import Repo as GitRepo
+
+from codegen.git.repo_operator.local_repo_operator import LocalRepoOperator
+from codegen.git.schemas.repo_config import RepoConfig
+from codegen.git.utils.clone_url import get_authenticated_clone_url_for_repo_config
+from codegen.sdk.codebase.config import ProjectConfig
+from codegen.sdk.core.codebase import Codebase
+from codegen.shared.configs.config import config
+
+
+@pytest.fixture
+def repo_config(tmpdir):
+    repo_config = RepoConfig(
+        name="Kevin-s-Adventure-Game",
+        full_name="codegen-sh/Kevin-s-Adventure-Game",
+        organization_name="codegen-sh",
+        base_dir=str(tmpdir),
+    )
+    yield repo_config
+
+
+@pytest.fixture
+def op(repo_config):
+    os.chdir(repo_config.base_dir)
+    GitRepo.clone_from(url=get_authenticated_clone_url_for_repo_config(repo_config, token=config.secrets.github_token), to_path=os.path.join(repo_config.base_dir, repo_config.name), depth=1)
+    op = LocalRepoOperator(repo_config=repo_config, github_api_key=config.secrets.github_token)
+    yield op
+
+
+@pytest.fixture
+def codebase(op: LocalRepoOperator):
+    project_config = ProjectConfig(repo_operator=op)
+    codebase = Codebase(projects=[project_config])
+    yield codebase

tests/integration/codegen/git/codebase/test_codebase_create_pr.py

@@ -0,0 +1,26 @@
+import uuid
+
+import pytest
+
+from codegen.sdk.core.codebase import Codebase
+
+
+def test_codebase_create_pr_active_branch(codebase: Codebase):
+    head = f"test-create-pr-{uuid.uuid4()}"
+    codebase.checkout(branch=head, create_if_missing=True)
+    codebase.files[0].remove()
+    codebase.commit()
+    pr = codebase.create_pr(title="test-create-pr title", body="test-create-pr body")
+    assert pr.title == "test-create-pr title"
+    assert pr.body == "test-create-pr body"
+    assert pr.draft is False
+    assert pr.state == "open"
+    assert pr.head.ref == head
+    assert pr.base.ref == "main"
+
+
+def test_codebase_create_pr_detached_head(codebase: Codebase):
+    codebase.checkout(commit=codebase._op.git_cli.head.commit)  # move to detached head state
+    with pytest.raises(ValueError) as exc_info:
+        codebase.create_pr(title="test-create-pr title", body="test-create-pr body")
+    assert "Cannot make a PR from a detached HEAD" in str(exc_info.value)

tests/integration/codegen/git/conftest.py

@@ -5,7 +5,7 @@
 from codegen.git.schemas.repo_config import RepoConfig
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture()
 def mock_config():
     """Mock Config instance to prevent actual environment variable access during tests."""
     mock_config = MagicMock()
@@ -14,7 +14,7 @@ def mock_config():
     yield mock_config
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture()
 def repo_config(tmpdir):
     repo_config = RepoConfig(
         name="Kevin-s-Adventure-Game",

tests/integration/codegen/git/repo_operator/test_remote_repo_operator.py

@@ -12,11 +12,7 @@
 
 @pytest.fixture
 def op(repo_config, request):
-    op = RemoteRepoOperator(
-        repo_config,
-        shallow=request.param if hasattr(request, "param") else True,
-        bot_commit=False,
-    )
+    op = RemoteRepoOperator(repo_config, shallow=request.param if hasattr(request, "param") else True, bot_commit=False)
     yield op
 
 

tests/integration/codegen/runner/conftest.py

@@ -22,7 +22,7 @@ def get_free_port():
         return port
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture()
 def repo_config(tmpdir) -> Generator[RepoConfig, None, None]:
     yield RepoConfig(
         name="Kevin-s-Adventure-Game",