mirror of
https://github.com/home-assistant/core.git
synced 2025-07-19 11:17:21 +00:00
Rewrite SQL tests to pytest style (#41016)
* Remove unittests and common mock from SQL component tests * Switch to async versions of setup_component and block_till_done
This commit is contained in:
parent
9611e5223a
commit
473afc4837
@ -1,28 +1,13 @@
|
|||||||
"""The test for the sql sensor platform."""
|
"""The test for the sql sensor platform."""
|
||||||
import unittest
|
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
import voluptuous as vol
|
import voluptuous as vol
|
||||||
|
|
||||||
from homeassistant.components.sql.sensor import validate_sql_select
|
from homeassistant.components.sql.sensor import validate_sql_select
|
||||||
from homeassistant.const import STATE_UNKNOWN
|
from homeassistant.const import STATE_UNKNOWN
|
||||||
from homeassistant.setup import setup_component
|
from homeassistant.setup import async_setup_component
|
||||||
|
|
||||||
from tests.common import get_test_home_assistant
|
|
||||||
|
|
||||||
|
|
||||||
class TestSQLSensor(unittest.TestCase):
|
async def test_query(hass):
|
||||||
"""Test the SQL sensor."""
|
|
||||||
|
|
||||||
def setUp(self):
|
|
||||||
"""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_query(self):
|
|
||||||
"""Test the SQL sensor."""
|
"""Test the SQL sensor."""
|
||||||
config = {
|
config = {
|
||||||
"sensor": {
|
"sensor": {
|
||||||
@ -38,14 +23,15 @@ class TestSQLSensor(unittest.TestCase):
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
assert setup_component(self.hass, "sensor", config)
|
assert await async_setup_component(hass, "sensor", config)
|
||||||
self.hass.block_till_done()
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
state = self.hass.states.get("sensor.count_tables")
|
state = hass.states.get("sensor.count_tables")
|
||||||
assert state.state == "5"
|
assert state.state == "5"
|
||||||
assert state.attributes["value"] == 5
|
assert state.attributes["value"] == 5
|
||||||
|
|
||||||
def test_invalid_query(self):
|
|
||||||
|
async def test_invalid_query(hass):
|
||||||
"""Test the SQL sensor for invalid queries."""
|
"""Test the SQL sensor for invalid queries."""
|
||||||
with pytest.raises(vol.Invalid):
|
with pytest.raises(vol.Invalid):
|
||||||
validate_sql_select("DROP TABLE *")
|
validate_sql_select("DROP TABLE *")
|
||||||
@ -64,8 +50,8 @@ class TestSQLSensor(unittest.TestCase):
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
assert setup_component(self.hass, "sensor", config)
|
assert await async_setup_component(hass, "sensor", config)
|
||||||
self.hass.block_till_done()
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
state = self.hass.states.get("sensor.count_tables")
|
state = hass.states.get("sensor.count_tables")
|
||||||
assert state.state == STATE_UNKNOWN
|
assert state.state == STATE_UNKNOWN
|
||||||
|
Loading…
x
Reference in New Issue
Block a user