Set SRCDIR and clean environment

1 files changed, 27 insertions, 1 deletions

diff --git a/yarn b/yarn
index 288e23b..6fbd63a 100755
--- a/yarn
+++ b/yarn
@@ -267,6 +267,31 @@ class YarnRunner(cliapp.Application):
 
         return ok
 
+    def clean_env(self):
+        '''Return a clean environment for running tests.'''
+
+        whitelisted = [
+            'TERM',
+            'USER',
+            'USERNAME',
+            'PATH',
+            'HOME',
+            'LOGNAME',
+        ]
+
+        hardcoded = {
+            'SHELL': '/bin/sh',
+            'LC_ALL': 'C',
+        }
+
+        env = {}
+        for key in whitelisted:
+            if key in os.environ:
+                env[key] = os.environ[key]
+        for key in hardcoded:
+            env[key] = hardcoded[key]
+        return env
+
     def run_step(self, datadir, scenario, step, shell_prelude, report_error):
         self.info('Running step "%s %s"' % (step.what, step.text))
         self.ts['step'] = step
@@ -274,8 +299,9 @@ class YarnRunner(cliapp.Application):
 
         m = re.match(step.implementation.regexp, step.text)
         assert m is not None
-        env = os.environ.copy()
+        env = self.clean_env()
         env['DATADIR'] = datadir
+        env['SRCDIR'] = os.getcwd()
         for i, match in enumerate(m.groups('')):
             env['MATCH_%d' % (i+1)] = match