Add prusalink test cases and fix config flow (#77544)

tests/components/prusalink/test_camera.py

@@ -20,6 +20,7 @@ async def test_camera_no_job(
     hass: HomeAssistant,
     mock_config_entry,
     mock_api,
+    hass_client,
 ) -> None:
     """Test sensors while no job active."""
     assert await async_setup_component(hass, "prusalink", {})
@@ -27,6 +28,10 @@ async def test_camera_no_job(
     assert state is not None
     assert state.state == "unavailable"
 
+    client = await hass_client()
+    resp = await client.get("/api/camera_proxy/camera.mock_title_job_preview")
+    assert resp.status == 500
+
 
 async def test_camera_active_job(
     hass: HomeAssistant, mock_config_entry, mock_api, mock_job_api_active, hass_client

tests/components/prusalink/test_config_flow.py

@@ -83,8 +83,8 @@ async def test_form_unknown(hass: HomeAssistant) -> None:
     assert result2["errors"] == {"base": "unknown"}
 
 
-async def test_form_invalid_version(hass: HomeAssistant, mock_version_api) -> None:
-    """Test we handle invalid auth."""
+async def test_form_too_low_version(hass: HomeAssistant, mock_version_api) -> None:
+    """Test we handle too low API version."""
     result = await hass.config_entries.flow.async_init(
         DOMAIN, context={"source": config_entries.SOURCE_USER}
     )
@@ -103,6 +103,26 @@ async def test_form_invalid_version(hass: HomeAssistant, mock_version_api) -> No
     assert result2["errors"] == {"base": "not_supported"}
 
 
+async def test_form_invalid_version_2(hass: HomeAssistant, mock_version_api) -> None:
+    """Test we handle invalid version."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": config_entries.SOURCE_USER}
+    )
+
+    mock_version_api["api"] = "i am not a version"
+
+    result2 = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        {
+            "host": "1.1.1.1",
+            "api_key": "abcdefg",
+        },
+    )
+
+    assert result2["type"] == FlowResultType.FORM
+    assert result2["errors"] == {"base": "not_supported"}
+
+
 async def test_form_cannot_connect(hass: HomeAssistant) -> None:
     """Test we handle cannot connect error."""
     result = await hass.config_entries.flow.async_init(

tests/components/prusalink/test_init.py

@@ -1,16 +1,23 @@
 """Test setting up and unloading PrusaLink."""
+from datetime import timedelta
+from unittest.mock import patch
 
+from pyprusalink import InvalidAuth, PrusaLinkError
+import pytest
 
 from homeassistant.config_entries import ConfigEntry, ConfigEntryState
 from homeassistant.core import HomeAssistant
+from homeassistant.util.dt import utcnow
+
+from tests.common import async_fire_time_changed
 
 
-async def test_sensors_no_job(
+async def test_unloading(
     hass: HomeAssistant,
     mock_config_entry: ConfigEntry,
     mock_api,
 ):
-    """Test sensors while no job active."""
+    """Test unloading prusalink."""
     assert await hass.config_entries.async_setup(mock_config_entry.entry_id)
     assert mock_config_entry.state == ConfigEntryState.LOADED
 
@@ -21,3 +28,25 @@ async def test_sensors_no_job(
 
     for state in hass.states.async_all():
         assert state.state == "unavailable"
+
+
+@pytest.mark.parametrize("exception", [InvalidAuth, PrusaLinkError])
+async def test_failed_update(
+    hass: HomeAssistant, mock_config_entry: ConfigEntry, mock_api, exception
+):
+    """Test failed update marks prusalink unavailable."""
+    assert await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    assert mock_config_entry.state == ConfigEntryState.LOADED
+
+    with patch(
+        "homeassistant.components.prusalink.PrusaLink.get_printer",
+        side_effect=exception,
+    ), patch(
+        "homeassistant.components.prusalink.PrusaLink.get_job",
+        side_effect=exception,
+    ):
+        async_fire_time_changed(hass, utcnow() + timedelta(seconds=30), fire_all=True)
+        await hass.async_block_till_done()
+
+    for state in hass.states.async_all():
+        assert state.state == "unavailable"