Small huawei_lte improvements

homeassistant/components/huawei_lte.py

@@ -55,8 +55,12 @@ def __getitem__(self, path: str):
         such as device_information, device_signal etc, and the remaining
         path points to a value in the member's data structure.
         """
-        cat, *path_ = path.split(".")
-        return reduce(operator.getitem, path_, getattr(self, cat))
+        root, *rest = path.split(".")
+        try:
+            data = getattr(self, root)
+        except AttributeError as err:
+            raise KeyError from err
+        return reduce(operator.getitem, rest, data)
 
     @Throttle(MIN_TIME_BETWEEN_UPDATES)
     def update(self) -> None:

homeassistant/components/sensor/huawei_lte.py

@@ -113,7 +113,7 @@ class HuaweiLteSensor(Entity):
     @property
     def unique_id(self) -> str:
         """Return unique ID for sensor."""
-        return "%s_%s" % (
+        return "{}_{}".format(
             self.path,
             self.data["device_information.SerialNumber"],
         )

tests/components/huawei_lte.py

@@ -0,0 +1,48 @@
+"""Huawei LTE component tests."""
+import pytest
+
+from homeassistant.components import huawei_lte
+
+
+@pytest.fixture(autouse=True)
+def routerdata():
+    """Set up a router data for testing."""
+    rd = huawei_lte.RouterData(None)
+    rd.device_information = {
+        'SoftwareVersion': '1.0',
+        'nested': {'foo': 'bar'},
+    }
+    return rd
+
+
+async def test_routerdata_get_nonexistent_root(routerdata):
+    """Test that accessing a nonexistent root element raises KeyError."""
+    with pytest.raises(KeyError):  # NOT AttributeError
+        routerdata["nonexistent_root.foo"]
+
+
+async def test_routerdata_get_nonexistent_leaf(routerdata):
+    """Test that accessing a nonexistent leaf element raises KeyError."""
+    with pytest.raises(KeyError):
+        routerdata["device_information.foo"]
+
+
+async def test_routerdata_get_nonexistent_leaf_path(routerdata):
+    """Test that accessing a nonexistent long path raises KeyError."""
+    with pytest.raises(KeyError):
+        routerdata["device_information.long.path.foo"]
+
+
+async def test_routerdata_get_simple(routerdata):
+    """Test that accessing a short, simple path works."""
+    assert routerdata["device_information.SoftwareVersion"] == "1.0"
+
+
+async def test_routerdata_get_longer(routerdata):
+    """Test that accessing a longer path works."""
+    assert routerdata["device_information.nested.foo"] == "bar"
+
+
+async def test_routerdata_get_dict(routerdata):
+    """Test that returning an intermediate dict works."""
+    assert routerdata["device_information.nested"] == {'foo': 'bar'}