Fix Z-Wave (#5955)

homeassistant/components/zwave/__init__.py

@@ -174,7 +174,7 @@ CONFIG_SCHEMA = vol.Schema({
         vol.Optional(CONF_AUTOHEAL, default=DEFAULT_CONF_AUTOHEAL): cv.boolean,
         vol.Optional(CONF_CONFIG_PATH): cv.string,
         vol.Optional(CONF_DEVICE_CONFIG, default={}):
-            _DEVICE_CONFIG_SCHEMA_ENTRY,
+            vol.Schema({cv.entity_id: _DEVICE_CONFIG_SCHEMA_ENTRY}),
         vol.Optional(CONF_DEBUG, default=DEFAULT_DEBUG): cv.boolean,
         vol.Optional(CONF_POLLING_INTERVAL, default=DEFAULT_POLLING_INTERVAL):
             cv.positive_int,

tests/components/zwave/__init__.py

@@ -0,0 +1 @@
+"""Tests for the  Z-Wave component."""

tests/components/zwave/test_init.py

@@ -0,0 +1,46 @@
+"""Tests for the Z-Wave init."""
+import asyncio
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from homeassistant.bootstrap import async_setup_component
+from homeassistant.components.zwave import (
+    DATA_DEVICE_CONFIG, _DEVICE_CONFIG_SCHEMA_ENTRY)
+
+
+@pytest.fixture(autouse=True)
+def mock_openzwave():
+    """Mock out Open Z-Wave."""
+    libopenzwave = MagicMock()
+    libopenzwave.__file__ = 'test'
+    with patch.dict('sys.modules', {
+        'libopenzwave': libopenzwave,
+        'openzwave.option': MagicMock(),
+        'openzwave.network': MagicMock(),
+        'openzwave.group': MagicMock(),
+    }):
+        yield
+
+
+@asyncio.coroutine
+def test_device_config(hass):
+    """Test device config stored in hass."""
+    device_config = {
+        'light.kitchen': {
+            'ignored': 'true'
+        }
+    }
+    yield from async_setup_component(hass, 'zwave', {
+        'zwave': {
+            'device_config': device_config
+        }})
+
+    assert DATA_DEVICE_CONFIG in hass.data
+
+    test_data = {
+        key: _DEVICE_CONFIG_SCHEMA_ENTRY(value)
+        for key, value in device_config.items()
+    }
+
+    assert hass.data[DATA_DEVICE_CONFIG] == test_data