'Test when message is received.'

def test_on_message(self):

    device = _get_vacuum_device_cleaning()
    component = Dyson360EyeDevice(device)
    component.entity_id = 'entity_id'
    component.schedule_update_ha_state = mock.Mock()
    component.on_message(mock.Mock())
    self.assertTrue(component.schedule_update_ha_state.called)
