Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00
Code Activity

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

95
tests/components/ffmpeg/test_sensor.py
View File

@@ -3,8 +3,7 @@ from unittest.mock import patch
from homeassistant.setup import setup_component
from tests.common import (
get_test_home_assistant, assert_setup_component, mock_coro)
from tests.common import get_test_home_assistant, assert_setup_component, mock_coro
class TestFFmpegNoiseSetup:
@@ -15,10 +14,7 @@ class TestFFmpegNoiseSetup:
self.hass = get_test_home_assistant()
self.config = {
'binary_sensor': {
'platform': 'ffmpeg_noise',
'input': 'testinputvideo',
},
"binary_sensor": {"platform": "ffmpeg_noise", "input": "testinputvideo"}
}
def teardown_method(self):
@@ -27,50 +23,49 @@ class TestFFmpegNoiseSetup:
def test_setup_component(self):
"""Set up ffmpeg component."""
with assert_setup_component(1, 'binary_sensor'):
setup_component(self.hass, 'binary_sensor', self.config)
with assert_setup_component(1, "binary_sensor"):
setup_component(self.hass, "binary_sensor", self.config)
self.hass.block_till_done()
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert self.hass.states.get('binary_sensor.ffmpeg_noise') is not None
assert self.hass.data["ffmpeg"].binary == "ffmpeg"
assert self.hass.states.get("binary_sensor.ffmpeg_noise") is not None
@patch('haffmpeg.sensor.SensorNoise.open_sensor',
return_value=mock_coro())
@patch("haffmpeg.sensor.SensorNoise.open_sensor", return_value=mock_coro())
def test_setup_component_start(self, mock_start):
"""Set up ffmpeg component."""
with assert_setup_component(1, 'binary_sensor'):
setup_component(self.hass, 'binary_sensor', self.config)
with assert_setup_component(1, "binary_sensor"):
setup_component(self.hass, "binary_sensor", self.config)
self.hass.block_till_done()
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert self.hass.states.get('binary_sensor.ffmpeg_noise') is not None
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'
entity = self.hass.states.get("binary_sensor.ffmpeg_noise")
assert entity.state == "unavailable"
@patch('haffmpeg.sensor.SensorNoise')
@patch("haffmpeg.sensor.SensorNoise")
def test_setup_component_start_callback(self, mock_ffmpeg):
"""Set up ffmpeg component."""
with assert_setup_component(1, 'binary_sensor'):
setup_component(self.hass, 'binary_sensor', self.config)
with assert_setup_component(1, "binary_sensor"):
setup_component(self.hass, "binary_sensor", self.config)
self.hass.block_till_done()
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert self.hass.states.get('binary_sensor.ffmpeg_noise') is not None
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'
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'
entity = self.hass.states.get("binary_sensor.ffmpeg_noise")
assert entity.state == "on"
class TestFFmpegMotionSetup:
@@ -81,10 +76,7 @@ class TestFFmpegMotionSetup:
self.hass = get_test_home_assistant()
self.config = {
'binary_sensor': {
'platform': 'ffmpeg_motion',
'input': 'testinputvideo',
},
"binary_sensor": {"platform": "ffmpeg_motion", "input": "testinputvideo"}
}
def teardown_method(self):
@@ -93,47 +85,46 @@ class TestFFmpegMotionSetup:
def test_setup_component(self):
"""Set up ffmpeg component."""
with assert_setup_component(1, 'binary_sensor'):
setup_component(self.hass, 'binary_sensor', self.config)
with assert_setup_component(1, "binary_sensor"):
setup_component(self.hass, "binary_sensor", self.config)
self.hass.block_till_done()
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert self.hass.states.get('binary_sensor.ffmpeg_motion') is not None
assert self.hass.data["ffmpeg"].binary == "ffmpeg"
assert self.hass.states.get("binary_sensor.ffmpeg_motion") is not None
@patch('haffmpeg.sensor.SensorMotion.open_sensor',
return_value=mock_coro())
@patch("haffmpeg.sensor.SensorMotion.open_sensor", return_value=mock_coro())
def test_setup_component_start(self, mock_start):
"""Set up ffmpeg component."""
with assert_setup_component(1, 'binary_sensor'):
setup_component(self.hass, 'binary_sensor', self.config)
with assert_setup_component(1, "binary_sensor"):
setup_component(self.hass, "binary_sensor", self.config)
self.hass.block_till_done()
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert self.hass.states.get('binary_sensor.ffmpeg_motion') is not None
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'
entity = self.hass.states.get("binary_sensor.ffmpeg_motion")
assert entity.state == "unavailable"
@patch('haffmpeg.sensor.SensorMotion')
@patch("haffmpeg.sensor.SensorMotion")
def test_setup_component_start_callback(self, mock_ffmpeg):
"""Set up ffmpeg component."""
with assert_setup_component(1, 'binary_sensor'):
setup_component(self.hass, 'binary_sensor', self.config)
with assert_setup_component(1, "binary_sensor"):
setup_component(self.hass, "binary_sensor", self.config)
self.hass.block_till_done()
assert self.hass.data['ffmpeg'].binary == 'ffmpeg'
assert self.hass.states.get('binary_sensor.ffmpeg_motion') is not None
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'
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'
entity = self.hass.states.get("binary_sensor.ffmpeg_motion")
assert entity.state == "on"