core/tests/components/uptime/test_sensor.py

"""The tests for the uptime sensor platform."""
import asyncio
from datetime import timedelta
import unittest
from unittest.mock import patch

from homeassistant.components.uptime.sensor import UptimeSensor
from homeassistant.setup import setup_component

from tests.common import get_test_home_assistant


class TestUptimeSensor(unittest.TestCase):
    """Test the uptime sensor."""

    def setUp(self):
        """Set up things to run when tests begin."""
        self.hass = get_test_home_assistant()

    def tearDown(self):
        """Stop everything that was started."""
        self.hass.stop()

    def test_uptime_min_config(self):
        """Test minimum uptime configuration."""
        config = {"sensor": {"platform": "uptime"}}
        assert setup_component(self.hass, "sensor", config)

    def test_uptime_sensor_name_change(self):
        """Test uptime sensor with different name."""
        config = {"sensor": {"platform": "uptime", "name": "foobar"}}
        assert setup_component(self.hass, "sensor", config)

    def test_uptime_sensor_config_hours(self):
        """Test uptime sensor with hours defined in config."""
        config = {"sensor": {"platform": "uptime", "unit_of_measurement": "hours"}}
        assert setup_component(self.hass, "sensor", config)

    def test_uptime_sensor_config_minutes(self):
        """Test uptime sensor with minutes defined in config."""
        config = {"sensor": {"platform": "uptime", "unit_of_measurement": "minutes"}}
        assert setup_component(self.hass, "sensor", config)

    def test_uptime_sensor_days_output(self):
        """Test uptime sensor output data."""
        sensor = UptimeSensor("test", "days")
        assert sensor.unit_of_measurement == "days"
        new_time = sensor.initial + timedelta(days=1)
        with patch("homeassistant.util.dt.now", return_value=new_time):
            asyncio.run_coroutine_threadsafe(
                sensor.async_update(), self.hass.loop
            ).result()
            assert sensor.state == 1.00
        new_time = sensor.initial + timedelta(days=111.499)
        with patch("homeassistant.util.dt.now", return_value=new_time):
            asyncio.run_coroutine_threadsafe(
                sensor.async_update(), self.hass.loop
            ).result()
            assert sensor.state == 111.50

    def test_uptime_sensor_hours_output(self):
        """Test uptime sensor output data."""
        sensor = UptimeSensor("test", "hours")
        assert sensor.unit_of_measurement == "hours"
        new_time = sensor.initial + timedelta(hours=16)
        with patch("homeassistant.util.dt.now", return_value=new_time):
            asyncio.run_coroutine_threadsafe(
                sensor.async_update(), self.hass.loop
            ).result()
            assert sensor.state == 16.00
        new_time = sensor.initial + timedelta(hours=72.499)
        with patch("homeassistant.util.dt.now", return_value=new_time):
            asyncio.run_coroutine_threadsafe(
                sensor.async_update(), self.hass.loop
            ).result()
            assert sensor.state == 72.50

    def test_uptime_sensor_minutes_output(self):
        """Test uptime sensor output data."""
        sensor = UptimeSensor("test", "minutes")
        assert sensor.unit_of_measurement == "minutes"
        new_time = sensor.initial + timedelta(minutes=16)
        with patch("homeassistant.util.dt.now", return_value=new_time):
            asyncio.run_coroutine_threadsafe(
                sensor.async_update(), self.hass.loop
            ).result()
            assert sensor.state == 16.00
        new_time = sensor.initial + timedelta(minutes=12.499)
        with patch("homeassistant.util.dt.now", return_value=new_time):
            asyncio.run_coroutine_threadsafe(
                sensor.async_update(), self.hass.loop
            ).result()
            assert sensor.state == 12.50