feat: add set_service_level function (#205)

* feat: add set_service_level function

* fix websocket logic

* linting

Author: firstof9

openevsehttp/__main__.py

@@ -235,7 +235,7 @@ def _update_status(self, msgtype, data, error):
             _LOGGER.debug("Websocket data: %s", data)
             if "wh" in data.keys():
                 data["watthour"] = data.pop("wh")
-            elif "config_version" in data.keys():
+            if "config_version" in data.keys():
                 self.update()
             self._status.update(data)
 
@@ -412,6 +412,25 @@ async def set_current(self, amps: int = 6) -> None:
             response, msg = await self.send_command(command)
             _LOGGER.debug("Set current response: %s", msg)
 
+    async def set_service_level(self, level: int = 2) -> None:
+        """Set the service level of the EVSE."""
+        if not isinstance(level, int) or not 0 <= level <= 2:
+            _LOGGER.error("Invalid service level: %s", level)
+            raise ValueError
+
+        url = f"{self.url}config"
+        data = {"service": level}
+
+        _LOGGER.debug("Set service level to: %s", level)
+        response = await self.process_request(
+            url=url, method="post", data=data
+        )  # noqa: E501
+        _LOGGER.debug("service response: %s", response)
+        result = response["msg"]
+        if result not in ["done", "no change"]:
+            _LOGGER.error("Problem issuing command: %s", response["msg"])
+            raise UnknownError
+
     # Restart OpenEVSE WiFi
     async def restart_wifi(self) -> None:
         """Restart OpenEVSE WiFi module."""

tests/test_main.py

@@ -1267,3 +1267,55 @@ async def test_charge_mode(fixture, expected, request):
     await charger.update()
     status = charger.charge_mode
     assert status == expected
+
+
+async def test_set_service_level(test_charger, mock_aioclient, caplog):
+    """Test set service level."""
+    await test_charger.update()
+    value = {"msg": "done"}
+    mock_aioclient.post(
+        TEST_URL_CONFIG,
+        status=200,
+        body=json.dumps(value),
+    )
+    with caplog.at_level(logging.DEBUG):
+        await test_charger.set_service_level(1)
+
+    mock_aioclient.get(
+        TEST_URL_STATUS,
+        status=200,
+        body=load_fixture("v4_json/status.json"),
+    )
+    mock_aioclient.get(
+        TEST_URL_CONFIG,
+        status=200,
+        body=load_fixture("v4_json/config.json"),
+    )
+    value = {"config_version": 2, "msg": "done"}
+    mock_aioclient.post(
+        TEST_URL_CONFIG,
+        status=200,
+        body=json.dumps(value),
+    )
+    with caplog.at_level(logging.DEBUG):
+        await test_charger.set_service_level(2)
+
+    value = {"msg": "error"}
+    mock_aioclient.post(
+        TEST_URL_CONFIG,
+        status=200,
+        body=json.dumps(value),
+    )
+    with caplog.at_level(logging.DEBUG):
+        with pytest.raises(UnknownError):
+            await test_charger.set_service_level(1)
+            assert "Problem issuing command: error" in caplog.text
+
+    value = {"msg": "done"}
+    mock_aioclient.post(
+        TEST_URL_CONFIG,
+        status=200,
+        body=json.dumps(value),
+    )
+    with pytest.raises(ValueError):
+        await test_charger.set_service_level("A")