mirror of
https://github.com/thegeeklab/prometheus-pve-sd.git
synced 2024-12-01 05:20:35 +00:00
88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
"""Test CLI class."""
|
|
import json
|
|
|
|
import pytest
|
|
from proxmoxer import ProxmoxAPI
|
|
|
|
import prometheuspvesd.exception
|
|
from prometheuspvesd.cli import PrometheusSD
|
|
from prometheuspvesd.client import ProxmoxClient
|
|
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(ProxmoxClient, "_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(ProxmoxClient, "_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 Loglevel Exception"))
|
|
mocker.patch.object(ProxmoxClient, "_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 Loglevel 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(ProxmoxClient, "_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
|
|
|
|
|
|
def test_cli_write(mocker, tmp_path, builtins, inventory, labels):
|
|
temp = tmp_path / "temp.txt"
|
|
out = tmp_path / "out.txt"
|
|
|
|
builtins["output_file"]["default"] = out.as_posix()
|
|
|
|
mocker.patch.dict(Config.SETTINGS, builtins)
|
|
mocker.patch.object(ProxmoxClient, "_auth", return_value=mocker.create_autospec(ProxmoxAPI))
|
|
mocker.patch.object(Discovery, "propagate", return_value=inventory)
|
|
mocker.patch("tempfile.NamedTemporaryFile", return_value=temp.open("w"))
|
|
|
|
psd = PrometheusSD()
|
|
assert json.loads(out.read_text()) == labels
|
|
assert oct(out.stat().st_mode & 0o777) == oct(int(psd.config.config["output_file_mode"], 8))
|