Update ha-ffmpeg 2.0 (#22427)

tests/components/ffmpeg/test_sensor.py

@@ -0,0 +1,133 @@
+"""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):
+        """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)
+
+        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)
+
+        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)
+
+        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):
+        """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)
+
+        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)
+
+        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)
+
+        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'