diff --git a/hassio/discovery/const.py b/hassio/discovery/const.py index c85f248fd..71ca1b020 100644 --- a/hassio/discovery/const.py +++ b/hassio/discovery/const.py @@ -6,3 +6,5 @@ ATTR_PORT = "port" ATTR_PROTOCOL = "protocol" ATTR_SSL = "ssl" ATTR_USERNAME = "username" +ATTR_API_KEY = "api_key" +ATTR_SERIAL = "serial" diff --git a/hassio/discovery/services/deconz.py b/hassio/discovery/services/deconz.py index d817aac1f..463e8474d 100644 --- a/hassio/discovery/services/deconz.py +++ b/hassio/discovery/services/deconz.py @@ -3,9 +3,14 @@ import voluptuous as vol from hassio.validate import NETWORK_PORT -from ..const import ATTR_HOST, ATTR_PORT +from ..const import ATTR_HOST, ATTR_PORT, ATTR_API_KEY, ATTR_SERIAL SCHEMA = vol.Schema( - {vol.Required(ATTR_HOST): vol.Coerce(str), vol.Required(ATTR_PORT): NETWORK_PORT} + { + vol.Required(ATTR_HOST): vol.Coerce(str), + vol.Required(ATTR_PORT): NETWORK_PORT, + vol.Required(ATTR_SERIAL): vol.Coerce(str), + vol.Required(ATTR_API_KEY): vol.Coerce(str), + } ) diff --git a/tests/discovery/test_deconz.py b/tests/discovery/test_deconz.py index 092cd35cd..4f76db326 100644 --- a/tests/discovery/test_deconz.py +++ b/tests/discovery/test_deconz.py @@ -9,11 +9,14 @@ from hassio.discovery.validate import valid_discovery_config def test_good_config(): """Test good deconz config.""" - valid_discovery_config("deconz", {"host": "test", "port": 3812}) + valid_discovery_config( + "deconz", + {"host": "test", "port": 3812, "api_key": "MY_api_KEY99", "serial": "xyz"}, + ) def test_bad_config(): """Test good deconz config.""" with pytest.raises(vol.Invalid): - valid_discovery_config("deconz", {"host": "test"}) + valid_discovery_config("deconz", {"host": "test", "port": 8080})