pytester: factor out testdir._env_run_update

Author: blueyed

src/_pytest/pytester.py

@@ -517,6 +517,10 @@ def __init__(self, request, tmpdir_factory):
         # Discard outer pytest options.
         mp.delenv("PYTEST_ADDOPTS", raising=False)
 
+        # Environment (updates) for inner runs.
+        tmphome = str(self.tmpdir)
+        self._env_run_update = {"HOME": tmphome, "USERPROFILE": tmphome}
+
     def __repr__(self):
         return "<Testdir %r>" % (self.tmpdir,)
 
@@ -813,8 +817,8 @@ def inline_run(self, *args, **kwargs):
         try:
             # Do not load user config (during runs only).
             mp_run = MonkeyPatch()
-            mp_run.setenv("HOME", str(self.tmpdir))
-            mp_run.setenv("USERPROFILE", str(self.tmpdir))
+            for k, v in self._env_run_update.items():
+                mp_run.setenv(k, v)
             finalizers.append(mp_run.undo)
 
             # When running pytest inline any plugins active in the main test
@@ -1054,9 +1058,7 @@ def popen(
         env["PYTHONPATH"] = os.pathsep.join(
             filter(None, [os.getcwd(), env.get("PYTHONPATH", "")])
         )
-        # Do not load user config.
-        env["HOME"] = str(self.tmpdir)
-        env["USERPROFILE"] = env["HOME"]
+        env.update(self._env_run_update)
         kw["env"] = env
 
         if stdin is Testdir.CLOSE_STDIN:
@@ -1245,8 +1247,7 @@ def spawn(self, cmd, expect_timeout=10.0):
 
         # Do not load user config.
         env = os.environ.copy()
-        env["HOME"] = str(self.tmpdir)
-        env["USERPROFILE"] = env["HOME"]
+        env.update(self._env_run_update)
 
         child = pexpect.spawn(cmd, logfile=logfile, env=env)
         self.request.addfinalizer(logfile.close)

testing/test_pytester.py

@@ -569,12 +569,15 @@ def test_spawn_uses_tmphome(testdir):
     # Does use HOME only during run.
     assert os.environ.get("HOME") != tmphome
 
+    testdir._env_run_update["CUSTOMENV"] = "42"
+
     p1 = testdir.makepyfile(
         """
         import os
 
         def test():
             assert os.environ["HOME"] == {tmphome!r}
+            assert os.environ["CUSTOMENV"] == "42"
         """.format(
             tmphome=tmphome
         )