"""Test cases for the API stream sensor."""

from homeassistant.bootstrap import async_setup_component

from tests.common import assert_setup_component
from .test_auth import test_auth_active_with_token


async def test_websocket_api(
    hass, no_auth_websocket_client, hass_access_token, legacy_auth
):
    """Test API streams."""
    with assert_setup_component(1):
        await async_setup_component(
            hass, "sensor", {"sensor": {"platform": "websocket_api"}}
        )

    state = hass.states.get("sensor.connected_clients")
    assert state.state == "0"

    await test_auth_active_with_token(hass, no_auth_websocket_client, hass_access_token)

    state = hass.states.get("sensor.connected_clients")
    assert state.state == "1"

    await no_auth_websocket_client.close()
    await hass.async_block_till_done()

    state = hass.states.get("sensor.connected_clients")
    assert state.state == "0"