feat: sudo prompt check test (#432)

freelabz · Sep 15, 2024 · f45b123 · f45b123
1 parent 0e26b55
commit f45b123
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/secator/runners/command.py b/secator/runners/command.py
@@ -477,8 +477,12 @@ def _prompt_sudo(self, command):
 			return None
 
 		# Check if sudo can be executed without a password
-		if subprocess.run(['sudo', '-n', 'true'], capture_output=False).returncode == 0:
-			return None
+		try:
+		        if subprocess.run(['sudo', '-n', 'true'], capture_output=False).returncode == 0:
+			        return None
+		except ValueError:
+			error = "Could not run sudo check test"
+			self.errors.append(error)
 
 		# Check if we have a tty
 		if not os.isatty(sys.stdin.fileno()):