Rewrite simulated unittest tests to pytest style test functions (#41115)

This commit is contained in:
Eliot Wong 2020-10-04 07:48:47 -04:00 committed by GitHub
parent c3ef05a37e
commit f7696fdb9f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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