Browse Source

Unset env vars behavior in 'run' mirroring engine

Unset env vars passed to `run` via command line options take the
value of the system's var with the same name.

Signed-off-by: Joffrey F <[email protected]>
Joffrey F 9 years ago
parent
commit
5d244ef6d8
3 changed files with 23 additions and 1 deletions
  1. 3 1
      compose/cli/main.py
  2. 12 0
      compose/config/environment.py
  3. 8 0
      tests/acceptance/cli_test.py

+ 3 - 1
compose/cli/main.py

@@ -891,7 +891,9 @@ def build_container_options(options, detach, command):
     }
 
     if options['-e']:
-        container_options['environment'] = parse_environment(options['-e'])
+        container_options['environment'] = Environment.from_command_line(
+            parse_environment(options['-e'])
+        )
 
     if options['--entrypoint']:
         container_options['entrypoint'] = options.get('--entrypoint')

+ 12 - 0
compose/config/environment.py

@@ -60,6 +60,18 @@ class Environment(dict):
         instance.update(os.environ)
         return instance
 
+    @classmethod
+    def from_command_line(cls, parsed_env_opts):
+        result = cls()
+        for k, v in parsed_env_opts.items():
+            # Values from the command line take priority, unless they're unset
+            # in which case they take the value from the system's environment
+            if v is None and k in os.environ:
+                result[k] = os.environ[k]
+            else:
+                result[k] = v
+        return result
+
     def __getitem__(self, key):
         try:
             return super(Environment, self).__getitem__(key)

+ 8 - 0
tests/acceptance/cli_test.py

@@ -1216,6 +1216,14 @@ class CLITestCase(DockerClientTestCase):
             'simplecomposefile_simple_run_1',
             'exited'))
 
+    @mock.patch.dict(os.environ)
+    def test_run_env_values_from_system(self):
+        os.environ['FOO'] = 'bar'
+        os.environ['BAR'] = 'baz'
+        result = self.dispatch(['run', '-e', 'FOO', 'simple', 'env'], None)
+        assert 'FOO=bar' in result.stdout
+        assert 'BAR=baz' not in result.stdout
+
     def test_rm(self):
         service = self.project.get_service('simple')
         service.create_container()