diff --git a/src/codegen/git/schemas/enums.py b/src/codegen/git/schemas/enums.py index 983f0f2d8..8219e97b4 100644 --- a/src/codegen/git/schemas/enums.py +++ b/src/codegen/git/schemas/enums.py @@ -27,3 +27,9 @@ class DiffChangeType(StrEnum): DELETED = "D" RENAMED = "R" MODIFIED = "M" + + +class RepoVisibility(StrEnum): + PRIVATE = auto() + PUBLIC = auto() + INTERNAL = auto() diff --git a/src/codegen/git/schemas/repo_config.py b/src/codegen/git/schemas/repo_config.py index b43de3954..8f9b2ce2e 100644 --- a/src/codegen/git/schemas/repo_config.py +++ b/src/codegen/git/schemas/repo_config.py @@ -3,6 +3,8 @@ from pydantic import BaseModel +from codegen.git.schemas.enums import RepoVisibility + logger = logging.getLogger(__name__) @@ -19,6 +21,7 @@ class RepoConfig(BaseModel): id: int name: str full_name: str + visibility: RepoVisibility | None = None # Org fields organization_id: int diff --git a/tests/integration/codegen/git/conftest.py b/tests/integration/codegen/git/conftest.py index ef06df1c3..7158095f0 100644 --- a/tests/integration/codegen/git/conftest.py +++ b/tests/integration/codegen/git/conftest.py @@ -18,8 +18,14 @@ def mock_config(): @pytest.fixture(autouse=True) -def repo_config() -> RepoConfig: +def repo_config(): with patch("codegen.git.utils.clone.get_authenticated_clone_url_for_repo_config") as mock_clone_url: mock_clone_url.return_value = "https://github.com/codegen-sh/Kevin-s-Adventure-Game.git" - repo_config = RepoConfig(id=321, name="Kevin-s-Adventure-Game", full_name="codegen-sh/Kevin-s-Adventure-Game", organization_id="123", organization_name="codegen-sh") + repo_config = RepoConfig( + id=321, + name="Kevin-s-Adventure-Game", + full_name="codegen-sh/Kevin-s-Adventure-Game", + organization_id="123", + organization_name="codegen-sh", + ) yield repo_config