123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921 |
- from __future__ import absolute_import
- from __future__ import unicode_literals
- import os
- import shutil
- import tempfile
- from os import path
- from docker.errors import APIError
- from six import StringIO
- from six import text_type
- from .. import mock
- from .testcases import DockerClientTestCase
- from compose import __version__
- from compose.const import LABEL_CONTAINER_NUMBER
- from compose.const import LABEL_ONE_OFF
- from compose.const import LABEL_PROJECT
- from compose.const import LABEL_SERVICE
- from compose.const import LABEL_VERSION
- from compose.container import Container
- from compose.service import build_extra_hosts
- from compose.service import ConfigError
- from compose.service import ConvergencePlan
- from compose.service import Service
- 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_volume_driver(self):
- service = self.create_service('db', volume_driver='foodriver')
- container = service.create_container()
- service.start_container(container)
- self.assertEqual('foodriver', container.get('Config.VolumeDriver'))
- 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.get('HostConfig.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.get('HostConfig.CpusetCpus'), '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_mac_address(self):
- service = self.create_service('db', mac_address='02:42:ac:11:65:43')
- container = service.create_container()
- service.start_container(container)
- self.assertEqual(container.inspect()['Config']['MacAddress'], '02:42:ac:11:65:43')
- 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)))
- def test_recreate_preserves_volume_with_trailing_slash(self):
- """
- When the Compose file specifies a trailing slash in the container path, make
- sure we copy the volume over when recreating.
- """
- service = self.create_service('data', volumes=['/data/'])
- old_container = create_and_start_container(service)
- volume_path = old_container.get('Volumes')['/data']
- new_container = service.recreate_container(old_container)
- self.assertEqual(new_container.get('Volumes')['/data'], volume_path)
- def test_duplicate_volume_trailing_slash(self):
- """
- When an image specifies a volume, and the Compose file specifies a host path
- but adds a trailing slash, make sure that we don't create duplicate binds.
- """
- host_path = '/tmp/data'
- container_path = '/data'
- volumes = ['{}:{}/'.format(host_path, container_path)]
- tmp_container = self.client.create_container(
- 'busybox', 'true',
- volumes={container_path: {}},
- labels={'com.docker.compose.test_image': 'true'},
- )
- image = self.client.commit(tmp_container)['Id']
- service = self.create_service('db', image=image, volumes=volumes)
- old_container = create_and_start_container(service)
- self.assertEqual(
- old_container.get('Config.Volumes'),
- {container_path: {}},
- )
- service = self.create_service('db', image=image, volumes=volumes)
- new_container = service.recreate_container(old_container)
- self.assertEqual(
- new_container.get('Config.Volumes'),
- {container_path: {}},
- )
- self.assertEqual(service.containers(stopped=False), [new_container])
- 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"],
- labels={LABEL_PROJECT: 'composetest'},
- )
- 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_execute_convergence_plan_recreate(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.execute_convergence_plan(
- ConvergencePlan('recreate', [old_container]))
- 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_execute_convergence_plan_when_containers_are_stopped(self):
- service = self.create_service(
- 'db',
- environment={'FOO': '1'},
- volumes=['/var/db'],
- entrypoint=['top'],
- command=['-d', '1']
- )
- service.create_container()
- containers = service.containers(stopped=True)
- self.assertEqual(len(containers), 1)
- container, = containers
- self.assertFalse(container.is_running)
- service.execute_convergence_plan(ConvergencePlan('start', [container]))
- containers = service.containers()
- self.assertEqual(len(containers), 1)
- container.inspect()
- self.assertEqual(container, containers[0])
- self.assertTrue(container.is_running)
- def test_execute_convergence_plan_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.execute_convergence_plan(
- ConvergencePlan('recreate', [old_container]))
- 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(b'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.check_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(b'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=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)
- @mock.patch('sys.stdout', new_callable=StringIO)
- def test_scale_with_stopped_containers(self, mock_stdout):
- """
- Given there are some stopped containers and scale is called with a
- desired number that is the same as the number of stopped containers,
- test that those containers are restarted and not removed/recreated.
- """
- service = self.create_service('web')
- next_number = service._next_container_number()
- valid_numbers = [next_number, next_number + 1]
- service.create_container(number=next_number, quiet=True)
- service.create_container(number=next_number + 1, quiet=True)
- for container in service.containers():
- self.assertFalse(container.is_running)
- service.scale(2)
- self.assertEqual(len(service.containers()), 2)
- for container in service.containers():
- self.assertTrue(container.is_running)
- self.assertTrue(container.number in valid_numbers)
- captured_output = mock_stdout.getvalue()
- self.assertNotIn('Creating', captured_output)
- self.assertIn('Starting', captured_output)
- @mock.patch('sys.stdout', new_callable=StringIO)
- def test_scale_with_stopped_containers_and_needing_creation(self, mock_stdout):
- """
- Given there are some stopped containers and scale is called with a
- desired number that is greater than the number of stopped containers,
- test that those containers are restarted and required number are created.
- """
- service = self.create_service('web')
- next_number = service._next_container_number()
- service.create_container(number=next_number, quiet=True)
- for container in service.containers():
- self.assertFalse(container.is_running)
- service.scale(2)
- self.assertEqual(len(service.containers()), 2)
- for container in service.containers():
- self.assertTrue(container.is_running)
- captured_output = mock_stdout.getvalue()
- self.assertIn('Creating', captured_output)
- self.assertIn('Starting', captured_output)
- @mock.patch('sys.stdout', new_callable=StringIO)
- def test_scale_with_api_returns_errors(self, mock_stdout):
- """
- Test that when scaling if the API returns an error, that error is handled
- and the remaining threads continue.
- """
- service = self.create_service('web')
- next_number = service._next_container_number()
- service.create_container(number=next_number, quiet=True)
- with mock.patch(
- 'compose.container.Container.create',
- side_effect=APIError(message="testing", response={}, explanation="Boom")):
- service.scale(3)
- self.assertEqual(len(service.containers()), 1)
- self.assertTrue(service.containers()[0].is_running)
- self.assertIn("ERROR: for 2 Boom", mock_stdout.getvalue())
- @mock.patch('sys.stdout', new_callable=StringIO)
- def test_scale_with_api_returns_unexpected_exception(self, mock_stdout):
- """
- Test that when scaling if the API returns an error, that is not of type
- APIError, that error is re-raised.
- """
- service = self.create_service('web')
- next_number = service._next_container_number()
- service.create_container(number=next_number, quiet=True)
- with mock.patch(
- 'compose.container.Container.create',
- side_effect=ValueError("BOOM")):
- with self.assertRaises(ValueError):
- service.scale(3)
- self.assertEqual(len(service.containers()), 1)
- self.assertTrue(service.containers()[0].is_running)
- @mock.patch('compose.service.log')
- def test_scale_with_desired_number_already_achieved(self, mock_log):
- """
- Test that calling scale with a desired number that is equal to the
- number of containers already running results in no change.
- """
- service = self.create_service('web')
- next_number = service._next_container_number()
- container = service.create_container(number=next_number, quiet=True)
- container.start()
- self.assertTrue(container.is_running)
- self.assertEqual(len(service.containers()), 1)
- service.scale(1)
- self.assertEqual(len(service.containers()), 1)
- container.inspect()
- self.assertTrue(container.is_running)
- captured_output = mock_log.info.call_args[0]
- self.assertIn('Desired container number already achieved', captured_output)
- @mock.patch('compose.service.log')
- def test_scale_with_custom_container_name_outputs_warning(self, mock_log):
- """
- Test that calling scale on a service that has a custom container name
- results in warning output.
- """
- service = self.create_service('web', container_name='custom-container')
- self.assertEqual(service.custom_container_name(), 'custom-container')
- service.scale(3)
- captured_output = mock_log.warn.call_args[0][0]
- self.assertEqual(len(service.containers()), 1)
- self.assertIn(
- "Remove the custom name to scale the service.",
- captured_output
- )
- 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_custom_container_name(self):
- service = self.create_service('web', container_name='my-web-container')
- self.assertEqual(service.custom_container_name(), 'my-web-container')
- container = create_and_start_container(service)
- self.assertEqual(container.name, 'my-web-container')
- one_off_container = service.create_container(one_off=True)
- self.assertNotEqual(one_off_container.name, 'my-web-container')
- 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])
- def test_duplicate_containers(self):
- service = self.create_service('web')
- options = service._get_container_create_options({}, 1)
- original = Container.create(service.client, **options)
- self.assertEqual(set(service.containers(stopped=True)), set([original]))
- self.assertEqual(set(service.duplicate_containers()), set())
- options['name'] = 'temporary_container_name'
- duplicate = Container.create(service.client, **options)
- self.assertEqual(set(service.containers(stopped=True)), set([original, duplicate]))
- self.assertEqual(set(service.duplicate_containers()), set([duplicate]))
|