"""Tests for the diagnostics data provided by the CPU Speed integration."""
from unittest.mock import patch

from homeassistant.core import HomeAssistant

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


async def test_diagnostics(
    hass: HomeAssistant,
    hass_client: ClientSessionGenerator,
    init_integration: MockConfigEntry,
) -> None:
    """Test diagnostics."""
    info = {
        "hz_actual": (3200000001, 0),
        "arch_string_raw": "aargh",
        "brand_raw": "Intel Ryzen 7",
        "hz_advertised": (3600000001, 0),
    }

    with patch(
        "homeassistant.components.cpuspeed.diagnostics.cpuinfo.get_cpu_info",
        return_value=info,
    ):
        assert await get_diagnostics_for_config_entry(
            hass, hass_client, init_integration
        ) == {
            "hz_actual": [3200000001, 0],
            "arch_string_raw": "aargh",
            "brand_raw": "Intel Ryzen 7",
            "hz_advertised": [3600000001, 0],
        }