From 8a08eb668876e73d5f18983fb591cce626ca4b27 Mon Sep 17 00:00:00 2001
From: Joffrey F <joffrey@docker.com>
Date: Mon, 16 Oct 2017 11:43:06 -0700
Subject: [PATCH] Move build labels parsing to config module

Signed-off-by: Joffrey F <joffrey@docker.com>
---
 compose/service.py                |  3 +--
 tests/integration/service_test.py | 19 +------------------
 tests/unit/service_test.py        |  4 ++--
 3 files changed, 4 insertions(+), 22 deletions(-)

diff --git a/compose/service.py b/compose/service.py
index 22a7ca53a..48d428cb8 100644
--- a/compose/service.py
+++ b/compose/service.py
@@ -23,7 +23,6 @@ from . import const
 from . import progress_stream
 from .config import DOCKER_CONFIG_KEYS
 from .config import merge_environment
-from .config.config import parse_labels
 from .config.errors import DependencyError
 from .config.types import ServicePort
 from .config.types import VolumeSpec
@@ -918,7 +917,7 @@ class Service(object):
             nocache=no_cache,
             dockerfile=build_opts.get('dockerfile', None),
             cache_from=build_opts.get('cache_from', None),
-            labels=parse_labels(build_opts.get('labels', None)),
+            labels=build_opts.get('labels', None),
             buildargs=build_args,
             network_mode=build_opts.get('network', None),
             target=build_opts.get('target', None),
diff --git a/tests/integration/service_test.py b/tests/integration/service_test.py
index 6cf8ddaa9..3ddf991b3 100644
--- a/tests/integration/service_test.py
+++ b/tests/integration/service_test.py
@@ -761,7 +761,7 @@ class ServiceTest(DockerClientTestCase):
         assert service.image()
         assert "build_version=2" in service.image()['ContainerConfig']['Cmd']
 
-    def test_build_with_build_labels_dict(self):
+    def test_build_with_build_labels(self):
         base_dir = tempfile.mkdtemp()
         self.addCleanup(shutil.rmtree, base_dir)
 
@@ -778,23 +778,6 @@ class ServiceTest(DockerClientTestCase):
         assert service.image()
         assert service.image()['Config']['Labels']['com.docker.compose.test'] == 'true'
 
-    def test_build_with_build_labels_list(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')
-
-        service = self.create_service('buildlabels', build={
-            'context': text_type(base_dir),
-            'labels': ['com.docker.compose.test=true']
-        })
-        service.build()
-        self.addCleanup(self.client.remove_image, service.image_name)
-
-        assert service.image()
-        assert service.image()['Config']['Labels']['com.docker.compose.test'] == 'true'
-
     @no_cluster('Container networks not on Swarm')
     def test_build_with_network(self):
         base_dir = tempfile.mkdtemp()
diff --git a/tests/unit/service_test.py b/tests/unit/service_test.py
index 50b09c87f..0bf0280de 100644
--- a/tests/unit/service_test.py
+++ b/tests/unit/service_test.py
@@ -473,7 +473,7 @@ class ServiceTest(unittest.TestCase):
             nocache=False,
             rm=True,
             buildargs={},
-            labels={},
+            labels=None,
             cache_from=None,
             network_mode=None,
             target=None,
@@ -514,7 +514,7 @@ class ServiceTest(unittest.TestCase):
             nocache=False,
             rm=True,
             buildargs={},
-            labels={},
+            labels=None,
             cache_from=None,
             network_mode=None,
             target=None,