Merge branch 'rightlib' into mergelibs-241120-1113

1 files changed, 40 insertions, 0 deletions

diff --git a/yql/essentials/tests/common/test_framework/solomon_runner.py b/yql/essentials/tests/common/test_framework/solomon_runner.py
new file mode 100644
index 0000000000..de6062a9ec
--- /dev/null
+++ b/yql/essentials/tests/common/test_framework/solomon_runner.py
@@ -0,0 +1,40 @@
+import os
+import pytest
+import requests
+
+
+class SolomonWrapper(object):
+    def __init__(self, url, endpoint):
+        self._url = url
+        self._endpoint = endpoint
+        self.table_prefix = ""
+
+    def is_valid(self):
+        return self._url is not None
+
+    def cleanup(self):
+        res = requests.post(self._url + "/cleanup")
+        res.raise_for_status()
+
+    def get_metrics(self):
+        res = requests.get(self._url + "/metrics?project=my_project&cluster=my_cluster&service=my_service")
+        res.raise_for_status()
+        return res.text
+
+    def prepare_program(self, program, program_file, res_dir, lang='sql'):
+        return program, program_file
+
+    @property
+    def url(self):
+        return self._url
+
+    @property
+    def endpoint(self):
+        return self._endpoint
+
+
+@pytest.fixture(scope='module')
+def solomon(request):
+    solomon_url = os.environ.get("SOLOMON_URL")
+    solomon_endpoint = os.environ.get("SOLOMON_ENDPOINT")
+    return SolomonWrapper(solomon_url, solomon_endpoint)