def test__get_code(mocker): """ Test "_get_code" method must return plugin exit code. """ mocker.patch("sys.argv", ["check_hddtemp.py", "-s", "127.0.0.1", "-p", "7634"]) checker = CheckHDDTemp() result = checker._get_code(status="ok") assert result == 0 # nosec: B101
def test__get_code__unknown_device_temperature(mocker): """ Test "_get_code" method must return plugin exit code (unknown device temperature case). """ mocker.patch("sys.argv", ["check_hddtemp.py", "-s", "127.0.0.1", "-p", "7634"]) checker = CheckHDDTemp() result = checker._get_code(status="unknown") assert result == 3 # nosec: B101