From fe9a254017dd11c40c1f5be71b0dff527430c372 Mon Sep 17 00:00:00 2001
From: Allen Porter <allen@thebends.org>
Date: Thu, 31 Dec 2020 10:22:24 -0800
Subject: [PATCH] Fix legacy nest api binary_sensor initialization (#44674)

---
 .../components/nest/binary_sensor.py          |  2 +-
 .../components/nest/legacy/binary_sensor.py   |  2 +-
 tests/components/nest/test_init_legacy.py     | 87 +++++++++++++++++++
 3 files changed, 89 insertions(+), 2 deletions(-)
 create mode 100644 tests/components/nest/test_init_legacy.py

diff --git a/homeassistant/components/nest/binary_sensor.py b/homeassistant/components/nest/binary_sensor.py
index dc58dd2856f..d49ec8535cc 100644
--- a/homeassistant/components/nest/binary_sensor.py
+++ b/homeassistant/components/nest/binary_sensor.py
@@ -4,7 +4,7 @@ from homeassistant.config_entries import ConfigEntry
 from homeassistant.helpers.typing import HomeAssistantType
 
 from .const import DATA_SDM
-from .legacy.sensor import async_setup_legacy_entry
+from .legacy.binary_sensor import async_setup_legacy_entry
 
 
 async def async_setup_entry(
diff --git a/homeassistant/components/nest/legacy/binary_sensor.py b/homeassistant/components/nest/legacy/binary_sensor.py
index 4470bd14676..32c30f747d2 100644
--- a/homeassistant/components/nest/legacy/binary_sensor.py
+++ b/homeassistant/components/nest/legacy/binary_sensor.py
@@ -61,7 +61,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
     """
 
 
-async def async_setup_entry(hass, entry, async_add_entities):
+async def async_setup_legacy_entry(hass, entry, async_add_entities):
     """Set up a Nest binary sensor based on a config entry."""
     nest = hass.data[DATA_NEST]
 
diff --git a/tests/components/nest/test_init_legacy.py b/tests/components/nest/test_init_legacy.py
new file mode 100644
index 00000000000..f85fcdaa749
--- /dev/null
+++ b/tests/components/nest/test_init_legacy.py
@@ -0,0 +1,87 @@
+"""Test basic initialization for the Legacy Nest API using mocks for the Nest python library."""
+
+import time
+
+from homeassistant.setup import async_setup_component
+
+from tests.async_mock import MagicMock, PropertyMock, patch
+from tests.common import MockConfigEntry
+
+DOMAIN = "nest"
+
+CONFIG = {
+    "nest": {
+        "client_id": "some-client-id",
+        "client_secret": "some-client-secret",
+    },
+}
+
+CONFIG_ENTRY_DATA = {
+    "auth_implementation": "local",
+    "tokens": {
+        "expires_at": time.time() + 86400,
+        "access_token": {
+            "token": "some-token",
+        },
+    },
+}
+
+
+def make_thermostat():
+    """Make a mock thermostat with dummy values."""
+    device = MagicMock()
+    type(device).device_id = PropertyMock(return_value="a.b.c.d.e.f.g")
+    type(device).name = PropertyMock(return_value="My Thermostat")
+    type(device).name_long = PropertyMock(return_value="My Thermostat")
+    type(device).serial = PropertyMock(return_value="serial-number")
+    type(device).mode = "off"
+    type(device).hvac_state = "off"
+    type(device).target = PropertyMock(return_value=31.0)
+    type(device).temperature = PropertyMock(return_value=30.1)
+    type(device).min_temperature = PropertyMock(return_value=10.0)
+    type(device).max_temperature = PropertyMock(return_value=50.0)
+    type(device).humidity = PropertyMock(return_value=40.4)
+    type(device).software_version = PropertyMock(return_value="a.b.c")
+    return device
+
+
+async def test_thermostat(hass):
+    """Test simple initialization for thermostat entities."""
+
+    thermostat = make_thermostat()
+
+    structure = MagicMock()
+    type(structure).name = PropertyMock(return_value="My Room")
+    type(structure).thermostats = PropertyMock(return_value=[thermostat])
+    type(structure).eta = PropertyMock(return_value="away")
+
+    nest = MagicMock()
+    type(nest).structures = PropertyMock(return_value=[structure])
+
+    config_entry = MockConfigEntry(domain=DOMAIN, data=CONFIG_ENTRY_DATA)
+    config_entry.add_to_hass(hass)
+    with patch("homeassistant.components.nest.legacy.Nest", return_value=nest), patch(
+        "homeassistant.components.nest.legacy.sensor._VALID_SENSOR_TYPES",
+        ["humidity", "temperature"],
+    ), patch(
+        "homeassistant.components.nest.legacy.binary_sensor._VALID_BINARY_SENSOR_TYPES",
+        {"fan": None},
+    ):
+        assert await async_setup_component(hass, DOMAIN, CONFIG)
+        await hass.async_block_till_done()
+
+    climate = hass.states.get("climate.my_thermostat")
+    assert climate is not None
+    assert climate.state == "off"
+
+    temperature = hass.states.get("sensor.my_thermostat_temperature")
+    assert temperature is not None
+    assert temperature.state == "-1.1"
+
+    humidity = hass.states.get("sensor.my_thermostat_humidity")
+    assert humidity is not None
+    assert humidity.state == "40.4"
+
+    fan = hass.states.get("binary_sensor.my_thermostat_fan")
+    assert fan is not None
+    assert fan.state == "on"