jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-06 21:07:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rewrite scene unittest tests to pytest style test function (#41172)

Browse Source
This commit is contained in:
Ariana Hlavaty 2020-10-15 22:35:30 +01:00 committed by GitHub
parent 162a5d99a2
commit e28e453c08
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 132 additions and 115 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

163
tests/components/scene/test_init.py
View File

@ -1,47 +1,25 @@
"""The tests for the Scene component.""" """The tests for the Scene component."""
import io import io
import unittest
import pytest
from homeassistant.components import light, scene from homeassistant.components import light, scene
from homeassistant.setup import async_setup_component, setup_component from homeassistant.const import ATTR_ENTITY_ID, ENTITY_MATCH_ALL, SERVICE_TURN_ON
from homeassistant.setup import async_setup_component
from homeassistant.util.yaml import loader as yaml_loader from homeassistant.util.yaml import loader as yaml_loader
from tests.common import get_test_home_assistant, mock_service from tests.common import async_mock_service
from tests.components.light import common as common_light
from tests.components.scene import common
class TestScene(unittest.TestCase): @pytest.fixture(autouse=True)
"""Test the scene component.""" def entities(hass):
"""Initialize the test light."""
platform = getattr(hass.components, "test.light")
platform.init()
yield platform.ENTITIES[0:2]
def setUp(self): # pylint: disable=invalid-name
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
test_light = getattr(self.hass.components, "test.light")
test_light.init()
assert setup_component( async def test_config_yaml_alias_anchor(hass, entities):
self.hass, light.DOMAIN, {light.DOMAIN: {"platform": "test"}}
)
self.hass.block_till_done()
self.light_1, self.light_2 = test_light.ENTITIES[0:2]
common_light.turn_off(
self.hass, [self.light_1.entity_id, self.light_2.entity_id]
)
self.hass.block_till_done()
assert not self.light_1.is_on
assert not self.light_2.is_on
self.addCleanup(self.tear_down_cleanup)
def tear_down_cleanup(self):
"""Stop everything that was started."""
self.hass.stop()
def test_config_yaml_alias_anchor(self):
"""Test the usage of YAML aliases and anchors. """Test the usage of YAML aliases and anchors.
The following test scene configuration is equivalent to: The following test scene configuration is equivalent to:
@ -58,40 +36,44 @@ class TestScene(unittest.TestCase):
reference to the original dictionary, instead of creating a copy, so reference to the original dictionary, instead of creating a copy, so
care needs to be taken to not modify the original. care needs to be taken to not modify the original.
""" """
light_1, light_2 = await setup_lights(hass, entities)
entity_state = {"state": "on", "brightness": 100} entity_state = {"state": "on", "brightness": 100}
assert setup_component(
self.hass, assert await async_setup_component(
hass,
scene.DOMAIN, scene.DOMAIN,
{ {
"scene": [ "scene": [
{ {
"name": "test", "name": "test",
"entities": { "entities": {
self.light_1.entity_id: entity_state, light_1.entity_id: entity_state,
self.light_2.entity_id: entity_state, light_2.entity_id: entity_state,
}, },
} }
] ]
}, },
) )
self.hass.block_till_done() await hass.async_block_till_done()
common.activate(self.hass, "scene.test") await activate(hass, "scene.test")
self.hass.block_till_done()
assert self.light_1.is_on assert light.is_on(hass, light_1.entity_id)
assert self.light_2.is_on assert light.is_on(hass, light_2.entity_id)
assert 100 == self.light_1.last_call("turn_on")[1].get("brightness") assert 100 == light_1.last_call("turn_on")[1].get("brightness")
assert 100 == self.light_2.last_call("turn_on")[1].get("brightness") assert 100 == light_2.last_call("turn_on")[1].get("brightness")
def test_config_yaml_bool(self):
async def test_config_yaml_bool(hass, entities):
"""Test parsing of booleans in yaml config.""" """Test parsing of booleans in yaml config."""
light_1, light_2 = await setup_lights(hass, entities)
config = ( config = (
"scene:\n" "scene:\n"
" - name: test\n" " - name: test\n"
" entities:\n" " entities:\n"
f" {self.light_1.entity_id}: on\n" f" {light_1.entity_id}: on\n"
f" {self.light_2.entity_id}:\n" f" {light_2.entity_id}:\n"
" state: on\n" " state: on\n"
" brightness: 100\n" " brightness: 100\n"
) )
@ -99,51 +81,63 @@ class TestScene(unittest.TestCase):
with io.StringIO(config) as file: with io.StringIO(config) as file:
doc = yaml_loader.yaml.safe_load(file) doc = yaml_loader.yaml.safe_load(file)
assert setup_component(self.hass, scene.DOMAIN, doc) assert await async_setup_component(hass, scene.DOMAIN, doc)
common.activate(self.hass, "scene.test") await hass.async_block_till_done()
self.hass.block_till_done()
assert self.light_1.is_on await activate(hass, "scene.test")
assert self.light_2.is_on
assert 100 == self.light_2.last_call("turn_on")[1].get("brightness")
def test_activate_scene(self): assert light.is_on(hass, light_1.entity_id)
assert light.is_on(hass, light_2.entity_id)
assert 100 == light_2.last_call("turn_on")[1].get("brightness")
async def test_activate_scene(hass, entities):
"""Test active scene.""" """Test active scene."""
assert setup_component( light_1, light_2 = await setup_lights(hass, entities)
self.hass,
assert await async_setup_component(
hass,
scene.DOMAIN, scene.DOMAIN,
{ {
"scene": [ "scene": [
{ {
"name": "test", "name": "test",
"entities": { "entities": {
self.light_1.entity_id: "on", light_1.entity_id: "on",
self.light_2.entity_id: {"state": "on", "brightness": 100}, light_2.entity_id: {"state": "on", "brightness": 100},
}, },
} }
] ]
}, },
) )
self.hass.block_till_done() await hass.async_block_till_done()
await activate(hass, "scene.test")
common.activate(self.hass, "scene.test") assert light.is_on(hass, light_1.entity_id)
self.hass.block_till_done() assert light.is_on(hass, light_2.entity_id)
assert light_2.last_call("turn_on")[1].get("brightness") == 100
assert self.light_1.is_on calls = async_mock_service(hass, "light", "turn_on")
assert self.light_2.is_on
assert self.light_2.last_call("turn_on")[1].get("brightness") == 100
turn_on_calls = mock_service(self.hass, "light", "turn_on") await hass.services.async_call(
self.hass.services.call(
scene.DOMAIN, "turn_on", {"transition": 42, "entity_id": "scene.test"} scene.DOMAIN, "turn_on", {"transition": 42, "entity_id": "scene.test"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
assert len(turn_on_calls) == 1 assert len(calls) == 1
assert turn_on_calls[0].domain == "light" assert calls[0].domain == "light"
assert turn_on_calls[0].service == "turn_on" assert calls[0].service == "turn_on"
assert turn_on_calls[0].data.get("transition") == 42 assert calls[0].data.get("transition") == 42
async def activate(hass, entity_id=ENTITY_MATCH_ALL):
"""Activate a scene."""
data = {}
if entity_id:
data[ATTR_ENTITY_ID] = entity_id
await hass.services.async_call(scene.DOMAIN, SERVICE_TURN_ON, data, blocking=True)
async def test_services_registered(hass): async def test_services_registered(hass):
@ -152,3 +146,26 @@ async def test_services_registered(hass):
assert hass.services.has_service("scene", "reload") assert hass.services.has_service("scene", "reload")
assert hass.services.has_service("scene", "turn_on") assert hass.services.has_service("scene", "turn_on")
assert hass.services.has_service("scene", "apply") assert hass.services.has_service("scene", "apply")
async def setup_lights(hass, entities):
"""Set up the light component."""
assert await async_setup_component(
hass, light.DOMAIN, {light.DOMAIN: {"platform": "test"}}
)
await hass.async_block_till_done()
light_1, light_2 = entities
await hass.services.async_call(
"light",
"turn_off",
{"entity_id": [light_1.entity_id, light_2.entity_id]},
blocking=True,
)
await hass.async_block_till_done()
assert not light.is_on(hass, light_1.entity_id)
assert not light.is_on(hass, light_2.entity_id)
return light_1, light_2