Skip to content

Commit 2bd4c3c

Browse files
chore: fix repo operator unit tests
1 parent 367420e commit 2bd4c3c

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

src/codegen/git/clients/github_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ class GithubClient:
2525
@classmethod
2626
def from_repo_config(cls, repo_config: RepoConfig) -> Self:
2727
gh_wrapper = cls()
28-
gh_wrapper.read_client = gh_wrapper._create_client_for_repo(repo_config)
28+
gh_wrapper.read_client = gh_wrapper._create_client_for_repo(repo_config) # TODO: the read/write separation isn't necessary since it's just going to use the lowside vs the highside token
2929
gh_wrapper._write_client = gh_wrapper._create_client_for_repo(repo_config)
3030
return gh_wrapper
3131

@@ -37,7 +37,7 @@ def from_token(cls, token: str | None = None) -> Self:
3737
gh_wrapper._write_client = Github(token, base_url=cls.base_url)
3838
return gh_wrapper
3939

40-
def _create_client_for_repo(self, repo_config: RepoConfig) -> Github:
40+
def _create_client_for_repo(self) -> Github:
4141
token = get_token_for_repo_config(github_type=self.type)
4242
return Github(token, base_url=self.base_url)
4343

src/codegen/git/configs/token.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,11 @@
22

33
from codegen.git.configs.config import config
44
from codegen.git.schemas.github import GithubType
5-
from codegen.git.schemas.repo_config import RepoConfig
65

76
logger = logging.getLogger(__name__)
87

98

10-
def get_token_for_repo_config(repo_config: RepoConfig, github_type: GithubType = GithubType.GithubEnterprise) -> str:
9+
def get_token_for_repo_config(github_type: GithubType = GithubType.GithubEnterprise) -> str:
1110
# TODO: implement config such that we can retrieve tokens for different repos
1211
if github_type == GithubType.GithubEnterprise:
1312
return config.LOWSIDE_TOKEN

src/codegen/git/utils/clone_url.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def get_authenticated_clone_url_for_repo_config(
2424
github_type: GithubType = GithubType.GithubEnterprise,
2525
) -> str:
2626
git_url = get_clone_url_for_repo_config(repo, github_type)
27-
token = get_token_for_repo_config(repo_config=repo, github_type=github_type)
27+
token = get_token_for_repo_config(github_type=github_type)
2828
return f"https://x-access-token:{token}@{git_url}"
2929

3030

0 commit comments

Comments
 (0)