async def test_parallel_updates_sync_platform(opp):
"""Test sync platform parallel_updates default set to 1."""
platform = MockPlatform()
mock_entity_platform(opp, "test_domain.platform", platform)
component = EntityComponent(_LOGGER, DOMAIN, opp)
component._platforms = {}
await component.async_setup({DOMAIN: {"platform": "platform"}})
await opp.async_block_till_done()
handle = list(component._platforms.values())[-1]
class SyncEntity(MockEntity):
"""Mock entity that has update."""
async def update(self):
pass
entity = SyncEntity()
await handle.async_add_entities([entity])
assert entity.parallel_updates is not None
assert entity.parallel_updates._value == 1
async def test_parallel_updates_async_platform(opp):
"""Test async platform does not have parallel_updates limit by default."""
platform = MockPlatform()
mock_entity_platform(opp, "test_domain.platform", platform)
component = EntityComponent(_LOGGER, DOMAIN, opp)
component._platforms = {}
await component.async_setup({DOMAIN: {"platform": "platform"}})
await opp.async_block_till_done()
handle = list(component._platforms.values())[-1]
assert handle.parallel_updates is None
class AsyncEntity(MockEntity):
"""Mock entity that has async_update."""
async def async_update(self):
pass
entity = AsyncEntity()
await handle.async_add_entities([entity])
assert entity.parallel_updates is None
async def test_parallel_updates_async_platform_with_constant(opp):
"""Test async platform can set parallel_updates limit."""
platform = MockPlatform()
platform.PARALLEL_UPDATES = 2
mock_entity_platform(opp, "test_domain.platform", platform)
component = EntityComponent(_LOGGER, DOMAIN, opp)
component._platforms = {}
await component.async_setup({DOMAIN: {"platform": "platform"}})
await opp.async_block_till_done()
handle = list(component._platforms.values())[-1]
class AsyncEntity(MockEntity):
"""Mock entity that has async_update."""
async def async_update(self):
pass
entity = AsyncEntity()
await handle.async_add_entities([entity])
assert entity.parallel_updates is not None
assert entity.parallel_updates._value == 2