add include parameter to On-Call team rules test (#2582)

Co-authored-by: ci.datadog-api-spec <[email protected]>

Author: api-clients-generation-pipeline[bot]

.apigentools-info

@@ -4,13 +4,13 @@
     "spec_versions": {
         "v1": {
             "apigentools_version": "1.6.6",
-            "regenerated": "2025-05-27 10:37:47.296878",
-            "spec_repo_commit": "c75940cb"
+            "regenerated": "2025-05-27 15:21:04.559526",
+            "spec_repo_commit": "97968943"
         },
         "v2": {
             "apigentools_version": "1.6.6",
-            "regenerated": "2025-05-27 10:37:47.323956",
-            "spec_repo_commit": "c75940cb"
+            "regenerated": "2025-05-27 15:21:04.576370",
+            "spec_repo_commit": "97968943"
         }
     }
 }

.generator/schemas/v2/openapi.yaml

@@ -30384,8 +30384,8 @@ components:
       description: Defines an action that is executed when a routing rule matches
         certain criteria.
       oneOf:
-      - $ref: '#/components/schemas/SlackAction'
-      - $ref: '#/components/schemas/TeamsAction'
+      - $ref: '#/components/schemas/SendSlackMessageAction'
+      - $ref: '#/components/schemas/SendTeamsMessageAction'
     RoutingRuleAttributes:
       description: Defines the configurable attributes of a routing rule, such as
         actions, query, time restriction, and urgency.
@@ -30401,6 +30401,7 @@ components:
           type: string
         time_restriction:
           $ref: '#/components/schemas/TimeRestrictions'
+          nullable: true
         urgency:
           $ref: '#/components/schemas/Urgency'
       type: object
@@ -30416,8 +30417,7 @@ components:
       properties:
         data:
           $ref: '#/components/schemas/RoutingRuleRelationshipsPolicyData'
-      required:
-      - data
+          nullable: true
       type: object
     RoutingRuleRelationshipsPolicyData:
       description: Represents the policy data reference, containing the policy's ID
@@ -34489,6 +34489,65 @@ components:
       required:
       - selfServiceTrigger
       type: object
+    SendSlackMessageAction:
+      description: Sends a message to a Slack channel.
+      properties:
+        channel:
+          description: The channel ID.
+          example: CHANNEL
+          type: string
+        type:
+          $ref: '#/components/schemas/SendSlackMessageActionType'
+        workspace:
+          description: The workspace ID.
+          example: WORKSPACE
+          type: string
+      required:
+      - type
+      - channel
+      - workspace
+      type: object
+    SendSlackMessageActionType:
+      default: send_slack_message
+      description: Indicates that the action is a send Slack message action.
+      enum:
+      - send_slack_message
+      example: send_slack_message
+      type: string
+      x-enum-varnames:
+      - SEND_SLACK_MESSAGE
+    SendTeamsMessageAction:
+      description: Sends a message to a Microsoft Teams channel.
+      properties:
+        channel:
+          description: The channel ID.
+          example: CHANNEL
+          type: string
+        team:
+          description: The team ID.
+          example: TEAM
+          type: string
+        tenant:
+          description: The tenant ID.
+          example: TENANT
+          type: string
+        type:
+          $ref: '#/components/schemas/SendTeamsMessageActionType'
+      required:
+      - type
+      - channel
+      - tenant
+      - team
+      type: object
+    SendTeamsMessageActionType:
+      default: send_teams_message
+      description: Indicates that the action is a send Microsoft Teams message action.
+      enum:
+      - send_teams_message
+      example: send_teams_message
+      type: string
+      x-enum-varnames:
+      - SEND_TEAMS_MESSAGE
     SensitiveDataScannerConfigRequest:
       description: Group reorder request.
       properties:
@@ -36396,26 +36455,6 @@ components:
       type: string
       x-enum-varnames:
       - AGGREGATED_CONNECTION
-    SlackAction:
-      description: Sends a message to a Slack channel.
-      properties:
-        channel:
-          description: The channel ID.
-          example: CHANNEL
-          type: string
-        type:
-          description: Must be set to "send_slack_message".
-          example: send_slack_message
-          type: string
-        workspace:
-          description: The workspace ID.
-          example: WORKSPACE
-          type: string
-      required:
-      - type
-      - channel
-      - workspace
-      type: object
     SlackIntegrationMetadata:
       description: Incident integration metadata for the Slack integration.
       properties:
@@ -38330,31 +38369,6 @@ components:
       required:
       - data
       type: object
-    TeamsAction:
-      description: Sends a message to a Microsoft Teams channel.
-      properties:
-        channel:
-          description: The channel ID.
-          example: CHANNEL
-          type: string
-        team:
-          description: The team ID.
-          example: TEAM
-          type: string
-        tenant:
-          description: The tenant ID.
-          example: TENANT
-          type: string
-        type:
-          description: Must be set to "send_teams_message".
-          example: send_teams_message
-          type: string
-      required:
-      - type
-      - channel
-      - tenant
-      - team
-      type: object
     TeamsField:
       description: Supported teams field.
       enum:

docs/datadog_api_client.v2.model.rst

@@ -14851,6 +14851,34 @@ datadog\_api\_client.v2.model.self\_service\_trigger\_wrapper module
    :members:
    :show-inheritance:
 
+datadog\_api\_client.v2.model.send\_slack\_message\_action module
+-----------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.send_slack_message_action
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.send\_slack\_message\_action\_type module
+-----------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.send_slack_message_action_type
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.send\_teams\_message\_action module
+-----------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.send_teams_message_action
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.send\_teams\_message\_action\_type module
+-----------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.send_teams_message_action_type
+   :members:
+   :show-inheritance:
+
 datadog\_api\_client.v2.model.sensitive\_data\_scanner\_config\_request module
 ------------------------------------------------------------------------------
 
@@ -15747,13 +15775,6 @@ datadog\_api\_client.v2.model.single\_aggregated\_connection\_response\_data\_ty
    :members:
    :show-inheritance:
 
-datadog\_api\_client.v2.model.slack\_action module
---------------------------------------------------
-
-.. automodule:: datadog_api_client.v2.model.slack_action
-   :members:
-   :show-inheritance:
-
 datadog\_api\_client.v2.model.slack\_integration\_metadata module
 -----------------------------------------------------------------
 
@@ -16699,13 +16720,6 @@ datadog\_api\_client.v2.model.team\_update\_request module
    :members:
    :show-inheritance:
 
-datadog\_api\_client.v2.model.teams\_action module
---------------------------------------------------
-
-.. automodule:: datadog_api_client.v2.model.teams_action
-   :members:
-   :show-inheritance:
-
 datadog\_api\_client.v2.model.teams\_field module
 -------------------------------------------------
 

examples/v2/on-call/SetOnCallTeamRoutingRules.py

@@ -5,7 +5,8 @@
 from os import environ
 from datadog_api_client import ApiClient, Configuration
 from datadog_api_client.v2.api.on_call_api import OnCallApi
-from datadog_api_client.v2.model.slack_action import SlackAction
+from datadog_api_client.v2.model.send_slack_message_action import SendSlackMessageAction
+from datadog_api_client.v2.model.send_slack_message_action_type import SendSlackMessageActionType
 from datadog_api_client.v2.model.team_routing_rules_request import TeamRoutingRulesRequest
 from datadog_api_client.v2.model.team_routing_rules_request_data import TeamRoutingRulesRequestData
 from datadog_api_client.v2.model.team_routing_rules_request_data_attributes import TeamRoutingRulesRequestDataAttributes
@@ -28,9 +29,9 @@
             rules=[
                 TeamRoutingRulesRequestRule(
                     actions=[
-                        SlackAction(
+                        SendSlackMessageAction(
                             channel="channel",
-                            type="send_slack_message",
+                            type=SendSlackMessageActionType.SEND_SLACK_MESSAGE,
                             workspace="workspace",
                         ),
                     ],
@@ -52,7 +53,6 @@
                             ),
                         ],
                     ),
