feat: support standard and kernel-modules

Signed-off-by: Callahan Kovacs <[email protected]>
canonical · Oct 7, 2024 · b97e356 · b97e356
1 parent df6ae07
commit b97e356
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/snapcraft/models/project.py b/snapcraft/models/project.py
@@ -572,7 +572,7 @@ class Component(models.CraftBaseModel):
 
     summary: SummaryStr
     description: str
-    type: Literal["test"]
+    type: Literal["test", "kernel-modules", "standard"]
     version: VersionStr | None = None
     hooks: dict[str, Hook] | None = None
 

diff --git a/tests/unit/models/test_projects.py b/tests/unit/models/test_projects.py
@@ -2600,16 +2600,17 @@ def test_components_valid(self, project, project_yaml_data, stub_component_data)
 
         assert test_project.components == components
 
+    @pytest.mark.parametrize("component_type", ["test", "kernel-modules", "standard"])
     def test_component_type_valid(
-        self, project, project_yaml_data, stub_component_data
+        self, component_type, project, project_yaml_data, stub_component_data
     ):
         component = {"foo": stub_component_data}
-        component["foo"]["type"] = "test"
+        component["foo"]["type"] = component_type
 
         test_project = project.unmarshal(project_yaml_data(components=component))
 
         assert test_project.components
-        assert test_project.components["foo"].type == "test"
+        assert test_project.components["foo"].type == component_type
 
     def test_component_type_invalid(
         self, project, project_yaml_data, stub_component_data