Merge pull request #134984 from blaggacao/fix-125992-2

fix 125992 2

nixos/tests/common/ec2.nix

@@ -23,6 +23,7 @@ with pkgs.lib;
       testScript = ''
         import os
         import subprocess
+        import tempfile
 
         image_dir = os.path.join(
             os.environ.get("TMPDIR", tempfile.gettempdir()), "tmp", "vm-state-machine"

nixos/tests/postgresql.nix

@@ -61,6 +61,7 @@ let
 
       with subtest("Postgresql survives restart (bug #1735)"):
           machine.shutdown()
+          import time
           time.sleep(2)
           machine.start()
           machine.wait_for_unit("postgresql")

nixos/tests/turbovnc-headless-server.nix

@@ -57,6 +57,7 @@ import ./make-test-python.nix ({ pkgs, lib, ... }: {
             else:
                 if check_success():
                     return
+            import time
             time.sleep(retry_sleep)
 
         if not check_success():