Organize tests (#60)

# Motivation
<!-- Why is this change necessary? -->

# Content
<!-- Please include a summary of the change -->
# Testing
<!-- How was the change tested? -->
# Please check the following before marking your PR as ready for review

- [x] I have added tests for my changes
- [x] I have updated the documentation or added new documentation as
needed
- [x] I have read and agree to the [Contributor License
Agreement](../CLA.md)

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(

src/codemods/canonical/classnames_to_backtick.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/convert_array_type_to_square_bracket/convert_array_type_to_square_bracket.py

@@ -4,7 +4,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/convert_attribute_to_decorator/convert_attribute_to_decorator.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/convert_comments_to_JSDoc_style/convert_comments_to_JSDoc_style.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/convert_docstring_to_google_style/convert_docstring_to_google_style.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/delete_unused_functions/delete_unused_functions.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/emojify_py_files_codemod/emojify_py_files_codemod.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/enum_mover/enum_mover.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/insert_arguments_to_decorator/insert_arguments_to_decorator.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/invite_factory_create_params/invite_factory_create_params.py

@@ -4,7 +4,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/js_to_esm_codemod/js_to_esm_codemod.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/mark_as_internal_codemod/mark_as_internal_codemod.py

@@ -5,7 +5,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/mark_internal_to_module/mark_internal_to_module.py

@@ -4,7 +4,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/mark_is_boolean/mark_is_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/migrate_class_attributes/migrate_class_attributes.py

@@ -6,7 +6,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
 
 logger = logging.getLogger(__name__)
 

src/codemods/canonical/move_enums_codemod/move_enums_codemod.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/move_functions_to_new_file/move_functions_to_new_file.py

@@ -4,7 +4,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/openapi_add_response_none/openapi_add_response_none.py

@@ -5,7 +5,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/openapi_no_reference_request/openapi_no_reference_request.py

@@ -5,7 +5,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/pascal_case_symbols/pascal_case_symbols.py

@@ -6,7 +6,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/pivot_return_types/pivot_return_types.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/refactor_react_components_into_separate_files/refactor_react_components_into_separate_files.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/remove_indirect_imports/remove_indirect_imports.py

@@ -6,7 +6,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/rename_function_parameters/rename_function_parameters.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/rename_local_variables/rename_local_variables.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/replace_prop_values/replace_prop_values.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/return_none_type_annotation/return_none_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/split_decorators/split_decorators.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/split_file/split_file.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/split_file_and_rename_symbols/split_file_and_rename_symbols.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/split_large_files/split_large_files.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/swap_call_site_imports/swap_call_site_imports.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/swap_class_attribute_usages/swap_class_attribute_usages.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(