Track entity sources (#37258)

Co-authored-by: David Mulcahey <david.mulcahey@me.com>
2020-08-19 14:57:38 +02:00
parent 24a16ff8fe
commit 3dc79aa60a
15 changed files with 379 additions and 154 deletions
@@ -1,85 +1,55 @@
 """The tests for the Switch component."""
-# pylint: disable=protected-access
-import unittest
+import pytest

 from homeassistant import core
 from homeassistant.components import switch
 from homeassistant.const import CONF_PLATFORM
-from homeassistant.setup import async_setup_component, setup_component
+from homeassistant.setup import async_setup_component

-from tests.common import get_test_home_assistant, mock_entity_platform
 from tests.components.switch import common


-class TestSwitch(unittest.TestCase):
-    """Test the switch module."""
-
-    # pylint: disable=invalid-name
-    def setUp(self):
-        """Set up things to be run when tests are started."""
-        self.hass = get_test_home_assistant()
-        platform = getattr(self.hass.components, "test.switch")
-        platform.init()
-        # Switch 1 is ON, switch 2 is OFF
-        self.switch_1, self.switch_2, self.switch_3 = platform.ENTITIES
-        self.addCleanup(self.hass.stop)
-
-    def test_methods(self):
-        """Test is_on, turn_on, turn_off methods."""
-        assert setup_component(
-            self.hass, switch.DOMAIN, {switch.DOMAIN: {CONF_PLATFORM: "test"}}
-        )
-        self.hass.block_till_done()
-        assert switch.is_on(self.hass, self.switch_1.entity_id)
-        assert not switch.is_on(self.hass, self.switch_2.entity_id)
-        assert not switch.is_on(self.hass, self.switch_3.entity_id)
-
-        common.turn_off(self.hass, self.switch_1.entity_id)
-        common.turn_on(self.hass, self.switch_2.entity_id)
-
-        self.hass.block_till_done()
-
-        assert not switch.is_on(self.hass, self.switch_1.entity_id)
-        assert switch.is_on(self.hass, self.switch_2.entity_id)
-
-        # Turn all off
-        common.turn_off(self.hass)
-
-        self.hass.block_till_done()
-
-        assert not switch.is_on(self.hass, self.switch_1.entity_id)
-        assert not switch.is_on(self.hass, self.switch_2.entity_id)
-        assert not switch.is_on(self.hass, self.switch_3.entity_id)
-
-        # Turn all on
-        common.turn_on(self.hass)
-
-        self.hass.block_till_done()
-
-        assert switch.is_on(self.hass, self.switch_1.entity_id)
-        assert switch.is_on(self.hass, self.switch_2.entity_id)
-        assert switch.is_on(self.hass, self.switch_3.entity_id)
-
-    def test_setup_two_platforms(self):
-        """Test with bad configuration."""
-        # Test if switch component returns 0 switches
-        test_platform = getattr(self.hass.components, "test.switch")
-        test_platform.init(True)
-
-        mock_entity_platform(self.hass, "switch.test2", test_platform)
-        test_platform.init(False)
-
-        assert setup_component(
-            self.hass,
-            switch.DOMAIN,
-            {
-                switch.DOMAIN: {CONF_PLATFORM: "test"},
-                f"{switch.DOMAIN} 2": {CONF_PLATFORM: "test2"},
-            },
-        )
+@pytest.fixture(autouse=True)
+def entities(hass):
+    """Initialize the test switch."""
+    platform = getattr(hass.components, "test.switch")
+    platform.init()
+    yield platform.ENTITIES


-async def test_switch_context(hass, hass_admin_user):
+async def test_methods(hass, entities):
+    """Test is_on, turn_on, turn_off methods."""
+    switch_1, switch_2, switch_3 = entities
+    assert await async_setup_component(
+        hass, switch.DOMAIN, {switch.DOMAIN: {CONF_PLATFORM: "test"}}
+    )
+    await hass.async_block_till_done()
+    assert switch.is_on(hass, switch_1.entity_id)
+    assert not switch.is_on(hass, switch_2.entity_id)
+    assert not switch.is_on(hass, switch_3.entity_id)
+
+    await common.async_turn_off(hass, switch_1.entity_id)
+    await common.async_turn_on(hass, switch_2.entity_id)
+
+    assert not switch.is_on(hass, switch_1.entity_id)
+    assert switch.is_on(hass, switch_2.entity_id)
+
+    # Turn all off
+    await common.async_turn_off(hass)
+
+    assert not switch.is_on(hass, switch_1.entity_id)
+    assert not switch.is_on(hass, switch_2.entity_id)
+    assert not switch.is_on(hass, switch_3.entity_id)
+
+    # Turn all on
+    await common.async_turn_on(hass)
+
+    assert switch.is_on(hass, switch_1.entity_id)
+    assert switch.is_on(hass, switch_2.entity_id)
+    assert switch.is_on(hass, switch_3.entity_id)
+
+
+async def test_switch_context(hass, entities, hass_admin_user):
    """Test that switch context works."""
    assert await async_setup_component(hass, "switch", {"switch": {"platform": "test"}})