"""Test the GitHub init file."""
from pytest import LogCaptureFixture

from homeassistant.components.github import CONF_REPOSITORIES
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr

from .common import setup_github_integration

from tests.common import MockConfigEntry
from tests.test_util.aiohttp import AiohttpClientMocker


async def test_device_registry_cleanup(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    aioclient_mock: AiohttpClientMocker,
    caplog: LogCaptureFixture,
) -> None:
    """Test that we remove untracked repositories from the decvice registry."""
    mock_config_entry.options = {CONF_REPOSITORIES: ["home-assistant/core"]}
    await setup_github_integration(hass, mock_config_entry, aioclient_mock)

    device_registry = dr.async_get(hass)
    devices = dr.async_entries_for_config_entry(
        registry=device_registry,
        config_entry_id=mock_config_entry.entry_id,
    )

    assert len(devices) == 1

    mock_config_entry.options = {CONF_REPOSITORIES: []}
    assert await hass.config_entries.async_reload(mock_config_entry.entry_id)
    await hass.async_block_till_done()

    assert (
        f"Unlinking device {devices[0].id} for untracked repository home-assistant/core from config entry {mock_config_entry.entry_id}"
        in caplog.text
    )

    devices = dr.async_entries_for_config_entry(
        registry=device_registry,
        config_entry_id=mock_config_entry.entry_id,
    )

    assert len(devices) == 0


async def test_subscription_setup(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    aioclient_mock: AiohttpClientMocker,
) -> None:
    """Test that we setup event subscription."""
    mock_config_entry.options = {CONF_REPOSITORIES: ["home-assistant/core"]}
    mock_config_entry.pref_disable_polling = False
    await setup_github_integration(hass, mock_config_entry, aioclient_mock)
    assert (
        "https://api.github.com/repos/home-assistant/core/events" in x[1]
        for x in aioclient_mock.mock_calls
    )


async def test_subscription_setup_polling_disabled(
    hass: HomeAssistant,
    mock_config_entry: MockConfigEntry,
    aioclient_mock: AiohttpClientMocker,
) -> None:
    """Test that we do not setup event subscription if polling is disabled."""
    mock_config_entry.options = {CONF_REPOSITORIES: ["home-assistant/core"]}
    mock_config_entry.pref_disable_polling = True
    await setup_github_integration(hass, mock_config_entry, aioclient_mock)
    assert (
        "https://api.github.com/repos/home-assistant/core/events" not in x[1]
        for x in aioclient_mock.mock_calls
    )

    # Prove that we subscribed if the user enabled polling again
    mock_config_entry.pref_disable_polling = False
    assert await hass.config_entries.async_reload(mock_config_entry.entry_id)
    await hass.async_block_till_done()
    assert (
        "https://api.github.com/repos/home-assistant/core/events" in x[1]
        for x in aioclient_mock.mock_calls
    )