131 lines
4.6 KiB
Python
131 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 assert_setup_component, get_test_home_assistant, mock_coro
|
|
|
|
|
|
class TestFFmpegNoiseSetup:
|
|
"""Test class for ffmpeg."""
|
|
|
|
def setup_method(self):
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
self.config = {
|
|
"binary_sensor": {"platform": "ffmpeg_noise", "input": "testinputvideo"}
|
|
}
|
|
|
|
def teardown_method(self):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_setup_component(self):
|
|
"""Set up ffmpeg component."""
|
|
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
|
|
|
|
@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)
|
|
self.hass.block_till_done()
|
|
|
|
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.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)
|
|
self.hass.block_till_done()
|
|
|
|
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][1], 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):
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
self.config = {
|
|
"binary_sensor": {"platform": "ffmpeg_motion", "input": "testinputvideo"}
|
|
}
|
|
|
|
def teardown_method(self):
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
def test_setup_component(self):
|
|
"""Set up ffmpeg component."""
|
|
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
|
|
|
|
@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)
|
|
self.hass.block_till_done()
|
|
|
|
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.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)
|
|
self.hass.block_till_done()
|
|
|
|
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][1], True)
|
|
self.hass.block_till_done()
|
|
|
|
entity = self.hass.states.get("binary_sensor.ffmpeg_motion")
|
|
assert entity.state == "on"
|