Add cwd to ops

vshepard · vshepard · commit d091eb15f04e · 2025-06-24T14:23:15.000+02:00
diff --git a/testgres/operations/local_ops.py b/testgres/operations/local_ops.py
@@ -101,6 +101,7 @@ def _run_command__nt(self, cmd, shell, input, stdin, stdout, stderr, get_process
             process = subprocess.Popen(
                 cmd,
                 shell=shell,
+                cwd=self.cwd,
                 stdin=stdin or subprocess.PIPE if input is not None else None,
                 stdout=stdout,
                 stderr=stderr,
@@ -151,6 +152,7 @@ def _run_command__generic(self, cmd, shell, input, stdin, stdout, stderr, get_pr
         process = subprocess.Popen(
             cmd,
             shell=shell,
+            cwd=self.cwd,
             stdin=stdin or subprocess.PIPE if input is not None else None,
             stdout=stdout or subprocess.PIPE,
             stderr=stderr or subprocess.PIPE,
diff --git a/testgres/operations/os_ops.py b/testgres/operations/os_ops.py
@@ -1,5 +1,6 @@
 import getpass
 import locale
+import os
 
 
 class ConnectionParams:
@@ -18,6 +19,7 @@ def get_default_encoding():
 
 class OsOperations:
     def __init__(self, username=None):
+        self.cwd = os.getcwd()
         self.ssh_key = None
         self.username = username or getpass.getuser()
 
@@ -133,3 +135,7 @@ def is_port_free(self, number: int):
 
     def get_tempdir(self) -> str:
         raise NotImplementedError()
+
+    def set_cwd(self, cwd):
+        if cwd:
+            self.cwd = cwd