-                    urgency=Urgency.HIGH,
                 ),
                 TeamRoutingRulesRequestRule(
                     policy_id=ESCALATION_POLICY_DATA_ID,
@@ -69,6 +69,6 @@
 configuration = Configuration()
 with ApiClient(configuration) as api_client:
     api_instance = OnCallApi(api_client)
-    response = api_instance.set_on_call_team_routing_rules(team_id=DD_TEAM_DATA_ID, body=body)
+    response = api_instance.set_on_call_team_routing_rules(team_id=DD_TEAM_DATA_ID, include="rules", body=body)
 
     print(response)

src/datadog_api_client/v2/model/routing_rule_action.py

@@ -18,8 +18,8 @@ def __init__(self, **kwargs):
         :param channel: The channel ID.
         :type channel: str
 
-        :param type: Must be set to "send_slack_message".
-        :type type: str
+        :param type: Indicates that the action is a send Slack message action.
+        :type type: SendSlackMessageActionType
 
         :param workspace: The workspace ID.
         :type workspace: str
@@ -41,12 +41,12 @@ def _composed_schemas(_):
         # code would be run when this module is imported, and these composed
         # classes don't exist yet because their module has not finished
         # loading
-        from datadog_api_client.v2.model.slack_action import SlackAction
-        from datadog_api_client.v2.model.teams_action import TeamsAction
+        from datadog_api_client.v2.model.send_slack_message_action import SendSlackMessageAction
+        from datadog_api_client.v2.model.send_teams_message_action import SendTeamsMessageAction
 
         return {
             "oneOf": [
-                SlackAction,
-                TeamsAction,
+                SendSlackMessageAction,
+                SendTeamsMessageAction,
             ],
         }

src/datadog_api_client/v2/model/routing_rule_attributes.py

@@ -17,8 +17,8 @@
     from datadog_api_client.v2.model.routing_rule_action import RoutingRuleAction
     from datadog_api_client.v2.model.time_restrictions import TimeRestrictions
     from datadog_api_client.v2.model.urgency import Urgency
-    from datadog_api_client.v2.model.slack_action import SlackAction
-    from datadog_api_client.v2.model.teams_action import TeamsAction
+    from datadog_api_client.v2.model.send_slack_message_action import SendSlackMessageAction
+    from datadog_api_client.v2.model.send_teams_message_action import SendTeamsMessageAction
 
 
 class RoutingRuleAttributes(ModelNormal):
@@ -44,7 +44,9 @@ def openapi_types(_):
 
     def __init__(
         self_,
-        actions: Union[List[Union[RoutingRuleAction, SlackAction, TeamsAction]], UnsetType] = unset,
+        actions: Union[
+            List[Union[RoutingRuleAction, SendSlackMessageAction, SendTeamsMessageAction]], UnsetType
+        ] = unset,
         query: Union[str, UnsetType] = unset,
         time_restriction: Union[TimeRestrictions, UnsetType] = unset,
         urgency: Union[Urgency, UnsetType] = unset,

src/datadog_api_client/v2/model/routing_rule_relationships_policy.py

@@ -3,11 +3,13 @@
 # Copyright 2019-Present Datadog, Inc.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
+from typing import Union, TYPE_CHECKING
 
 from datadog_api_client.model_utils import (
     ModelNormal,
     cached_property,
+    unset,
+    UnsetType,
 )
 
 
@@ -30,13 +32,13 @@ def openapi_types(_):
         "data": "data",
     }
 
-    def __init__(self_, data: RoutingRuleRelationshipsPolicyData, **kwargs):
+    def __init__(self_, data: Union[RoutingRuleRelationshipsPolicyData, UnsetType] = unset, **kwargs):
         """
         Defines the relationship that links a routing rule to a policy.
 
         :param data: Represents the policy data reference, containing the policy's ID and resource type.
-        :type data: RoutingRuleRelationshipsPolicyData
+        :type data: RoutingRuleRelationshipsPolicyData, optional
         """
+        if data is not unset:
+            kwargs["data"] = data
         super().__init__(kwargs)
-
-        self_.data = data