Update device registry of MQTT cover (#20443)

* Update device registry of MQTT cover

* Move config_entry to constructor
This commit is contained in:
emontnemery 2019-01-26 19:52:41 +01:00 committed by Paulus Schoutsen
parent 85c72fbca6
commit 60dc337f3f
2 changed files with 62 additions and 5 deletions

View File

@ -140,7 +140,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
try: try:
discovery_hash = discovery_payload[ATTR_DISCOVERY_HASH] discovery_hash = discovery_payload[ATTR_DISCOVERY_HASH]
config = PLATFORM_SCHEMA(discovery_payload) config = PLATFORM_SCHEMA(discovery_payload)
await _async_setup_entity(config, async_add_entities, await _async_setup_entity(config, async_add_entities, config_entry,
discovery_hash) discovery_hash)
except Exception: except Exception:
if discovery_hash: if discovery_hash:
@ -152,16 +152,17 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
async_discover) async_discover)
async def _async_setup_entity(config, async_add_entities, discovery_hash=None): async def _async_setup_entity(config, async_add_entities, config_entry=None,
discovery_hash=None):
"""Set up the MQTT Cover.""" """Set up the MQTT Cover."""
async_add_entities([MqttCover(config, discovery_hash)]) async_add_entities([MqttCover(config, config_entry, discovery_hash)])
class MqttCover(MqttAttributes, MqttAvailability, MqttDiscoveryUpdate, class MqttCover(MqttAttributes, MqttAvailability, MqttDiscoveryUpdate,
MqttEntityDeviceInfo, CoverDevice): MqttEntityDeviceInfo, CoverDevice):
"""Representation of a cover that can be controlled using MQTT.""" """Representation of a cover that can be controlled using MQTT."""
def __init__(self, config, discovery_hash): def __init__(self, config, config_entry, discovery_hash):
"""Initialize the cover.""" """Initialize the cover."""
self._unique_id = config.get(CONF_UNIQUE_ID) self._unique_id = config.get(CONF_UNIQUE_ID)
self._position = None self._position = None
@ -181,7 +182,7 @@ class MqttCover(MqttAttributes, MqttAvailability, MqttDiscoveryUpdate,
MqttAvailability.__init__(self, config) MqttAvailability.__init__(self, config)
MqttDiscoveryUpdate.__init__(self, discovery_hash, MqttDiscoveryUpdate.__init__(self, discovery_hash,
self.discovery_update) self.discovery_update)
MqttEntityDeviceInfo.__init__(self, device_config) MqttEntityDeviceInfo.__init__(self, device_config, config_entry)
async def async_added_to_hass(self): async def async_added_to_hass(self):
"""Subscribe MQTT events.""" """Subscribe MQTT events."""
@ -194,6 +195,7 @@ class MqttCover(MqttAttributes, MqttAvailability, MqttDiscoveryUpdate,
self._setup_from_config(config) self._setup_from_config(config)
await self.attributes_discovery_update(config) await self.attributes_discovery_update(config)
await self.availability_discovery_update(config) await self.availability_discovery_update(config)
await self.device_info_discovery_update(config)
await self._subscribe_topics() await self._subscribe_topics()
self.async_schedule_update_ha_state() self.async_schedule_update_ha_state()

View File

@ -758,6 +758,7 @@ class TestCoverMQTT(unittest.TestCase):
'set_position_topic': None, 'set_position_template': None, 'set_position_topic': None, 'set_position_template': None,
'unique_id': None, 'device_config': None, 'unique_id': None, 'device_config': None,
}, },
None,
None) None)
assert 44 == mqtt_cover.find_percentage_in_range(44) assert 44 == mqtt_cover.find_percentage_in_range(44)
@ -788,6 +789,7 @@ class TestCoverMQTT(unittest.TestCase):
'set_position_topic': None, 'set_position_template': None, 'set_position_topic': None, 'set_position_template': None,
'unique_id': None, 'device_config': None, 'unique_id': None, 'device_config': None,
}, },
None,
None) None)
assert 40 == mqtt_cover.find_percentage_in_range(120) assert 40 == mqtt_cover.find_percentage_in_range(120)
@ -818,6 +820,7 @@ class TestCoverMQTT(unittest.TestCase):
'set_position_topic': None, 'set_position_template': None, 'set_position_topic': None, 'set_position_template': None,
'unique_id': None, 'device_config': None, 'unique_id': None, 'device_config': None,
}, },
None,
None) None)
assert 56 == mqtt_cover.find_percentage_in_range(44) assert 56 == mqtt_cover.find_percentage_in_range(44)
@ -848,6 +851,7 @@ class TestCoverMQTT(unittest.TestCase):
'set_position_topic': None, 'set_position_template': None, 'set_position_topic': None, 'set_position_template': None,
'unique_id': None, 'device_config': None, 'unique_id': None, 'device_config': None,
}, },
None,
None) None)
assert 60 == mqtt_cover.find_percentage_in_range(120) assert 60 == mqtt_cover.find_percentage_in_range(120)
@ -878,6 +882,7 @@ class TestCoverMQTT(unittest.TestCase):
'set_position_topic': None, 'set_position_template': None, 'set_position_topic': None, 'set_position_template': None,
'unique_id': None, 'device_config': None, 'unique_id': None, 'device_config': None,
}, },
None,
None) None)
assert 44 == mqtt_cover.find_in_range_from_percent(44) assert 44 == mqtt_cover.find_in_range_from_percent(44)
@ -908,6 +913,7 @@ class TestCoverMQTT(unittest.TestCase):
'set_position_topic': None, 'set_position_template': None, 'set_position_topic': None, 'set_position_template': None,
'unique_id': None, 'device_config': None, 'unique_id': None, 'device_config': None,
}, },
None,
None) None)
assert 120 == mqtt_cover.find_in_range_from_percent(40) assert 120 == mqtt_cover.find_in_range_from_percent(40)
@ -938,6 +944,7 @@ class TestCoverMQTT(unittest.TestCase):
'set_position_topic': None, 'set_position_template': None, 'set_position_topic': None, 'set_position_template': None,
'unique_id': None, 'device_config': None, 'unique_id': None, 'device_config': None,
}, },
None,
None) None)
assert 44 == mqtt_cover.find_in_range_from_percent(56) assert 44 == mqtt_cover.find_in_range_from_percent(56)
@ -968,6 +975,7 @@ class TestCoverMQTT(unittest.TestCase):
'set_position_topic': None, 'set_position_template': None, 'set_position_topic': None, 'set_position_template': None,
'unique_id': None, 'device_config': None, 'unique_id': None, 'device_config': None,
}, },
None,
None) None)
assert 120 == mqtt_cover.find_in_range_from_percent(60) assert 120 == mqtt_cover.find_in_range_from_percent(60)
@ -1293,6 +1301,53 @@ async def test_entity_device_info_with_identifier(hass, mqtt_mock):
assert device.sw_version == '0.1-beta' assert device.sw_version == '0.1-beta'
async def test_entity_device_info_update(hass, mqtt_mock):
"""Test device registry update."""
entry = MockConfigEntry(domain=mqtt.DOMAIN)
entry.add_to_hass(hass)
await async_start(hass, 'homeassistant', {}, entry)
registry = await hass.helpers.device_registry.async_get_registry()
config = {
'platform': 'mqtt',
'name': 'Test 1',
'state_topic': 'test-topic',
'command_topic': 'test-command-topic',
'device': {
'identifiers': ['helloworld'],
'connections': [
["mac", "02:5b:26:a8:dc:12"],
],
'manufacturer': 'Whatever',
'name': 'Beer',
'model': 'Glass',
'sw_version': '0.1-beta',
},
'unique_id': 'veryunique'
}
data = json.dumps(config)
async_fire_mqtt_message(hass, 'homeassistant/cover/bla/config',
data)
await hass.async_block_till_done()
await hass.async_block_till_done()
device = registry.async_get_device({('mqtt', 'helloworld')}, set())
assert device is not None
assert device.name == 'Beer'
config['device']['name'] = 'Milk'
data = json.dumps(config)
async_fire_mqtt_message(hass, 'homeassistant/cover/bla/config',
data)
await hass.async_block_till_done()
await hass.async_block_till_done()
device = registry.async_get_device({('mqtt', 'helloworld')}, set())
assert device is not None
assert device.name == 'Milk'
async def test_entity_id_update(hass, mqtt_mock): async def test_entity_id_update(hass, mqtt_mock):
"""Test MQTT subscriptions are managed when entity_id is updated.""" """Test MQTT subscriptions are managed when entity_id is updated."""
registry = mock_registry(hass, {}) registry = mock_registry(hass, {})