feat: set running divertmode (#430)

* feat: set running `divertmode`

* formatting and linting

* update tests

* formatting

Author: firstof9

openevsehttp/__main__.py

@@ -64,6 +64,11 @@
     255: "disabled",
 }
 
+divert_mode = {
+    "fast": 1,
+    "eco": 2,
+}
+
 ERROR_TIMEOUT = "Timeout while updating"
 INFO_LOOP_RUNNING = "Event loop already running, not creating new one."
 UPDATE_TRIGGERS = [
@@ -328,7 +333,7 @@ async def get_schedule(self) -> Union[Dict[str, str], Dict[str, Any]]:
         return response
 
     async def set_charge_mode(self, mode: str = "fast") -> None:
-        """Set the charge mode."""
+        """Set the charge mode at startup setting."""
         url = f"{self.url}config"
 
         if mode not in ["fast", "eco"]:
@@ -868,6 +873,26 @@ async def set_led_brightness(self, level: int) -> None:
         _LOGGER.debug("Setting LED brightness to %s", level)
         await self.process_request(url=url, method="post", data=data)  # noqa: E501
 
+    async def set_divert_mode(self, mode: str = "fast") -> None:
+        """Set the divert mode."""
+        url = f"{self.url}divertmode"
+
+        if mode not in ["fast", "eco"]:
+            _LOGGER.error("Invalid value for charge_mode: %s", mode)
+            raise ValueError
+
+        _LOGGER.debug("Setting divert mode to %s", mode)
+
+        # convert text to int
+        new_mode = divert_mode[mode]
+        data = {"divertmode": new_mode}
+        response = await self.process_request(
+            url=url, method="get", data=data
+        )  # noqa: E501
+        if response != "Divert Mode changed":
+            _LOGGER.error("Problem issuing command: %s", response)
+            raise UnknownError
+
     @property
     def led_brightness(self) -> str:
         """Return charger led_brightness."""
@@ -1221,7 +1246,7 @@ def divertmode(self) -> str:
         assert self._status is not None
         mode = self._status["divertmode"]
         if mode == 1:
-            return "normal"
+            return "fast"
         return "eco"
 
     @property

tests/test_main.py

@@ -641,18 +641,6 @@ async def test_get_relayt(fixture, expected, request):
     await charger.ws_disconnect()
 
 
-@pytest.mark.parametrize(
-    "fixture, expected", [("test_charger", "eco"), ("test_charger_v2", "normal")]
-)
-async def test_get_divertmode(fixture, expected, request):
-    """Test v4 Status reply."""
-    charger = request.getfixturevalue(fixture)
-    await charger.update()
-    status = charger.divertmode
-    assert status == expected
-    await charger.ws_disconnect()
-
-
 @pytest.mark.parametrize(
     "fixture, expected", [("test_charger", 0), ("test_charger_v2", 0)]
 )
@@ -2159,3 +2147,62 @@ async def test_get_status(test_charger_timeout, caplog):
     assert "Updating data from http://openevse.test.tld/status" in caplog.text
     assert "Status update:" not in caplog.text
     assert "Config update:" not in caplog.text
+
+
+@pytest.mark.parametrize(
+    "fixture, expected",
+    [
+        ("test_charger", "eco"),
+        ("test_charger_v2", "fast"),
+        ("test_charger_broken", "eco"),
+        ("test_charger_new", "fast"),
+    ],
+)
+async def test_divertmode(fixture, expected, request):
+    """Test divertmode property."""
+    charger = request.getfixturevalue(fixture)
+    await charger.update()
+    status = charger.divertmode
+    assert status == expected
+    await charger.ws_disconnect()
+
+
+async def test_set_divert_mode(
+    test_charger_new, test_charger_v2, mock_aioclient, caplog
+):
+    """Test set_divert_mode reply."""
+    await test_charger_new.update()
+    value = "Divert Mode changed"
+    mock_aioclient.get(
+        TEST_URL_DIVERT,
+        status=200,
+        body=value,
+    )
+    with caplog.at_level(logging.DEBUG):
+        await test_charger_new.set_divert_mode("fast")
+    assert "Setting divert mode to fast" in caplog.text
+
+    mock_aioclient.get(
+        TEST_URL_DIVERT,
+        status=200,
+        body=value,
+    )
+    await test_charger_v2.update()
+    with caplog.at_level(logging.DEBUG):
+        await test_charger_v2.set_divert_mode("eco")
+    assert "Setting divert mode to eco" in caplog.text
+
+    with pytest.raises(ValueError):
+        with caplog.at_level(logging.DEBUG):
+            await test_charger_new.set_divert_mode("test")
+    assert "Invalid value for charge_mode: test" in caplog.text
+
+    mock_aioclient.get(
+        TEST_URL_DIVERT,
+        status=200,
+        body="error",
+    )
+    with pytest.raises(UnknownError):
+        with caplog.at_level(logging.DEBUG):
+            await test_charger_new.set_divert_mode("fast")
+    assert "Problem issuing command: error" in caplog.text