28 lines
974 B
Python
28 lines
974 B
Python
"""Test ping id allocation."""
|
|
|
|
from homeassistant.components.ping import async_get_next_ping_id
|
|
from homeassistant.components.ping.const import (
|
|
DEFAULT_START_ID,
|
|
DOMAIN,
|
|
MAX_PING_ID,
|
|
PING_ID,
|
|
)
|
|
|
|
|
|
async def test_async_get_next_ping_id(hass):
|
|
"""Verify we allocate ping ids as expected."""
|
|
hass.data[DOMAIN] = {PING_ID: DEFAULT_START_ID}
|
|
|
|
assert async_get_next_ping_id(hass) == DEFAULT_START_ID + 1
|
|
assert async_get_next_ping_id(hass) == DEFAULT_START_ID + 2
|
|
assert async_get_next_ping_id(hass, 2) == DEFAULT_START_ID + 3
|
|
assert async_get_next_ping_id(hass) == DEFAULT_START_ID + 5
|
|
|
|
hass.data[DOMAIN][PING_ID] = MAX_PING_ID
|
|
assert async_get_next_ping_id(hass) == DEFAULT_START_ID + 1
|
|
assert async_get_next_ping_id(hass) == DEFAULT_START_ID + 2
|
|
|
|
hass.data[DOMAIN][PING_ID] = MAX_PING_ID
|
|
assert async_get_next_ping_id(hass, 2) == DEFAULT_START_ID + 1
|
|
assert async_get_next_ping_id(hass) == DEFAULT_START_ID + 3
|