65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
"""The test for the sql sensor platform."""
|
|
import unittest
|
|
import pytest
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.sensor.sql import validate_sql_select
|
|
from homeassistant.setup import setup_component
|
|
from homeassistant.const import STATE_UNKNOWN
|
|
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
class TestSQLSensor(unittest.TestCase):
|
|
"""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."""
|
|
config = {
|
|
'sensor': {
|
|
'platform': 'sql',
|
|
'db_url': 'sqlite://',
|
|
'queries': [{
|
|
'name': 'count_tables',
|
|
'query': 'SELECT 5 as value',
|
|
'column': 'value',
|
|
}]
|
|
}
|
|
}
|
|
|
|
assert setup_component(self.hass, 'sensor', config)
|
|
|
|
state = self.hass.states.get('sensor.count_tables')
|
|
assert state.state == '5'
|
|
assert state.attributes['value'] == 5
|
|
|
|
def test_invalid_query(self):
|
|
"""Test the SQL sensor for invalid queries."""
|
|
with pytest.raises(vol.Invalid):
|
|
validate_sql_select("DROP TABLE *")
|
|
|
|
config = {
|
|
'sensor': {
|
|
'platform': 'sql',
|
|
'db_url': 'sqlite://',
|
|
'queries': [{
|
|
'name': 'count_tables',
|
|
'query': 'SELECT * value FROM sqlite_master;',
|
|
'column': 'value',
|
|
}]
|
|
}
|
|
}
|
|
|
|
assert setup_component(self.hass, 'sensor', config)
|
|
|
|
state = self.hass.states.get('sensor.count_tables')
|
|
self.assertEqual(state.state, STATE_UNKNOWN)
|