jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects Packages Wiki Activity Actions

Move ffmpeg to dispatcher from hass.data entity store. (#6211)

Browse source 
* Move ffmpeg to dispatcher from hass.data entity store.

* fix lint

* address paulus comments

* add more unittest for better coverage
This commit is contained in:
Pascal Vizeli 2017-02-26 23:31:46 +01:00 • committed by Paulus Schoutsen
parent 9490cb4c8f
commit 48cf7a4af9
5 changed files with 252 additions and 206 deletions
Download patch file Download diff file Expand all files Collapse all files

55
tests/components/binary_sensor/test_ffmpeg.py
View file

@ -2,7 +2,6 @@
from unittest.mock import patch
from homeassistant.bootstrap import setup_component
from homeassistant.util.async import run_callback_threadsafe
from tests.common import (
get_test_home_assistant, assert_setup_component, mock_coro)
@ -35,7 +34,7 @@ class TestFFmpegNoiseSetup(object):
setup_component(self.hass, 'binary_sensor', self.config)
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert len(self.hass.data['ffmpeg'].entities) == 1
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):
@ -44,15 +43,32 @@ class TestFFmpegNoiseSetup(object):
setup_component(self.hass, 'binary_sensor', self.config)
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert len(self.hass.data['ffmpeg'].entities) == 1
assert self.hass.states.get('binary_sensor.ffmpeg_noise') is not None
entity = self.hass.data['ffmpeg'].entities[0]
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'
run_callback_threadsafe(
self.hass.loop, entity._async_callback, True).result()
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'
@ -83,7 +99,7 @@ class TestFFmpegMotionSetup(object):
setup_component(self.hass, 'binary_sensor', self.config)
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert len(self.hass.data['ffmpeg'].entities) == 1
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):
@ -92,13 +108,30 @@ class TestFFmpegMotionSetup(object):
setup_component(self.hass, 'binary_sensor', self.config)
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert len(self.hass.data['ffmpeg'].entities) == 1
assert self.hass.states.get('binary_sensor.ffmpeg_motion') is not None
entity = self.hass.data['ffmpeg'].entities[0]
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'
run_callback_threadsafe(
self.hass.loop, entity._async_callback, True).result()
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'