From f7696fdb9f00007aa0989be5cbf83e8487247a7b Mon Sep 17 00:00:00 2001 From: Eliot Wong Date: Sun, 4 Oct 2020 07:48:47 -0400 Subject: [PATCH] Rewrite simulated unittest tests to pytest style test functions (#41115) --- tests/components/simulated/test_sensor.py | 49 ++++++++--------------- 1 file changed, 17 insertions(+), 32 deletions(-) diff --git a/tests/components/simulated/test_sensor.py b/tests/components/simulated/test_sensor.py index 09e77f7b283..781b5d494ba 100644 --- a/tests/components/simulated/test_sensor.py +++ b/tests/components/simulated/test_sensor.py @@ -1,6 +1,4 @@ """The tests for the simulated sensor.""" -import unittest - from homeassistant.components.simulated.sensor import ( CONF_AMP, CONF_FWHM, @@ -19,37 +17,24 @@ from homeassistant.components.simulated.sensor import ( DEFAULT_SEED, ) from homeassistant.const import CONF_FRIENDLY_NAME -from homeassistant.setup import setup_component - -from tests.common import get_test_home_assistant +from homeassistant.setup import async_setup_component -class TestSimulatedSensor(unittest.TestCase): - """Test the simulated sensor.""" +async def test_simulated_sensor_default_config(hass): + """Test default config.""" + config = {"sensor": {"platform": "simulated"}} + assert await async_setup_component(hass, "sensor", config) + await hass.async_block_till_done() - def setup_method(self, method): - """Set up things to be run when tests are started.""" - self.hass = get_test_home_assistant() + assert len(hass.states.async_entity_ids()) == 1 + state = hass.states.get("sensor.simulated") - def teardown_method(self, method): - """Stop everything that was started.""" - self.hass.stop() - - def test_default_config(self): - """Test default config.""" - config = {"sensor": {"platform": "simulated"}} - assert setup_component(self.hass, "sensor", config) - self.hass.block_till_done() - - assert len(self.hass.states.entity_ids()) == 1 - state = self.hass.states.get("sensor.simulated") - - assert state.attributes.get(CONF_FRIENDLY_NAME) == DEFAULT_NAME - assert state.attributes.get(CONF_AMP) == DEFAULT_AMP - assert state.attributes.get(CONF_UNIT) is None - assert state.attributes.get(CONF_MEAN) == DEFAULT_MEAN - assert state.attributes.get(CONF_PERIOD) == 60.0 - assert state.attributes.get(CONF_PHASE) == DEFAULT_PHASE - assert state.attributes.get(CONF_FWHM) == DEFAULT_FWHM - assert state.attributes.get(CONF_SEED) == DEFAULT_SEED - assert state.attributes.get(CONF_RELATIVE_TO_EPOCH) == DEFAULT_RELATIVE_TO_EPOCH + assert state.attributes.get(CONF_FRIENDLY_NAME) == DEFAULT_NAME + assert state.attributes.get(CONF_AMP) == DEFAULT_AMP + assert state.attributes.get(CONF_UNIT) is None + assert state.attributes.get(CONF_MEAN) == DEFAULT_MEAN + assert state.attributes.get(CONF_PERIOD) == 60.0 + assert state.attributes.get(CONF_PHASE) == DEFAULT_PHASE + assert state.attributes.get(CONF_FWHM) == DEFAULT_FWHM + assert state.attributes.get(CONF_SEED) == DEFAULT_SEED + assert state.attributes.get(CONF_RELATIVE_TO_EPOCH) == DEFAULT_RELATIVE_TO_EPOCH