Initial commit

tests/api/test_power.py

@@ -0,0 +1,104 @@
+"""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)