"""The tests for the Binary sensor component."""
import unittest
from unittest import mock

from homeassistant.components import binary_sensor
from homeassistant.const import STATE_ON, STATE_OFF


class TestBinarySensor(unittest.TestCase):
    """Test the binary_sensor base class."""

    def test_state(self):
        """Test binary sensor state."""
        sensor = binary_sensor.BinarySensorDevice()
        assert STATE_OFF == sensor.state
        with mock.patch('homeassistant.components.binary_sensor.'
                        'BinarySensorDevice.is_on',
                        new=False):
            assert STATE_OFF == \
                binary_sensor.BinarySensorDevice().state
        with mock.patch('homeassistant.components.binary_sensor.'
                        'BinarySensorDevice.is_on',
                        new=True):
            assert STATE_ON == \
                binary_sensor.BinarySensorDevice().state