Organize tests directory

Author: caroljung-cg

src/codegen/git/configs/token.py

@@ -2,11 +2,13 @@
 
 from codegen.git.configs.config import config
 from codegen.git.schemas.github import GithubType
+from codegen.git.schemas.repo_config import RepoConfig
 
 logger = logging.getLogger(__name__)
 
 
-def get_token_for_repo_config(github_type: GithubType = GithubType.GithubEnterprise) -> str:
+def get_token_for_repo_config(repo_config: RepoConfig, github_type: GithubType = GithubType.GithubEnterprise) -> str:
+    # TODO: implement config such that we can retrieve tokens for different repos
     if github_type == GithubType.GithubEnterprise:
         return config.LOWSIDE_TOKEN
     elif github_type == GithubType.Github:

src/codegen/git/utils/clone_url.py

@@ -24,7 +24,7 @@ def get_authenticated_clone_url_for_repo_config(
     github_type: GithubType = GithubType.GithubEnterprise,
 ) -> str:
     git_url = get_clone_url_for_repo_config(repo, github_type)
-    token = get_token_for_repo_config(github_type=github_type)
+    token = get_token_for_repo_config(repo_config=repo, github_type=github_type)
     return f"https://x-access-token:{token}@{git_url}"
 
 

src/codegen/sdk/code_generation/doc_utils/skills.py

@@ -10,7 +10,7 @@
 from codegen.sdk.skills.core.skill import Skill
 from codegen.sdk.skills.core.skill_implementation import SkillImplementation
 from codegen.sdk.skills.core.utils import get_all_evaluation_skills
-from codemods.canonical.codemod import Codemod
+from codemods.codemod import Codemod
 
 
 def remove_leading_tab_or_spaces(text: str) -> str:

src/codegen/sdk/testing/models.py

@@ -29,7 +29,7 @@
 except (FileNotFoundError, json.decoder.JSONDecodeError):
     REPO_ID_TO_URL = {}
 if TYPE_CHECKING:
-    from codemods.canonical.codemod import Codemod
+    from codemods.codemod import Codemod
 
 
 @unique

src/codegen/sdk/testing/test_discovery.py

@@ -14,7 +14,7 @@
 from codegen.sdk.testing.constants import DIFF_FILEPATH
 from codegen.sdk.testing.models import BASE_PATH, CODEMOD_PATH, REPO_ID_TO_URL, TEST_DIR, VERIFIED_CODEMOD_DATA_DIR, VERIFIED_CODEMOD_DIFFS, ClonedRepoTestCase, CodemodMetadata, Repo, Size
 from codegen.sdk.testing.verified_codemod_utils import CodemodAPI, RepoCodemodMetadata, SkillTestConfig, anonymize_id
-from codemods.canonical.codemod import Codemod
+from codemods.codemod import Codemod
 
 
 def find_repos(

src/codemods/canonical/add_function_parameter_type_annotations/add_function_parameter_type_annotations.py

@@ -3,7 +3,7 @@
 from codegen.sdk.skills.core.skill import Skill
 from codegen.sdk.skills.core.utils import skill, skill_impl
 from codegen.sdk.writer_decorators import canonical
-from codemods.canonical.codemod import Codemod
+from codemods.codemod import Codemod
 
 
 @skill(

src/codemods/canonical/add_internal_to_non_exported_components/add_internal_to_non_exported_components.py

@@ -3,7 +3,7 @@
 from codegen.sdk.skills.core.skill import Skill
 from codegen.sdk.skills.core.utils import skill, skill_impl
 from codegen.sdk.writer_decorators import canonical
-from codemods.canonical.codemod import Codemod
+from codemods.codemod import Codemod
 
 
 @skill(

src/codemods/canonical/bang_bang_to_boolean/bang_bang_to_boolean.py

@@ -3,7 +3,7 @@
 from codegen.sdk.skills.core.skill import Skill
 from codegen.sdk.skills.core.utils import skill, skill_impl
 from codegen.sdk.writer_decorators import canonical
-from codemods.canonical.codemod import Codemod
+from codemods.codemod import Codemod
 
 
 @skill(

src/codemods/canonical/built_in_type_annotation/built_in_type_annotation.py

@@ -3,7 +3,7 @@
 from codegen.sdk.skills.core.skill import Skill
 from codegen.sdk.skills.core.utils import skill, skill_impl
 from codegen.sdk.writer_decorators import canonical
-from codemods.canonical.codemod import Codemod
+from codemods.codemod import Codemod
 
 
 @skill(

src/codemods/canonical/change_component_tag_names/change_component_tag_names.py

@@ -3,7 +3,7 @@
 from codegen.sdk.skills.core.skill import Skill
 from codegen.sdk.skills.core.utils import skill, skill_impl
 from codegen.sdk.writer_decorators import canonical
-from codemods.canonical.codemod import Codemod
+from codemods.codemod import Codemod
 
 
 @skill(