core/tests/components/ring/test_sensor.py

"""The tests for the Ring sensor platform."""
import os
import unittest
import requests_mock

import homeassistant.components.ring.sensor as ring
from homeassistant.components import ring as base_ring
from homeassistant.helpers.icon import icon_for_battery_level
from tests.components.ring.test_init import ATTRIBUTION, VALID_CONFIG
from tests.common import get_test_config_dir, get_test_home_assistant, load_fixture


class TestRingSensorSetup(unittest.TestCase):
    """Test the Ring platform."""

    DEVICES = []

    def add_entities(self, devices, action):
        """Mock add devices."""
        for device in devices:
            self.DEVICES.append(device)

    def cleanup(self):
        """Cleanup any data created from the tests."""
        if os.path.isfile(self.cache):
            os.remove(self.cache)

    def setUp(self):
        """Initialize values for this testcase class."""
        self.hass = get_test_home_assistant()
        self.cache = get_test_config_dir(base_ring.DEFAULT_CACHEDB)
        self.config = {
            "username": "foo",
            "password": "bar",
            "monitored_conditions": [
                "battery",
                "last_activity",
                "last_ding",
                "last_motion",
                "volume",
                "wifi_signal_category",
                "wifi_signal_strength",
            ],
        }

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

    @requests_mock.Mocker()
    def test_sensor(self, mock):
        """Test the Ring sensor class and methods."""
        mock.post(
            "https://oauth.ring.com/oauth/token", text=load_fixture("ring_oauth.json")
        )
        mock.post(
            "https://api.ring.com/clients_api/session",
            text=load_fixture("ring_session.json"),
        )
        mock.get(
            "https://api.ring.com/clients_api/ring_devices",
            text=load_fixture("ring_devices.json"),
        )
        mock.get(
            "https://api.ring.com/clients_api/doorbots/987652/history",
            text=load_fixture("ring_doorbots.json"),
        )
        mock.get(
            "https://api.ring.com/clients_api/doorbots/987652/health",
            text=load_fixture("ring_doorboot_health_attrs.json"),
        )
        mock.get(
            "https://api.ring.com/clients_api/chimes/999999/health",
            text=load_fixture("ring_chime_health_attrs.json"),
        )
        base_ring.setup(self.hass, VALID_CONFIG)
        ring.setup_platform(self.hass, self.config, self.add_entities, None)

        for device in self.DEVICES:
            device.update()
            if device.name == "Front Battery":
                expected_icon = icon_for_battery_level(
                    battery_level=int(device.state), charging=False
                )
                assert device.icon == expected_icon
                assert 80 == device.state
                assert "hp_cam_v1" == device.device_state_attributes["kind"]
                assert "stickup_cams" == device.device_state_attributes["type"]
            if device.name == "Front Door Battery":
                assert 100 == device.state
                assert "lpd_v1" == device.device_state_attributes["kind"]
                assert "chimes" != device.device_state_attributes["type"]
            if device.name == "Downstairs Volume":
                assert 2 == device.state
                assert "1.2.3" == device.device_state_attributes["firmware"]
                assert "ring_mock_wifi" == device.device_state_attributes["wifi_name"]
                assert "mdi:bell-ring" == device.icon
                assert "chimes" == device.device_state_attributes["type"]
            if device.name == "Front Door Last Activity":
                assert not device.device_state_attributes["answered"]
                assert "America/New_York" == device.device_state_attributes["timezone"]

            if device.name == "Downstairs WiFi Signal Strength":
                assert -39 == device.state

            if device.name == "Front Door WiFi Signal Category":
                assert "good" == device.state

            if device.name == "Front Door WiFi Signal Strength":
                assert -58 == device.state

            assert device.entity_picture is None
            assert ATTRIBUTION == device.device_state_attributes["attribution"]
            assert not device.should_poll