chore: fix init command test

Signed-off-by: Claudio Matsuoka <claudio.matsuoka@canonical.com>
canonical · Apr 22, 2024 · cfda0e4 · cfda0e4
1 parent 7e2c483
commit cfda0e4
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/tests/integration/commands/test_init.py b/tests/integration/commands/test_init.py
@@ -46,16 +46,34 @@
         "pyproject.toml",
         "README.md",
         "requirements.txt",
+        "spread.yaml",
         "src",
         "src/charm.py",
         "tests",
         "tests/integration",
         "tests/integration/test_charm.py",
+        "tests/spread",
+        "tests/spread/lib",
+        "tests/spread/lib/cloud-init.yaml",
+        "tests/spread/lib/test-helpers.sh",
+        "tests/spread/general",
+        "tests/spread/general/integration",
+        "tests/spread/general/integration/task.yaml",
         "tests/unit",
         "tests/unit/test_charm.py",
         "tox.ini",
     )
 )
+BASIC_INIT_FILES_WITH_TOOLS = frozenset(
+    [p for p in BASIC_INIT_FILES]
+    + [
+        pathlib.Path(p)
+        for p in (
+            "tests/spread/lib/tools",
+            "tests/spread/lib/tools/retry",
+        )
+    ],
+)
 UNKNOWN_AUTHOR_REGEX = re.compile(
     r"^Unable to automatically determine author's name, specify it with --author$"
 )
@@ -96,9 +114,9 @@ def create_namespace(
 @pytest.mark.parametrize(
     ("profile", "expected_files"),
     [
-        pytest.param("simple", BASIC_INIT_FILES, id="simple"),
+        pytest.param("simple", BASIC_INIT_FILES_WITH_TOOLS, id="simple"),
         pytest.param("machine", BASIC_INIT_FILES, id="machine"),
-        pytest.param("kubernetes", BASIC_INIT_FILES, id="kubernetes"),
+        pytest.param("kubernetes", BASIC_INIT_FILES_WITH_TOOLS, id="kubernetes"),
     ],
 )
 @pytest.mark.parametrize("charm_name", ["my-charm", "charm123"])