binarydev · binarydev · Jul 25, 2025 · Jul 14, 2025 · Jul 14, 2025 · Jul 14, 2025
diff --git a/.github/workflows/constraints.txt b/.github/workflows/constraints.txt
diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
@@ -11,7 +11,7 @@ on:
     - cron: "0 0 * * *"
 
 env:
-  DEFAULT_PYTHON: 3.11
+  DEFAULT_PYTHON: 3.13
 
 jobs:
   pre-commit:
@@ -28,12 +28,12 @@ jobs:
 
       - name: Upgrade pip
         run: |
-          pip install --constraint=.github/workflows/constraints.txt pip
+          pip install pip
           pip --version
 
       - name: Install Python modules
         run: |
-          pip install --constraint=.github/workflows/constraints.txt pre-commit black flake8 reorder-python-imports pipreqs
+          pip install pre-commit black flake8 reorder-python-imports pipreqs
 
       - name: Run pre-commit on all files
         run: |
@@ -61,3 +61,28 @@ jobs:
 
       - name: Hassfest validation
         uses: "home-assistant/actions/hassfest@master"
+
+  pytest:
+    runs-on: "ubuntu-latest"
+    name: Pytest
+    steps:
+      - name: Check out the repository
+        uses: actions/checkout@v4.2.2
+
+      - name: Set up Python ${{ env.DEFAULT_PYTHON }}
+        uses: actions/setup-python@v5.6.0
+        with:
+          python-version: ${{ env.DEFAULT_PYTHON }}
+
+      - name: Upgrade pip
+        run: |
+          pip install pip
+          pip --version
+
+      - name: Install dependencies
+        run: |
+          pip install -r requirements.txt
+
+      - name: Run pytest
+        run: |
+          pytest --maxfail=3 --disable-warnings --color=yes
diff --git a/local_test.py → _test_local_access.py b/local_test.py → _test_local_access.py
diff --git a/custom_components/generac/__init__.py b/custom_components/generac/__init__.py
@@ -19,7 +19,6 @@
 from .coordinator import GeneracDataUpdateCoordinator
 from .utils import async_client_session
 
-
 _LOGGER: logging.Logger = logging.getLogger(__package__)
 
 

diff --git a/pytest.ini b/pytest.ini
@@ -0,0 +1,4 @@
+[pytest]
+asyncio_mode = auto
+markers =
+    socket: Enables socket connections
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,13 @@
 beautifulsoup4==4.13.4
 dacite==1.9.2
-homeassistant==2025.3.3
+homeassistant==2025.7.3
 voluptuous==0.15.2
 pre-commit==4.2.0
 reorder-python-imports==3.15.0
