core/tests/components/ffmpeg/test_sensor.py

"""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)
        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][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)
        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][2], True)
        self.hass.block_till_done()

        entity = self.hass.states.get('binary_sensor.ffmpeg_motion')
        assert entity.state == 'on'