diff --git a/tests/components/random/test_binary_sensor.py b/tests/components/random/test_binary_sensor.py index 2f15243c71e..85a8e32018c 100644 --- a/tests/components/random/test_binary_sensor.py +++ b/tests/components/random/test_binary_sensor.py @@ -1,45 +1,44 @@ """The test for the Random binary sensor platform.""" -import unittest - -from homeassistant.setup import setup_component +from homeassistant.setup import async_setup_component from tests.async_mock import patch -from tests.common import get_test_home_assistant -class TestRandomSensor(unittest.TestCase): +async def test_random_binary_sensor_on(hass): """Test the Random binary sensor.""" + config = {"binary_sensor": {"platform": "random", "name": "test"}} - def setup_method(self, method): - """Set up things to be run when tests are started.""" - self.hass = get_test_home_assistant() + with patch( + "homeassistant.components.random.binary_sensor.getrandbits", + return_value=1, + ): + assert await async_setup_component( + hass, + "binary_sensor", + config, + ) + await hass.async_block_till_done() - def teardown_method(self, method): - """Stop everything that was started.""" - self.hass.stop() + state = hass.states.get("binary_sensor.test") - @patch("homeassistant.components.random.binary_sensor.getrandbits", return_value=1) - def test_random_binary_sensor_on(self, mocked): - """Test the Random binary sensor.""" - config = {"binary_sensor": {"platform": "random", "name": "test"}} + assert state.state == "on" - assert setup_component(self.hass, "binary_sensor", config) - self.hass.block_till_done() - state = self.hass.states.get("binary_sensor.test") +async def test_random_binary_sensor_off(hass): + """Test the Random binary sensor.""" + config = {"binary_sensor": {"platform": "random", "name": "test"}} - assert state.state == "on" + with patch( + "homeassistant.components.random.binary_sensor.getrandbits", + return_value=False, + ): + assert await async_setup_component( + hass, + "binary_sensor", + config, + ) + await hass.async_block_till_done() - @patch( - "homeassistant.components.random.binary_sensor.getrandbits", return_value=False - ) - def test_random_binary_sensor_off(self, mocked): - """Test the Random binary sensor.""" - config = {"binary_sensor": {"platform": "random", "name": "test"}} + state = hass.states.get("binary_sensor.test") - assert setup_component(self.hass, "binary_sensor", config) - self.hass.block_till_done() - - state = self.hass.states.get("binary_sensor.test") - - assert state.state == "off" + assert state.state == "off" diff --git a/tests/components/random/test_sensor.py b/tests/components/random/test_sensor.py index 657efc9a0cc..9ac2588ff0a 100644 --- a/tests/components/random/test_sensor.py +++ b/tests/components/random/test_sensor.py @@ -1,37 +1,26 @@ """The test for the random number sensor platform.""" -import unittest - -from homeassistant.setup import setup_component - -from tests.common import get_test_home_assistant +from homeassistant.setup import async_setup_component -class TestRandomSensor(unittest.TestCase): +async def test_random_sensor(hass): """Test the Random number sensor.""" - - def setup_method(self, method): - """Set up things to be run when tests are started.""" - self.hass = get_test_home_assistant() - - def teardown_method(self, method): - """Stop everything that was started.""" - self.hass.stop() - - def test_random_sensor(self): - """Test the Random number sensor.""" - config = { - "sensor": { - "platform": "random", - "name": "test", - "minimum": 10, - "maximum": 20, - } + config = { + "sensor": { + "platform": "random", + "name": "test", + "minimum": 10, + "maximum": 20, } + } - assert setup_component(self.hass, "sensor", config) - self.hass.block_till_done() + assert await async_setup_component( + hass, + "sensor", + config, + ) + await hass.async_block_till_done() - state = self.hass.states.get("sensor.test") + state = hass.states.get("sensor.test") - assert int(state.state) <= config["sensor"]["maximum"] - assert int(state.state) >= config["sensor"]["minimum"] + assert int(state.state) <= config["sensor"]["maximum"] + assert int(state.state) >= config["sensor"]["minimum"]