cloudtests: add sugar to ControllerDefinition

MaterializeInc · Aug 24, 2023 · eb0e7e0 · eb0e7e0
1 parent aee4c0b
commit eb0e7e0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/misc/python/materialize/cloudtest/util/controller.py b/misc/python/materialize/cloudtest/util/controller.py
@@ -14,7 +14,9 @@
 from dataclasses import dataclass
 from typing import Any, List, Optional, Tuple, Union
 
+from materialize.cloudtest.util.authentication import AuthConfig
 from materialize.cloudtest.util.common import eprint, retry
+from materialize.cloudtest.util.web_request import WebRequests
 
 
 @dataclass
@@ -55,6 +57,9 @@ def configured_base_url(self) -> str:
 
         return self.endpoint.base_url
 
+    def requests(self, auth: Optional[AuthConfig]) -> WebRequests:
+        return WebRequests(auth, self.configured_base_url())
+
 
 def wait_for_connectable(
     address: Union[Tuple[Any, int], str],