aurora watch tests

agentgonzo · Steve · commit ba7f7705fa91 · 2026-03-01T23:25:23.000Z
diff --git a/tests/components/aurorawatch/__init__.py b/tests/components/aurorawatch/__init__.py
@@ -0,0 +1,13 @@
+"""The tests for the AuroraWatch UK integration."""
+
+from homeassistant.core import HomeAssistant
+
+from tests.common import MockConfigEntry
+
+
+async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
+    """Fixture for setting up the component."""
+    config_entry.add_to_hass(hass)
+
+    await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
diff --git a/tests/components/aurorawatch/conftest.py b/tests/components/aurorawatch/conftest.py
@@ -0,0 +1,97 @@
+"""Common fixtures for the AuroraWatch UK tests."""
+
+from collections.abc import Generator
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from homeassistant.components.aurorawatch.const import DOMAIN
+
+from tests.common import MockConfigEntry
+
+# Sample XML responses for testing
+MOCK_STATUS_XML = """<?xml version="1.0" encoding="UTF-8"?>
+<current_status api_version="0.2">
+    <updated>
+        <datetime>2024-01-15T12:00:00Z</datetime>
+    </updated>
+    <site_status status_id="green" project_id="awn" site_id="lancaster" site_url="http://aurorawatch.lancs.ac.uk">
+        <status_description>No significant activity</status_description>
+    </site_status>
+</current_status>
+"""
+
+MOCK_ACTIVITY_XML = """<?xml version="1.0" encoding="UTF-8"?>
+<activity_data api_version="0.2">
+    <activity>
+        <datetime>2024-01-15T11:00:00Z</datetime>
+        <value>45.2</value>
+    </activity>
+    <activity>
+        <datetime>2024-01-15T12:00:00Z</datetime>
+        <value>52.7</value>
+    </activity>
+</activity_data>
+"""
+
+MOCK_INVALID_STATUS_XML = """<?xml version="1.0" encoding="UTF-8"?>
+<invalid_status>
+    <missing_required_fields />
+</invalid_status>
+"""
+
+MOCK_MALFORMED_XML = """<?xml version="1.0" encoding="UTF-8"?>
+<unclosed_tag>
+"""
+
+
+@pytest.fixture
+def mock_setup_entry() -> Generator[AsyncMock]:
+    """Override async_setup_entry."""
+    with patch(
+        "homeassistant.components.aurorawatch.async_setup_entry",
+        return_value=True,
+    ) as mock_setup_entry:
+        yield mock_setup_entry
+
+
+@pytest.fixture
+def mock_aiohttp_session() -> Generator[AsyncMock]:
+    """Mock aiohttp session."""
+    with patch(
+        "homeassistant.components.aurorawatch.coordinator.async_get_clientsession"
+    ) as mock_session_factory:
+        mock_session = AsyncMock()
+        mock_session_factory.return_value = mock_session
+
+        # Create mock responses
+        mock_status_response = AsyncMock()
+        mock_status_response.text = AsyncMock(return_value=MOCK_STATUS_XML)
+        mock_status_response.raise_for_status = AsyncMock()
+
+        mock_activity_response = AsyncMock()
+        mock_activity_response.text = AsyncMock(return_value=MOCK_ACTIVITY_XML)
+        mock_activity_response.raise_for_status = AsyncMock()
+
+        # Mock session.get to return different responses based on URL
+        async def mock_get(url):
+            if "current-status.xml" in url:
+                return mock_status_response
+            if "sum-activity.xml" in url:
+                return mock_activity_response
+            return AsyncMock()
+
+        mock_session.get = mock_get
+
+        yield mock_session
+
+
+@pytest.fixture
+def mock_config_entry() -> MockConfigEntry:
+    """Mock a config entry."""
+    return MockConfigEntry(
+        domain=DOMAIN,
+        title="AuroraWatch UK",
+        data={},
+        unique_id=DOMAIN,
+    )
diff --git a/tests/components/aurorawatch/test_config_flow.py b/tests/components/aurorawatch/test_config_flow.py
@@ -0,0 +1,58 @@
+"""Test the AuroraWatch UK config flow."""
+
+from unittest.mock import AsyncMock
+
+import pytest
+
+from homeassistant.components.aurorawatch.const import DOMAIN
+from homeassistant.config_entries import SOURCE_USER
+from homeassistant.core import HomeAssistant
+from homeassistant.data_entry_flow import FlowResultType
+
+from tests.common import MockConfigEntry
+
+
+async def test_full_flow(
+    hass: HomeAssistant,
+    mock_setup_entry: AsyncMock,
+    mock_aiohttp_session: AsyncMock,
+) -> None:
+    """Test full config flow."""
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_USER}
+    )
+    assert result["type"] is FlowResultType.FORM
+    assert result["errors"] == {}
+    assert result["step_id"] == "user"
+
+    result = await hass.config_entries.flow.async_configure(
+        result["flow_id"],
+        user_input={},
+    )
+    await hass.async_block_till_done()
+
+    assert result["type"] is FlowResultType.CREATE_ENTRY
+    assert result["title"] == "AuroraWatch UK"
+    assert result["data"] == {}
+    assert len(mock_setup_entry.mock_calls) == 1
+
+
+async def test_flow_already_configured(
+    hass: HomeAssistant,
+    mock_aiohttp_session: AsyncMock,
+) -> None:
+    """Test config flow aborts if already configured."""
+    mock_entry = MockConfigEntry(
+        domain=DOMAIN,
+        title="AuroraWatch UK",
+        data={},
+        unique_id=DOMAIN,
+    )
+    mock_entry.add_to_hass(hass)
+
+    result = await hass.config_entries.flow.async_init(
+        DOMAIN, context={"source": SOURCE_USER}
+    )
+
+    assert result["type"] is FlowResultType.ABORT
+    assert result["reason"] == "already_configured"
diff --git a/tests/components/aurorawatch/test_sensor.py b/tests/components/aurorawatch/test_sensor.py
@@ -0,0 +1,206 @@
+"""Test the AuroraWatch UK sensor platform."""
+
+from unittest.mock import AsyncMock
+
+from aiohttp import ClientError
+import pytest
+
+from homeassistant.components.aurorawatch.const import (
+    ATTR_API_VERSION,
+    ATTR_LAST_UPDATED,
+    ATTR_PROJECT_ID,
+    ATTR_SITE_ID,
+    ATTR_SITE_URL,
+    DOMAIN,
+)
+from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
+from homeassistant.const import STATE_UNAVAILABLE
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import entity_registry as er
+
+from . import setup_integration
+
+from tests.common import MockConfigEntry
+
+
+async def test_sensor_setup(
+    hass: HomeAssistant,
+    mock_aiohttp_session: AsyncMock,
+    mock_config_entry: MockConfigEntry,
+) -> None:
+    """Test sensor setup."""
+    await setup_integration(hass, mock_config_entry)
+
+    entity_registry = er.async_get(hass)
+
+    # Check that both sensors are created
+    status_sensor = entity_registry.async_get("sensor.aurorawatch_uk_aurora_status")
+    assert status_sensor
+    assert status_sensor.unique_id == "aurorawatch_aurora_status"
+    assert status_sensor.translation_key == "aurora_status"
+
+    activity_sensor = entity_registry.async_get(
+        "sensor.aurorawatch_uk_geomagnetic_activity"
+    )
+    assert activity_sensor
+    assert activity_sensor.unique_id == "aurorawatch_geomagnetic_activity"
+    assert activity_sensor.translation_key == "geomagnetic_activity"
+
+
+async def test_sensor_states(
+    hass: HomeAssistant,
+    mock_aiohttp_session: AsyncMock,
+    mock_config_entry: MockConfigEntry,
+) -> None:
+    """Test sensor states."""
+    await setup_integration(hass, mock_config_entry)
+
+    # Check status sensor state
+    status_state = hass.states.get("sensor.aurorawatch_uk_aurora_status")
+    assert status_state
+    assert status_state.state == "green"
+
+    # Check activity sensor state
+    activity_state = hass.states.get("sensor.aurorawatch_uk_geomagnetic_activity")
+    assert activity_state
+    assert activity_state.state == "52.7"
+    assert activity_state.attributes.get("unit_of_measurement") == "nT"
+    assert activity_state.attributes.get("state_class") == "measurement"
+
+
+async def test_sensor_attributes(
+    hass: HomeAssistant,
+    mock_aiohttp_session: AsyncMock,
+    mock_config_entry: MockConfigEntry,
+) -> None:
+    """Test sensor attributes."""
+    await setup_integration(hass, mock_config_entry)
+
+    # Check status sensor attributes
+    status_state = hass.states.get("sensor.aurorawatch_uk_aurora_status")
+    assert status_state
+    assert status_state.attributes.get(ATTR_LAST_UPDATED) == "2024-01-15T12:00:00Z"
+    assert status_state.attributes.get(ATTR_PROJECT_ID) == "awn"
+    assert status_state.attributes.get(ATTR_SITE_ID) == "lancaster"
+    assert (
+        status_state.attributes.get(ATTR_SITE_URL) == "http://aurorawatch.lancs.ac.uk"
+    )
+    assert status_state.attributes.get(ATTR_API_VERSION) == "0.2"
+    assert (
+        status_state.attributes.get("attribution") == "Data provided by AuroraWatch UK"
+    )
+
+
+async def test_sensor_update_failure_network_error(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+) -> None:
+    """Test sensor update with network error."""
+    from unittest.mock import patch
+
+    with patch(
+        "homeassistant.components.aurorawatch.coordinator.async_get_clientsession"
+    ) as mock_session_factory:
+        mock_session = AsyncMock()
+        mock_session_factory.return_value = mock_session
+
+        # Mock network error
+        mock_session.get.side_effect = ClientError("Connection failed")
+
+        mock_config_entry.add_to_hass(hass)
+        await hass.config_entries.async_setup(mock_config_entry.entry_id)
+        await hass.async_block_till_done()
+
+        # Sensors should be unavailable
+        status_state = hass.states.get("sensor.aurorawatch_uk_aurora_status")
+        assert status_state
+        assert status_state.state == STATE_UNAVAILABLE
+
+        activity_state = hass.states.get("sensor.aurorawatch_uk_geomagnetic_activity")
+        assert activity_state
+        assert activity_state.state == STATE_UNAVAILABLE
+
+
+async def test_sensor_update_failure_invalid_xml(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+) -> None:
+    """Test sensor update with invalid XML."""
+    from unittest.mock import patch
+
+    from tests.components.aurorawatch.conftest import MOCK_MALFORMED_XML
+
+    with patch(
+        "homeassistant.components.aurorawatch.coordinator.async_get_clientsession"
+    ) as mock_session_factory:
+        mock_session = AsyncMock()
+        mock_session_factory.return_value = mock_session
+
+        # Mock malformed XML response
+        mock_response = AsyncMock()
+        mock_response.text = AsyncMock(return_value=MOCK_MALFORMED_XML)
+        mock_response.raise_for_status = AsyncMock()
+        mock_session.get.return_value = mock_response
+
+        mock_config_entry.add_to_hass(hass)
+        await hass.config_entries.async_setup(mock_config_entry.entry_id)
+        await hass.async_block_till_done()
+
+        # Sensors should be unavailable due to parsing error
+        status_state = hass.states.get("sensor.aurorawatch_uk_aurora_status")
+        assert status_state
+        assert status_state.state == STATE_UNAVAILABLE
+
+
+async def test_sensor_update_failure_missing_fields(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+) -> None:
+    """Test sensor update with missing required fields."""
+    from unittest.mock import patch
+
+    from tests.components.aurorawatch.conftest import MOCK_INVALID_STATUS_XML
+
+    with patch(
+        "homeassistant.components.aurorawatch.coordinator.async_get_clientsession"
+    ) as mock_session_factory:
+        mock_session = AsyncMock()
+        mock_session_factory.return_value = mock_session
+
+        # Mock invalid XML response (missing required fields)
+        mock_response = AsyncMock()
+        mock_response.text = AsyncMock(return_value=MOCK_INVALID_STATUS_XML)
+        mock_response.raise_for_status = AsyncMock()
+        mock_session.get.return_value = mock_response
+
+        mock_config_entry.add_to_hass(hass)
+        await hass.config_entries.async_setup(mock_config_entry.entry_id)
+        await hass.async_block_till_done()
+
+        # Sensors should be unavailable due to missing fields
+        status_state = hass.states.get("sensor.aurorawatch_uk_aurora_status")
+        assert status_state
+        assert status_state.state == STATE_UNAVAILABLE
+
+
+async def test_sensor_device_info(
+    hass: HomeAssistant,
+    mock_aiohttp_session: AsyncMock,
+    mock_config_entry: MockConfigEntry,
+) -> None:
+    """Test sensor device info."""
+    await setup_integration(hass, mock_config_entry)
+
+    entity_registry = er.async_get(hass)
+    status_sensor = entity_registry.async_get("sensor.aurorawatch_uk_aurora_status")
+    assert status_sensor
+    assert status_sensor.device_id
+
+    from homeassistant.helpers import device_registry as dr
+
+    device_registry = dr.async_get(hass)
+    device = device_registry.async_get(status_sensor.device_id)
+    assert device
+    assert device.manufacturer == "Lancaster University"
+    assert device.model == "AuroraWatch UK"
+    assert device.name == "AuroraWatch UK"
