From eb0e7e0c0b3a10be4a126727604488b8d618ca13 Mon Sep 17 00:00:00 2001 From: Rainer Niedermayr Date: Thu, 24 Aug 2023 10:42:39 +0200 Subject: [PATCH] cloudtests: add sugar to ControllerDefinition --- misc/python/materialize/cloudtest/util/controller.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/misc/python/materialize/cloudtest/util/controller.py b/misc/python/materialize/cloudtest/util/controller.py index 45b42744ebbba..cd608dd9de29b 100644 --- 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],