ydb/yql/essentials/tests/common/test_framework/solomon_runner.py

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)