"""Test Hue diagnostics."""
from homeassistant.core import HomeAssistant

from .conftest import setup_platform

from tests.components.diagnostics import get_diagnostics_for_config_entry
from tests.typing import ClientSessionGenerator


async def test_diagnostics_v1(
    hass: HomeAssistant, hass_client: ClientSessionGenerator, mock_bridge_v1
) -> None:
    """Test diagnostics v1."""
    await setup_platform(hass, mock_bridge_v1, [])
    config_entry = hass.config_entries.async_entries("hue")[0]
    result = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
    assert result == {}


async def test_diagnostics_v2(
    hass: HomeAssistant, hass_client: ClientSessionGenerator, mock_bridge_v2
) -> None:
    """Test diagnostics v2."""
    mock_bridge_v2.api.get_diagnostics.return_value = {"hello": "world"}
    await setup_platform(hass, mock_bridge_v2, [])
    config_entry = hass.config_entries.async_entries("hue")[0]
    result = await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
    assert result == {"hello": "world"}