generator: add support for "graph_config" verifications

Author: jcristau

src/taskgraph/generator.py

@@ -378,6 +378,7 @@ def _run(self):
 
         # Initial verifications that don't depend on any generation state.
         self.verify("initial")
+        self.verify("graph_config", graph_config)
 
         if callable(self._parameters):
             parameters = self._parameters(graph_config)

src/taskgraph/util/verify.py

@@ -36,6 +36,12 @@ def verify(self):
         self.func()
 
 
+@dataclass(frozen=True)
+class GraphConfigVerification(Verification):
+    def verify(self, graph_config: GraphConfig):
+        self.func(graph_config)
+
+
 @dataclass(frozen=True)
 class GraphVerification(Verification):
     """Verification for a TaskGraph object."""
@@ -95,6 +101,7 @@ class VerificationSequence:
     _verifications: Dict = field(default_factory=dict)
     _verification_types = {
         "graph": GraphVerification,
+        "graph_config": GraphConfigVerification,
         "initial": InitialVerification,
         "kinds": KindsVerification,
         "parameters": ParametersVerification,

test/test_util_verify.py

@@ -11,6 +11,7 @@
 from taskgraph.task import Task
 from taskgraph.util.treeherder import split_symbol
 from taskgraph.util.verify import (
+    GraphConfigVerification,
     GraphVerification,
     ParametersVerification,
     VerificationSequence,
@@ -38,6 +39,8 @@ def inner(name, **kwargs):
 
         if isinstance(v, GraphVerification):
             assert "graph" in kwargs
+
+        if isinstance(v, (GraphVerification, GraphConfigVerification)):
             kwargs.setdefault("graph_config", graph_config)
 
         if isinstance(v, (GraphVerification, ParametersVerification)):
@@ -94,6 +97,13 @@ def assert_simple_verification(arg):
             1,
             id="ParametersVerification",
         ),
+        pytest.param(
+            "graph_config",
+            ("passed-thru",),
+            assert_simple_verification,
+            1,
+            id="GraphConfigVerification",
+        ),
     ),
 )
 def test_verification_types(name, input, run_assertions, expected_called):