tests: set env for run_binary

gsmolk · gsmolk · commit 23b00b1ddfec · 2021-04-26T12:52:37.000+03:00
diff --git a/tests/archive.py b/tests/archive.py
@@ -2030,7 +2030,9 @@ def test_archive_pg_receivexlog_partial_handling(self):
         if self.archive_compress and node.major_version >= 10:
             cmdline += ['-Z', '1']
 
-        pg_receivexlog = self.run_binary(cmdline, asynchronous=True)
+        env = self.test_env
+        env["PGAPPNAME"] = app_name
+        pg_receivexlog = self.run_binary(cmdline, asynchronous=True, env)
 
         if pg_receivexlog.returncode:
             self.assertFalse(
diff --git a/tests/helpers/ptrack_helpers.py b/tests/helpers/ptrack_helpers.py
@@ -779,7 +779,11 @@ def run_pb(self, command, asynchronous=False, gdb=False, old_binary=False, retur
         except subprocess.CalledProcessError as e:
             raise ProbackupException(e.output.decode('utf-8'), self.cmd)
 
-    def run_binary(self, command, asynchronous=False):
+    def run_binary(self, command, asynchronous=False, env=None):
+
+        if not env:
+            env = self.test_env
+
         if self.verbose:
                 print([' '.join(map(str, command))])
         try:
@@ -789,13 +793,13 @@ def run_binary(self, command, asynchronous=False):
                     stdin=subprocess.PIPE,
                     stdout=subprocess.PIPE,
                     stderr=subprocess.PIPE,
-                    env=self.test_env
+                    env=env
                 )
             else:
                 self.output = subprocess.check_output(
                     command,
                     stderr=subprocess.STDOUT,
-                    env=self.test_env
+                    env=env
                     ).decode('utf-8')
                 return self.output
         except subprocess.CalledProcessError as e:
