Fix sudo after reboot tmp clear (#1033)

* cleanup sudo file data * logic fixed * Fix `server.reboot` test --------- Co-authored-by: Nick Mills-Barrett <[email protected]>
pyinfra-dev · Nov 26, 2023 · 9256e6a · 9256e6a
1 parent 84fca9a
commit 9256e6a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pyinfra/operations/server.py b/pyinfra/operations/server.py
@@ -99,6 +99,12 @@ def wait_and_reconnect(state, host):  # pragma: no cover
 
     yield FunctionCommand(wait_and_reconnect, (), {})
 
+    # On certain systems sudo files are lost on reboot
+    def clean_sudo_info(state, host):
+        host.connector_data["sudo_askpass_path"] = None
+
+    yield FunctionCommand(clean_sudo_info, (), {})
+
 
 @operation(is_idempotent=False)
 def wait(port: int):

diff --git a/tests/operations/server.reboot/reboot.json b/tests/operations/server.reboot/reboot.json
@@ -6,7 +6,8 @@
             "command": "reboot",
             "success_exit_codes": [0, -1]
         },
-        ["wait_and_reconnect", [], {}]
+        ["wait_and_reconnect", [], {}],
+        ["clean_sudo_info", [], {}]
     ],
     "idempotent": false
 }