2f7b79764a
## Description: More fixes flagged by pylint 2 that don't hurt to have before the actual pylint 2 upgrade (which I'll submit soon). ## Checklist: - [ ] The code change is tested and works locally. - [x] Local tests pass with `tox`. **Your PR cannot be merged unless tests pass**
137 lines
4.6 KiB
Python
137 lines
4.6 KiB
Python
"""The tests for Home Assistant ffmpeg binary sensor."""
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.setup import setup_component
|
|
|
|
from tests.common import (
|
|
get_test_home_assistant, assert_setup_component, mock_coro)
|
|
|
|
|
|
class TestFFmpegNoiseSetup:
|
|
"""Test class for ffmpeg."""
|
|
|
|
def setup_method(self):
|
|
"""Setup things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
self.config = {
|
|
'ffmpeg': {
|
|
'run_test': False,
|
|
},
|
|
'binary_sensor': {
|
|
'platform': 'ffmpeg_noise',
|
|
'input': 'testinputvideo',
|
|
},
|
|
}
|
|
|
|
def teardown_method(self):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_setup_component(self):
|
|
"""Setup ffmpeg component."""
|
|
with assert_setup_component(1, 'binary_sensor'):
|
|
setup_component(self.hass, 'binary_sensor', self.config)
|
|
|
|
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
|
|
assert self.hass.states.get('binary_sensor.ffmpeg_noise') is not None
|
|
|
|
@patch('haffmpeg.SensorNoise.open_sensor', return_value=mock_coro())
|
|
def test_setup_component_start(self, mock_start):
|
|
"""Setup ffmpeg component."""
|
|
with assert_setup_component(1, 'binary_sensor'):
|
|
setup_component(self.hass, 'binary_sensor', self.config)
|
|
|
|
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
|
|
assert self.hass.states.get('binary_sensor.ffmpeg_noise') is not None
|
|
|
|
self.hass.start()
|
|
assert mock_start.called
|
|
|
|
entity = self.hass.states.get('binary_sensor.ffmpeg_noise')
|
|
assert entity.state == 'unavailable'
|
|
|
|
@patch('haffmpeg.SensorNoise')
|
|
def test_setup_component_start_callback(self, mock_ffmpeg):
|
|
"""Setup ffmpeg component."""
|
|
with assert_setup_component(1, 'binary_sensor'):
|
|
setup_component(self.hass, 'binary_sensor', self.config)
|
|
|
|
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
|
|
assert self.hass.states.get('binary_sensor.ffmpeg_noise') is not None
|
|
|
|
self.hass.start()
|
|
|
|
entity = self.hass.states.get('binary_sensor.ffmpeg_noise')
|
|
assert entity.state == 'off'
|
|
|
|
self.hass.add_job(mock_ffmpeg.call_args[0][2], True)
|
|
self.hass.block_till_done()
|
|
|
|
entity = self.hass.states.get('binary_sensor.ffmpeg_noise')
|
|
assert entity.state == 'on'
|
|
|
|
|
|
class TestFFmpegMotionSetup:
|
|
"""Test class for ffmpeg."""
|
|
|
|
def setup_method(self):
|
|
"""Setup things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
self.config = {
|
|
'ffmpeg': {
|
|
'run_test': False,
|
|
},
|
|
'binary_sensor': {
|
|
'platform': 'ffmpeg_motion',
|
|
'input': 'testinputvideo',
|
|
},
|
|
}
|
|
|
|
def teardown_method(self):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_setup_component(self):
|
|
"""Setup ffmpeg component."""
|
|
with assert_setup_component(1, 'binary_sensor'):
|
|
setup_component(self.hass, 'binary_sensor', self.config)
|
|
|
|
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
|
|
assert self.hass.states.get('binary_sensor.ffmpeg_motion') is not None
|
|
|
|
@patch('haffmpeg.SensorMotion.open_sensor', return_value=mock_coro())
|
|
def test_setup_component_start(self, mock_start):
|
|
"""Setup ffmpeg component."""
|
|
with assert_setup_component(1, 'binary_sensor'):
|
|
setup_component(self.hass, 'binary_sensor', self.config)
|
|
|
|
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
|
|
assert self.hass.states.get('binary_sensor.ffmpeg_motion') is not None
|
|
|
|
self.hass.start()
|
|
assert mock_start.called
|
|
|
|
entity = self.hass.states.get('binary_sensor.ffmpeg_motion')
|
|
assert entity.state == 'unavailable'
|
|
|
|
@patch('haffmpeg.SensorMotion')
|
|
def test_setup_component_start_callback(self, mock_ffmpeg):
|
|
"""Setup ffmpeg component."""
|
|
with assert_setup_component(1, 'binary_sensor'):
|
|
setup_component(self.hass, 'binary_sensor', self.config)
|
|
|
|
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
|
|
assert self.hass.states.get('binary_sensor.ffmpeg_motion') is not None
|
|
|
|
self.hass.start()
|
|
|
|
entity = self.hass.states.get('binary_sensor.ffmpeg_motion')
|
|
assert entity.state == 'off'
|
|
|
|
self.hass.add_job(mock_ffmpeg.call_args[0][2], True)
|
|
self.hass.block_till_done()
|
|
|
|
entity = self.hass.states.get('binary_sensor.ffmpeg_motion')
|
|
assert entity.state == 'on'
|