feat: add repository provider as a part of the testing fixture options and add some basic tests to ensure files are removed properly in the post gen hook

Author: 56kyle

tests/conftest.py

@@ -3,6 +3,7 @@
 import subprocess
 from pathlib import Path
 from typing import Any
+from typing import Literal
 
 import pytest
 import toml
@@ -84,10 +85,15 @@ def robust_demo__name(robust_demo__add_rust_extension: str, robust_demo__is_setu
 
 
 @pytest.fixture(scope="session")
-def robust_demo__extra_context(robust_demo__name: str, robust_demo__add_rust_extension: bool) -> dict[str, Any]:
+def robust_demo__extra_context(
+    robust_demo__name: str,
+    robust_demo__add_rust_extension: bool,
+    robust_demo__repository_provider: Literal["github", "gitlab", "bitbucket"]
+) -> dict[str, Any]:
     return {
         "project_name": robust_demo__name,
-        "add_rust_extension": robust_demo__add_rust_extension
+        "add_rust_extension": robust_demo__add_rust_extension,
+        "repository_provider": robust_demo__repository_provider,
     }
 
 
@@ -96,6 +102,11 @@ def robust_demo__add_rust_extension(request: FixtureRequest) -> bool:
     return getattr(request, "param", False)
 
 
+@pytest.fixture(scope="session")
+def robust_demo__repository_provider(request: FixtureRequest) -> Literal["github", "gitlab", "bitbucket"]:
+    return getattr(request, "param", "github")
+
+
 @pytest.fixture(scope="session")
 def robust_demo__is_setup(request: FixtureRequest) -> bool:
     return getattr(request, "param", True)

tests/unit_tests/test_post_gen_hook.py

@@ -0,0 +1,70 @@
+from pathlib import Path
+
+import pytest
+
+
+@pytest.mark.parametrize(
+    argnames="removed_relative_path",
+    argvalues=[".gitlab-ci.yml", "bitbucket-pipelines.yml"]
+)
+@pytest.mark.parametrize(
+    argnames="robust_demo__repository_provider",
+    argvalues=["github"],
+    indirect=True
+)
+def test_files_removed_for_github(robust_demo: Path, removed_relative_path: str) -> None:
+    path: Path = robust_demo / removed_relative_path
+    assert not path.exists()
+
+
+@pytest.mark.parametrize(
+    argnames="removed_relative_path",
+    argvalues=[".github", "bitbucket-pipelines.yml"]
+)
+@pytest.mark.parametrize(
+    argnames="robust_demo__repository_provider",
+    argvalues=["gitlab"],
+    indirect=True
+)
+def test_files_removed_for_gitlab(robust_demo: Path, removed_relative_path: str) -> None:
+    path: Path = robust_demo / removed_relative_path
+    assert not path.exists()
+
+
+@pytest.mark.parametrize(
+    argnames="removed_relative_path",
+    argvalues=[".github", ".gitlab-ci.yml"]
+)
+@pytest.mark.parametrize(
+    argnames="robust_demo__repository_provider",
+    argvalues=["bitbucket"],
+    indirect=True
+)
+def test_files_removed_for_bitbucket(robust_demo: Path, removed_relative_path: str) -> None:
+    path: Path = robust_demo / removed_relative_path
+    assert not path.exists()
+
+
+@pytest.mark.parametrize(
+    argnames="removed_relative_path",
+    argvalues=[
+        "rust",
+        ".github/workflows/lint-rust.yml",
+        ".github/workflows/build-rust.yml",
+        ".github/workflows/test-rust.yml"
+    ]
+)
+@pytest.mark.parametrize(
+    argnames="robust_demo__add_rust_extension",
+    argvalues=[False],
+    indirect=True,
+    ids=["no-rust"]
+)
+@pytest.mark.parametrize(
+    argnames="robust_demo__repository_provider",
+    argvalues=["github"],
+    indirect=True,
+)
+def test_files_removed_for_no_rust_extension(robust_demo: Path, removed_relative_path: str) -> None:
+    path: Path = robust_demo / removed_relative_path
+    assert not path.exists()