sql_server: enable restart action in zippy tests

Author: ptravers

misc/python/materialize/mzcompose/service.py

@@ -53,6 +53,39 @@ class ServiceDependency(TypedDict, total=False):
     """Condition under which a dependency is considered satisfied."""
 
 
+class ServiceDeployResourceLimits(TypedDict, total=False):
+    """Resource limits for a service deployment."""
+
+    memory: str
+    """Memory limit (e.g., '6G', '512M')."""
+
+    cpus: str
+    """CPU limit (e.g., '0.5', '2')."""
+
+
+class ServiceDeployResources(TypedDict, total=False):
+    """Resource configuration for a service deployment."""
+
+    limits: ServiceDeployResourceLimits
+    """Resource limits."""
+
+    reservations: ServiceDeployResourceLimits
+    """Resource reservations."""
+
+
+class ServiceDeploy(TypedDict, total=False):
+    """Deployment configuration for a service."""
+
+    resources: ServiceDeployResources
+    """Resource configuration."""
+
+    replicas: int
+    """Number of container replicas to deploy."""
+
+    restart_policy: dict[str, Any]
+    """Restart policy configuration."""
+
+
 class ServiceConfig(TypedDict, total=False):
     """The definition of a service in Docker Compose.
 
@@ -141,11 +174,8 @@ class ServiceConfig(TypedDict, total=False):
     TODO(benesch): this should use a nested TypedDict.
     """
 
-    deploy: dict[str, dict[str, dict[str, str]]]
-    """Additional deployment configuration, like resource limits.
-
-    TODO(benesch): this should use a nested TypedDict.
-    """
+    deploy: ServiceDeploy
+    """Deployment configuration, including resource limits."""
 
     ulimits: dict[str, Any]
     """Override the default ulimits for a container."""
@@ -178,6 +208,9 @@ class ServiceConfig(TypedDict, total=False):
     user: str | None
     """The user for the container."""
 
+    cap_add: list[str] | None
+    """cap_add specifies additional container capabilities as strings."""
+
 
 class Service:
     """A Docker Compose service in a `Composition`.

misc/python/materialize/mzcompose/services/sql_server.py

@@ -44,6 +44,8 @@ def __init__(
                     "interval": "1s",
                     "start_period": "30s",
                 },
+                "deploy": {"resources": {"limits": {"memory": "3G"}}},
+                "cap_add": ["SYS_PTRACE"],
             },
         )
         self.sa_password = sa_password

misc/python/materialize/zippy/scenarios.py

@@ -37,6 +37,7 @@
     CreateSqlServerTable,
     SqlServerDML,
     SqlServerStart,
+    SqlServerRestart,
 )
 from materialize.zippy.sql_server_cdc_actions import CreateSqlServerCdcTable
 from materialize.zippy.mz_actions import (
@@ -252,8 +253,7 @@ def actions_with_weight(self) -> dict[ActionOrFactory, float]:
             KillClusterd: 5,
             StoragedKill: 5,
             StoragedStart: 5,
-            # TODO: Reenable when database-issues#9624 is fixed
-            # SqlServerRestart: 10,
+            SqlServerRestart: 10,
             CreateViewParameterized(): 10,
             ValidateView: 20,
             SqlServerDML: 100,