"""Define tests for the GeoJSON Events general setup."""

from unittest.mock import patch

from homeassistant.components.geo_location import DOMAIN as GEO_LOCATION_DOMAIN
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er

from . import _generate_mock_feed_entry

from tests.common import MockConfigEntry


async def test_component_unload_config_entry(
    hass: HomeAssistant, config_entry: MockConfigEntry
) -> None:
    """Test that loading and unloading of a config entry works."""
    config_entry.add_to_hass(hass)
    with patch(
        "aio_geojson_generic_client.GenericFeedManager.update"
    ) as mock_feed_manager_update:
        # Load config entry.
        assert await hass.config_entries.async_setup(config_entry.entry_id)
        await hass.async_block_till_done()
        assert mock_feed_manager_update.call_count == 1
        assert config_entry.state is ConfigEntryState.LOADED
        # Unload config entry.
        assert await hass.config_entries.async_unload(config_entry.entry_id)
        await hass.async_block_till_done()
        assert config_entry.state is ConfigEntryState.NOT_LOADED


async def test_remove_orphaned_entities(
    hass: HomeAssistant,
    entity_registry: er.EntityRegistry,
    config_entry: MockConfigEntry,
) -> None:
    """Test removing orphaned geolocation entities."""
    config_entry.add_to_hass(hass)

    entity_registry.async_get_or_create(
        GEO_LOCATION_DOMAIN, "geo_json_events", "1", config_entry=config_entry
    )
    entity_registry.async_get_or_create(
        GEO_LOCATION_DOMAIN, "geo_json_events", "2", config_entry=config_entry
    )
    entity_registry.async_get_or_create(
        GEO_LOCATION_DOMAIN, "geo_json_events", "3", config_entry=config_entry
    )

    # There should now be 3 "orphaned" entries available which will be removed
    # when the component is set up.
    entries = er.async_entries_for_config_entry(entity_registry, config_entry.entry_id)
    assert len(entries) == 3

    # Set up a mock feed entry for this test.
    mock_entry_1 = _generate_mock_feed_entry(
        "1234",
        "Title 1",
        15.5,
        (38.0, -3.0),
    )

    with patch(
        "aio_geojson_client.feed.GeoJsonFeed.update",
        return_value=("OK", [mock_entry_1]),
    ):
        assert await hass.config_entries.async_setup(config_entry.entry_id)
        await hass.async_block_till_done()

        # 1 geolocation entity.
        entries = er.async_entries_for_config_entry(
            entity_registry, config_entry.entry_id
        )
        assert len(entries) == 1

        assert len(hass.states.async_entity_ids(GEO_LOCATION_DOMAIN)) == 1
