Pass scenario to UseClusterdCompute

Author: def-

doc/developer/platform-checks.md

@@ -199,5 +199,5 @@ later in the execution. This will confirm that the database is able to perform a
 
 When you add new features or have SQL syntax changes, this can be handled in platform checks by comparing against the `base_version`. In a `Check` this can be done directly, see for example `UUID`:
 https://github.com/MaterializeInc/materialize/blob/fef9d6a00de4ff2bc6a92968470a2c88f85cab12/misc/python/materialize/checks/uuid.py#L17-L19
-For an action you have to pass the `base_version` parameter explicitly, see for example `UseClusterdCompute`:
+For an action you have to access the `scenario.base_version`, see for example `UseClusterdCompute`:
 https://github.com/MaterializeInc/materialize/blob/fef9d6a00de4ff2bc6a92968470a2c88f85cab12/misc/python/materialize/checks/mzcompose_actions.py#L74-L97

misc/python/materialize/checks/mzcompose_actions.py

@@ -7,13 +7,16 @@
 # the Business Source License, use of this software will be governed
 # by the Apache License, Version 2.0.
 
-from typing import List, Optional
+from typing import TYPE_CHECKING, List, Optional
 
 from materialize.checks.actions import Action
 from materialize.checks.executors import Executor
 from materialize.mzcompose.services import Clusterd, Materialized
 from materialize.util import MzVersion
 
+if TYPE_CHECKING:
+    from materialize.checks.scenarios import Scenario
+
 
 class MzcomposeAction(Action):
     pass
@@ -72,8 +75,8 @@ def execute(self, e: Executor) -> None:
 
 
 class UseClusterdCompute(MzcomposeAction):
-    def __init__(self, base_version: MzVersion) -> None:
-        self.base_version = base_version
+    def __init__(self, scenario: "Scenario") -> None:
+        self.base_version = scenario.base_version()
 
     def execute(self, e: Executor) -> None:
         c = e.mzcompose_composition()

misc/python/materialize/checks/scenarios.py

@@ -110,7 +110,7 @@ def actions(self) -> List[Action]:
         return [
             StartMz(),
             StartClusterdCompute(),
-            UseClusterdCompute(base_version=self.base_version()),
+            UseClusterdCompute(self),
             Initialize(self),
             KillClusterdCompute(),
             StartClusterdCompute(),
@@ -131,7 +131,7 @@ def actions(self) -> List[Action]:
         return [
             StartMz(),
             StartClusterdCompute(),
-            UseClusterdCompute(base_version=self.base_version()),
+            UseClusterdCompute(self),
             Initialize(self),
             KillMz(),
             StartMz(),
@@ -152,7 +152,7 @@ def actions(self) -> List[Action]:
         return [
             StartMz(),
             StartClusterdCompute(),
-            UseClusterdCompute(base_version=self.base_version()),
+            UseClusterdCompute(self),
             Initialize(self),
             KillClusterdStorageAction(),
             Manipulate(self, phase=1),

misc/python/materialize/checks/scenarios_upgrade.py

@@ -78,7 +78,7 @@ def actions(self) -> List[Action]:
         return [
             StartMz(tag=self.base_version()),
             StartClusterdCompute(tag=self.base_version()),
-            UseClusterdCompute(base_version=self.base_version()),
+            UseClusterdCompute(self),
             Initialize(self),
             Manipulate(self, phase=1),
             KillMz(),
@@ -110,7 +110,7 @@ def actions(self) -> List[Action]:
         return [
             StartMz(tag=self.base_version()),
             StartClusterdCompute(tag=self.base_version()),
-            UseClusterdCompute(base_version=self.base_version()),
+            UseClusterdCompute(self),
             Initialize(self),
             Manipulate(self, phase=1),
             KillClusterdCompute(),