wait_for_login() related update

Redhat ID: 1080834



Hi @FengYang here the session is the serial session but not a ssh session. I am thinking that maybe it is better to return a ssh session here if restart_network set to True,