[context] Split "env" values and prefixes.

recipe_engine/unittests/step_runner_test.py

 #!/usr/bin/env python
 # Copyright 2013 The LUCI Authors. All rights reserved.
 # Use of this source code is governed under the Apache License, Version 2.0
 # that can be found in the LICENSE file.
 
+import os
 import unittest
 
 import test_env
 
 from recipe_engine import step_runner
 
 
 class TestLinebuf(unittest.TestCase):
   def test_add_partial(self):
     lb = step_runner._streamingLinebuf()
@@ skipping 30 matching lines @@
 
     lb.ingest("doop\n")
     self.assertEqual(["nerdsdoop"], lb.get_buffered())
 
     self.assertEqual([], lb.get_buffered())
 
     self.assertEqual([], lb.buffedlines)
     self.assertEqual("", lb.extra.getvalue())
 
 
+class TestMergeEnvs(unittest.TestCase):
+  def setUp(self):
+    self.original = {
+        'FOO': 'foo',
+    }
+
+  @staticmethod
+  def pathjoin(*parts):
+    return os.pathsep.join(parts)
+
+  def test_nothing_to_do(self):
+    self.assertEqual(
+        step_runner._merge_envs(self.original, {}, {}),
+        self.original)
+
+  def test_merge_with_empty_path_tuple(self):
+    self.assertEqual(step_runner._merge_envs(
+      self.original, {'FOO': ''}, {'FOO': ()}),
+      {'FOO': ''})
+
+  def test_merge_with_path_tuple(self):
+    self.assertEqual(step_runner._merge_envs(
+      self.original, {}, {'BAR': ('bar', 'baz')}),
+      {'FOO': 'foo', 'BAR': self.pathjoin('bar', 'baz')})
+
+  def test_merge_with_path_tuple_and_orig_env(self):
+    self.assertEqual(step_runner._merge_envs(
+      self.original, {}, {'FOO': ('bar', 'baz')}),
+      {'FOO': self.pathjoin('bar', 'baz', 'foo')})
+
+  def test_merge_with_path_tuple_and_env(self):
+    self.assertEqual(step_runner._merge_envs(
+      self.original, {'FOO': 'override'}, {'FOO': ('bar', 'baz')}),
+      {'FOO': self.pathjoin('bar', 'baz', 'override')})
+
+  def test_merge_with_path_tuple_and_env_subst(self):
+    self.assertEqual(step_runner._merge_envs(
+      self.original, {'FOO': 'a-%(FOO)s-b'}, {'FOO': ('bar', 'baz')}),
+      {'FOO': self.pathjoin('bar', 'baz', 'a-foo-b')})
+
+  def test_merge_with_path_tuple_and_env_empty_subst(self):
+    self.original['FOO'] = ''
+    self.assertEqual(step_runner._merge_envs(
+      self.original, {'FOO': '%(FOO)s'}, {'FOO': ('bar', 'baz')}),
+      {'FOO': self.pathjoin('bar', 'baz')})
+
+  def test_merge_with_path_tuple_and_env_clear(self):
+    self.assertEqual(step_runner._merge_envs(
+      self.original, {'FOO': None}, {'FOO': ('bar', 'baz')}),
+      {'FOO': self.pathjoin('bar', 'baz')})
+
+  def test_merge_with_subst(self):
+    self.assertEqual(step_runner._merge_envs(
+        self.original, {'FOO': 'a-%(FOO)s-b'}, {}),
+        {'FOO': 'a-foo-b'})
+
+  def test_merge_with_empty_subst(self):
+    self.original['FOO'] = ''
+    self.assertEqual(step_runner._merge_envs(
+        self.original, {'FOO': '%(FOO)s'}, {}),
+        {'FOO': ''})
+
+  def test_merge_with_clear(self):
+    self.assertEqual(step_runner._merge_envs(
+        self.original, {'FOO': None}, {}),
+        {})
+
+
 if __name__ == '__main__':
   unittest.main()