12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097 |
- from __future__ import absolute_import
- from __future__ import unicode_literals
- import random
- import py
- import pytest
- from docker.errors import NotFound
- from .. import mock
- from ..helpers import build_config
- from .testcases import DockerClientTestCase
- from compose.config import config
- from compose.config import ConfigurationError
- from compose.config.config import V2_0
- from compose.config.types import VolumeFromSpec
- from compose.config.types import VolumeSpec
- from compose.const import LABEL_PROJECT
- from compose.const import LABEL_SERVICE
- from compose.container import Container
- from compose.project import Project
- from compose.service import ConvergenceStrategy
- from tests.integration.testcases import v2_only
- class ProjectTest(DockerClientTestCase):
- def test_containers(self):
- web = self.create_service('web')
- db = self.create_service('db')
- project = Project('composetest', [web, db], self.client)
- project.up()
- containers = project.containers()
- self.assertEqual(len(containers), 2)
- def test_containers_with_service_names(self):
- web = self.create_service('web')
- db = self.create_service('db')
- project = Project('composetest', [web, db], self.client)
- project.up()
- containers = project.containers(['web'])
- self.assertEqual(
- [c.name for c in containers],
- ['composetest_web_1'])
- def test_containers_with_extra_service(self):
- web = self.create_service('web')
- web_1 = web.create_container()
- db = self.create_service('db')
- db_1 = db.create_container()
- self.create_service('extra').create_container()
- project = Project('composetest', [web, db], self.client)
- self.assertEqual(
- set(project.containers(stopped=True)),
- set([web_1, db_1]),
- )
- def test_volumes_from_service(self):
- project = Project.from_config(
- name='composetest',
- config_data=build_config({
- 'data': {
- 'image': 'busybox:latest',
- 'volumes': ['/var/data'],
- },
- 'db': {
- 'image': 'busybox:latest',
- 'volumes_from': ['data'],
- },
- }),
- client=self.client,
- )
- db = project.get_service('db')
- data = project.get_service('data')
- self.assertEqual(db.volumes_from, [VolumeFromSpec(data, 'rw', 'service')])
- def test_volumes_from_container(self):
- data_container = Container.create(
- self.client,
- image='busybox:latest',
- volumes=['/var/data'],
- name='composetest_data_container',
- labels={LABEL_PROJECT: 'composetest'},
- )
- project = Project.from_config(
- name='composetest',
- config_data=build_config({
- 'db': {
- 'image': 'busybox:latest',
- 'volumes_from': ['composetest_data_container'],
- },
- }),
- client=self.client,
- )
- db = project.get_service('db')
- self.assertEqual(db._get_volumes_from(), [data_container.id + ':rw'])
- @v2_only()
- def test_network_mode_from_service(self):
- project = Project.from_config(
- name='composetest',
- client=self.client,
- config_data=build_config({
- 'version': V2_0,
- 'services': {
- 'net': {
- 'image': 'busybox:latest',
- 'command': ["top"]
- },
- 'web': {
- 'image': 'busybox:latest',
- 'network_mode': 'service:net',
- 'command': ["top"]
- },
- },
- }),
- )
- project.up()
- web = project.get_service('web')
- net = project.get_service('net')
- self.assertEqual(web.network_mode.mode, 'container:' + net.containers()[0].id)
- @v2_only()
- def test_network_mode_from_container(self):
- def get_project():
- return Project.from_config(
- name='composetest',
- config_data=build_config({
- 'version': V2_0,
- 'services': {
- 'web': {
- 'image': 'busybox:latest',
- 'network_mode': 'container:composetest_net_container'
- },
- },
- }),
- client=self.client,
- )
- with pytest.raises(ConfigurationError) as excinfo:
- get_project()
- assert "container 'composetest_net_container' which does not exist" in excinfo.exconly()
- net_container = Container.create(
- self.client,
- image='busybox:latest',
- name='composetest_net_container',
- command='top',
- labels={LABEL_PROJECT: 'composetest'},
- )
- net_container.start()
- project = get_project()
- project.up()
- web = project.get_service('web')
- self.assertEqual(web.network_mode.mode, 'container:' + net_container.id)
- def test_net_from_service_v1(self):
- project = Project.from_config(
- name='composetest',
- config_data=build_config({
- 'net': {
- 'image': 'busybox:latest',
- 'command': ["top"]
- },
- 'web': {
- 'image': 'busybox:latest',
- 'net': 'container:net',
- 'command': ["top"]
- },
- }),
- client=self.client,
- )
- project.up()
- web = project.get_service('web')
- net = project.get_service('net')
- self.assertEqual(web.network_mode.mode, 'container:' + net.containers()[0].id)
- def test_net_from_container_v1(self):
- def get_project():
- return Project.from_config(
- name='composetest',
- config_data=build_config({
- 'web': {
- 'image': 'busybox:latest',
- 'net': 'container:composetest_net_container'
- },
- }),
- client=self.client,
- )
- with pytest.raises(ConfigurationError) as excinfo:
- get_project()
- assert "container 'composetest_net_container' which does not exist" in excinfo.exconly()
- net_container = Container.create(
- self.client,
- image='busybox:latest',
- name='composetest_net_container',
- command='top',
- labels={LABEL_PROJECT: 'composetest'},
- )
- net_container.start()
- project = get_project()
- project.up()
- web = project.get_service('web')
- self.assertEqual(web.network_mode.mode, 'container:' + net_container.id)
- def test_start_pause_unpause_stop_kill_remove(self):
- web = self.create_service('web')
- db = self.create_service('db')
- project = Project('composetest', [web, db], self.client)
- project.start()
- self.assertEqual(len(web.containers()), 0)
- self.assertEqual(len(db.containers()), 0)
- web_container_1 = web.create_container()
- web_container_2 = web.create_container()
- db_container = db.create_container()
- project.start(service_names=['web'])
- self.assertEqual(
- set(c.name for c in project.containers()),
- set([web_container_1.name, web_container_2.name]))
- project.start()
- self.assertEqual(
- set(c.name for c in project.containers()),
- set([web_container_1.name, web_container_2.name, db_container.name]))
- project.pause(service_names=['web'])
- self.assertEqual(
- set([c.name for c in project.containers() if c.is_paused]),
- set([web_container_1.name, web_container_2.name]))
- project.pause()
- self.assertEqual(
- set([c.name for c in project.containers() if c.is_paused]),
- set([web_container_1.name, web_container_2.name, db_container.name]))
- project.unpause(service_names=['db'])
- self.assertEqual(len([c.name for c in project.containers() if c.is_paused]), 2)
- project.unpause()
- self.assertEqual(len([c.name for c in project.containers() if c.is_paused]), 0)
- project.stop(service_names=['web'], timeout=1)
- self.assertEqual(set(c.name for c in project.containers()), set([db_container.name]))
- project.kill(service_names=['db'])
- self.assertEqual(len(project.containers()), 0)
- self.assertEqual(len(project.containers(stopped=True)), 3)
- project.remove_stopped(service_names=['web'])
- self.assertEqual(len(project.containers(stopped=True)), 1)
- project.remove_stopped()
- self.assertEqual(len(project.containers(stopped=True)), 0)
- def test_create(self):
- web = self.create_service('web')
- db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
- project = Project('composetest', [web, db], self.client)
- project.create(['db'])
- self.assertEqual(len(project.containers()), 0)
- self.assertEqual(len(project.containers(stopped=True)), 1)
- self.assertEqual(len(db.containers()), 0)
- self.assertEqual(len(db.containers(stopped=True)), 1)
- self.assertEqual(len(web.containers(stopped=True)), 0)
- def test_create_twice(self):
- web = self.create_service('web')
- db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
- project = Project('composetest', [web, db], self.client)
- project.create(['db', 'web'])
- project.create(['db', 'web'])
- self.assertEqual(len(project.containers()), 0)
- self.assertEqual(len(project.containers(stopped=True)), 2)
- self.assertEqual(len(db.containers()), 0)
- self.assertEqual(len(db.containers(stopped=True)), 1)
- self.assertEqual(len(web.containers()), 0)
- self.assertEqual(len(web.containers(stopped=True)), 1)
- def test_create_with_links(self):
- db = self.create_service('db')
- web = self.create_service('web', links=[(db, 'db')])
- project = Project('composetest', [db, web], self.client)
- project.create(['web'])
- self.assertEqual(len(project.containers()), 0)
- self.assertEqual(len(project.containers(stopped=True)), 2)
- self.assertEqual(len(db.containers()), 0)
- self.assertEqual(len(db.containers(stopped=True)), 1)
- self.assertEqual(len(web.containers()), 0)
- self.assertEqual(len(web.containers(stopped=True)), 1)
- def test_create_strategy_always(self):
- db = self.create_service('db')
- project = Project('composetest', [db], self.client)
- project.create(['db'])
- old_id = project.containers(stopped=True)[0].id
- project.create(['db'], strategy=ConvergenceStrategy.always)
- self.assertEqual(len(project.containers()), 0)
- self.assertEqual(len(project.containers(stopped=True)), 1)
- db_container = project.containers(stopped=True)[0]
- self.assertNotEqual(db_container.id, old_id)
- def test_create_strategy_never(self):
- db = self.create_service('db')
- project = Project('composetest', [db], self.client)
- project.create(['db'])
- old_id = project.containers(stopped=True)[0].id
- project.create(['db'], strategy=ConvergenceStrategy.never)
- self.assertEqual(len(project.containers()), 0)
- self.assertEqual(len(project.containers(stopped=True)), 1)
- db_container = project.containers(stopped=True)[0]
- self.assertEqual(db_container.id, old_id)
- def test_project_up(self):
- web = self.create_service('web')
- db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
- project = Project('composetest', [web, db], self.client)
- project.start()
- self.assertEqual(len(project.containers()), 0)
- project.up(['db'])
- self.assertEqual(len(project.containers()), 1)
- self.assertEqual(len(db.containers()), 1)
- self.assertEqual(len(web.containers()), 0)
- def test_project_up_starts_uncreated_services(self):
- db = self.create_service('db')
- web = self.create_service('web', links=[(db, 'db')])
- project = Project('composetest', [db, web], self.client)
- project.up(['db'])
- self.assertEqual(len(project.containers()), 1)
- project.up()
- self.assertEqual(len(project.containers()), 2)
- self.assertEqual(len(db.containers()), 1)
- self.assertEqual(len(web.containers()), 1)
- def test_recreate_preserves_volumes(self):
- web = self.create_service('web')
- db = self.create_service('db', volumes=[VolumeSpec.parse('/etc')])
- project = Project('composetest', [web, db], self.client)
- project.start()
- self.assertEqual(len(project.containers()), 0)
- project.up(['db'])
- self.assertEqual(len(project.containers()), 1)
- old_db_id = project.containers()[0].id
- db_volume_path = project.containers()[0].get('Volumes./etc')
- project.up(strategy=ConvergenceStrategy.always)
- self.assertEqual(len(project.containers()), 2)
- db_container = [c for c in project.containers() if 'db' in c.name][0]
- self.assertNotEqual(db_container.id, old_db_id)
- self.assertEqual(db_container.get('Volumes./etc'), db_volume_path)
- def test_project_up_with_no_recreate_running(self):
- web = self.create_service('web')
- db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
- project = Project('composetest', [web, db], self.client)
- project.start()
- self.assertEqual(len(project.containers()), 0)
- project.up(['db'])
- self.assertEqual(len(project.containers()), 1)
- old_db_id = project.containers()[0].id
- container, = project.containers()
- db_volume_path = container.get_mount('/var/db')['Source']
- project.up(strategy=ConvergenceStrategy.never)
- self.assertEqual(len(project.containers()), 2)
- db_container = [c for c in project.containers() if 'db' in c.name][0]
- self.assertEqual(db_container.id, old_db_id)
- self.assertEqual(
- db_container.get_mount('/var/db')['Source'],
- db_volume_path)
- def test_project_up_with_no_recreate_stopped(self):
- web = self.create_service('web')
- db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
- project = Project('composetest', [web, db], self.client)
- project.start()
- self.assertEqual(len(project.containers()), 0)
- project.up(['db'])
- project.kill()
- old_containers = project.containers(stopped=True)
- self.assertEqual(len(old_containers), 1)
- old_container, = old_containers
- old_db_id = old_container.id
- db_volume_path = old_container.get_mount('/var/db')['Source']
- project.up(strategy=ConvergenceStrategy.never)
- new_containers = project.containers(stopped=True)
- self.assertEqual(len(new_containers), 2)
- self.assertEqual([c.is_running for c in new_containers], [True, True])
- db_container = [c for c in new_containers if 'db' in c.name][0]
- self.assertEqual(db_container.id, old_db_id)
- self.assertEqual(
- db_container.get_mount('/var/db')['Source'],
- db_volume_path)
- def test_project_up_without_all_services(self):
- console = self.create_service('console')
- db = self.create_service('db')
- project = Project('composetest', [console, db], self.client)
- project.start()
- self.assertEqual(len(project.containers()), 0)
- project.up()
- self.assertEqual(len(project.containers()), 2)
- self.assertEqual(len(db.containers()), 1)
- self.assertEqual(len(console.containers()), 1)
- def test_project_up_starts_links(self):
- console = self.create_service('console')
- db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
- web = self.create_service('web', links=[(db, 'db')])
- project = Project('composetest', [web, db, console], self.client)
- project.start()
- self.assertEqual(len(project.containers()), 0)
- project.up(['web'])
- self.assertEqual(len(project.containers()), 2)
- self.assertEqual(len(web.containers()), 1)
- self.assertEqual(len(db.containers()), 1)
- self.assertEqual(len(console.containers()), 0)
- def test_project_up_starts_depends(self):
- project = Project.from_config(
- name='composetest',
- config_data=build_config({
- 'console': {
- 'image': 'busybox:latest',
- 'command': ["top"],
- },
- 'data': {
- 'image': 'busybox:latest',
- 'command': ["top"]
- },
- 'db': {
- 'image': 'busybox:latest',
- 'command': ["top"],
- 'volumes_from': ['data'],
- },
- 'web': {
- 'image': 'busybox:latest',
- 'command': ["top"],
- 'links': ['db'],
- },
- }),
- client=self.client,
- )
- project.start()
- self.assertEqual(len(project.containers()), 0)
- project.up(['web'])
- self.assertEqual(len(project.containers()), 3)
- self.assertEqual(len(project.get_service('web').containers()), 1)
- self.assertEqual(len(project.get_service('db').containers()), 1)
- self.assertEqual(len(project.get_service('data').containers()), 1)
- self.assertEqual(len(project.get_service('console').containers()), 0)
- def test_project_up_with_no_deps(self):
- project = Project.from_config(
- name='composetest',
- config_data=build_config({
- 'console': {
- 'image': 'busybox:latest',
- 'command': ["top"],
- },
- 'data': {
- 'image': 'busybox:latest',
- 'command': ["top"]
- },
- 'db': {
- 'image': 'busybox:latest',
- 'command': ["top"],
- 'volumes_from': ['data'],
- },
- 'web': {
- 'image': 'busybox:latest',
- 'command': ["top"],
- 'links': ['db'],
- },
- }),
- client=self.client,
- )
- project.start()
- self.assertEqual(len(project.containers()), 0)
- project.up(['db'], start_deps=False)
- self.assertEqual(len(project.containers(stopped=True)), 2)
- self.assertEqual(len(project.get_service('web').containers()), 0)
- self.assertEqual(len(project.get_service('db').containers()), 1)
- self.assertEqual(len(project.get_service('data').containers()), 0)
- self.assertEqual(len(project.get_service('data').containers(stopped=True)), 1)
- self.assertEqual(len(project.get_service('console').containers()), 0)
- def test_unscale_after_restart(self):
- web = self.create_service('web')
- project = Project('composetest', [web], self.client)
- project.start()
- service = project.get_service('web')
- service.scale(1)
- self.assertEqual(len(service.containers()), 1)
- service.scale(3)
- self.assertEqual(len(service.containers()), 3)
- project.up()
- service = project.get_service('web')
- self.assertEqual(len(service.containers()), 3)
- service.scale(1)
- self.assertEqual(len(service.containers()), 1)
- project.up()
- service = project.get_service('web')
- self.assertEqual(len(service.containers()), 1)
- # does scale=0 ,makes any sense? after recreating at least 1 container is running
- service.scale(0)
- project.up()
- service = project.get_service('web')
- self.assertEqual(len(service.containers()), 1)
- @v2_only()
- def test_project_up_networks(self):
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top',
- 'networks': {'foo': None, 'bar': None, 'baz': None},
- }],
- volumes={},
- networks={
- 'foo': {'driver': 'bridge'},
- 'bar': {'driver': None},
- 'baz': {},
- },
- )
- project = Project.from_config(
- client=self.client,
- name='composetest',
- config_data=config_data,
- )
- project.up()
- self.assertEqual(len(project.containers()), 1)
- for net_name in ['foo', 'bar', 'baz']:
- full_net_name = 'composetest_{}'.format(net_name)
- network_data = self.client.inspect_network(full_net_name)
- self.assertEqual(network_data['Name'], full_net_name)
- foo_data = self.client.inspect_network('composetest_foo')
- self.assertEqual(foo_data['Driver'], 'bridge')
- @v2_only()
- def test_up_with_ipam_config(self):
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'networks': {'front': None},
- }],
- volumes={},
- networks={
- 'front': {
- 'driver': 'bridge',
- 'driver_opts': {
- "com.docker.network.bridge.enable_icc": "false",
- },
- 'ipam': {
- 'driver': 'default',
- 'config': [{
- "subnet": "172.28.0.0/16",
- "ip_range": "172.28.5.0/24",
- "gateway": "172.28.5.254",
- "aux_addresses": {
- "a": "172.28.1.5",
- "b": "172.28.1.6",
- "c": "172.28.1.7",
- },
- }],
- },
- },
- },
- )
- project = Project.from_config(
- client=self.client,
- name='composetest',
- config_data=config_data,
- )
- project.up()
- network = self.client.networks(names=['composetest_front'])[0]
- assert network['Options'] == {
- "com.docker.network.bridge.enable_icc": "false"
- }
- assert network['IPAM'] == {
- 'Driver': 'default',
- 'Options': None,
- 'Config': [{
- 'Subnet': "172.28.0.0/16",
- 'IPRange': "172.28.5.0/24",
- 'Gateway': "172.28.5.254",
- 'AuxiliaryAddresses': {
- 'a': '172.28.1.5',
- 'b': '172.28.1.6',
- 'c': '172.28.1.7',
- },
- }],
- }
- @v2_only()
- def test_up_with_network_static_addresses(self):
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top',
- 'networks': {
- 'static_test': {
- 'ipv4_address': '172.16.100.100',
- 'ipv6_address': 'fe80::1001:102'
- }
- },
- }],
- volumes={},
- networks={
- 'static_test': {
- 'driver': 'bridge',
- 'driver_opts': {
- "com.docker.network.enable_ipv6": "true",
- },
- 'ipam': {
- 'driver': 'default',
- 'config': [
- {"subnet": "172.16.100.0/24",
- "gateway": "172.16.100.1"},
- {"subnet": "fe80::/64",
- "gateway": "fe80::1001:1"}
- ]
- }
- }
- }
- )
- project = Project.from_config(
- client=self.client,
- name='composetest',
- config_data=config_data,
- )
- project.up(detached=True)
- network = self.client.networks(names=['static_test'])[0]
- service_container = project.get_service('web').containers()[0]
- assert network['Options'] == {
- "com.docker.network.enable_ipv6": "true"
- }
- IPAMConfig = (service_container.inspect().get('NetworkSettings', {}).
- get('Networks', {}).get('composetest_static_test', {}).
- get('IPAMConfig', {}))
- assert IPAMConfig.get('IPv4Address') == '172.16.100.100'
- assert IPAMConfig.get('IPv6Address') == 'fe80::1001:102'
- @v2_only()
- def test_up_with_network_static_addresses_missing_subnet(self):
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'networks': {
- 'static_test': {
- 'ipv4_address': '172.16.100.100',
- 'ipv6_address': 'fe80::1001:101'
- }
- },
- }],
- volumes={},
- networks={
- 'static_test': {
- 'driver': 'bridge',
- 'driver_opts': {
- "com.docker.network.enable_ipv6": "true",
- },
- 'ipam': {
- 'driver': 'default',
- },
- },
- },
- )
- project = Project.from_config(
- client=self.client,
- name='composetest',
- config_data=config_data,
- )
- assert len(project.up()) == 0
- @v2_only()
- def test_project_up_volumes(self):
- vol_name = '{0:x}'.format(random.getrandbits(32))
- full_vol_name = 'composetest_{0}'.format(vol_name)
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top'
- }],
- volumes={vol_name: {'driver': 'local'}},
- networks={},
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data, client=self.client
- )
- project.up()
- self.assertEqual(len(project.containers()), 1)
- volume_data = self.client.inspect_volume(full_vol_name)
- self.assertEqual(volume_data['Name'], full_vol_name)
- self.assertEqual(volume_data['Driver'], 'local')
- @v2_only()
- def test_project_up_logging_with_multiple_files(self):
- base_file = config.ConfigFile(
- 'base.yml',
- {
- 'version': V2_0,
- 'services': {
- 'simple': {'image': 'busybox:latest', 'command': 'top'},
- 'another': {
- 'image': 'busybox:latest',
- 'command': 'top',
- 'logging': {
- 'driver': "json-file",
- 'options': {
- 'max-size': "10m"
- }
- }
- }
- }
- })
- override_file = config.ConfigFile(
- 'override.yml',
- {
- 'version': V2_0,
- 'services': {
- 'another': {
- 'logging': {
- 'driver': "none"
- }
- }
- }
- })
- details = config.ConfigDetails('.', [base_file, override_file])
- tmpdir = py.test.ensuretemp('logging_test')
- self.addCleanup(tmpdir.remove)
- with tmpdir.as_cwd():
- config_data = config.load(details)
- project = Project.from_config(
- name='composetest', config_data=config_data, client=self.client
- )
- project.up()
- containers = project.containers()
- self.assertEqual(len(containers), 2)
- another = project.get_service('another').containers()[0]
- log_config = another.get('HostConfig.LogConfig')
- self.assertTrue(log_config)
- self.assertEqual(log_config.get('Type'), 'none')
- @v2_only()
- def test_initialize_volumes(self):
- vol_name = '{0:x}'.format(random.getrandbits(32))
- full_vol_name = 'composetest_{0}'.format(vol_name)
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top'
- }],
- volumes={vol_name: {}},
- networks={},
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data, client=self.client
- )
- project.volumes.initialize()
- volume_data = self.client.inspect_volume(full_vol_name)
- self.assertEqual(volume_data['Name'], full_vol_name)
- self.assertEqual(volume_data['Driver'], 'local')
- @v2_only()
- def test_project_up_implicit_volume_driver(self):
- vol_name = '{0:x}'.format(random.getrandbits(32))
- full_vol_name = 'composetest_{0}'.format(vol_name)
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top'
- }],
- volumes={vol_name: {}},
- networks={},
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data, client=self.client
- )
- project.up()
- volume_data = self.client.inspect_volume(full_vol_name)
- self.assertEqual(volume_data['Name'], full_vol_name)
- self.assertEqual(volume_data['Driver'], 'local')
- @v2_only()
- def test_initialize_volumes_invalid_volume_driver(self):
- vol_name = '{0:x}'.format(random.getrandbits(32))
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top'
- }],
- volumes={vol_name: {'driver': 'foobar'}},
- networks={},
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data, client=self.client
- )
- with self.assertRaises(config.ConfigurationError):
- project.volumes.initialize()
- @v2_only()
- def test_initialize_volumes_updated_driver(self):
- vol_name = '{0:x}'.format(random.getrandbits(32))
- full_vol_name = 'composetest_{0}'.format(vol_name)
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top'
- }],
- volumes={vol_name: {'driver': 'local'}},
- networks={},
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data, client=self.client
- )
- project.volumes.initialize()
- volume_data = self.client.inspect_volume(full_vol_name)
- self.assertEqual(volume_data['Name'], full_vol_name)
- self.assertEqual(volume_data['Driver'], 'local')
- config_data = config_data._replace(
- volumes={vol_name: {'driver': 'smb'}}
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data,
- client=self.client
- )
- with self.assertRaises(config.ConfigurationError) as e:
- project.volumes.initialize()
- assert 'Configuration for volume {0} specifies driver smb'.format(
- vol_name
- ) in str(e.exception)
- @v2_only()
- def test_initialize_volumes_updated_blank_driver(self):
- vol_name = '{0:x}'.format(random.getrandbits(32))
- full_vol_name = 'composetest_{0}'.format(vol_name)
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top'
- }],
- volumes={vol_name: {'driver': 'local'}},
- networks={},
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data, client=self.client
- )
- project.volumes.initialize()
- volume_data = self.client.inspect_volume(full_vol_name)
- self.assertEqual(volume_data['Name'], full_vol_name)
- self.assertEqual(volume_data['Driver'], 'local')
- config_data = config_data._replace(
- volumes={vol_name: {}}
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data,
- client=self.client
- )
- project.volumes.initialize()
- volume_data = self.client.inspect_volume(full_vol_name)
- self.assertEqual(volume_data['Name'], full_vol_name)
- self.assertEqual(volume_data['Driver'], 'local')
- @v2_only()
- def test_initialize_volumes_external_volumes(self):
- # Use composetest_ prefix so it gets garbage-collected in tearDown()
- vol_name = 'composetest_{0:x}'.format(random.getrandbits(32))
- full_vol_name = 'composetest_{0}'.format(vol_name)
- self.client.create_volume(vol_name)
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top'
- }],
- volumes={
- vol_name: {'external': True, 'external_name': vol_name}
- },
- networks=None,
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data, client=self.client
- )
- project.volumes.initialize()
- with self.assertRaises(NotFound):
- self.client.inspect_volume(full_vol_name)
- @v2_only()
- def test_initialize_volumes_inexistent_external_volume(self):
- vol_name = '{0:x}'.format(random.getrandbits(32))
- config_data = config.Config(
- version=V2_0,
- services=[{
- 'name': 'web',
- 'image': 'busybox:latest',
- 'command': 'top'
- }],
- volumes={
- vol_name: {'external': True, 'external_name': vol_name}
- },
- networks=None,
- )
- project = Project.from_config(
- name='composetest',
- config_data=config_data, client=self.client
- )
- with self.assertRaises(config.ConfigurationError) as e:
- project.volumes.initialize()
- assert 'Volume {0} declared as external'.format(
- vol_name
- ) in str(e.exception)
- @v2_only()
- def test_project_up_named_volumes_in_binds(self):
- vol_name = '{0:x}'.format(random.getrandbits(32))
- full_vol_name = 'composetest_{0}'.format(vol_name)
- base_file = config.ConfigFile(
- 'base.yml',
- {
- 'version': V2_0,
- 'services': {
- 'simple': {
- 'image': 'busybox:latest',
- 'command': 'top',
- 'volumes': ['{0}:/data'.format(vol_name)]
- },
- },
- 'volumes': {
- vol_name: {'driver': 'local'}
- }
- })
- config_details = config.ConfigDetails('.', [base_file])
- config_data = config.load(config_details)
- project = Project.from_config(
- name='composetest', config_data=config_data, client=self.client
- )
- service = project.services[0]
- self.assertEqual(service.name, 'simple')
- volumes = service.options.get('volumes')
- self.assertEqual(len(volumes), 1)
- self.assertEqual(volumes[0].external, full_vol_name)
- project.up()
- engine_volumes = self.client.volumes()['Volumes']
- container = service.get_container()
- assert [mount['Name'] for mount in container.get('Mounts')] == [full_vol_name]
- assert next((v for v in engine_volumes if v['Name'] == vol_name), None) is None
- def test_project_up_orphans(self):
- config_dict = {
- 'service1': {
- 'image': 'busybox:latest',
- 'command': 'top',
- }
- }
- config_data = build_config(config_dict)
- project = Project.from_config(
- name='composetest', config_data=config_data, client=self.client
- )
- project.up()
- config_dict['service2'] = config_dict['service1']
- del config_dict['service1']
- config_data = build_config(config_dict)
- project = Project.from_config(
- name='composetest', config_data=config_data, client=self.client
- )
- with mock.patch('compose.project.log') as mock_log:
- project.up()
- mock_log.warning.assert_called_once_with(mock.ANY)
- assert len([
- ctnr for ctnr in project._labeled_containers()
- if ctnr.labels.get(LABEL_SERVICE) == 'service1'
- ]) == 1
- project.up(remove_orphans=True)
- assert len([
- ctnr for ctnr in project._labeled_containers()
- if ctnr.labels.get(LABEL_SERVICE) == 'service1'
- ]) == 0
|