[SILO-700] chore: improve tests for module and cycles feature check (#10)

Author: Saurabhkmr98

plane/models/projects.py

@@ -140,10 +140,10 @@ class ProjectFeature(BaseModel):
 
   model_config = ConfigDict(extra="allow", populate_by_name=True)
 
-  epics: bool
-  modules: bool
-  cycles: bool
-  views: bool
-  pages: bool
-  intakes: bool
-  work_item_types: bool
+  epics: bool | None = None
+  modules: bool | None = None
+  cycles: bool | None = None
+  views: bool | None = None
+  pages: bool | None = None
+  intakes: bool | None = None
+  work_item_types: bool | None = None

tests/unit/test_cycles.py

@@ -6,7 +6,7 @@
 
 from plane.client import PlaneClient
 from plane.models.cycles import CreateCycle, UpdateCycle
-from plane.models.projects import Project
+from plane.models.projects import Project, ProjectFeature
 
 
 class TestCyclesAPI:
@@ -60,6 +60,7 @@ def cycle(
         cycle_data: CreateCycle,
     ):
         """Create a test cycle and yield it, then delete it."""
+        client.projects.update_features(workspace_slug, project.id, ProjectFeature(cycles=True))
         cycle = client.cycles.create(workspace_slug, project.id, cycle_data)
         yield cycle
         try:
@@ -75,6 +76,7 @@ def test_create_cycle(
         cycle_data: CreateCycle,
     ) -> None:
         """Test creating a cycle."""
+        client.projects.update_features(workspace_slug, project.id, ProjectFeature(cycles=True))
         cycle = client.cycles.create(workspace_slug, project.id, cycle_data)
         assert cycle is not None
         assert cycle.id is not None

tests/unit/test_intake.py

@@ -4,7 +4,7 @@
 
 from plane.client import PlaneClient
 from plane.models.intake import CreateIntakeWorkItem
-from plane.models.projects import Project
+from plane.models.projects import Project, ProjectFeature
 
 
 class TestIntakeAPI:

tests/unit/test_modules.py

@@ -7,7 +7,7 @@
 from plane.client import PlaneClient
 from plane.models.enums import ModuleStatus
 from plane.models.modules import CreateModule, UpdateModule
-from plane.models.projects import Project
+from plane.models.projects import Project, ProjectFeature
 
 
 class TestModulesAPI:
@@ -61,6 +61,7 @@ def module(
         module_data: CreateModule,
     ):
         """Create a test module and yield it, then delete it."""
+        client.projects.update_features(workspace_slug, project.id, ProjectFeature(modules=True))
         module = client.modules.create(workspace_slug, project.id, module_data)
         yield module
         try:
@@ -76,6 +77,7 @@ def test_create_module(
         module_data: CreateModule,
     ) -> None:
         """Test creating a module."""
+        client.projects.update_features(workspace_slug, project.id, ProjectFeature(modules=True))
         module = client.modules.create(workspace_slug, project.id, module_data)
         assert module is not None
         assert module.id is not None