mirror of
https://github.com/thegeeklab/prometheus-pve-sd.git
synced 2024-12-01 05:20:35 +00:00
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
|
"""Test CLI class."""
|
||
|
|
import pytest
|
||
|
|
from proxmoxer import ProxmoxAPI
|
||
|
|
|
||
|
|
import prometheuspvesd.exception
|
||
|
|
from prometheuspvesd.cli import PrometheusSD
|
||
|
|
from prometheuspvesd.config import Config
|
||
|
|
from prometheuspvesd.discovery import Discovery
|
||
|
|
from prometheuspvesd.exception import APIError
|
||
|
|
from prometheuspvesd.logger import Log
|
||
|
|
|
||
|
|
pytest_plugins = [
|
||
|
|
"prometheuspvesd.test.fixtures.fixtures",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def test_cli_required_error(mocker, capsys):
|
||
|
|
mocker.patch.object(Discovery, "_auth", return_value=mocker.create_autospec(ProxmoxAPI))
|
||
|
|
mocker.patch.object(PrometheusSD, "_fetch", return_value=True)
|
||
|
|
|
||
|
|
with pytest.raises(SystemExit) as e:
|
||
|
|
PrometheusSD()
|
||
|
|
|
||
|
|
stdout, stderr = capsys.readouterr()
|
||
|
|
assert "Option 'pve.server' is required but not set" in stderr
|
||
|
|
assert e.value.code == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_cli_config_error(mocker, capsys):
|
||
|
|
mocker.patch(
|
||
|
|
"prometheuspvesd.config.SingleConfig.__init__",
|
||
|
|
side_effect=prometheuspvesd.exception.ConfigError("Dummy Config Exception")
|
||
|
|
)
|
||
|
|
mocker.patch.object(Discovery, "_auth", return_value=mocker.create_autospec(ProxmoxAPI))
|
||
|
|
mocker.patch.object(PrometheusSD, "_fetch", return_value=True)
|
||
|
|
|
||
|
|
with pytest.raises(SystemExit) as e:
|
||
|
|
PrometheusSD()
|
||
|
|
|
||
|
|
stdout, stderr = capsys.readouterr()
|
||
|
|
assert "Dummy Config Exception" in stderr
|
||
|
|
assert e.value.code == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_cli_log_error(mocker, capsys):
|
||
|
|
mocker.patch.object(Log, "update_logger", side_effect=ValueError("Dummy Logleve Exception"))
|
||
|
|
mocker.patch.object(Discovery, "_auth", return_value=mocker.create_autospec(ProxmoxAPI))
|
||
|
|
mocker.patch.object(PrometheusSD, "_fetch", return_value=True)
|
||
|
|
|
||
|
|
with pytest.raises(SystemExit) as e:
|
||
|
|
PrometheusSD()
|
||
|
|
|
||
|
|
stdout, stderr = capsys.readouterr()
|
||
|
|
assert "Dummy Logleve Exception" in stderr
|
||
|
|
assert e.value.code == 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_cli_api_error(mocker, builtins, capsys):
|
||
|
|
mocker.patch.dict(Config.SETTINGS, builtins)
|
||
|
|
mocker.patch.object(Discovery, "_auth", side_effect=APIError("Dummy API Exception"))
|
||
|
|
mocker.patch.object(PrometheusSD, "_fetch", return_value=True)
|
||
|
|
|
||
|
|
with pytest.raises(SystemExit) as e:
|
||
|
|
PrometheusSD()
|
||
|
|
|
||
|
|
stdout, stderr = capsys.readouterr()
|
||
|
|
assert "Proxmoxer API error: Dummy API Exception" in stderr
|
||
|
|
assert e.value.code == 1
|