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
|
from os import getenv
from time import sleep
import pytest
from requests.exceptions import ConnectionError
from pyzabbix import ZabbixAPI, ZabbixAPIException
ZABBIX_SERVER = "http://localhost:8888"
ZABBIX_VERSION = getenv("ZABBIX_VERSION", "6.2")
@pytest.fixture(scope="session", autouse=True)
def wait_for_zabbix() -> None:
max_attempts = 30
while max_attempts > 0:
try:
ZabbixAPI(ZABBIX_SERVER).login("Admin", "zabbix")
except (ConnectionError, ZabbixAPIException):
sleep(2)
max_attempts -= 1
continue
break
if max_attempts <= 0:
pytest.exit("waiting for zabbix failed!", 1)
# extra sleep if zabbix wasn't ready on first attempt
if max_attempts < 30:
sleep(5)
@pytest.fixture()
def zapi() -> ZabbixAPI:
api = ZabbixAPI(ZABBIX_SERVER)
api.login("Admin", "zabbix")
return api
|
