server.reboot: properly disconnect before trying to reconnect

Otherwise the `reboot` operation will assume we are already connected
after the initial delay and will go on even though we don't have a
valid connection.

Author: wackou

pyinfra/api/host.py

@@ -412,6 +412,7 @@ def disconnect(self):
         remove_any_sudo_askpass_file(self)
 
         self.state.trigger_callbacks("host_disconnect", self)
+        self.connected = False
 
     def run_shell_command(self, *args, **kwargs) -> tuple[bool, CommandOutput]:
         """

pyinfra/operations/server.py

@@ -89,7 +89,7 @@ def wait_and_reconnect(state, host):  # pragma: no cover
         sleep(delay)
         max_retries = round(reboot_timeout / interval)
 
-        host.connection = None  # remove the connection object
+        host.disconnect()  # make sure we are properly disconnected
         retries = 0
 
         while True: