mirror of
				https://github.com/docker/compose.git
				synced 2025-11-04 05:34:09 +01:00 
			
		
		
		
	It's only used in a single test method. Signed-off-by: Aanand Prasad <aanand.prasad@gmail.com>
		
			
				
	
	
		
			708 lines
		
	
	
		
			29 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			708 lines
		
	
	
		
			29 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from __future__ import unicode_literals
 | 
						|
from __future__ import absolute_import
 | 
						|
import os
 | 
						|
from os import path
 | 
						|
import mock
 | 
						|
 | 
						|
import tempfile
 | 
						|
import shutil
 | 
						|
import six
 | 
						|
 | 
						|
from compose import __version__
 | 
						|
from compose.const import (
 | 
						|
    LABEL_CONTAINER_NUMBER,
 | 
						|
    LABEL_ONE_OFF,
 | 
						|
    LABEL_PROJECT,
 | 
						|
    LABEL_SERVICE,
 | 
						|
    LABEL_VERSION,
 | 
						|
)
 | 
						|
from compose.service import (
 | 
						|
    ConfigError,
 | 
						|
    Service,
 | 
						|
    build_extra_hosts,
 | 
						|
)
 | 
						|
from compose.container import Container
 | 
						|
from docker.errors import APIError
 | 
						|
from .testcases import DockerClientTestCase
 | 
						|
 | 
						|
 | 
						|
def create_and_start_container(service, **override_options):
 | 
						|
    container = service.create_container(**override_options)
 | 
						|
    return service.start_container(container)
 | 
						|
 | 
						|
 | 
						|
class ServiceTest(DockerClientTestCase):
 | 
						|
    def test_containers(self):
 | 
						|
        foo = self.create_service('foo')
 | 
						|
        bar = self.create_service('bar')
 | 
						|
 | 
						|
        create_and_start_container(foo)
 | 
						|
 | 
						|
        self.assertEqual(len(foo.containers()), 1)
 | 
						|
        self.assertEqual(foo.containers()[0].name, 'composetest_foo_1')
 | 
						|
        self.assertEqual(len(bar.containers()), 0)
 | 
						|
 | 
						|
        create_and_start_container(bar)
 | 
						|
        create_and_start_container(bar)
 | 
						|
 | 
						|
        self.assertEqual(len(foo.containers()), 1)
 | 
						|
        self.assertEqual(len(bar.containers()), 2)
 | 
						|
 | 
						|
        names = [c.name for c in bar.containers()]
 | 
						|
        self.assertIn('composetest_bar_1', names)
 | 
						|
        self.assertIn('composetest_bar_2', names)
 | 
						|
 | 
						|
    def test_containers_one_off(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
        container = db.create_container(one_off=True)
 | 
						|
        self.assertEqual(db.containers(stopped=True), [])
 | 
						|
        self.assertEqual(db.containers(one_off=True, stopped=True), [container])
 | 
						|
 | 
						|
    def test_project_is_added_to_container_name(self):
 | 
						|
        service = self.create_service('web')
 | 
						|
        create_and_start_container(service)
 | 
						|
        self.assertEqual(service.containers()[0].name, 'composetest_web_1')
 | 
						|
 | 
						|
    def test_start_stop(self):
 | 
						|
        service = self.create_service('scalingtest')
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 0)
 | 
						|
 | 
						|
        service.create_container()
 | 
						|
        self.assertEqual(len(service.containers()), 0)
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 1)
 | 
						|
 | 
						|
        service.start()
 | 
						|
        self.assertEqual(len(service.containers()), 1)
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 1)
 | 
						|
 | 
						|
        service.stop(timeout=1)
 | 
						|
        self.assertEqual(len(service.containers()), 0)
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 1)
 | 
						|
 | 
						|
        service.stop(timeout=1)
 | 
						|
        self.assertEqual(len(service.containers()), 0)
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 1)
 | 
						|
 | 
						|
    def test_kill_remove(self):
 | 
						|
        service = self.create_service('scalingtest')
 | 
						|
 | 
						|
        create_and_start_container(service)
 | 
						|
        self.assertEqual(len(service.containers()), 1)
 | 
						|
 | 
						|
        service.remove_stopped()
 | 
						|
        self.assertEqual(len(service.containers()), 1)
 | 
						|
 | 
						|
        service.kill()
 | 
						|
        self.assertEqual(len(service.containers()), 0)
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 1)
 | 
						|
 | 
						|
        service.remove_stopped()
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 0)
 | 
						|
 | 
						|
    def test_create_container_with_one_off(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
        container = db.create_container(one_off=True)
 | 
						|
        self.assertEqual(container.name, 'composetest_db_run_1')
 | 
						|
 | 
						|
    def test_create_container_with_one_off_when_existing_container_is_running(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
        db.start()
 | 
						|
        container = db.create_container(one_off=True)
 | 
						|
        self.assertEqual(container.name, 'composetest_db_run_1')
 | 
						|
 | 
						|
    def test_create_container_with_unspecified_volume(self):
 | 
						|
        service = self.create_service('db', volumes=['/var/db'])
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
        self.assertIn('/var/db', container.get('Volumes'))
 | 
						|
 | 
						|
    def test_create_container_with_cpu_shares(self):
 | 
						|
        service = self.create_service('db', cpu_shares=73)
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
        self.assertEqual(container.inspect()['Config']['CpuShares'], 73)
 | 
						|
 | 
						|
    def test_build_extra_hosts(self):
 | 
						|
        # string
 | 
						|
        self.assertRaises(ConfigError, lambda: build_extra_hosts("www.example.com: 192.168.0.17"))
 | 
						|
 | 
						|
        # list of strings
 | 
						|
        self.assertEqual(build_extra_hosts(
 | 
						|
            ["www.example.com:192.168.0.17"]),
 | 
						|
            {'www.example.com': '192.168.0.17'})
 | 
						|
        self.assertEqual(build_extra_hosts(
 | 
						|
            ["www.example.com: 192.168.0.17"]),
 | 
						|
            {'www.example.com': '192.168.0.17'})
 | 
						|
        self.assertEqual(build_extra_hosts(
 | 
						|
            ["www.example.com: 192.168.0.17",
 | 
						|
             "static.example.com:192.168.0.19",
 | 
						|
             "api.example.com: 192.168.0.18"]),
 | 
						|
            {'www.example.com': '192.168.0.17',
 | 
						|
             'static.example.com': '192.168.0.19',
 | 
						|
             'api.example.com': '192.168.0.18'})
 | 
						|
 | 
						|
        # list of dictionaries
 | 
						|
        self.assertRaises(ConfigError, lambda: build_extra_hosts(
 | 
						|
            [{'www.example.com': '192.168.0.17'},
 | 
						|
             {'api.example.com': '192.168.0.18'}]))
 | 
						|
 | 
						|
        # dictionaries
 | 
						|
        self.assertEqual(build_extra_hosts(
 | 
						|
            {'www.example.com': '192.168.0.17',
 | 
						|
             'api.example.com': '192.168.0.18'}),
 | 
						|
            {'www.example.com': '192.168.0.17',
 | 
						|
             'api.example.com': '192.168.0.18'})
 | 
						|
 | 
						|
    def test_create_container_with_extra_hosts_list(self):
 | 
						|
        extra_hosts = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
 | 
						|
        service = self.create_service('db', extra_hosts=extra_hosts)
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
        self.assertEqual(set(container.get('HostConfig.ExtraHosts')), set(extra_hosts))
 | 
						|
 | 
						|
    def test_create_container_with_extra_hosts_string(self):
 | 
						|
        extra_hosts = 'somehost:162.242.195.82'
 | 
						|
        service = self.create_service('db', extra_hosts=extra_hosts)
 | 
						|
        self.assertRaises(ConfigError, lambda: service.create_container())
 | 
						|
 | 
						|
    def test_create_container_with_extra_hosts_list_of_dicts(self):
 | 
						|
        extra_hosts = [{'somehost': '162.242.195.82'}, {'otherhost': '50.31.209.229'}]
 | 
						|
        service = self.create_service('db', extra_hosts=extra_hosts)
 | 
						|
        self.assertRaises(ConfigError, lambda: service.create_container())
 | 
						|
 | 
						|
    def test_create_container_with_extra_hosts_dicts(self):
 | 
						|
        extra_hosts = {'somehost': '162.242.195.82', 'otherhost': '50.31.209.229'}
 | 
						|
        extra_hosts_list = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
 | 
						|
        service = self.create_service('db', extra_hosts=extra_hosts)
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
        self.assertEqual(set(container.get('HostConfig.ExtraHosts')), set(extra_hosts_list))
 | 
						|
 | 
						|
    def test_create_container_with_cpu_set(self):
 | 
						|
        service = self.create_service('db', cpuset='0')
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
        self.assertEqual(container.inspect()['Config']['Cpuset'], '0')
 | 
						|
 | 
						|
    def test_create_container_with_read_only_root_fs(self):
 | 
						|
        read_only = True
 | 
						|
        service = self.create_service('db', read_only=read_only)
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
        self.assertEqual(container.get('HostConfig.ReadonlyRootfs'), read_only, container.get('HostConfig'))
 | 
						|
 | 
						|
    def test_create_container_with_security_opt(self):
 | 
						|
        security_opt = ['label:disable']
 | 
						|
        service = self.create_service('db', security_opt=security_opt)
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
        self.assertEqual(set(container.get('HostConfig.SecurityOpt')), set(security_opt))
 | 
						|
 | 
						|
    def test_create_container_with_specified_volume(self):
 | 
						|
        host_path = '/tmp/host-path'
 | 
						|
        container_path = '/container-path'
 | 
						|
 | 
						|
        service = self.create_service('db', volumes=['%s:%s' % (host_path, container_path)])
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
 | 
						|
        volumes = container.inspect()['Volumes']
 | 
						|
        self.assertIn(container_path, volumes)
 | 
						|
 | 
						|
        # Match the last component ("host-path"), because boot2docker symlinks /tmp
 | 
						|
        actual_host_path = volumes[container_path]
 | 
						|
        self.assertTrue(path.basename(actual_host_path) == path.basename(host_path),
 | 
						|
                        msg=("Last component differs: %s, %s" % (actual_host_path, host_path)))
 | 
						|
 | 
						|
    @mock.patch.dict(os.environ)
 | 
						|
    def test_create_container_with_home_and_env_var_in_volume_path(self):
 | 
						|
        os.environ['VOLUME_NAME'] = 'my-volume'
 | 
						|
        os.environ['HOME'] = '/tmp/home-dir'
 | 
						|
        expected_host_path = os.path.join(os.environ['HOME'], os.environ['VOLUME_NAME'])
 | 
						|
 | 
						|
        host_path = '~/${VOLUME_NAME}'
 | 
						|
        container_path = '/container-path'
 | 
						|
 | 
						|
        service = self.create_service('db', volumes=['%s:%s' % (host_path, container_path)])
 | 
						|
        container = service.create_container()
 | 
						|
        service.start_container(container)
 | 
						|
 | 
						|
        actual_host_path = container.get('Volumes')[container_path]
 | 
						|
        components = actual_host_path.split('/')
 | 
						|
        self.assertTrue(components[-2:] == ['home-dir', 'my-volume'],
 | 
						|
                        msg="Last two components differ: %s, %s" % (actual_host_path, expected_host_path))
 | 
						|
 | 
						|
    def test_create_container_with_volumes_from(self):
 | 
						|
        volume_service = self.create_service('data')
 | 
						|
        volume_container_1 = volume_service.create_container()
 | 
						|
        volume_container_2 = Container.create(self.client, image='busybox:latest', command=["top"])
 | 
						|
        host_service = self.create_service('host', volumes_from=[volume_service, volume_container_2])
 | 
						|
        host_container = host_service.create_container()
 | 
						|
        host_service.start_container(host_container)
 | 
						|
        self.assertIn(volume_container_1.id,
 | 
						|
                      host_container.get('HostConfig.VolumesFrom'))
 | 
						|
        self.assertIn(volume_container_2.id,
 | 
						|
                      host_container.get('HostConfig.VolumesFrom'))
 | 
						|
 | 
						|
    def test_converge(self):
 | 
						|
        service = self.create_service(
 | 
						|
            'db',
 | 
						|
            environment={'FOO': '1'},
 | 
						|
            volumes=['/etc'],
 | 
						|
            entrypoint=['top'],
 | 
						|
            command=['-d', '1']
 | 
						|
        )
 | 
						|
        old_container = service.create_container()
 | 
						|
        self.assertEqual(old_container.get('Config.Entrypoint'), ['top'])
 | 
						|
        self.assertEqual(old_container.get('Config.Cmd'), ['-d', '1'])
 | 
						|
        self.assertIn('FOO=1', old_container.get('Config.Env'))
 | 
						|
        self.assertEqual(old_container.name, 'composetest_db_1')
 | 
						|
        service.start_container(old_container)
 | 
						|
        old_container.inspect()  # reload volume data
 | 
						|
        volume_path = old_container.get('Volumes')['/etc']
 | 
						|
 | 
						|
        num_containers_before = len(self.client.containers(all=True))
 | 
						|
 | 
						|
        service.options['environment']['FOO'] = '2'
 | 
						|
        new_container = service.converge()[0]
 | 
						|
 | 
						|
        self.assertEqual(new_container.get('Config.Entrypoint'), ['top'])
 | 
						|
        self.assertEqual(new_container.get('Config.Cmd'), ['-d', '1'])
 | 
						|
        self.assertIn('FOO=2', new_container.get('Config.Env'))
 | 
						|
        self.assertEqual(new_container.name, 'composetest_db_1')
 | 
						|
        self.assertEqual(new_container.get('Volumes')['/etc'], volume_path)
 | 
						|
        self.assertIn(
 | 
						|
            'affinity:container==%s' % old_container.id,
 | 
						|
            new_container.get('Config.Env'))
 | 
						|
 | 
						|
        self.assertEqual(len(self.client.containers(all=True)), num_containers_before)
 | 
						|
        self.assertNotEqual(old_container.id, new_container.id)
 | 
						|
        self.assertRaises(APIError,
 | 
						|
                          self.client.inspect_container,
 | 
						|
                          old_container.id)
 | 
						|
 | 
						|
    def test_converge_when_containers_are_stopped(self):
 | 
						|
        service = self.create_service(
 | 
						|
            'db',
 | 
						|
            environment={'FOO': '1'},
 | 
						|
            volumes=['/var/db'],
 | 
						|
            entrypoint=['top'],
 | 
						|
            command=['-d', '1']
 | 
						|
        )
 | 
						|
        service.create_container()
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 1)
 | 
						|
        service.converge()
 | 
						|
        self.assertEqual(len(service.containers(stopped=True)), 1)
 | 
						|
 | 
						|
    def test_converge_with_image_declared_volume(self):
 | 
						|
        service = Service(
 | 
						|
            project='composetest',
 | 
						|
            name='db',
 | 
						|
            client=self.client,
 | 
						|
            build='tests/fixtures/dockerfile-with-volume',
 | 
						|
        )
 | 
						|
 | 
						|
        old_container = create_and_start_container(service)
 | 
						|
        self.assertEqual(old_container.get('Volumes').keys(), ['/data'])
 | 
						|
        volume_path = old_container.get('Volumes')['/data']
 | 
						|
 | 
						|
        new_container = service.converge()[0]
 | 
						|
        self.assertEqual(new_container.get('Volumes').keys(), ['/data'])
 | 
						|
        self.assertEqual(new_container.get('Volumes')['/data'], volume_path)
 | 
						|
 | 
						|
    def test_start_container_passes_through_options(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
        create_and_start_container(db, environment={'FOO': 'BAR'})
 | 
						|
        self.assertEqual(db.containers()[0].environment['FOO'], 'BAR')
 | 
						|
 | 
						|
    def test_start_container_inherits_options_from_constructor(self):
 | 
						|
        db = self.create_service('db', environment={'FOO': 'BAR'})
 | 
						|
        create_and_start_container(db)
 | 
						|
        self.assertEqual(db.containers()[0].environment['FOO'], 'BAR')
 | 
						|
 | 
						|
    def test_start_container_creates_links(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
        web = self.create_service('web', links=[(db, None)])
 | 
						|
 | 
						|
        create_and_start_container(db)
 | 
						|
        create_and_start_container(db)
 | 
						|
        create_and_start_container(web)
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
            set(web.containers()[0].links()),
 | 
						|
            set([
 | 
						|
                'composetest_db_1', 'db_1',
 | 
						|
                'composetest_db_2', 'db_2',
 | 
						|
                'db'])
 | 
						|
        )
 | 
						|
 | 
						|
    def test_start_container_creates_links_with_names(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
        web = self.create_service('web', links=[(db, 'custom_link_name')])
 | 
						|
 | 
						|
        create_and_start_container(db)
 | 
						|
        create_and_start_container(db)
 | 
						|
        create_and_start_container(web)
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
            set(web.containers()[0].links()),
 | 
						|
            set([
 | 
						|
                'composetest_db_1', 'db_1',
 | 
						|
                'composetest_db_2', 'db_2',
 | 
						|
                'custom_link_name'])
 | 
						|
        )
 | 
						|
 | 
						|
    def test_start_container_with_external_links(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
        web = self.create_service('web', external_links=['composetest_db_1',
 | 
						|
                                                         'composetest_db_2',
 | 
						|
                                                         'composetest_db_3:db_3'])
 | 
						|
 | 
						|
        for _ in range(3):
 | 
						|
            create_and_start_container(db)
 | 
						|
        create_and_start_container(web)
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
            set(web.containers()[0].links()),
 | 
						|
            set([
 | 
						|
                'composetest_db_1',
 | 
						|
                'composetest_db_2',
 | 
						|
                'db_3']),
 | 
						|
        )
 | 
						|
 | 
						|
    def test_start_normal_container_does_not_create_links_to_its_own_service(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
 | 
						|
        create_and_start_container(db)
 | 
						|
        create_and_start_container(db)
 | 
						|
 | 
						|
        c = create_and_start_container(db)
 | 
						|
        self.assertEqual(set(c.links()), set([]))
 | 
						|
 | 
						|
    def test_start_one_off_container_creates_links_to_its_own_service(self):
 | 
						|
        db = self.create_service('db')
 | 
						|
 | 
						|
        create_and_start_container(db)
 | 
						|
        create_and_start_container(db)
 | 
						|
 | 
						|
        c = create_and_start_container(db, one_off=True)
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
            set(c.links()),
 | 
						|
            set([
 | 
						|
                'composetest_db_1', 'db_1',
 | 
						|
                'composetest_db_2', 'db_2',
 | 
						|
                'db'])
 | 
						|
        )
 | 
						|
 | 
						|
    def test_start_container_builds_images(self):
 | 
						|
        service = Service(
 | 
						|
            name='test',
 | 
						|
            client=self.client,
 | 
						|
            build='tests/fixtures/simple-dockerfile',
 | 
						|
            project='composetest',
 | 
						|
        )
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        container.wait()
 | 
						|
        self.assertIn('success', container.logs())
 | 
						|
        self.assertEqual(len(self.client.images(name='composetest_test')), 1)
 | 
						|
 | 
						|
    def test_start_container_uses_tagged_image_if_it_exists(self):
 | 
						|
        self.client.build('tests/fixtures/simple-dockerfile', tag='composetest_test')
 | 
						|
        service = Service(
 | 
						|
            name='test',
 | 
						|
            client=self.client,
 | 
						|
            build='this/does/not/exist/and/will/throw/error',
 | 
						|
            project='composetest',
 | 
						|
        )
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        container.wait()
 | 
						|
        self.assertIn('success', container.logs())
 | 
						|
 | 
						|
    def test_start_container_creates_ports(self):
 | 
						|
        service = self.create_service('web', ports=[8000])
 | 
						|
        container = create_and_start_container(service).inspect()
 | 
						|
        self.assertEqual(list(container['NetworkSettings']['Ports'].keys()), ['8000/tcp'])
 | 
						|
        self.assertNotEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8000')
 | 
						|
 | 
						|
    def test_build(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")
 | 
						|
 | 
						|
        self.create_service('web', build=base_dir).build()
 | 
						|
        self.assertEqual(len(self.client.images(name='composetest_web')), 1)
 | 
						|
 | 
						|
    def test_build_non_ascii_filename(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")
 | 
						|
 | 
						|
        with open(os.path.join(base_dir, b'foo\xE2bar'), 'w') as f:
 | 
						|
            f.write("hello world\n")
 | 
						|
 | 
						|
        self.create_service('web', build=six.text_type(base_dir)).build()
 | 
						|
        self.assertEqual(len(self.client.images(name='composetest_web')), 1)
 | 
						|
 | 
						|
    def test_start_container_stays_unpriviliged(self):
 | 
						|
        service = self.create_service('web')
 | 
						|
        container = create_and_start_container(service).inspect()
 | 
						|
        self.assertEqual(container['HostConfig']['Privileged'], False)
 | 
						|
 | 
						|
    def test_start_container_becomes_priviliged(self):
 | 
						|
        service = self.create_service('web', privileged=True)
 | 
						|
        container = create_and_start_container(service).inspect()
 | 
						|
        self.assertEqual(container['HostConfig']['Privileged'], True)
 | 
						|
 | 
						|
    def test_expose_does_not_publish_ports(self):
 | 
						|
        service = self.create_service('web', expose=[8000])
 | 
						|
        container = create_and_start_container(service).inspect()
 | 
						|
        self.assertEqual(container['NetworkSettings']['Ports'], {'8000/tcp': None})
 | 
						|
 | 
						|
    def test_start_container_creates_port_with_explicit_protocol(self):
 | 
						|
        service = self.create_service('web', ports=['8000/udp'])
 | 
						|
        container = create_and_start_container(service).inspect()
 | 
						|
        self.assertEqual(list(container['NetworkSettings']['Ports'].keys()), ['8000/udp'])
 | 
						|
 | 
						|
    def test_start_container_creates_fixed_external_ports(self):
 | 
						|
        service = self.create_service('web', ports=['8000:8000'])
 | 
						|
        container = create_and_start_container(service).inspect()
 | 
						|
        self.assertIn('8000/tcp', container['NetworkSettings']['Ports'])
 | 
						|
        self.assertEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8000')
 | 
						|
 | 
						|
    def test_start_container_creates_fixed_external_ports_when_it_is_different_to_internal_port(self):
 | 
						|
        service = self.create_service('web', ports=['8001:8000'])
 | 
						|
        container = create_and_start_container(service).inspect()
 | 
						|
        self.assertIn('8000/tcp', container['NetworkSettings']['Ports'])
 | 
						|
        self.assertEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8001')
 | 
						|
 | 
						|
    def test_port_with_explicit_interface(self):
 | 
						|
        service = self.create_service('web', ports=[
 | 
						|
            '127.0.0.1:8001:8000',
 | 
						|
            '0.0.0.0:9001:9000/udp',
 | 
						|
        ])
 | 
						|
        container = create_and_start_container(service).inspect()
 | 
						|
        self.assertEqual(container['NetworkSettings']['Ports'], {
 | 
						|
            '8000/tcp': [
 | 
						|
                {
 | 
						|
                    'HostIp': '127.0.0.1',
 | 
						|
                    'HostPort': '8001',
 | 
						|
                },
 | 
						|
            ],
 | 
						|
            '9000/udp': [
 | 
						|
                {
 | 
						|
                    'HostIp': '0.0.0.0',
 | 
						|
                    'HostPort': '9001',
 | 
						|
                },
 | 
						|
            ],
 | 
						|
        })
 | 
						|
 | 
						|
    def test_create_with_image_id(self):
 | 
						|
        # Image id for the current busybox:latest
 | 
						|
        service = self.create_service('foo', image='8c2e06607696')
 | 
						|
        service.create_container()
 | 
						|
 | 
						|
    def test_scale(self):
 | 
						|
        service = self.create_service('web')
 | 
						|
        service.scale(1)
 | 
						|
        self.assertEqual(len(service.containers()), 1)
 | 
						|
 | 
						|
        # Ensure containers don't have stdout or stdin connected
 | 
						|
        container = service.containers()[0]
 | 
						|
        config = container.inspect()['Config']
 | 
						|
        self.assertFalse(config['AttachStderr'])
 | 
						|
        self.assertFalse(config['AttachStdout'])
 | 
						|
        self.assertFalse(config['AttachStdin'])
 | 
						|
 | 
						|
        service.scale(3)
 | 
						|
        self.assertEqual(len(service.containers()), 3)
 | 
						|
        service.scale(1)
 | 
						|
        self.assertEqual(len(service.containers()), 1)
 | 
						|
        service.scale(0)
 | 
						|
        self.assertEqual(len(service.containers()), 0)
 | 
						|
 | 
						|
    def test_scale_sets_ports(self):
 | 
						|
        service = self.create_service('web', ports=['8000'])
 | 
						|
        service.scale(2)
 | 
						|
        containers = service.containers()
 | 
						|
        self.assertEqual(len(containers), 2)
 | 
						|
        for container in containers:
 | 
						|
            self.assertEqual(list(container.inspect()['HostConfig']['PortBindings'].keys()), ['8000/tcp'])
 | 
						|
 | 
						|
    def test_network_mode_none(self):
 | 
						|
        service = self.create_service('web', net='none')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.NetworkMode'), 'none')
 | 
						|
 | 
						|
    def test_network_mode_bridged(self):
 | 
						|
        service = self.create_service('web', net='bridge')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.NetworkMode'), 'bridge')
 | 
						|
 | 
						|
    def test_network_mode_host(self):
 | 
						|
        service = self.create_service('web', net='host')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.NetworkMode'), 'host')
 | 
						|
 | 
						|
    def test_pid_mode_none_defined(self):
 | 
						|
        service = self.create_service('web', pid=None)
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.PidMode'), '')
 | 
						|
 | 
						|
    def test_pid_mode_host(self):
 | 
						|
        service = self.create_service('web', pid='host')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.PidMode'), 'host')
 | 
						|
 | 
						|
    def test_dns_no_value(self):
 | 
						|
        service = self.create_service('web')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertIsNone(container.get('HostConfig.Dns'))
 | 
						|
 | 
						|
    def test_dns_single_value(self):
 | 
						|
        service = self.create_service('web', dns='8.8.8.8')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.Dns'), ['8.8.8.8'])
 | 
						|
 | 
						|
    def test_dns_list(self):
 | 
						|
        service = self.create_service('web', dns=['8.8.8.8', '9.9.9.9'])
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.Dns'), ['8.8.8.8', '9.9.9.9'])
 | 
						|
 | 
						|
    def test_restart_always_value(self):
 | 
						|
        service = self.create_service('web', restart='always')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'always')
 | 
						|
 | 
						|
    def test_restart_on_failure_value(self):
 | 
						|
        service = self.create_service('web', restart='on-failure:5')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'on-failure')
 | 
						|
        self.assertEqual(container.get('HostConfig.RestartPolicy.MaximumRetryCount'), 5)
 | 
						|
 | 
						|
    def test_cap_add_list(self):
 | 
						|
        service = self.create_service('web', cap_add=['SYS_ADMIN', 'NET_ADMIN'])
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.CapAdd'), ['SYS_ADMIN', 'NET_ADMIN'])
 | 
						|
 | 
						|
    def test_cap_drop_list(self):
 | 
						|
        service = self.create_service('web', cap_drop=['SYS_ADMIN', 'NET_ADMIN'])
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.CapDrop'), ['SYS_ADMIN', 'NET_ADMIN'])
 | 
						|
 | 
						|
    def test_dns_search_no_value(self):
 | 
						|
        service = self.create_service('web')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertIsNone(container.get('HostConfig.DnsSearch'))
 | 
						|
 | 
						|
    def test_dns_search_single_value(self):
 | 
						|
        service = self.create_service('web', dns_search='example.com')
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.DnsSearch'), ['example.com'])
 | 
						|
 | 
						|
    def test_dns_search_list(self):
 | 
						|
        service = self.create_service('web', dns_search=['dc1.example.com', 'dc2.example.com'])
 | 
						|
        container = create_and_start_container(service)
 | 
						|
        self.assertEqual(container.get('HostConfig.DnsSearch'), ['dc1.example.com', 'dc2.example.com'])
 | 
						|
 | 
						|
    def test_working_dir_param(self):
 | 
						|
        service = self.create_service('container', working_dir='/working/dir/sample')
 | 
						|
        container = service.create_container()
 | 
						|
        self.assertEqual(container.get('Config.WorkingDir'), '/working/dir/sample')
 | 
						|
 | 
						|
    def test_split_env(self):
 | 
						|
        service = self.create_service('web', environment=['NORMAL=F1', 'CONTAINS_EQUALS=F=2', 'TRAILING_EQUALS='])
 | 
						|
        env = create_and_start_container(service).environment
 | 
						|
        for k, v in {'NORMAL': 'F1', 'CONTAINS_EQUALS': 'F=2', 'TRAILING_EQUALS': ''}.items():
 | 
						|
            self.assertEqual(env[k], v)
 | 
						|
 | 
						|
    def test_env_from_file_combined_with_env(self):
 | 
						|
        service = self.create_service('web', environment=['ONE=1', 'TWO=2', 'THREE=3'], env_file=['tests/fixtures/env/one.env', 'tests/fixtures/env/two.env'])
 | 
						|
        env = create_and_start_container(service).environment
 | 
						|
        for k, v in {'ONE': '1', 'TWO': '2', 'THREE': '3', 'FOO': 'baz', 'DOO': 'dah'}.items():
 | 
						|
            self.assertEqual(env[k], v)
 | 
						|
 | 
						|
    @mock.patch.dict(os.environ)
 | 
						|
    def test_resolve_env(self):
 | 
						|
        os.environ['FILE_DEF'] = 'E1'
 | 
						|
        os.environ['FILE_DEF_EMPTY'] = 'E2'
 | 
						|
        os.environ['ENV_DEF'] = 'E3'
 | 
						|
        service = self.create_service('web', environment={'FILE_DEF': 'F1', 'FILE_DEF_EMPTY': '', 'ENV_DEF': None, 'NO_DEF': None})
 | 
						|
        env = create_and_start_container(service).environment
 | 
						|
        for k, v in {'FILE_DEF': 'F1', 'FILE_DEF_EMPTY': '', 'ENV_DEF': 'E3', 'NO_DEF': ''}.items():
 | 
						|
            self.assertEqual(env[k], v)
 | 
						|
 | 
						|
    def test_labels(self):
 | 
						|
        labels_dict = {
 | 
						|
            'com.example.description': "Accounting webapp",
 | 
						|
            'com.example.department': "Finance",
 | 
						|
            'com.example.label-with-empty-value': "",
 | 
						|
        }
 | 
						|
 | 
						|
        compose_labels = {
 | 
						|
            LABEL_CONTAINER_NUMBER: '1',
 | 
						|
            LABEL_ONE_OFF: 'False',
 | 
						|
            LABEL_PROJECT: 'composetest',
 | 
						|
            LABEL_SERVICE: 'web',
 | 
						|
            LABEL_VERSION: __version__,
 | 
						|
        }
 | 
						|
        expected = dict(labels_dict, **compose_labels)
 | 
						|
 | 
						|
        service = self.create_service('web', labels=labels_dict)
 | 
						|
        labels = create_and_start_container(service).labels.items()
 | 
						|
        for pair in expected.items():
 | 
						|
            self.assertIn(pair, labels)
 | 
						|
 | 
						|
        service.kill()
 | 
						|
        service.remove_stopped()
 | 
						|
 | 
						|
        labels_list = ["%s=%s" % pair for pair in labels_dict.items()]
 | 
						|
 | 
						|
        service = self.create_service('web', labels=labels_list)
 | 
						|
        labels = create_and_start_container(service).labels.items()
 | 
						|
        for pair in expected.items():
 | 
						|
            self.assertIn(pair, labels)
 | 
						|
 | 
						|
    def test_empty_labels(self):
 | 
						|
        labels_list = ['foo', 'bar']
 | 
						|
 | 
						|
        service = self.create_service('web', labels=labels_list)
 | 
						|
        labels = create_and_start_container(service).labels.items()
 | 
						|
        for name in labels_list:
 | 
						|
            self.assertIn((name, ''), labels)
 | 
						|
 | 
						|
    def test_log_drive_invalid(self):
 | 
						|
        service = self.create_service('web', log_driver='xxx')
 | 
						|
        self.assertRaises(ValueError, lambda: create_and_start_container(service))
 | 
						|
 | 
						|
    def test_log_drive_empty_default_jsonfile(self):
 | 
						|
        service = self.create_service('web')
 | 
						|
        log_config = create_and_start_container(service).log_config
 | 
						|
 | 
						|
        self.assertEqual('json-file', log_config['Type'])
 | 
						|
        self.assertFalse(log_config['Config'])
 | 
						|
 | 
						|
    def test_log_drive_none(self):
 | 
						|
        service = self.create_service('web', log_driver='none')
 | 
						|
        log_config = create_and_start_container(service).log_config
 | 
						|
 | 
						|
        self.assertEqual('none', log_config['Type'])
 | 
						|
        self.assertFalse(log_config['Config'])
 | 
						|
 | 
						|
    def test_devices(self):
 | 
						|
        service = self.create_service('web', devices=["/dev/random:/dev/mapped-random"])
 | 
						|
        device_config = create_and_start_container(service).get('HostConfig.Devices')
 | 
						|
 | 
						|
        device_dict = {
 | 
						|
            'PathOnHost': '/dev/random',
 | 
						|
            'CgroupPermissions': 'rwm',
 | 
						|
            'PathInContainer': '/dev/mapped-random'
 | 
						|
        }
 | 
						|
 | 
						|
        self.assertEqual(1, len(device_config))
 | 
						|
        self.assertDictEqual(device_dict, device_config[0])
 |