'Test when message is received.'

def test_on_message(self):

    device = _get_device_on()
    component = dyson.DysonPureCoolLinkDevice(self.hass, device)
    component.entity_id = 'entity_id'
    component.schedule_update_ha_state = mock.Mock()
    component.on_message(MockDysonState())
    component.schedule_update_ha_state.assert_called_with()
