Widget test inconsistencies

- Use `wait_for_idle` and new `wait_for_widget` instead of sleeping.