mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 21:06:19 +00:00
Black
This commit is contained in:
Paulus Schoutsen
@@ -31,95 +31,104 @@ class TestRestSensorSetup(unittest.TestCase):
|
||||
def test_setup_missing_config(self):
|
||||
"""Test setup with configuration missing required entries."""
|
||||
with assert_setup_component(0):
|
||||
assert setup_component(self.hass, sensor.DOMAIN, {
|
||||
'sensor': {'platform': 'rest'}})
|
||||
assert setup_component(
|
||||
self.hass, sensor.DOMAIN, {"sensor": {"platform": "rest"}}
|
||||
)
|
||||
|
||||
def test_setup_missing_schema(self):
|
||||
"""Test setup with resource missing schema."""
|
||||
with pytest.raises(MissingSchema):
|
||||
rest.setup_platform(self.hass, {
|
||||
'platform': 'rest',
|
||||
'resource': 'localhost',
|
||||
'method': 'GET'
|
||||
}, None)
|
||||
rest.setup_platform(
|
||||
self.hass,
|
||||
{"platform": "rest", "resource": "localhost", "method": "GET"},
|
||||
None,
|
||||
)
|
||||
|
||||
@patch('requests.Session.send',
|
||||
side_effect=requests.exceptions.ConnectionError())
|
||||
@patch("requests.Session.send", side_effect=requests.exceptions.ConnectionError())
|
||||
def test_setup_failed_connect(self, mock_req):
|
||||
"""Test setup when connection error occurs."""
|
||||
with raises(PlatformNotReady):
|
||||
rest.setup_platform(self.hass, {
|
||||
'platform': 'rest',
|
||||
'resource': 'http://localhost',
|
||||
}, lambda devices, update=True: None)
|
||||
rest.setup_platform(
|
||||
self.hass,
|
||||
{"platform": "rest", "resource": "http://localhost"},
|
||||
lambda devices, update=True: None,
|
||||
)
|
||||
|
||||
@patch('requests.Session.send', side_effect=Timeout())
|
||||
@patch("requests.Session.send", side_effect=Timeout())
|
||||
def test_setup_timeout(self, mock_req):
|
||||
"""Test setup when connection timeout occurs."""
|
||||
with raises(PlatformNotReady):
|
||||
rest.setup_platform(self.hass, {
|
||||
'platform': 'rest',
|
||||
'resource': 'http://localhost',
|
||||
}, lambda devices, update=True: None)
|
||||
rest.setup_platform(
|
||||
self.hass,
|
||||
{"platform": "rest", "resource": "http://localhost"},
|
||||
lambda devices, update=True: None,
|
||||
)
|
||||
|
||||
@requests_mock.Mocker()
|
||||
def test_setup_minimum(self, mock_req):
|
||||
"""Test setup with minimum configuration."""
|
||||
mock_req.get('http://localhost', status_code=200)
|
||||
with assert_setup_component(1, 'sensor'):
|
||||
assert setup_component(self.hass, 'sensor', {
|
||||
'sensor': {
|
||||
'platform': 'rest',
|
||||
'resource': 'http://localhost'
|
||||
}
|
||||
})
|
||||
mock_req.get("http://localhost", status_code=200)
|
||||
with assert_setup_component(1, "sensor"):
|
||||
assert setup_component(
|
||||
self.hass,
|
||||
"sensor",
|
||||
{"sensor": {"platform": "rest", "resource": "http://localhost"}},
|
||||
)
|
||||
assert 2 == mock_req.call_count
|
||||
|
||||
@requests_mock.Mocker()
|
||||
def test_setup_get(self, mock_req):
|
||||
"""Test setup with valid configuration."""
|
||||
mock_req.get('http://localhost', status_code=200)
|
||||
with assert_setup_component(1, 'sensor'):
|
||||
assert setup_component(self.hass, 'sensor', {
|
||||
'sensor': {
|
||||
'platform': 'rest',
|
||||
'resource': 'http://localhost',
|
||||
'method': 'GET',
|
||||
'value_template': '{{ value_json.key }}',
|
||||
'name': 'foo',
|
||||
'unit_of_measurement': 'MB',
|
||||
'verify_ssl': 'true',
|
||||
'timeout': 30,
|
||||
'authentication': 'basic',
|
||||
'username': 'my username',
|
||||
'password': 'my password',
|
||||
'headers': {'Accept': 'application/json'}
|
||||
}
|
||||
})
|
||||
mock_req.get("http://localhost", status_code=200)
|
||||
with assert_setup_component(1, "sensor"):
|
||||
assert setup_component(
|
||||
self.hass,
|
||||
"sensor",
|
||||
{
|
||||
"sensor": {
|
||||
"platform": "rest",
|
||||
"resource": "http://localhost",
|
||||
"method": "GET",
|
||||
"value_template": "{{ value_json.key }}",
|
||||
"name": "foo",
|
||||
"unit_of_measurement": "MB",
|
||||
"verify_ssl": "true",
|
||||
"timeout": 30,
|
||||
"authentication": "basic",
|
||||
"username": "my username",
|
||||
"password": "my password",
|
||||
"headers": {"Accept": "application/json"},
|
||||
}
|
||||
},
|
||||
)
|
||||
assert 2 == mock_req.call_count
|
||||
|
||||
@requests_mock.Mocker()
|
||||
def test_setup_post(self, mock_req):
|
||||
"""Test setup with valid configuration."""
|
||||
mock_req.post('http://localhost', status_code=200)
|
||||
with assert_setup_component(1, 'sensor'):
|
||||
assert setup_component(self.hass, 'sensor', {
|
||||
'sensor': {
|
||||
'platform': 'rest',
|
||||
'resource': 'http://localhost',
|
||||
'method': 'POST',
|
||||
'value_template': '{{ value_json.key }}',
|
||||
'payload': '{ "device": "toaster"}',
|
||||
'name': 'foo',
|
||||
'unit_of_measurement': 'MB',
|
||||
'verify_ssl': 'true',
|
||||
'timeout': 30,
|
||||
'authentication': 'basic',
|
||||
'username': 'my username',
|
||||
'password': 'my password',
|
||||
'headers': {'Accept': 'application/json'}
|
||||
}
|
||||
})
|
||||
mock_req.post("http://localhost", status_code=200)
|
||||
with assert_setup_component(1, "sensor"):
|
||||
assert setup_component(
|
||||
self.hass,
|
||||
"sensor",
|
||||
{
|
||||
"sensor": {
|
||||
"platform": "rest",
|
||||
"resource": "http://localhost",
|
||||
"method": "POST",
|
||||
"value_template": "{{ value_json.key }}",
|
||||
"payload": '{ "device": "toaster"}',
|
||||
"name": "foo",
|
||||
"unit_of_measurement": "MB",
|
||||
"verify_ssl": "true",
|
||||
"timeout": 30,
|
||||
"authentication": "basic",
|
||||
"username": "my username",
|
||||
"password": "my password",
|
||||
"headers": {"Accept": "application/json"},
|
||||
}
|
||||
},
|
||||
)
|
||||
assert 2 == mock_req.call_count
|
||||
|
||||
|
||||
@@ -129,21 +138,30 @@ class TestRestSensor(unittest.TestCase):
|
||||
def setUp(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
self.initial_state = 'initial_state'
|
||||
self.rest = Mock('rest.RestData')
|
||||
self.rest.update = Mock('rest.RestData.update',
|
||||
side_effect=self.update_side_effect(
|
||||
'{ "key": "' + self.initial_state + '" }'))
|
||||
self.name = 'foo'
|
||||
self.unit_of_measurement = 'MB'
|
||||
self.initial_state = "initial_state"
|
||||
self.rest = Mock("rest.RestData")
|
||||
self.rest.update = Mock(
|
||||
"rest.RestData.update",
|
||||
side_effect=self.update_side_effect(
|
||||
'{ "key": "' + self.initial_state + '" }'
|
||||
),
|
||||
)
|
||||
self.name = "foo"
|
||||
self.unit_of_measurement = "MB"
|
||||
self.device_class = None
|
||||
self.value_template = template('{{ value_json.key }}')
|
||||
self.value_template = template("{{ value_json.key }}")
|
||||
self.value_template.hass = self.hass
|
||||
self.force_update = False
|
||||
|
||||
self.sensor = rest.RestSensor(
|
||||
self.hass, self.rest, self.name, self.unit_of_measurement,
|
||||
self.device_class, self.value_template, [], self.force_update
|
||||
self.hass,
|
||||
self.rest,
|
||||
self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class,
|
||||
self.value_template,
|
||||
[],
|
||||
self.force_update,
|
||||
)
|
||||
|
||||
def tearDown(self):
|
||||
@@ -174,83 +192,118 @@ class TestRestSensor(unittest.TestCase):
|
||||
def test_update_when_value_is_none(self):
|
||||
"""Test state gets updated to unknown when sensor returns no data."""
|
||||
self.rest.update = Mock(
|
||||
'rest.RestData.update', side_effect=self.update_side_effect(None))
|
||||
"rest.RestData.update", side_effect=self.update_side_effect(None)
|
||||
)
|
||||
self.sensor.update()
|
||||
assert self.sensor.state is None
|
||||
assert not self.sensor.available
|
||||
|
||||
def test_update_when_value_changed(self):
|
||||
"""Test state gets updated when sensor returns a new status."""
|
||||
self.rest.update = Mock('rest.RestData.update',
|
||||
side_effect=self.update_side_effect(
|
||||
'{ "key": "updated_state" }'))
|
||||
self.rest.update = Mock(
|
||||
"rest.RestData.update",
|
||||
side_effect=self.update_side_effect('{ "key": "updated_state" }'),
|
||||
)
|
||||
self.sensor.update()
|
||||
assert 'updated_state' == self.sensor.state
|
||||
assert "updated_state" == self.sensor.state
|
||||
assert self.sensor.available
|
||||
|
||||
def test_update_with_no_template(self):
|
||||
"""Test update when there is no value template."""
|
||||
self.rest.update = Mock('rest.RestData.update',
|
||||
side_effect=self.update_side_effect(
|
||||
'plain_state'))
|
||||
self.sensor = rest.RestSensor(self.hass, self.rest, self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class, None, [],
|
||||
self.force_update)
|
||||
self.rest.update = Mock(
|
||||
"rest.RestData.update", side_effect=self.update_side_effect("plain_state")
|
||||
)
|
||||
self.sensor = rest.RestSensor(
|
||||
self.hass,
|
||||
self.rest,
|
||||
self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class,
|
||||
None,
|
||||
[],
|
||||
self.force_update,
|
||||
)
|
||||
self.sensor.update()
|
||||
assert 'plain_state' == self.sensor.state
|
||||
assert "plain_state" == self.sensor.state
|
||||
assert self.sensor.available
|
||||
|
||||
def test_update_with_json_attrs(self):
|
||||
"""Test attributes get extracted from a JSON result."""
|
||||
self.rest.update = Mock('rest.RestData.update',
|
||||
side_effect=self.update_side_effect(
|
||||
'{ "key": "some_json_value" }'))
|
||||
self.sensor = rest.RestSensor(self.hass, self.rest, self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class, None, ['key'],
|
||||
self.force_update)
|
||||
self.rest.update = Mock(
|
||||
"rest.RestData.update",
|
||||
side_effect=self.update_side_effect('{ "key": "some_json_value" }'),
|
||||
)
|
||||
self.sensor = rest.RestSensor(
|
||||
self.hass,
|
||||
self.rest,
|
||||
self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class,
|
||||
None,
|
||||
["key"],
|
||||
self.force_update,
|
||||
)
|
||||
self.sensor.update()
|
||||
assert 'some_json_value' == \
|
||||
self.sensor.device_state_attributes['key']
|
||||
assert "some_json_value" == self.sensor.device_state_attributes["key"]
|
||||
|
||||
@patch('homeassistant.components.rest.sensor._LOGGER')
|
||||
@patch("homeassistant.components.rest.sensor._LOGGER")
|
||||
def test_update_with_json_attrs_no_data(self, mock_logger):
|
||||
"""Test attributes when no JSON result fetched."""
|
||||
self.rest.update = Mock('rest.RestData.update',
|
||||
side_effect=self.update_side_effect(None))
|
||||
self.sensor = rest.RestSensor(self.hass, self.rest, self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class, None, ['key'],
|
||||
self.force_update)
|
||||
self.rest.update = Mock(
|
||||
"rest.RestData.update", side_effect=self.update_side_effect(None)
|
||||
)
|
||||
self.sensor = rest.RestSensor(
|
||||
self.hass,
|
||||
self.rest,
|
||||
self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class,
|
||||
None,
|
||||
["key"],
|
||||
self.force_update,
|
||||
)
|
||||
self.sensor.update()
|
||||
assert {} == self.sensor.device_state_attributes
|
||||
assert mock_logger.warning.called
|
||||
|
||||
@patch('homeassistant.components.rest.sensor._LOGGER')
|
||||
@patch("homeassistant.components.rest.sensor._LOGGER")
|
||||
def test_update_with_json_attrs_not_dict(self, mock_logger):
|
||||
"""Test attributes get extracted from a JSON result."""
|
||||
self.rest.update = Mock('rest.RestData.update',
|
||||
side_effect=self.update_side_effect(
|
||||
'["list", "of", "things"]'))
|
||||
self.sensor = rest.RestSensor(self.hass, self.rest, self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class, None, ['key'],
|
||||
self.force_update)
|
||||
self.rest.update = Mock(
|
||||
"rest.RestData.update",
|
||||
side_effect=self.update_side_effect('["list", "of", "things"]'),
|
||||
)
|
||||
self.sensor = rest.RestSensor(
|
||||
self.hass,
|
||||
self.rest,
|
||||
self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class,
|
||||
None,
|
||||
["key"],
|
||||
self.force_update,
|
||||
)
|
||||
self.sensor.update()
|
||||
assert {} == self.sensor.device_state_attributes
|
||||
assert mock_logger.warning.called
|
||||
|
||||
@patch('homeassistant.components.rest.sensor._LOGGER')
|
||||
@patch("homeassistant.components.rest.sensor._LOGGER")
|
||||
def test_update_with_json_attrs_bad_JSON(self, mock_logger):
|
||||
"""Test attributes get extracted from a JSON result."""
|
||||
self.rest.update = Mock('rest.RestData.update',
|
||||
side_effect=self.update_side_effect(
|
||||
'This is text rather than JSON data.'))
|
||||
self.sensor = rest.RestSensor(self.hass, self.rest, self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class, None, ['key'],
|
||||
self.force_update)
|
||||
self.rest.update = Mock(
|
||||
"rest.RestData.update",
|
||||
side_effect=self.update_side_effect("This is text rather than JSON data."),
|
||||
)
|
||||
self.sensor = rest.RestSensor(
|
||||
self.hass,
|
||||
self.rest,
|
||||
self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class,
|
||||
None,
|
||||
["key"],
|
||||
self.force_update,
|
||||
)
|
||||
self.sensor.update()
|
||||
assert {} == self.sensor.device_state_attributes
|
||||
assert mock_logger.warning.called
|
||||
@@ -258,20 +311,28 @@ class TestRestSensor(unittest.TestCase):
|
||||
|
||||
def test_update_with_json_attrs_and_template(self):
|
||||
"""Test attributes get extracted from a JSON result."""
|
||||
self.rest.update = Mock('rest.RestData.update',
|
||||
side_effect=self.update_side_effect(
|
||||
'{ "key": "json_state_updated_value" }'))
|
||||
self.sensor = rest.RestSensor(self.hass, self.rest, self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class,
|
||||
self.value_template, ['key'],
|
||||
self.force_update)
|
||||
self.rest.update = Mock(
|
||||
"rest.RestData.update",
|
||||
side_effect=self.update_side_effect(
|
||||
'{ "key": "json_state_updated_value" }'
|
||||
),
|
||||
)
|
||||
self.sensor = rest.RestSensor(
|
||||
self.hass,
|
||||
self.rest,
|
||||
self.name,
|
||||
self.unit_of_measurement,
|
||||
self.device_class,
|
||||
self.value_template,
|
||||
["key"],
|
||||
self.force_update,
|
||||
)
|
||||
self.sensor.update()
|
||||
|
||||
assert 'json_state_updated_value' == self.sensor.state
|
||||
assert 'json_state_updated_value' == \
|
||||
self.sensor.device_state_attributes['key'], \
|
||||
self.force_update
|
||||
assert "json_state_updated_value" == self.sensor.state
|
||||
assert (
|
||||
"json_state_updated_value" == self.sensor.device_state_attributes["key"]
|
||||
), self.force_update
|
||||
|
||||
|
||||
class TestRestData(unittest.TestCase):
|
||||
@@ -284,17 +345,17 @@ class TestRestData(unittest.TestCase):
|
||||
self.verify_ssl = True
|
||||
self.timeout = 10
|
||||
self.rest = rest.RestData(
|
||||
self.method, self.resource, None, None, None, self.verify_ssl,
|
||||
self.timeout)
|
||||
self.method, self.resource, None, None, None, self.verify_ssl, self.timeout
|
||||
)
|
||||
|
||||
@requests_mock.Mocker()
|
||||
def test_update(self, mock_req):
|
||||
"""Test update."""
|
||||
mock_req.get('http://localhost', text='test data')
|
||||
mock_req.get("http://localhost", text="test data")
|
||||
self.rest.update()
|
||||
assert 'test data' == self.rest.data
|
||||
assert "test data" == self.rest.data
|
||||
|
||||
@patch('requests.Session', side_effect=RequestException)
|
||||
@patch("requests.Session", side_effect=RequestException)
|
||||
def test_update_request_exception(self, mock_req):
|
||||
"""Test update when a request exception occurs."""
|
||||
self.rest.update()
|
||||
|
||||
Reference in New Issue
Block a user