104 lines
2.7 KiB
Python
104 lines
2.7 KiB
Python
"""Tests for the power related API functions"""
|
|
import pytest
|
|
import requests
|
|
|
|
from redfish_cli.api import power
|
|
from redfish_cli.api.exceptions import FailedToSetPowerState
|
|
from tests.api import args
|
|
|
|
from ..utils import get_test_json, MockRedfishResponse, MockResponse
|
|
|
|
|
|
def test_power_state(monkeypatch):
|
|
"""Test the get power_state API call"""
|
|
|
|
def mock_get(*aargs, **kwargs):
|
|
# pylint: disable=unused-argument
|
|
mock_redfish = MockRedfishResponse()
|
|
mock_redfish.json_dict = get_test_json("api_power_state.json")
|
|
return MockResponse(content=mock_redfish.text)
|
|
|
|
monkeypatch.setattr(requests, "get", mock_get)
|
|
|
|
_args = args()
|
|
_args.system = "System.Embedded.1"
|
|
|
|
power_state = power.system_power_state(_args)
|
|
assert power_state == "Off"
|
|
|
|
|
|
def test_power_states_allowed(monkeypatch):
|
|
"""Test the poert starts allowed API function"""
|
|
|
|
def mock_get(*aargs, **kwargs):
|
|
# pylint: disable=unused-argument
|
|
mock_redfish = MockRedfishResponse()
|
|
mock_redfish.json_dict = get_test_json("power_states_allowed.json")
|
|
return MockResponse(content=mock_redfish.text)
|
|
|
|
monkeypatch.setattr(requests, "get", mock_get)
|
|
|
|
_args = args()
|
|
_args.system = "system"
|
|
|
|
power_states = power.system_power_states_allowed(_args)
|
|
assert power_states == [
|
|
"On",
|
|
"ForceOff",
|
|
"ForceRestart",
|
|
"GracefulShutdown",
|
|
"PushPowerButton",
|
|
"Nmi",
|
|
]
|
|
|
|
|
|
def test_set_power_state(monkeypatch):
|
|
"""Test the set power state API call"""
|
|
|
|
def mock_post(*aargs, **kwargs):
|
|
# pylint: disable=unused-argument
|
|
return MockResponse(content="{}")
|
|
|
|
monkeypatch.setattr(requests, "post", mock_post)
|
|
|
|
_args = args()
|
|
_args.system = "System.Embedded.1"
|
|
_args.power_state = "On"
|
|
|
|
result = power.system_set_power_state(_args)
|
|
assert result == "Powerstate set to On"
|
|
|
|
monkeypatch.setattr(
|
|
requests, "post", lambda *args, **kwargs: MockResponse(ok=False)
|
|
)
|
|
with pytest.raises(FailedToSetPowerState):
|
|
_ = power.system_set_power_state(_args)
|
|
|
|
|
|
def test_system_reset(monkeypatch):
|
|
"""Test resetting a system"""
|
|
monkeypatch.setattr(
|
|
requests,
|
|
"post",
|
|
lambda args, auth=None, verify=None, data=None, headers=None, timeout=None: MockResponse(
|
|
content="{}"
|
|
),
|
|
)
|
|
|
|
_args = args(
|
|
system="System.Embedded.1",
|
|
)
|
|
|
|
result = power.system_reset(_args)
|
|
assert result == "System reset"
|
|
|
|
monkeypatch.setattr(
|
|
requests,
|
|
"post",
|
|
lambda args, auth=None, verify=None, data=None, headers=None, timeout=None: MockResponse(
|
|
ok=False
|
|
),
|
|
)
|
|
|
|
with pytest.raises(FailedToSetPowerState):
|
|
result = power.system_reset(_args)
|