Properly resolve build args against host environment values

Signed-off-by: Joffrey F <joffrey@docker.com>
This commit is contained in:
Joffrey F 2017-03-07 17:47:41 -08:00
parent 449dcc9d7b
commit d5a2d37d05
7 changed files with 43 additions and 82 deletions

View file

@ -597,12 +597,30 @@ class ServiceTest(DockerClientTestCase):
with open(os.path.join(base_dir, 'Dockerfile'), 'w') as f:
f.write("FROM busybox\n")
f.write("ARG build_version\n")
f.write("RUN echo ${build_version}\n")
service = self.create_service('buildwithargs',
build={'context': text_type(base_dir),
'args': {"build_version": "1"}})
service.build()
assert service.image()
assert "build_version=1" in service.image()['ContainerConfig']['Cmd']
def test_build_with_build_args_override(self):
base_dir = tempfile.mkdtemp()
self.addCleanup(shutil.rmtree, base_dir)
with open(os.path.join(base_dir, 'Dockerfile'), 'w') as f:
f.write("FROM busybox\n")
f.write("ARG build_version\n")
f.write("RUN echo ${build_version}\n")
service = self.create_service('buildwithargs',
build={'context': text_type(base_dir),
'args': {"build_version": "1"}})
service.build(build_args_override={'build_version': '2'})
assert service.image()
assert "build_version=2" in service.image()['ContainerConfig']['Cmd']
def test_start_container_stays_unprivileged(self):
service = self.create_service('web')
@ -1057,7 +1075,7 @@ class ServiceTest(DockerClientTestCase):
one_off_container = service.create_container(one_off=True)
self.assertNotEqual(one_off_container.name, 'my-web-container')
@pytest.mark.skipif(True, reason="Broken on 1.11.0rc1")
@pytest.mark.skipif(True, reason="Broken on 1.11.0 - 17.03.0")
def test_log_drive_invalid(self):
service = self.create_service('web', logging={'driver': 'xxx'})
expected_error_msg = "logger: no log driver named 'xxx' is registered"

View file

@ -15,7 +15,6 @@ import yaml
from ...helpers import build_config_details
from compose.config import config
from compose.config import types
from compose.config.config import merge_build_args
from compose.config.config import resolve_build_args
from compose.config.config import resolve_environment
from compose.config.config import V1
@ -1523,7 +1522,7 @@ class ConfigTest(unittest.TestCase):
assert actual == {
'image': 'alpine:edge',
'volumes': ['.:/app'],
'ports': ['5432']
'ports': types.ServicePort.parse('5432')
}
def test_merge_service_dicts_heterogeneous_2(self):
@ -1542,40 +1541,7 @@ class ConfigTest(unittest.TestCase):
assert actual == {
'image': 'alpine:edge',
'volumes': ['.:/app'],
'ports': ['5432']
}
def test_merge_build_args(self):
base = {
'build': {
'context': '.',
'args': {
'ONE': '1',
'TWO': '2',
},
}
}
override = {
'build': {
'args': {
'TWO': 'dos',
'THREE': '3',
},
}
}
actual = config.merge_service_dicts(
base,
override,
DEFAULT_VERSION)
assert actual == {
'build': {
'context': '.',
'args': {
'ONE': '1',
'TWO': 'dos',
'THREE': '3',
},
}
'ports': types.ServicePort.parse('5432')
}
def test_merge_logging_v1(self):
@ -2878,28 +2844,10 @@ class EnvTest(unittest.TestCase):
}
}
self.assertEqual(
resolve_build_args(build, Environment.from_env_file(build['context'])),
resolve_build_args(build['args'], Environment.from_env_file(build['context'])),
{'arg1': 'value1', 'empty_arg': '', 'env_arg': 'value2', 'no_env': None},
)
@mock.patch.dict(os.environ)
def test_merge_build_args(self):
os.environ['env_arg'] = 'value2'
base = {
'arg1': 'arg1_value',
'arg2': 'arg2_value'
}
override = {
'arg1': 'arg1_new_value',
'arg2': 'arg2_value'
}
self.assertEqual(base['arg1'], 'arg1_value')
merge_build_args(base, override, os.environ)
self.assertEqual(base['arg1'], 'arg1_new_value')
@pytest.mark.xfail(IS_WINDOWS_PLATFORM, reason='paths use slash')
@mock.patch.dict(os.environ)
def test_resolve_path(self):

View file

@ -461,7 +461,7 @@ class ServiceTest(unittest.TestCase):
forcerm=False,
nocache=False,
rm=True,
buildargs=None,
buildargs={},
cache_from=None,
)
@ -498,7 +498,7 @@ class ServiceTest(unittest.TestCase):
forcerm=False,
nocache=False,
rm=True,
buildargs=None,
buildargs={},
cache_from=None,
)
@ -518,19 +518,17 @@ class ServiceTest(unittest.TestCase):
b'{"stream": "Successfully built 12345"}',
]
build_args = [
'arg1=arg1_new_value',
'arg2=arg2_value'
]
build_args = {
'arg1': 'arg1_new_value',
}
service = Service('foo', client=self.mock_client,
build={'context': '.', 'args': {'arg1': 'arg1', 'arg2': 'arg2'}})
service.build(build_args=build_args)
service.build(build_args_override=build_args)
called_build_args = self.mock_client.build.call_args[1]['buildargs']
for arg in called_build_args:
if "arg1=" in arg:
self.assertEquals(arg, 'arg1=arg1_new_value')
assert called_build_args['arg1'] == build_args['arg1']
assert called_build_args['arg2'] == 'arg2'
def test_config_dict(self):
self.mock_client.inspect_image.return_value = {'Id': 'abcd'}