get_process_children is updated

dmitry-lipetsk · dmitry-lipetsk · commit cc4361c6fce7 · 2025-03-14T18:44:28.000+03:00
diff --git a/testgres/operations/local_ops.py b/testgres/operations/local_ops.py
@@ -418,6 +418,7 @@ def get_pid(self):
         return os.getpid()
 
     def get_process_children(self, pid):
+        assert type(pid) == int  # noqa: E721
         return psutil.Process(pid).children()
 
     # Database control
diff --git a/testgres/operations/remote_ops.py b/testgres/operations/remote_ops.py
@@ -599,15 +599,16 @@ def get_pid(self):
         return int(self.exec_command("echo $$", encoding=get_default_encoding()))
 
     def get_process_children(self, pid):
-        command = ["ssh"] + self.ssh_args + [self.ssh_dest, f"pgrep -P {pid}"]
+        assert type(pid) == int  # noqa: E721
+        command = ["ssh"] + self.ssh_args + [self.ssh_dest, "pgrep", "-P", str(pid)]
 
         result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
 
         if result.returncode == 0:
             children = result.stdout.strip().splitlines()
             return [PsUtilProcessProxy(self, int(child_pid.strip())) for child_pid in children]
-        else:
-            raise ExecUtilException(f"Error in getting process children. Error: {result.stderr}")
+
+        raise ExecUtilException(f"Error in getting process children. Error: {result.stderr}")
 
     # Database control
     def db_connect(self, dbname, user, password=None, host="localhost", port=5432):
