blob: de6062a9ec32b0bb0748a1f7c59ad45c880e6c2a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
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)
|