+pytest==8.4.0
+pytest-homeassistant-custom-component==0.13.263
+pytest-asyncio==1.0.0
+pytest-socket==0.7.0
+respx==0.22.0
+aioresponses==0.7.8
 Brotli==1.1.0
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -0,0 +1 @@
+"""Tests for the Generac integration."""
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -0,0 +1,10 @@
+"""Global fixtures for generac integration."""
+import pytest
+
+pytest_plugins = "pytest_homeassistant_custom_component"
+
+
+@pytest.fixture(autouse=True)
+def auto_enable_custom_integrations(enable_custom_integrations):
+    """Enable custom integrations defined in the test dir."""
+    yield
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -0,0 +1,86 @@
+"""Test the Generac API."""
+import re
+
+import aiohttp
+import pytest
+from aioresponses import aioresponses
+from custom_components.generac.api import GeneracApiClient
+from custom_components.generac.api import get_setting_json
+
+
+def test_get_setting_json():
+    """Test the get_setting_json function."""
+    html = """<html>
+<head>
+<script>
+var SETTINGS = {"key": "value"};
+</script>
+</head>
+<body>
+</body>
+</html>"""
+    assert get_setting_json(html) == {"key": "value"}
+
+
+def test_get_setting_json_no_settings():
+    """Test the get_setting_json function when there are no settings."""
+    html = """<html>
+<head>
+</head>
+<body>
+</body>
+</html>"""
+    assert get_setting_json(html) is None
+
+
+@pytest.mark.asyncio
+async def test_api_flow():
+    """Test the full API flow."""
+    with aioresponses() as m:
+        m.get(
+            "https://app.mobilelinkgen.com/api/Auth/SignIn?email=test-username",
+            status=200,
+            body="""<html><head><script>
+var SETTINGS = {"csrf": "test-csrf", "transId": "test-trans-id", "config": {}, "hosts": {}};
+</script></head><body></body></html>""",
+        )
+        m.post(
+            re.compile(
+                r"https://generacconnectivity.b2clogin.com/generacconnectivity.onmicrosoft.com/B2C_1A_MobileLink_SignIn/SelfAsserted.*"
+            ),
+            status=200,
+            payload={"status": "200"},
+        )
+        m.get(
+            re.compile(
+                r"https://generacconnectivity.b2clogin.com/generacconnectivity.onmicrosoft.com/B2C_1A_MobileLink_SignIn/api/CombinedSigninAndSignup/confirmed.*"
+            ),
+            status=200,
+            body="""<html><body><form action="https://app.mobilelinkgen.com/test-action"><input name="state" value="test-state"><input name="code" value="test-code"></form></body></html>""",
+        )
+        m.post("https://app.mobilelinkgen.com/test-action", status=200)
+        m.get(
+            "https://app.mobilelinkgen.com/api/v2/Apparatus/list",
+            status=200,
+            payload=[
+                {
+                    "apparatusId": 12345,
+                    "type": 0,
+                    "name": "test-name",
+                }
+            ],
+        )
+        m.get(
+            "https://app.mobilelinkgen.com/api/v1/Apparatus/details/12345",
+            status=200,
+            payload={"key": "value"},
+        )
+
+        async with aiohttp.ClientSession() as session:
+            client = GeneracApiClient("test-username", "test-password", session)
+            data = await client.async_get_data()
+            assert data is not None
+            assert data["12345"].apparatus.apparatusId == 12345
+            # This is a bit of a hack, but we don't have the ApparatusDetail model fully defined for this test
+            # assert data["12345"].detail.raw == {"key": "value"}
+            assert client.csrf == "test-csrf"
diff --git a/tests/test_binary_sensor.py b/tests/test_binary_sensor.py
@@ -0,0 +1,100 @@
+"""Test the Generac binary sensor platform."""
+from unittest.mock import MagicMock
+
+from custom_components.generac.binary_sensor import GeneracConnectedSensor
+from custom_components.generac.binary_sensor import GeneracConnectingSensor
+from custom_components.generac.binary_sensor import GeneracMaintenanceAlertSensor
+from custom_components.generac.binary_sensor import GeneracWarningSensor
+from custom_components.generac.models import Apparatus
+from custom_components.generac.models import ApparatusDetail
+from custom_components.generac.models import Item
+
+
+def get_mock_item(
+    is_connected: bool,
+    is_connecting: bool,
+    has_maintenance_alert: bool,
+    show_warning: bool,
+) -> Item:
+    """Return a mock Item object."""
+    return Item(
+        apparatus=Apparatus(),
+        apparatusDetail=ApparatusDetail(
+            isConnected=is_connected,
+            isConnecting=is_connecting,
+            hasMaintenanceAlert=has_maintenance_alert,
+            showWarning=show_warning,
+        ),
+    )
+
+
+async def test_connected_sensor(hass):
+    """Test the connected sensor."""
+    coordinator = MagicMock()
+    entry = MagicMock()
+
+    # Test when connected
+    item = get_mock_item(True, False, False, False)
+    sensor = GeneracConnectedSensor(coordinator, entry, "12345", item)
+    assert sensor.is_on is True
+    assert sensor.name == "generac_12345_is_connected"
+    assert sensor.device_class == "connectivity"
+
+    # Test when not connected
+    item = get_mock_item(False, False, False, False)
+    sensor = GeneracConnectedSensor(coordinator, entry, "12345", item)
+    assert sensor.is_on is False
+
+
+async def test_connecting_sensor(hass):
+    """Test the connecting sensor."""
+    coordinator = MagicMock()
+    entry = MagicMock()
+
+    # Test when connecting
+    item = get_mock_item(False, True, False, False)
+    sensor = GeneracConnectingSensor(coordinator, entry, "12345", item)
+    assert sensor.is_on is True
+    assert sensor.name == "generac_12345_is_connecting"
+    assert sensor.device_class == "connectivity"
+
+    # Test when not connecting
+    item = get_mock_item(False, False, False, False)
+    sensor = GeneracConnectingSensor(coordinator, entry, "12345", item)
+    assert sensor.is_on is False
+
+
+async def test_maintenance_alert_sensor(hass):
+    """Test the maintenance alert sensor."""
+    coordinator = MagicMock()
+    entry = MagicMock()
+
+    # Test when maintenance alert is active
+    item = get_mock_item(False, False, True, False)
+    sensor = GeneracMaintenanceAlertSensor(coordinator, entry, "12345", item)
+    assert sensor.is_on is True
+    assert sensor.name == "generac_12345_has_maintenance_alert"
+    assert sensor.device_class == "safety"
+
+    # Test when maintenance alert is not active
+    item = get_mock_item(False, False, False, False)
+    sensor = GeneracMaintenanceAlertSensor(coordinator, entry, "12345", item)
+    assert sensor.is_on is False
+
+
+async def test_warning_sensor(hass):
+    """Test the warning sensor."""
+    coordinator = MagicMock()
+    entry = MagicMock()
+
+    # Test when warning is active
+    item = get_mock_item(False, False, False, True)
+    sensor = GeneracWarningSensor(coordinator, entry, "12345", item)
+    assert sensor.is_on is True
+    assert sensor.name == "generac_12345_show_warning"
+    assert sensor.device_class == "safety"
+
+    # Test when warning is not active
+    item = get_mock_item(False, False, False, False)
+    sensor = GeneracWarningSensor(coordinator, entry, "12345", item)
+    assert sensor.is_on is False
diff --git a/tests/test_config_flow.py b/tests/test_config_flow.py
@@ -0,0 +1,41 @@
+"""Test the Generac config flow."""
+from unittest.mock import patch
+
+from custom_components.generac.const import DOMAIN
+from homeassistant import config_entries
+from homeassistant import setup
+from homeassistant.core import HomeAssistant
+
+
+async def test_form(hass: HomeAssistant) -> None:
+    """Test we get the form."""
+    await setup.async_setup_component(hass, "persistent_notification", {})
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": config_entries.SOURCE_USER}
+    )
+    assert result["type"] == "form"
+    assert result["errors"] == {}
+
+    with patch(
+        "custom_components.generac.config_flow.GeneracApiClient.async_get_data",
+        return_value=True,
+    ), patch(
+        "custom_components.generac.async_setup_entry",
+        return_value=True,
+    ) as mock_setup_entry:
+        result2 = await hass.config_entries.flow.async_configure(
+            result["flow_id"],
+            {
+                "username": "test-username",
+                "password": "test-password",
+            },
+        )
+        await hass.async_block_till_done()
+
+    assert result2["type"] == "create_entry"
+    assert result2["title"] == "test-username"
+    assert result2["data"] == {
+        "username": "test-username",
+        "password": "test-password",
+    }
+    assert len(mock_setup_entry.mock_calls) == 1
diff --git a/tests/test_coordinator.py b/tests/test_coordinator.py
@@ -0,0 +1,43 @@
+"""Test the Generac data update coordinator."""
+from unittest.mock import AsyncMock
+from unittest.mock import MagicMock
+
+import pytest
+from custom_components.generac.coordinator import GeneracDataUpdateCoordinator
+from homeassistant.helpers.update_coordinator import UpdateFailed
+
+
+async def test_coordinator_init(hass):
+    """Test the coordinator initialization."""
+    config_entry = MagicMock()
+    config_entry.options = {}
+    client = MagicMock()
+    coordinator = GeneracDataUpdateCoordinator(hass, client, config_entry)
+    assert coordinator.hass is hass
+    assert coordinator.api is client
+    assert coordinator._config_entry is config_entry
+    assert not coordinator.is_online
+
+
+async def test_coordinator_update_data(hass):
+    """Test the coordinator update data."""
+    config_entry = MagicMock()
+    config_entry.options = {}
+    client = MagicMock()
+    client.async_get_data = AsyncMock(return_value={"foo": "bar"})
+    coordinator = GeneracDataUpdateCoordinator(hass, client, config_entry)
+    coordinator.data = await coordinator._async_update_data()
+    assert coordinator.data == {"foo": "bar"}
+    assert coordinator.is_online
+
+
+async def test_coordinator_update_data_fails(hass):
+    """Test the coordinator update data fails."""
+    config_entry = MagicMock()
+    config_entry.options = {}
+    client = MagicMock()
+    client.async_get_data = AsyncMock(side_effect=Exception)
+    coordinator = GeneracDataUpdateCoordinator(hass, client, config_entry)
+    with pytest.raises(UpdateFailed):
+        await coordinator._async_update_data()
+    assert not coordinator.is_online
Original file line number	Diff line number	Diff line change
Expand Up		@@ -19,7 +19,6 @@
		from .coordinator import GeneracDataUpdateCoordinator
		from .utils import async_client_session


		_LOGGER: logging.Logger = logging.getLogger(__package__)


Expand Down