allow for @intersect_status to be sufficient for building a capability

Signed-off-by: Lance Drane <[email protected]>

Author: Lance-Drane

src/intersect_sdk/_internal/schema.py

@@ -143,9 +143,9 @@ def _get_functions(
         logger.warning(
             f"Class '{capability.__name__}' has no function annotated with the @intersect_status() decorator. No status information will be provided when sending status messages."
         )
-    if not intersect_messages and not intersect_events:
+    if not intersect_messages and not intersect_events and not intersect_status:
         die(
-            f"No intersect annotations detected on class '{capability.__name__}'. Please annotate at least one entrypoint with '@intersect_message()', or one event-emitting function with '@intersect_event()' ."
+            f"No intersect annotations detected on class '{capability.__name__}'. Please annotate at least one entrypoint with '@intersect_message()', or one event-emitting function with '@intersect_event()', or create an '@intersect_status' function."
         )
     return intersect_status, intersect_messages, intersect_events
 

tests/unit/test_schema_valid.py

@@ -1,7 +1,15 @@
 import json
 from pathlib import Path
 
-from intersect_sdk import IntersectDataHandler, IntersectMimeType
+from intersect_sdk import (
+    IntersectBaseCapabilityImplementation,
+    IntersectDataHandler,
+    IntersectEventDefinition,
+    IntersectMimeType,
+    intersect_event,
+    intersect_message,
+    intersect_status,
+)
 from intersect_sdk._internal.constants import (
     REQUEST_CONTENT,
     RESPONSE_CONTENT,
@@ -22,7 +30,43 @@ def get_fixture_path(fixture: str):
     return Path(__file__).absolute().parents[1] / 'fixtures' / fixture
 
 
-# TESTS ##################
+# MINIMAL ANNOTATION TESTS ######################
+
+
+def test_minimal_intersect_annotations():
+    class CapWithMessage(IntersectBaseCapabilityImplementation):
+        intersect_sdk_capability_name = 'CapWithMessage'
+
+        @intersect_message
+        def message_function(self, theinput: int) -> int:
+            return theinput * 4
+
+    class CapWithEvent(IntersectBaseCapabilityImplementation):
+        intersect_sdk_capability_name = 'CapWithEvent'
+
+        @intersect_event(events={'event': IntersectEventDefinition(event_type=str)})
+        def event_function(self):
+            self.intersect_sdk_emit_event('event', 'emitted_value')
+
+    class CapWithStatus(IntersectBaseCapabilityImplementation):
+        intersect_sdk_capability_name = 'CapWithStatus'
+
+        @intersect_status
+        def status_function(self) -> str:
+            return 'Up'
+
+    schemas = get_schema_from_capability_implementations(
+        [
+            CapWithEvent,
+            CapWithMessage,
+            CapWithStatus,
+        ],
+        FAKE_HIERARCHY_CONFIG,
+    )
+    assert len(schemas['capabilities']) == 3
+
+
+# FIXTURE TESTS ##################
 
 
 def test_schema_comparison():