Adding compatibility tests for client and server triggers and actions (#12778)

Author: chrisguidry

src/prefect/events/__init__.py

@@ -4,6 +4,7 @@
     Automation,
     AutomationCore,
     Posture,
+    TriggerTypes,
     Trigger,
     ResourceTrigger,
     EventTrigger,
@@ -21,6 +22,26 @@
     DeploymentCompoundTrigger,
     DeploymentSequenceTrigger,
 )
+from .actions import (
+    ActionTypes,
+    Action,
+    DoNothing,
+    RunDeployment,
+    PauseDeployment,
+    ResumeDeployment,
+    ChangeFlowRunState,
+    CancelFlowRun,
+    SuspendFlowRun,
+    CallWebhook,
+    SendNotification,
+    PauseWorkPool,
+    ResumeWorkPool,
+    PauseWorkQueue,
+    ResumeWorkQueue,
+    PauseAutomation,
+    ResumeAutomation,
+    DeclareIncident,
+)
 from .clients import get_events_client, get_events_subscriber
 from .utilities import emit_event
 
@@ -33,6 +54,7 @@
     "Automation",
     "AutomationCore",
     "Posture",
+    "TriggerTypes",
     "Trigger",
     "ResourceTrigger",
     "EventTrigger",
@@ -47,6 +69,24 @@
     "DeploymentMetricTrigger",
     "DeploymentCompoundTrigger",
     "DeploymentSequenceTrigger",
+    "ActionTypes",
+    "Action",
+    "DoNothing",
+    "RunDeployment",
+    "PauseDeployment",
+    "ResumeDeployment",
+    "ChangeFlowRunState",
+    "CancelFlowRun",
+    "SuspendFlowRun",
+    "CallWebhook",
+    "SendNotification",
+    "PauseWorkPool",
+    "ResumeWorkPool",
+    "PauseWorkQueue",
+    "ResumeWorkQueue",
+    "PauseAutomation",
+    "ResumeAutomation",
+    "DeclareIncident",
     "emit_event",
     "get_events_client",
     "get_events_subscriber",

src/prefect/events/schemas/deployment_triggers.py

@@ -146,6 +146,8 @@ class DeploymentEventTrigger(DeploymentResourceTrigger):
     period of time.
     """
 
+    trigger_type = EventTrigger
+
     type: Literal["event"] = "event"
 
     after: Set[str] = Field(
@@ -226,7 +228,7 @@ def enforce_minimum_within_for_proactive_triggers(cls, values: Dict[str, Any]):
         return values
 
     def as_trigger(self) -> Trigger:
-        return EventTrigger(
+        return self.trigger_type(
             match=self.match,
             match_related=self.match_related,
             after=self.after,
@@ -243,6 +245,8 @@ class DeploymentMetricTrigger(DeploymentResourceTrigger):
     A trigger that fires based on the results of a metric query.
     """
 
+    trigger_type = MetricTrigger
+
     type: Literal["metric"] = "metric"
 
     posture: Literal[Posture.Metric] = Field(  # type: ignore[valid-type]
@@ -256,7 +260,7 @@ class DeploymentMetricTrigger(DeploymentResourceTrigger):
     )
 
     def as_trigger(self) -> Trigger:
-        return MetricTrigger(
+        return self.trigger_type(
             match=self.match,
             match_related=self.match_related,
             posture=self.posture,
@@ -279,6 +283,8 @@ class DeploymentCompoundTrigger(DeploymentCompositeTrigger):
     """A composite trigger that requires some number of triggers to have
     fired within the given time period"""
 
+    trigger_type = CompoundTrigger
+
     type: Literal["compound"] = "compound"
     require: Union[int, Literal["any", "all"]]
 
@@ -297,7 +303,7 @@ def validate_require(cls, values: Dict[str, Any]) -> Dict[str, Any]:
         return values
 
     def as_trigger(self) -> Trigger:
-        return CompoundTrigger(
+        return self.trigger_type(
             require=self.require,
             triggers=self.triggers,
             within=self.within,
@@ -309,10 +315,12 @@ class DeploymentSequenceTrigger(DeploymentCompositeTrigger):
     """A composite trigger that requires some number of triggers to have fired
     within the given time period in a specific order"""
 
+    trigger_type = SequenceTrigger
+
     type: Literal["sequence"] = "sequence"
 
     def as_trigger(self) -> Trigger:
-        return SequenceTrigger(
+        return self.trigger_type(
             triggers=self.triggers,
             within=self.within,
             job_variables=self.job_variables,