1
0

cli_test.py 87 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260
  1. # -*- coding: utf-8 -*-
  2. from __future__ import absolute_import
  3. from __future__ import unicode_literals
  4. import datetime
  5. import json
  6. import os
  7. import os.path
  8. import signal
  9. import subprocess
  10. import time
  11. from collections import Counter
  12. from collections import namedtuple
  13. from operator import attrgetter
  14. import py
  15. import six
  16. import yaml
  17. from docker import errors
  18. from .. import mock
  19. from ..helpers import create_host_file
  20. from ..helpers import is_cluster
  21. from ..helpers import no_cluster
  22. from compose.cli.command import get_project
  23. from compose.config.errors import DuplicateOverrideFileFound
  24. from compose.container import Container
  25. from compose.project import OneOffFilter
  26. from compose.utils import nanoseconds_from_time_seconds
  27. from tests.integration.testcases import DockerClientTestCase
  28. from tests.integration.testcases import get_links
  29. from tests.integration.testcases import pull_busybox
  30. from tests.integration.testcases import SWARM_SKIP_RM_VOLUMES
  31. from tests.integration.testcases import v2_1_only
  32. from tests.integration.testcases import v2_only
  33. from tests.integration.testcases import v3_only
  34. ProcessResult = namedtuple('ProcessResult', 'stdout stderr')
  35. BUILD_CACHE_TEXT = 'Using cache'
  36. BUILD_PULL_TEXT = 'Status: Image is up to date for busybox:latest'
  37. def start_process(base_dir, options):
  38. proc = subprocess.Popen(
  39. ['docker-compose'] + options,
  40. stdout=subprocess.PIPE,
  41. stderr=subprocess.PIPE,
  42. cwd=base_dir)
  43. print("Running process: %s" % proc.pid)
  44. return proc
  45. def wait_on_process(proc, returncode=0):
  46. stdout, stderr = proc.communicate()
  47. if proc.returncode != returncode:
  48. print("Stderr: {}".format(stderr))
  49. print("Stdout: {}".format(stdout))
  50. assert proc.returncode == returncode
  51. return ProcessResult(stdout.decode('utf-8'), stderr.decode('utf-8'))
  52. def wait_on_condition(condition, delay=0.1, timeout=40):
  53. start_time = time.time()
  54. while not condition():
  55. if time.time() - start_time > timeout:
  56. raise AssertionError("Timeout: %s" % condition)
  57. time.sleep(delay)
  58. def kill_service(service):
  59. for container in service.containers():
  60. if container.is_running:
  61. container.kill()
  62. class ContainerCountCondition(object):
  63. def __init__(self, project, expected):
  64. self.project = project
  65. self.expected = expected
  66. def __call__(self):
  67. return len([c for c in self.project.containers() if c.is_running]) == self.expected
  68. def __str__(self):
  69. return "waiting for counter count == %s" % self.expected
  70. class ContainerStateCondition(object):
  71. def __init__(self, client, name, status):
  72. self.client = client
  73. self.name = name
  74. self.status = status
  75. def __call__(self):
  76. try:
  77. container = self.client.inspect_container(self.name)
  78. return container['State']['Status'] == self.status
  79. except errors.APIError:
  80. return False
  81. def __str__(self):
  82. return "waiting for container to be %s" % self.status
  83. class CLITestCase(DockerClientTestCase):
  84. def setUp(self):
  85. super(CLITestCase, self).setUp()
  86. self.base_dir = 'tests/fixtures/simple-composefile'
  87. self.override_dir = None
  88. def tearDown(self):
  89. if self.base_dir:
  90. self.project.kill()
  91. self.project.remove_stopped()
  92. for container in self.project.containers(stopped=True, one_off=OneOffFilter.only):
  93. container.remove(force=True)
  94. networks = self.client.networks()
  95. for n in networks:
  96. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name)):
  97. self.client.remove_network(n['Name'])
  98. volumes = self.client.volumes().get('Volumes') or []
  99. for v in volumes:
  100. if v['Name'].split('/')[-1].startswith('{}_'.format(self.project.name)):
  101. self.client.remove_volume(v['Name'])
  102. if hasattr(self, '_project'):
  103. del self._project
  104. super(CLITestCase, self).tearDown()
  105. @property
  106. def project(self):
  107. # Hack: allow project to be overridden
  108. if not hasattr(self, '_project'):
  109. self._project = get_project(self.base_dir, override_dir=self.override_dir)
  110. return self._project
  111. def dispatch(self, options, project_options=None, returncode=0):
  112. project_options = project_options or []
  113. proc = start_process(self.base_dir, project_options + options)
  114. return wait_on_process(proc, returncode=returncode)
  115. def execute(self, container, cmd):
  116. # Remove once Hijack and CloseNotifier sign a peace treaty
  117. self.client.close()
  118. exc = self.client.exec_create(container.id, cmd)
  119. self.client.exec_start(exc)
  120. return self.client.exec_inspect(exc)['ExitCode']
  121. def lookup(self, container, hostname):
  122. return self.execute(container, ["nslookup", hostname]) == 0
  123. def test_help(self):
  124. self.base_dir = 'tests/fixtures/no-composefile'
  125. result = self.dispatch(['help', 'up'], returncode=0)
  126. assert 'Usage: up [options] [--scale SERVICE=NUM...] [SERVICE...]' in result.stdout
  127. # Prevent tearDown from trying to create a project
  128. self.base_dir = None
  129. def test_help_nonexistent(self):
  130. self.base_dir = 'tests/fixtures/no-composefile'
  131. result = self.dispatch(['help', 'foobar'], returncode=1)
  132. assert 'No such command' in result.stderr
  133. self.base_dir = None
  134. def test_shorthand_host_opt(self):
  135. self.dispatch(
  136. ['-H={0}'.format(os.environ.get('DOCKER_HOST', 'unix://')),
  137. 'up', '-d'],
  138. returncode=0
  139. )
  140. def test_host_not_reachable(self):
  141. result = self.dispatch(['-H=tcp://doesnotexist:8000', 'ps'], returncode=1)
  142. assert "Couldn't connect to Docker daemon" in result.stderr
  143. def test_host_not_reachable_volumes_from_container(self):
  144. self.base_dir = 'tests/fixtures/volumes-from-container'
  145. container = self.client.create_container(
  146. 'busybox', 'true', name='composetest_data_container',
  147. host_config={}
  148. )
  149. self.addCleanup(self.client.remove_container, container)
  150. result = self.dispatch(['-H=tcp://doesnotexist:8000', 'ps'], returncode=1)
  151. assert "Couldn't connect to Docker daemon" in result.stderr
  152. def test_config_list_services(self):
  153. self.base_dir = 'tests/fixtures/v2-full'
  154. result = self.dispatch(['config', '--services'])
  155. assert set(result.stdout.rstrip().split('\n')) == {'web', 'other'}
  156. def test_config_list_volumes(self):
  157. self.base_dir = 'tests/fixtures/v2-full'
  158. result = self.dispatch(['config', '--volumes'])
  159. assert set(result.stdout.rstrip().split('\n')) == {'data'}
  160. def test_config_quiet_with_error(self):
  161. self.base_dir = None
  162. result = self.dispatch([
  163. '-f', 'tests/fixtures/invalid-composefile/invalid.yml',
  164. 'config', '-q'
  165. ], returncode=1)
  166. assert "'notaservice' must be a mapping" in result.stderr
  167. def test_config_quiet(self):
  168. self.base_dir = 'tests/fixtures/v2-full'
  169. assert self.dispatch(['config', '-q']).stdout == ''
  170. def test_config_default(self):
  171. self.base_dir = 'tests/fixtures/v2-full'
  172. result = self.dispatch(['config'])
  173. # assert there are no python objects encoded in the output
  174. assert '!!' not in result.stdout
  175. output = yaml.load(result.stdout)
  176. expected = {
  177. 'version': '2.0',
  178. 'volumes': {'data': {'driver': 'local'}},
  179. 'networks': {'front': {}},
  180. 'services': {
  181. 'web': {
  182. 'build': {
  183. 'context': os.path.abspath(self.base_dir),
  184. },
  185. 'networks': {'front': None, 'default': None},
  186. 'volumes_from': ['service:other:rw'],
  187. },
  188. 'other': {
  189. 'image': 'busybox:latest',
  190. 'command': 'top',
  191. 'volumes': ['/data'],
  192. },
  193. },
  194. }
  195. assert output == expected
  196. def test_config_restart(self):
  197. self.base_dir = 'tests/fixtures/restart'
  198. result = self.dispatch(['config'])
  199. assert yaml.load(result.stdout) == {
  200. 'version': '2.0',
  201. 'services': {
  202. 'never': {
  203. 'image': 'busybox',
  204. 'restart': 'no',
  205. },
  206. 'always': {
  207. 'image': 'busybox',
  208. 'restart': 'always',
  209. },
  210. 'on-failure': {
  211. 'image': 'busybox',
  212. 'restart': 'on-failure',
  213. },
  214. 'on-failure-5': {
  215. 'image': 'busybox',
  216. 'restart': 'on-failure:5',
  217. },
  218. 'restart-null': {
  219. 'image': 'busybox',
  220. 'restart': ''
  221. },
  222. },
  223. }
  224. def test_config_external_network(self):
  225. self.base_dir = 'tests/fixtures/networks'
  226. result = self.dispatch(['-f', 'external-networks.yml', 'config'])
  227. json_result = yaml.load(result.stdout)
  228. assert 'networks' in json_result
  229. assert json_result['networks'] == {
  230. 'networks_foo': {
  231. 'external': True # {'name': 'networks_foo'}
  232. },
  233. 'bar': {
  234. 'external': {'name': 'networks_bar'}
  235. }
  236. }
  237. def test_config_external_volume(self):
  238. self.base_dir = 'tests/fixtures/volumes'
  239. result = self.dispatch(['-f', 'external-volumes.yml', 'config'])
  240. json_result = yaml.load(result.stdout)
  241. assert 'volumes' in json_result
  242. assert json_result['volumes'] == {
  243. 'foo': {
  244. 'external': True
  245. },
  246. 'bar': {
  247. 'external': {'name': 'some_bar'}
  248. }
  249. }
  250. def test_config_v1(self):
  251. self.base_dir = 'tests/fixtures/v1-config'
  252. result = self.dispatch(['config'])
  253. assert yaml.load(result.stdout) == {
  254. 'version': '2.1',
  255. 'services': {
  256. 'net': {
  257. 'image': 'busybox',
  258. 'network_mode': 'bridge',
  259. },
  260. 'volume': {
  261. 'image': 'busybox',
  262. 'volumes': ['/data'],
  263. 'network_mode': 'bridge',
  264. },
  265. 'app': {
  266. 'image': 'busybox',
  267. 'volumes_from': ['service:volume:rw'],
  268. 'network_mode': 'service:net',
  269. },
  270. },
  271. }
  272. @v3_only()
  273. def test_config_v3(self):
  274. self.base_dir = 'tests/fixtures/v3-full'
  275. result = self.dispatch(['config'])
  276. assert yaml.load(result.stdout) == {
  277. 'version': '3.2',
  278. 'volumes': {
  279. 'foobar': {
  280. 'labels': {
  281. 'com.docker.compose.test': 'true',
  282. },
  283. },
  284. },
  285. 'services': {
  286. 'web': {
  287. 'image': 'busybox',
  288. 'deploy': {
  289. 'mode': 'replicated',
  290. 'replicas': 6,
  291. 'labels': ['FOO=BAR'],
  292. 'update_config': {
  293. 'parallelism': 3,
  294. 'delay': '10s',
  295. 'failure_action': 'continue',
  296. 'monitor': '60s',
  297. 'max_failure_ratio': 0.3,
  298. },
  299. 'resources': {
  300. 'limits': {
  301. 'cpus': '0.001',
  302. 'memory': '50M',
  303. },
  304. 'reservations': {
  305. 'cpus': '0.0001',
  306. 'memory': '20M',
  307. },
  308. },
  309. 'restart_policy': {
  310. 'condition': 'on_failure',
  311. 'delay': '5s',
  312. 'max_attempts': 3,
  313. 'window': '120s',
  314. },
  315. 'placement': {
  316. 'constraints': ['node=foo'],
  317. },
  318. },
  319. 'healthcheck': {
  320. 'test': 'cat /etc/passwd',
  321. 'interval': '10s',
  322. 'timeout': '1s',
  323. 'retries': 5,
  324. },
  325. 'volumes': [
  326. '/host/path:/container/path:ro',
  327. 'foobar:/container/volumepath:rw',
  328. '/anonymous',
  329. 'foobar:/container/volumepath2:nocopy'
  330. ],
  331. 'stop_grace_period': '20s',
  332. },
  333. },
  334. }
  335. def test_ps(self):
  336. self.project.get_service('simple').create_container()
  337. result = self.dispatch(['ps'])
  338. assert 'simplecomposefile_simple_1' in result.stdout
  339. def test_ps_default_composefile(self):
  340. self.base_dir = 'tests/fixtures/multiple-composefiles'
  341. self.dispatch(['up', '-d'])
  342. result = self.dispatch(['ps'])
  343. self.assertIn('multiplecomposefiles_simple_1', result.stdout)
  344. self.assertIn('multiplecomposefiles_another_1', result.stdout)
  345. self.assertNotIn('multiplecomposefiles_yetanother_1', result.stdout)
  346. def test_ps_alternate_composefile(self):
  347. config_path = os.path.abspath(
  348. 'tests/fixtures/multiple-composefiles/compose2.yml')
  349. self._project = get_project(self.base_dir, [config_path])
  350. self.base_dir = 'tests/fixtures/multiple-composefiles'
  351. self.dispatch(['-f', 'compose2.yml', 'up', '-d'])
  352. result = self.dispatch(['-f', 'compose2.yml', 'ps'])
  353. self.assertNotIn('multiplecomposefiles_simple_1', result.stdout)
  354. self.assertNotIn('multiplecomposefiles_another_1', result.stdout)
  355. self.assertIn('multiplecomposefiles_yetanother_1', result.stdout)
  356. def test_pull(self):
  357. result = self.dispatch(['pull'])
  358. assert sorted(result.stderr.split('\n'))[1:] == [
  359. 'Pulling another (busybox:latest)...',
  360. 'Pulling simple (busybox:latest)...',
  361. ]
  362. def test_pull_with_digest(self):
  363. result = self.dispatch(['-f', 'digest.yml', 'pull'])
  364. assert 'Pulling simple (busybox:latest)...' in result.stderr
  365. assert ('Pulling digest (busybox@'
  366. 'sha256:38a203e1986cf79639cfb9b2e1d6e773de84002feea2d4eb006b520'
  367. '04ee8502d)...') in result.stderr
  368. def test_pull_with_ignore_pull_failures(self):
  369. result = self.dispatch([
  370. '-f', 'ignore-pull-failures.yml',
  371. 'pull', '--ignore-pull-failures']
  372. )
  373. assert 'Pulling simple (busybox:latest)...' in result.stderr
  374. assert 'Pulling another (nonexisting-image:latest)...' in result.stderr
  375. assert ('repository nonexisting-image not found' in result.stderr or
  376. 'image library/nonexisting-image:latest not found' in result.stderr)
  377. def test_pull_with_quiet(self):
  378. assert self.dispatch(['pull', '--quiet']).stderr == ''
  379. assert self.dispatch(['pull', '--quiet']).stdout == ''
  380. def test_build_plain(self):
  381. self.base_dir = 'tests/fixtures/simple-dockerfile'
  382. self.dispatch(['build', 'simple'])
  383. result = self.dispatch(['build', 'simple'])
  384. assert BUILD_CACHE_TEXT in result.stdout
  385. assert BUILD_PULL_TEXT not in result.stdout
  386. def test_build_no_cache(self):
  387. self.base_dir = 'tests/fixtures/simple-dockerfile'
  388. self.dispatch(['build', 'simple'])
  389. result = self.dispatch(['build', '--no-cache', 'simple'])
  390. assert BUILD_CACHE_TEXT not in result.stdout
  391. assert BUILD_PULL_TEXT not in result.stdout
  392. def test_build_pull(self):
  393. # Make sure we have the latest busybox already
  394. pull_busybox(self.client)
  395. self.base_dir = 'tests/fixtures/simple-dockerfile'
  396. self.dispatch(['build', 'simple'], None)
  397. result = self.dispatch(['build', '--pull', 'simple'])
  398. assert BUILD_CACHE_TEXT in result.stdout
  399. assert BUILD_PULL_TEXT in result.stdout
  400. def test_build_no_cache_pull(self):
  401. # Make sure we have the latest busybox already
  402. pull_busybox(self.client)
  403. self.base_dir = 'tests/fixtures/simple-dockerfile'
  404. self.dispatch(['build', 'simple'])
  405. result = self.dispatch(['build', '--no-cache', '--pull', 'simple'])
  406. assert BUILD_CACHE_TEXT not in result.stdout
  407. assert BUILD_PULL_TEXT in result.stdout
  408. def test_build_failed(self):
  409. self.base_dir = 'tests/fixtures/simple-failing-dockerfile'
  410. self.dispatch(['build', 'simple'], returncode=1)
  411. labels = ["com.docker.compose.test_failing_image=true"]
  412. containers = [
  413. Container.from_ps(self.project.client, c)
  414. for c in self.project.client.containers(
  415. all=True,
  416. filters={"label": labels})
  417. ]
  418. assert len(containers) == 1
  419. def test_build_failed_forcerm(self):
  420. self.base_dir = 'tests/fixtures/simple-failing-dockerfile'
  421. self.dispatch(['build', '--force-rm', 'simple'], returncode=1)
  422. labels = ["com.docker.compose.test_failing_image=true"]
  423. containers = [
  424. Container.from_ps(self.project.client, c)
  425. for c in self.project.client.containers(
  426. all=True,
  427. filters={"label": labels})
  428. ]
  429. assert not containers
  430. def test_bundle_with_digests(self):
  431. self.base_dir = 'tests/fixtures/bundle-with-digests/'
  432. tmpdir = py.test.ensuretemp('cli_test_bundle')
  433. self.addCleanup(tmpdir.remove)
  434. filename = str(tmpdir.join('example.dab'))
  435. self.dispatch(['bundle', '--output', filename])
  436. with open(filename, 'r') as fh:
  437. bundle = json.load(fh)
  438. assert bundle == {
  439. 'Version': '0.1',
  440. 'Services': {
  441. 'web': {
  442. 'Image': ('dockercloud/hello-world@sha256:fe79a2cfbd17eefc3'
  443. '44fb8419420808df95a1e22d93b7f621a7399fd1e9dca1d'),
  444. 'Networks': ['default'],
  445. },
  446. 'redis': {
  447. 'Image': ('redis@sha256:a84cb8f53a70e19f61ff2e1d5e73fb7ae62d'
  448. '374b2b7392de1e7d77be26ef8f7b'),
  449. 'Networks': ['default'],
  450. }
  451. },
  452. }
  453. def test_build_override_dir(self):
  454. self.base_dir = 'tests/fixtures/build-path-override-dir'
  455. self.override_dir = os.path.abspath('tests/fixtures')
  456. result = self.dispatch([
  457. '--project-directory', self.override_dir,
  458. 'build'])
  459. assert 'Successfully built' in result.stdout
  460. def test_build_override_dir_invalid_path(self):
  461. config_path = os.path.abspath('tests/fixtures/build-path-override-dir/docker-compose.yml')
  462. result = self.dispatch([
  463. '-f', config_path,
  464. 'build'], returncode=1)
  465. assert 'does not exist, is not accessible, or is not a valid URL' in result.stderr
  466. def test_create(self):
  467. self.dispatch(['create'])
  468. service = self.project.get_service('simple')
  469. another = self.project.get_service('another')
  470. service_containers = service.containers(stopped=True)
  471. another_containers = another.containers(stopped=True)
  472. assert len(service_containers) == 1
  473. assert len(another_containers) == 1
  474. assert not service_containers[0].is_running
  475. assert not another_containers[0].is_running
  476. def test_create_with_force_recreate(self):
  477. self.dispatch(['create'], None)
  478. service = self.project.get_service('simple')
  479. service_containers = service.containers(stopped=True)
  480. assert len(service_containers) == 1
  481. assert not service_containers[0].is_running
  482. old_ids = [c.id for c in service.containers(stopped=True)]
  483. self.dispatch(['create', '--force-recreate'], None)
  484. service_containers = service.containers(stopped=True)
  485. assert len(service_containers) == 1
  486. assert not service_containers[0].is_running
  487. new_ids = [c.id for c in service_containers]
  488. assert old_ids != new_ids
  489. def test_create_with_no_recreate(self):
  490. self.dispatch(['create'], None)
  491. service = self.project.get_service('simple')
  492. service_containers = service.containers(stopped=True)
  493. assert len(service_containers) == 1
  494. assert not service_containers[0].is_running
  495. old_ids = [c.id for c in service.containers(stopped=True)]
  496. self.dispatch(['create', '--no-recreate'], None)
  497. service_containers = service.containers(stopped=True)
  498. assert len(service_containers) == 1
  499. assert not service_containers[0].is_running
  500. new_ids = [c.id for c in service_containers]
  501. assert old_ids == new_ids
  502. def test_run_one_off_with_volume(self):
  503. self.base_dir = 'tests/fixtures/simple-composefile-volume-ready'
  504. volume_path = os.path.abspath(os.path.join(os.getcwd(), self.base_dir, 'files'))
  505. create_host_file(self.client, os.path.join(volume_path, 'example.txt'))
  506. self.dispatch([
  507. 'run',
  508. '-v', '{}:/data'.format(volume_path),
  509. 'simple',
  510. 'test', '-f', '/data/example.txt'
  511. ], returncode=0)
  512. service = self.project.get_service('simple')
  513. container_data = service.containers(one_off=OneOffFilter.only, stopped=True)[0]
  514. mount = container_data.get('Mounts')[0]
  515. assert mount['Source'] == volume_path
  516. assert mount['Destination'] == '/data'
  517. assert mount['Type'] == 'bind'
  518. def test_run_one_off_with_multiple_volumes(self):
  519. self.base_dir = 'tests/fixtures/simple-composefile-volume-ready'
  520. volume_path = os.path.abspath(os.path.join(os.getcwd(), self.base_dir, 'files'))
  521. create_host_file(self.client, os.path.join(volume_path, 'example.txt'))
  522. self.dispatch([
  523. 'run',
  524. '-v', '{}:/data'.format(volume_path),
  525. '-v', '{}:/data1'.format(volume_path),
  526. 'simple',
  527. 'test', '-f', '/data/example.txt'
  528. ], returncode=0)
  529. self.dispatch([
  530. 'run',
  531. '-v', '{}:/data'.format(volume_path),
  532. '-v', '{}:/data1'.format(volume_path),
  533. 'simple',
  534. 'test', '-f' '/data1/example.txt'
  535. ], returncode=0)
  536. def test_run_one_off_with_volume_merge(self):
  537. self.base_dir = 'tests/fixtures/simple-composefile-volume-ready'
  538. volume_path = os.path.abspath(os.path.join(os.getcwd(), self.base_dir, 'files'))
  539. create_host_file(self.client, os.path.join(volume_path, 'example.txt'))
  540. self.dispatch([
  541. '-f', 'docker-compose.merge.yml',
  542. 'run',
  543. '-v', '{}:/data'.format(volume_path),
  544. 'simple',
  545. 'test', '-f', '/data/example.txt'
  546. ], returncode=0)
  547. service = self.project.get_service('simple')
  548. container_data = service.containers(one_off=OneOffFilter.only, stopped=True)[0]
  549. mounts = container_data.get('Mounts')
  550. assert len(mounts) == 2
  551. config_mount = [m for m in mounts if m['Destination'] == '/data1'][0]
  552. override_mount = [m for m in mounts if m['Destination'] == '/data'][0]
  553. assert config_mount['Type'] == 'volume'
  554. assert override_mount['Source'] == volume_path
  555. assert override_mount['Type'] == 'bind'
  556. def test_create_with_force_recreate_and_no_recreate(self):
  557. self.dispatch(
  558. ['create', '--force-recreate', '--no-recreate'],
  559. returncode=1)
  560. def test_down_invalid_rmi_flag(self):
  561. result = self.dispatch(['down', '--rmi', 'bogus'], returncode=1)
  562. assert '--rmi flag must be' in result.stderr
  563. @v2_only()
  564. def test_down(self):
  565. self.base_dir = 'tests/fixtures/v2-full'
  566. self.dispatch(['up', '-d'])
  567. wait_on_condition(ContainerCountCondition(self.project, 2))
  568. self.dispatch(['run', 'web', 'true'])
  569. self.dispatch(['run', '-d', 'web', 'tail', '-f', '/dev/null'])
  570. assert len(self.project.containers(one_off=OneOffFilter.only, stopped=True)) == 2
  571. result = self.dispatch(['down', '--rmi=local', '--volumes'])
  572. assert 'Stopping v2full_web_1' in result.stderr
  573. assert 'Stopping v2full_other_1' in result.stderr
  574. assert 'Stopping v2full_web_run_2' in result.stderr
  575. assert 'Removing v2full_web_1' in result.stderr
  576. assert 'Removing v2full_other_1' in result.stderr
  577. assert 'Removing v2full_web_run_1' in result.stderr
  578. assert 'Removing v2full_web_run_2' in result.stderr
  579. assert 'Removing volume v2full_data' in result.stderr
  580. assert 'Removing image v2full_web' in result.stderr
  581. assert 'Removing image busybox' not in result.stderr
  582. assert 'Removing network v2full_default' in result.stderr
  583. assert 'Removing network v2full_front' in result.stderr
  584. def test_up_detached(self):
  585. self.dispatch(['up', '-d'])
  586. service = self.project.get_service('simple')
  587. another = self.project.get_service('another')
  588. self.assertEqual(len(service.containers()), 1)
  589. self.assertEqual(len(another.containers()), 1)
  590. # Ensure containers don't have stdin and stdout connected in -d mode
  591. container, = service.containers()
  592. self.assertFalse(container.get('Config.AttachStderr'))
  593. self.assertFalse(container.get('Config.AttachStdout'))
  594. self.assertFalse(container.get('Config.AttachStdin'))
  595. def test_up_attached(self):
  596. self.base_dir = 'tests/fixtures/echo-services'
  597. result = self.dispatch(['up', '--no-color'])
  598. assert 'simple_1 | simple' in result.stdout
  599. assert 'another_1 | another' in result.stdout
  600. assert 'simple_1 exited with code 0' in result.stdout
  601. assert 'another_1 exited with code 0' in result.stdout
  602. @v2_only()
  603. def test_up(self):
  604. self.base_dir = 'tests/fixtures/v2-simple'
  605. self.dispatch(['up', '-d'], None)
  606. services = self.project.get_services()
  607. network_name = self.project.networks.networks['default'].full_name
  608. networks = self.client.networks(names=[network_name])
  609. self.assertEqual(len(networks), 1)
  610. assert networks[0]['Driver'] == 'bridge' if not is_cluster(self.client) else 'overlay'
  611. assert 'com.docker.network.bridge.enable_icc' not in networks[0]['Options']
  612. network = self.client.inspect_network(networks[0]['Id'])
  613. for service in services:
  614. containers = service.containers()
  615. self.assertEqual(len(containers), 1)
  616. container = containers[0]
  617. self.assertIn(container.id, network['Containers'])
  618. networks = container.get('NetworkSettings.Networks')
  619. self.assertEqual(list(networks), [network['Name']])
  620. self.assertEqual(
  621. sorted(networks[network['Name']]['Aliases']),
  622. sorted([service.name, container.short_id]))
  623. for service in services:
  624. assert self.lookup(container, service.name)
  625. @v2_only()
  626. def test_up_with_default_network_config(self):
  627. filename = 'default-network-config.yml'
  628. self.base_dir = 'tests/fixtures/networks'
  629. self._project = get_project(self.base_dir, [filename])
  630. self.dispatch(['-f', filename, 'up', '-d'], None)
  631. network_name = self.project.networks.networks['default'].full_name
  632. networks = self.client.networks(names=[network_name])
  633. assert networks[0]['Options']['com.docker.network.bridge.enable_icc'] == 'false'
  634. @v2_only()
  635. def test_up_with_network_aliases(self):
  636. filename = 'network-aliases.yml'
  637. self.base_dir = 'tests/fixtures/networks'
  638. self.dispatch(['-f', filename, 'up', '-d'], None)
  639. back_name = '{}_back'.format(self.project.name)
  640. front_name = '{}_front'.format(self.project.name)
  641. networks = [
  642. n for n in self.client.networks()
  643. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  644. ]
  645. # Two networks were created: back and front
  646. assert sorted(n['Name'].split('/')[-1] for n in networks) == [back_name, front_name]
  647. web_container = self.project.get_service('web').containers()[0]
  648. back_aliases = web_container.get(
  649. 'NetworkSettings.Networks.{}.Aliases'.format(back_name)
  650. )
  651. assert 'web' in back_aliases
  652. front_aliases = web_container.get(
  653. 'NetworkSettings.Networks.{}.Aliases'.format(front_name)
  654. )
  655. assert 'web' in front_aliases
  656. assert 'forward_facing' in front_aliases
  657. assert 'ahead' in front_aliases
  658. @v2_only()
  659. def test_up_with_network_internal(self):
  660. self.require_api_version('1.23')
  661. filename = 'network-internal.yml'
  662. self.base_dir = 'tests/fixtures/networks'
  663. self.dispatch(['-f', filename, 'up', '-d'], None)
  664. internal_net = '{}_internal'.format(self.project.name)
  665. networks = [
  666. n for n in self.client.networks()
  667. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  668. ]
  669. # One network was created: internal
  670. assert sorted(n['Name'].split('/')[-1] for n in networks) == [internal_net]
  671. assert networks[0]['Internal'] is True
  672. @v2_only()
  673. def test_up_with_network_static_addresses(self):
  674. filename = 'network-static-addresses.yml'
  675. ipv4_address = '172.16.100.100'
  676. ipv6_address = 'fe80::1001:100'
  677. self.base_dir = 'tests/fixtures/networks'
  678. self.dispatch(['-f', filename, 'up', '-d'], None)
  679. static_net = '{}_static_test'.format(self.project.name)
  680. networks = [
  681. n for n in self.client.networks()
  682. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  683. ]
  684. # One networks was created: front
  685. assert sorted(n['Name'].split('/')[-1] for n in networks) == [static_net]
  686. web_container = self.project.get_service('web').containers()[0]
  687. ipam_config = web_container.get(
  688. 'NetworkSettings.Networks.{}.IPAMConfig'.format(static_net)
  689. )
  690. assert ipv4_address in ipam_config.values()
  691. assert ipv6_address in ipam_config.values()
  692. @v2_only()
  693. def test_up_with_networks(self):
  694. self.base_dir = 'tests/fixtures/networks'
  695. self.dispatch(['up', '-d'], None)
  696. back_name = '{}_back'.format(self.project.name)
  697. front_name = '{}_front'.format(self.project.name)
  698. networks = [
  699. n for n in self.client.networks()
  700. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  701. ]
  702. # Two networks were created: back and front
  703. assert sorted(n['Name'].split('/')[-1] for n in networks) == [back_name, front_name]
  704. back_network = [n for n in networks if n['Name'] == back_name][0]
  705. front_network = [n for n in networks if n['Name'] == front_name][0]
  706. web_container = self.project.get_service('web').containers()[0]
  707. app_container = self.project.get_service('app').containers()[0]
  708. db_container = self.project.get_service('db').containers()[0]
  709. for net_name in [front_name, back_name]:
  710. links = app_container.get('NetworkSettings.Networks.{}.Links'.format(net_name))
  711. assert '{}:database'.format(db_container.name) in links
  712. # db and app joined the back network
  713. assert sorted(back_network['Containers']) == sorted([db_container.id, app_container.id])
  714. # web and app joined the front network
  715. assert sorted(front_network['Containers']) == sorted([web_container.id, app_container.id])
  716. # web can see app but not db
  717. assert self.lookup(web_container, "app")
  718. assert not self.lookup(web_container, "db")
  719. # app can see db
  720. assert self.lookup(app_container, "db")
  721. # app has aliased db to "database"
  722. assert self.lookup(app_container, "database")
  723. @v2_only()
  724. def test_up_missing_network(self):
  725. self.base_dir = 'tests/fixtures/networks'
  726. result = self.dispatch(
  727. ['-f', 'missing-network.yml', 'up', '-d'],
  728. returncode=1)
  729. assert 'Service "web" uses an undefined network "foo"' in result.stderr
  730. @v2_only()
  731. @no_cluster('container networks not supported in Swarm')
  732. def test_up_with_network_mode(self):
  733. c = self.client.create_container(
  734. 'busybox', 'top', name='composetest_network_mode_container',
  735. host_config={}
  736. )
  737. self.addCleanup(self.client.remove_container, c, force=True)
  738. self.client.start(c)
  739. container_mode_source = 'container:{}'.format(c['Id'])
  740. filename = 'network-mode.yml'
  741. self.base_dir = 'tests/fixtures/networks'
  742. self._project = get_project(self.base_dir, [filename])
  743. self.dispatch(['-f', filename, 'up', '-d'], None)
  744. networks = [
  745. n for n in self.client.networks()
  746. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  747. ]
  748. assert not networks
  749. for name in ['bridge', 'host', 'none']:
  750. container = self.project.get_service(name).containers()[0]
  751. assert list(container.get('NetworkSettings.Networks')) == [name]
  752. assert container.get('HostConfig.NetworkMode') == name
  753. service_mode_source = 'container:{}'.format(
  754. self.project.get_service('bridge').containers()[0].id)
  755. service_mode_container = self.project.get_service('service').containers()[0]
  756. assert not service_mode_container.get('NetworkSettings.Networks')
  757. assert service_mode_container.get('HostConfig.NetworkMode') == service_mode_source
  758. container_mode_container = self.project.get_service('container').containers()[0]
  759. assert not container_mode_container.get('NetworkSettings.Networks')
  760. assert container_mode_container.get('HostConfig.NetworkMode') == container_mode_source
  761. @v2_only()
  762. def test_up_external_networks(self):
  763. filename = 'external-networks.yml'
  764. self.base_dir = 'tests/fixtures/networks'
  765. self._project = get_project(self.base_dir, [filename])
  766. result = self.dispatch(['-f', filename, 'up', '-d'], returncode=1)
  767. assert 'declared as external, but could not be found' in result.stderr
  768. networks = [
  769. n['Name'] for n in self.client.networks()
  770. if n['Name'].startswith('{}_'.format(self.project.name))
  771. ]
  772. assert not networks
  773. network_names = ['{}_{}'.format(self.project.name, n) for n in ['foo', 'bar']]
  774. for name in network_names:
  775. self.client.create_network(name, attachable=True)
  776. self.dispatch(['-f', filename, 'up', '-d'])
  777. container = self.project.containers()[0]
  778. assert sorted(list(container.get('NetworkSettings.Networks'))) == sorted(network_names)
  779. @v2_only()
  780. def test_up_with_external_default_network(self):
  781. filename = 'external-default.yml'
  782. self.base_dir = 'tests/fixtures/networks'
  783. self._project = get_project(self.base_dir, [filename])
  784. result = self.dispatch(['-f', filename, 'up', '-d'], returncode=1)
  785. assert 'declared as external, but could not be found' in result.stderr
  786. networks = [
  787. n['Name'] for n in self.client.networks()
  788. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  789. ]
  790. assert not networks
  791. network_name = 'composetest_external_network'
  792. self.client.create_network(network_name, attachable=True)
  793. self.dispatch(['-f', filename, 'up', '-d'])
  794. container = self.project.containers()[0]
  795. assert list(container.get('NetworkSettings.Networks')) == [network_name]
  796. @v2_1_only()
  797. def test_up_with_network_labels(self):
  798. filename = 'network-label.yml'
  799. self.base_dir = 'tests/fixtures/networks'
  800. self._project = get_project(self.base_dir, [filename])
  801. self.dispatch(['-f', filename, 'up', '-d'], returncode=0)
  802. network_with_label = '{}_network_with_label'.format(self.project.name)
  803. networks = [
  804. n for n in self.client.networks()
  805. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  806. ]
  807. assert [n['Name'].split('/')[-1] for n in networks] == [network_with_label]
  808. assert 'label_key' in networks[0]['Labels']
  809. assert networks[0]['Labels']['label_key'] == 'label_val'
  810. @v2_1_only()
  811. def test_up_with_volume_labels(self):
  812. filename = 'volume-label.yml'
  813. self.base_dir = 'tests/fixtures/volumes'
  814. self._project = get_project(self.base_dir, [filename])
  815. self.dispatch(['-f', filename, 'up', '-d'], returncode=0)
  816. volume_with_label = '{}_volume_with_label'.format(self.project.name)
  817. volumes = [
  818. v for v in self.client.volumes().get('Volumes', [])
  819. if v['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  820. ]
  821. assert set([v['Name'].split('/')[-1] for v in volumes]) == set([volume_with_label])
  822. assert 'label_key' in volumes[0]['Labels']
  823. assert volumes[0]['Labels']['label_key'] == 'label_val'
  824. @v2_only()
  825. def test_up_no_services(self):
  826. self.base_dir = 'tests/fixtures/no-services'
  827. self.dispatch(['up', '-d'], None)
  828. network_names = [
  829. n['Name'] for n in self.client.networks()
  830. if n['Name'].split('/')[-1].startswith('{}_'.format(self.project.name))
  831. ]
  832. assert network_names == []
  833. def test_up_with_links_v1(self):
  834. self.base_dir = 'tests/fixtures/links-composefile'
  835. self.dispatch(['up', '-d', 'web'], None)
  836. # No network was created
  837. network_name = self.project.networks.networks['default'].full_name
  838. networks = self.client.networks(names=[network_name])
  839. assert networks == []
  840. web = self.project.get_service('web')
  841. db = self.project.get_service('db')
  842. console = self.project.get_service('console')
  843. # console was not started
  844. self.assertEqual(len(web.containers()), 1)
  845. self.assertEqual(len(db.containers()), 1)
  846. self.assertEqual(len(console.containers()), 0)
  847. # web has links
  848. web_container = web.containers()[0]
  849. self.assertTrue(web_container.get('HostConfig.Links'))
  850. def test_up_with_net_is_invalid(self):
  851. self.base_dir = 'tests/fixtures/net-container'
  852. result = self.dispatch(
  853. ['-f', 'v2-invalid.yml', 'up', '-d'],
  854. returncode=1)
  855. assert "Unsupported config option for services.bar: 'net'" in result.stderr
  856. @no_cluster("Legacy networking not supported on Swarm")
  857. def test_up_with_net_v1(self):
  858. self.base_dir = 'tests/fixtures/net-container'
  859. self.dispatch(['up', '-d'], None)
  860. bar = self.project.get_service('bar')
  861. bar_container = bar.containers()[0]
  862. foo = self.project.get_service('foo')
  863. foo_container = foo.containers()[0]
  864. assert foo_container.get('HostConfig.NetworkMode') == \
  865. 'container:{}'.format(bar_container.id)
  866. @v3_only()
  867. def test_up_with_healthcheck(self):
  868. def wait_on_health_status(container, status):
  869. def condition():
  870. container.inspect()
  871. return container.get('State.Health.Status') == status
  872. return wait_on_condition(condition, delay=0.5)
  873. self.base_dir = 'tests/fixtures/healthcheck'
  874. self.dispatch(['up', '-d'], None)
  875. passes = self.project.get_service('passes')
  876. passes_container = passes.containers()[0]
  877. assert passes_container.get('Config.Healthcheck') == {
  878. "Test": ["CMD-SHELL", "/bin/true"],
  879. "Interval": nanoseconds_from_time_seconds(1),
  880. "Timeout": nanoseconds_from_time_seconds(30 * 60),
  881. "Retries": 1,
  882. }
  883. wait_on_health_status(passes_container, 'healthy')
  884. fails = self.project.get_service('fails')
  885. fails_container = fails.containers()[0]
  886. assert fails_container.get('Config.Healthcheck') == {
  887. "Test": ["CMD", "/bin/false"],
  888. "Interval": nanoseconds_from_time_seconds(2.5),
  889. "Retries": 2,
  890. }
  891. wait_on_health_status(fails_container, 'unhealthy')
  892. disabled = self.project.get_service('disabled')
  893. disabled_container = disabled.containers()[0]
  894. assert disabled_container.get('Config.Healthcheck') == {
  895. "Test": ["NONE"],
  896. }
  897. assert 'Health' not in disabled_container.get('State')
  898. def test_up_with_no_deps(self):
  899. self.base_dir = 'tests/fixtures/links-composefile'
  900. self.dispatch(['up', '-d', '--no-deps', 'web'], None)
  901. web = self.project.get_service('web')
  902. db = self.project.get_service('db')
  903. console = self.project.get_service('console')
  904. self.assertEqual(len(web.containers()), 1)
  905. self.assertEqual(len(db.containers()), 0)
  906. self.assertEqual(len(console.containers()), 0)
  907. def test_up_with_force_recreate(self):
  908. self.dispatch(['up', '-d'], None)
  909. service = self.project.get_service('simple')
  910. self.assertEqual(len(service.containers()), 1)
  911. old_ids = [c.id for c in service.containers()]
  912. self.dispatch(['up', '-d', '--force-recreate'], None)
  913. self.assertEqual(len(service.containers()), 1)
  914. new_ids = [c.id for c in service.containers()]
  915. self.assertNotEqual(old_ids, new_ids)
  916. def test_up_with_no_recreate(self):
  917. self.dispatch(['up', '-d'], None)
  918. service = self.project.get_service('simple')
  919. self.assertEqual(len(service.containers()), 1)
  920. old_ids = [c.id for c in service.containers()]
  921. self.dispatch(['up', '-d', '--no-recreate'], None)
  922. self.assertEqual(len(service.containers()), 1)
  923. new_ids = [c.id for c in service.containers()]
  924. self.assertEqual(old_ids, new_ids)
  925. def test_up_with_force_recreate_and_no_recreate(self):
  926. self.dispatch(
  927. ['up', '-d', '--force-recreate', '--no-recreate'],
  928. returncode=1)
  929. def test_up_with_timeout(self):
  930. self.dispatch(['up', '-d', '-t', '1'])
  931. service = self.project.get_service('simple')
  932. another = self.project.get_service('another')
  933. self.assertEqual(len(service.containers()), 1)
  934. self.assertEqual(len(another.containers()), 1)
  935. # Ensure containers don't have stdin and stdout connected in -d mode
  936. config = service.containers()[0].inspect()['Config']
  937. self.assertFalse(config['AttachStderr'])
  938. self.assertFalse(config['AttachStdout'])
  939. self.assertFalse(config['AttachStdin'])
  940. def test_up_handles_sigint(self):
  941. proc = start_process(self.base_dir, ['up', '-t', '2'])
  942. wait_on_condition(ContainerCountCondition(self.project, 2))
  943. os.kill(proc.pid, signal.SIGINT)
  944. wait_on_condition(ContainerCountCondition(self.project, 0))
  945. def test_up_handles_sigterm(self):
  946. proc = start_process(self.base_dir, ['up', '-t', '2'])
  947. wait_on_condition(ContainerCountCondition(self.project, 2))
  948. os.kill(proc.pid, signal.SIGTERM)
  949. wait_on_condition(ContainerCountCondition(self.project, 0))
  950. @v2_only()
  951. def test_up_handles_force_shutdown(self):
  952. self.base_dir = 'tests/fixtures/sleeps-composefile'
  953. proc = start_process(self.base_dir, ['up', '-t', '200'])
  954. wait_on_condition(ContainerCountCondition(self.project, 2))
  955. os.kill(proc.pid, signal.SIGTERM)
  956. time.sleep(0.1)
  957. os.kill(proc.pid, signal.SIGTERM)
  958. wait_on_condition(ContainerCountCondition(self.project, 0))
  959. def test_up_handles_abort_on_container_exit(self):
  960. self.base_dir = 'tests/fixtures/abort-on-container-exit-0'
  961. proc = start_process(self.base_dir, ['up', '--abort-on-container-exit'])
  962. wait_on_condition(ContainerCountCondition(self.project, 0))
  963. proc.wait()
  964. self.assertEqual(proc.returncode, 0)
  965. def test_up_handles_abort_on_container_exit_code(self):
  966. self.base_dir = 'tests/fixtures/abort-on-container-exit-1'
  967. proc = start_process(self.base_dir, ['up', '--abort-on-container-exit'])
  968. wait_on_condition(ContainerCountCondition(self.project, 0))
  969. proc.wait()
  970. self.assertEqual(proc.returncode, 1)
  971. @v2_only()
  972. def test_up_with_pid_mode(self):
  973. c = self.client.create_container(
  974. 'busybox', 'top', name='composetest_pid_mode_container',
  975. host_config={}
  976. )
  977. self.addCleanup(self.client.remove_container, c, force=True)
  978. self.client.start(c)
  979. container_mode_source = 'container:{}'.format(c['Id'])
  980. self.base_dir = 'tests/fixtures/pid-mode'
  981. self.dispatch(['up', '-d'], None)
  982. service_mode_source = 'container:{}'.format(
  983. self.project.get_service('container').containers()[0].id)
  984. service_mode_container = self.project.get_service('service').containers()[0]
  985. assert service_mode_container.get('HostConfig.PidMode') == service_mode_source
  986. container_mode_container = self.project.get_service('container').containers()[0]
  987. assert container_mode_container.get('HostConfig.PidMode') == container_mode_source
  988. host_mode_container = self.project.get_service('host').containers()[0]
  989. assert host_mode_container.get('HostConfig.PidMode') == 'host'
  990. def test_exec_without_tty(self):
  991. self.base_dir = 'tests/fixtures/links-composefile'
  992. self.dispatch(['up', '-d', 'console'])
  993. self.assertEqual(len(self.project.containers()), 1)
  994. stdout, stderr = self.dispatch(['exec', '-T', 'console', 'ls', '-1d', '/'])
  995. self.assertEqual(stdout, "/\n")
  996. self.assertEqual(stderr, "")
  997. def test_exec_custom_user(self):
  998. self.base_dir = 'tests/fixtures/links-composefile'
  999. self.dispatch(['up', '-d', 'console'])
  1000. self.assertEqual(len(self.project.containers()), 1)
  1001. stdout, stderr = self.dispatch(['exec', '-T', '--user=operator', 'console', 'whoami'])
  1002. self.assertEqual(stdout, "operator\n")
  1003. self.assertEqual(stderr, "")
  1004. def test_run_service_without_links(self):
  1005. self.base_dir = 'tests/fixtures/links-composefile'
  1006. self.dispatch(['run', 'console', '/bin/true'])
  1007. self.assertEqual(len(self.project.containers()), 0)
  1008. # Ensure stdin/out was open
  1009. container = self.project.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1010. config = container.inspect()['Config']
  1011. self.assertTrue(config['AttachStderr'])
  1012. self.assertTrue(config['AttachStdout'])
  1013. self.assertTrue(config['AttachStdin'])
  1014. def test_run_service_with_links(self):
  1015. self.base_dir = 'tests/fixtures/links-composefile'
  1016. self.dispatch(['run', 'web', '/bin/true'], None)
  1017. db = self.project.get_service('db')
  1018. console = self.project.get_service('console')
  1019. self.assertEqual(len(db.containers()), 1)
  1020. self.assertEqual(len(console.containers()), 0)
  1021. @v2_only()
  1022. def test_run_service_with_dependencies(self):
  1023. self.base_dir = 'tests/fixtures/v2-dependencies'
  1024. self.dispatch(['run', 'web', '/bin/true'], None)
  1025. db = self.project.get_service('db')
  1026. console = self.project.get_service('console')
  1027. self.assertEqual(len(db.containers()), 1)
  1028. self.assertEqual(len(console.containers()), 0)
  1029. def test_run_service_with_scaled_dependencies(self):
  1030. self.base_dir = 'tests/fixtures/v2-dependencies'
  1031. self.dispatch(['up', '-d', '--scale', 'db=2', '--scale', 'console=0'])
  1032. db = self.project.get_service('db')
  1033. console = self.project.get_service('console')
  1034. assert len(db.containers()) == 2
  1035. assert len(console.containers()) == 0
  1036. self.dispatch(['run', 'web', '/bin/true'], None)
  1037. assert len(db.containers()) == 2
  1038. assert len(console.containers()) == 0
  1039. def test_run_with_no_deps(self):
  1040. self.base_dir = 'tests/fixtures/links-composefile'
  1041. self.dispatch(['run', '--no-deps', 'web', '/bin/true'])
  1042. db = self.project.get_service('db')
  1043. self.assertEqual(len(db.containers()), 0)
  1044. def test_run_does_not_recreate_linked_containers(self):
  1045. self.base_dir = 'tests/fixtures/links-composefile'
  1046. self.dispatch(['up', '-d', 'db'])
  1047. db = self.project.get_service('db')
  1048. self.assertEqual(len(db.containers()), 1)
  1049. old_ids = [c.id for c in db.containers()]
  1050. self.dispatch(['run', 'web', '/bin/true'], None)
  1051. self.assertEqual(len(db.containers()), 1)
  1052. new_ids = [c.id for c in db.containers()]
  1053. self.assertEqual(old_ids, new_ids)
  1054. def test_run_without_command(self):
  1055. self.base_dir = 'tests/fixtures/commands-composefile'
  1056. self.check_build('tests/fixtures/simple-dockerfile', tag='composetest_test')
  1057. self.dispatch(['run', 'implicit'])
  1058. service = self.project.get_service('implicit')
  1059. containers = service.containers(stopped=True, one_off=OneOffFilter.only)
  1060. self.assertEqual(
  1061. [c.human_readable_command for c in containers],
  1062. [u'/bin/sh -c echo "success"'],
  1063. )
  1064. self.dispatch(['run', 'explicit'])
  1065. service = self.project.get_service('explicit')
  1066. containers = service.containers(stopped=True, one_off=OneOffFilter.only)
  1067. self.assertEqual(
  1068. [c.human_readable_command for c in containers],
  1069. [u'/bin/true'],
  1070. )
  1071. @py.test.mark.skipif(SWARM_SKIP_RM_VOLUMES, reason='Swarm DELETE /containers/<id> bug')
  1072. def test_run_rm(self):
  1073. self.base_dir = 'tests/fixtures/volume'
  1074. proc = start_process(self.base_dir, ['run', '--rm', 'test'])
  1075. wait_on_condition(ContainerStateCondition(
  1076. self.project.client,
  1077. 'volume_test_run_1',
  1078. 'running'))
  1079. service = self.project.get_service('test')
  1080. containers = service.containers(one_off=OneOffFilter.only)
  1081. self.assertEqual(len(containers), 1)
  1082. mounts = containers[0].get('Mounts')
  1083. for mount in mounts:
  1084. if mount['Destination'] == '/container-path':
  1085. anonymous_name = mount['Name']
  1086. break
  1087. os.kill(proc.pid, signal.SIGINT)
  1088. wait_on_process(proc, 1)
  1089. self.assertEqual(len(service.containers(stopped=True, one_off=OneOffFilter.only)), 0)
  1090. volumes = self.client.volumes()['Volumes']
  1091. assert volumes is not None
  1092. for volume in service.options.get('volumes'):
  1093. if volume.internal == '/container-named-path':
  1094. name = volume.external
  1095. break
  1096. volume_names = [v['Name'].split('/')[-1] for v in volumes]
  1097. assert name in volume_names
  1098. if not is_cluster(self.client):
  1099. # The `-v` flag for `docker rm` in Swarm seems to be broken
  1100. assert anonymous_name not in volume_names
  1101. def test_run_service_with_dockerfile_entrypoint(self):
  1102. self.base_dir = 'tests/fixtures/entrypoint-dockerfile'
  1103. self.dispatch(['run', 'test'])
  1104. container = self.project.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1105. assert container.get('Config.Entrypoint') == ['printf']
  1106. assert container.get('Config.Cmd') == ['default', 'args']
  1107. def test_run_service_with_dockerfile_entrypoint_overridden(self):
  1108. self.base_dir = 'tests/fixtures/entrypoint-dockerfile'
  1109. self.dispatch(['run', '--entrypoint', 'echo', 'test'])
  1110. container = self.project.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1111. assert container.get('Config.Entrypoint') == ['echo']
  1112. assert not container.get('Config.Cmd')
  1113. def test_run_service_with_dockerfile_entrypoint_and_command_overridden(self):
  1114. self.base_dir = 'tests/fixtures/entrypoint-dockerfile'
  1115. self.dispatch(['run', '--entrypoint', 'echo', 'test', 'foo'])
  1116. container = self.project.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1117. assert container.get('Config.Entrypoint') == ['echo']
  1118. assert container.get('Config.Cmd') == ['foo']
  1119. def test_run_service_with_compose_file_entrypoint(self):
  1120. self.base_dir = 'tests/fixtures/entrypoint-composefile'
  1121. self.dispatch(['run', 'test'])
  1122. container = self.project.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1123. assert container.get('Config.Entrypoint') == ['printf']
  1124. assert container.get('Config.Cmd') == ['default', 'args']
  1125. def test_run_service_with_compose_file_entrypoint_overridden(self):
  1126. self.base_dir = 'tests/fixtures/entrypoint-composefile'
  1127. self.dispatch(['run', '--entrypoint', 'echo', 'test'])
  1128. container = self.project.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1129. assert container.get('Config.Entrypoint') == ['echo']
  1130. assert not container.get('Config.Cmd')
  1131. def test_run_service_with_compose_file_entrypoint_and_command_overridden(self):
  1132. self.base_dir = 'tests/fixtures/entrypoint-composefile'
  1133. self.dispatch(['run', '--entrypoint', 'echo', 'test', 'foo'])
  1134. container = self.project.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1135. assert container.get('Config.Entrypoint') == ['echo']
  1136. assert container.get('Config.Cmd') == ['foo']
  1137. def test_run_service_with_compose_file_entrypoint_and_empty_string_command(self):
  1138. self.base_dir = 'tests/fixtures/entrypoint-composefile'
  1139. self.dispatch(['run', '--entrypoint', 'echo', 'test', ''])
  1140. container = self.project.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1141. assert container.get('Config.Entrypoint') == ['echo']
  1142. assert container.get('Config.Cmd') == ['']
  1143. def test_run_service_with_user_overridden(self):
  1144. self.base_dir = 'tests/fixtures/user-composefile'
  1145. name = 'service'
  1146. user = 'sshd'
  1147. self.dispatch(['run', '--user={user}'.format(user=user), name], returncode=1)
  1148. service = self.project.get_service(name)
  1149. container = service.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1150. self.assertEqual(user, container.get('Config.User'))
  1151. def test_run_service_with_user_overridden_short_form(self):
  1152. self.base_dir = 'tests/fixtures/user-composefile'
  1153. name = 'service'
  1154. user = 'sshd'
  1155. self.dispatch(['run', '-u', user, name], returncode=1)
  1156. service = self.project.get_service(name)
  1157. container = service.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1158. self.assertEqual(user, container.get('Config.User'))
  1159. def test_run_service_with_environment_overridden(self):
  1160. name = 'service'
  1161. self.base_dir = 'tests/fixtures/environment-composefile'
  1162. self.dispatch([
  1163. 'run', '-e', 'foo=notbar',
  1164. '-e', 'allo=moto=bobo',
  1165. '-e', 'alpha=beta',
  1166. name,
  1167. '/bin/true',
  1168. ])
  1169. service = self.project.get_service(name)
  1170. container = service.containers(stopped=True, one_off=OneOffFilter.only)[0]
  1171. # env overridden
  1172. self.assertEqual('notbar', container.environment['foo'])
  1173. # keep environment from yaml
  1174. self.assertEqual('world', container.environment['hello'])
  1175. # added option from command line
  1176. self.assertEqual('beta', container.environment['alpha'])
  1177. # make sure a value with a = don't crash out
  1178. self.assertEqual('moto=bobo', container.environment['allo'])
  1179. def test_run_service_without_map_ports(self):
  1180. # create one off container
  1181. self.base_dir = 'tests/fixtures/ports-composefile'
  1182. self.dispatch(['run', '-d', 'simple'])
  1183. container = self.project.get_service('simple').containers(one_off=OneOffFilter.only)[0]
  1184. # get port information
  1185. port_random = container.get_local_port(3000)
  1186. port_assigned = container.get_local_port(3001)
  1187. # close all one off containers we just created
  1188. container.stop()
  1189. # check the ports
  1190. self.assertEqual(port_random, None)
  1191. self.assertEqual(port_assigned, None)
  1192. def test_run_service_with_map_ports(self):
  1193. # create one off container
  1194. self.base_dir = 'tests/fixtures/ports-composefile'
  1195. self.dispatch(['run', '-d', '--service-ports', 'simple'])
  1196. container = self.project.get_service('simple').containers(one_off=OneOffFilter.only)[0]
  1197. # get port information
  1198. port_random = container.get_local_port(3000)
  1199. port_assigned = container.get_local_port(3001)
  1200. port_range = container.get_local_port(3002), container.get_local_port(3003)
  1201. # close all one off containers we just created
  1202. container.stop()
  1203. # check the ports
  1204. assert port_random is not None
  1205. assert port_assigned.endswith(':49152')
  1206. assert port_range[0].endswith(':49153')
  1207. assert port_range[1].endswith(':49154')
  1208. def test_run_service_with_explicitly_mapped_ports(self):
  1209. # create one off container
  1210. self.base_dir = 'tests/fixtures/ports-composefile'
  1211. self.dispatch(['run', '-d', '-p', '30000:3000', '--publish', '30001:3001', 'simple'])
  1212. container = self.project.get_service('simple').containers(one_off=OneOffFilter.only)[0]
  1213. # get port information
  1214. port_short = container.get_local_port(3000)
  1215. port_full = container.get_local_port(3001)
  1216. # close all one off containers we just created
  1217. container.stop()
  1218. # check the ports
  1219. assert port_short.endswith(':30000')
  1220. assert port_full.endswith(':30001')
  1221. def test_run_service_with_explicitly_mapped_ip_ports(self):
  1222. # create one off container
  1223. self.base_dir = 'tests/fixtures/ports-composefile'
  1224. self.dispatch([
  1225. 'run', '-d',
  1226. '-p', '127.0.0.1:30000:3000',
  1227. '--publish', '127.0.0.1:30001:3001',
  1228. 'simple'
  1229. ])
  1230. container = self.project.get_service('simple').containers(one_off=OneOffFilter.only)[0]
  1231. # get port information
  1232. port_short = container.get_local_port(3000)
  1233. port_full = container.get_local_port(3001)
  1234. # close all one off containers we just created
  1235. container.stop()
  1236. # check the ports
  1237. self.assertEqual(port_short, "127.0.0.1:30000")
  1238. self.assertEqual(port_full, "127.0.0.1:30001")
  1239. def test_run_with_expose_ports(self):
  1240. # create one off container
  1241. self.base_dir = 'tests/fixtures/expose-composefile'
  1242. self.dispatch(['run', '-d', '--service-ports', 'simple'])
  1243. container = self.project.get_service('simple').containers(one_off=OneOffFilter.only)[0]
  1244. ports = container.ports
  1245. self.assertEqual(len(ports), 9)
  1246. # exposed ports are not mapped to host ports
  1247. assert ports['3000/tcp'] is None
  1248. assert ports['3001/tcp'] is None
  1249. assert ports['3001/udp'] is None
  1250. assert ports['3002/tcp'] is None
  1251. assert ports['3003/tcp'] is None
  1252. assert ports['3004/tcp'] is None
  1253. assert ports['3005/tcp'] is None
  1254. assert ports['3006/udp'] is None
  1255. assert ports['3007/udp'] is None
  1256. # close all one off containers we just created
  1257. container.stop()
  1258. def test_run_with_custom_name(self):
  1259. self.base_dir = 'tests/fixtures/environment-composefile'
  1260. name = 'the-container-name'
  1261. self.dispatch(['run', '--name', name, 'service', '/bin/true'])
  1262. service = self.project.get_service('service')
  1263. container, = service.containers(stopped=True, one_off=OneOffFilter.only)
  1264. self.assertEqual(container.name, name)
  1265. def test_run_service_with_workdir_overridden(self):
  1266. self.base_dir = 'tests/fixtures/run-workdir'
  1267. name = 'service'
  1268. workdir = '/var'
  1269. self.dispatch(['run', '--workdir={workdir}'.format(workdir=workdir), name])
  1270. service = self.project.get_service(name)
  1271. container = service.containers(stopped=True, one_off=True)[0]
  1272. self.assertEqual(workdir, container.get('Config.WorkingDir'))
  1273. def test_run_service_with_workdir_overridden_short_form(self):
  1274. self.base_dir = 'tests/fixtures/run-workdir'
  1275. name = 'service'
  1276. workdir = '/var'
  1277. self.dispatch(['run', '-w', workdir, name])
  1278. service = self.project.get_service(name)
  1279. container = service.containers(stopped=True, one_off=True)[0]
  1280. self.assertEqual(workdir, container.get('Config.WorkingDir'))
  1281. @v2_only()
  1282. def test_run_interactive_connects_to_network(self):
  1283. self.base_dir = 'tests/fixtures/networks'
  1284. self.dispatch(['up', '-d'])
  1285. self.dispatch(['run', 'app', 'nslookup', 'app'])
  1286. self.dispatch(['run', 'app', 'nslookup', 'db'])
  1287. containers = self.project.get_service('app').containers(
  1288. stopped=True, one_off=OneOffFilter.only)
  1289. assert len(containers) == 2
  1290. for container in containers:
  1291. networks = container.get('NetworkSettings.Networks')
  1292. assert sorted(list(networks)) == [
  1293. '{}_{}'.format(self.project.name, name)
  1294. for name in ['back', 'front']
  1295. ]
  1296. for _, config in networks.items():
  1297. # TODO: once we drop support for API <1.24, this can be changed to:
  1298. # assert config['Aliases'] == [container.short_id]
  1299. aliases = set(config['Aliases'] or []) - set([container.short_id])
  1300. assert not aliases
  1301. @v2_only()
  1302. def test_run_detached_connects_to_network(self):
  1303. self.base_dir = 'tests/fixtures/networks'
  1304. self.dispatch(['up', '-d'])
  1305. self.dispatch(['run', '-d', 'app', 'top'])
  1306. container = self.project.get_service('app').containers(one_off=OneOffFilter.only)[0]
  1307. networks = container.get('NetworkSettings.Networks')
  1308. assert sorted(list(networks)) == [
  1309. '{}_{}'.format(self.project.name, name)
  1310. for name in ['back', 'front']
  1311. ]
  1312. for _, config in networks.items():
  1313. # TODO: once we drop support for API <1.24, this can be changed to:
  1314. # assert config['Aliases'] == [container.short_id]
  1315. aliases = set(config['Aliases'] or []) - set([container.short_id])
  1316. assert not aliases
  1317. assert self.lookup(container, 'app')
  1318. assert self.lookup(container, 'db')
  1319. def test_run_handles_sigint(self):
  1320. proc = start_process(self.base_dir, ['run', '-T', 'simple', 'top'])
  1321. wait_on_condition(ContainerStateCondition(
  1322. self.project.client,
  1323. 'simplecomposefile_simple_run_1',
  1324. 'running'))
  1325. os.kill(proc.pid, signal.SIGINT)
  1326. wait_on_condition(ContainerStateCondition(
  1327. self.project.client,
  1328. 'simplecomposefile_simple_run_1',
  1329. 'exited'))
  1330. def test_run_handles_sigterm(self):
  1331. proc = start_process(self.base_dir, ['run', '-T', 'simple', 'top'])
  1332. wait_on_condition(ContainerStateCondition(
  1333. self.project.client,
  1334. 'simplecomposefile_simple_run_1',
  1335. 'running'))
  1336. os.kill(proc.pid, signal.SIGTERM)
  1337. wait_on_condition(ContainerStateCondition(
  1338. self.project.client,
  1339. 'simplecomposefile_simple_run_1',
  1340. 'exited'))
  1341. @mock.patch.dict(os.environ)
  1342. def test_run_unicode_env_values_from_system(self):
  1343. value = 'ą, ć, ę, ł, ń, ó, ś, ź, ż'
  1344. if six.PY2: # os.environ doesn't support unicode values in Py2
  1345. os.environ['BAR'] = value.encode('utf-8')
  1346. else: # ... and doesn't support byte values in Py3
  1347. os.environ['BAR'] = value
  1348. self.base_dir = 'tests/fixtures/unicode-environment'
  1349. result = self.dispatch(['run', 'simple'])
  1350. if six.PY2: # Can't retrieve output on Py3. See issue #3670
  1351. assert value == result.stdout.strip()
  1352. container = self.project.containers(one_off=OneOffFilter.only, stopped=True)[0]
  1353. environment = container.get('Config.Env')
  1354. assert 'FOO={}'.format(value) in environment
  1355. @mock.patch.dict(os.environ)
  1356. def test_run_env_values_from_system(self):
  1357. os.environ['FOO'] = 'bar'
  1358. os.environ['BAR'] = 'baz'
  1359. self.dispatch(['run', '-e', 'FOO', 'simple', 'true'], None)
  1360. container = self.project.containers(one_off=OneOffFilter.only, stopped=True)[0]
  1361. environment = container.get('Config.Env')
  1362. assert 'FOO=bar' in environment
  1363. assert 'BAR=baz' not in environment
  1364. def test_rm(self):
  1365. service = self.project.get_service('simple')
  1366. service.create_container()
  1367. kill_service(service)
  1368. self.assertEqual(len(service.containers(stopped=True)), 1)
  1369. self.dispatch(['rm', '--force'], None)
  1370. self.assertEqual(len(service.containers(stopped=True)), 0)
  1371. service = self.project.get_service('simple')
  1372. service.create_container()
  1373. kill_service(service)
  1374. self.assertEqual(len(service.containers(stopped=True)), 1)
  1375. self.dispatch(['rm', '-f'], None)
  1376. self.assertEqual(len(service.containers(stopped=True)), 0)
  1377. service = self.project.get_service('simple')
  1378. service.create_container()
  1379. self.dispatch(['rm', '-fs'], None)
  1380. self.assertEqual(len(service.containers(stopped=True)), 0)
  1381. def test_rm_stop(self):
  1382. self.dispatch(['up', '-d'], None)
  1383. simple = self.project.get_service('simple')
  1384. another = self.project.get_service('another')
  1385. assert len(simple.containers()) == 1
  1386. assert len(another.containers()) == 1
  1387. self.dispatch(['rm', '-fs'], None)
  1388. assert len(simple.containers(stopped=True)) == 0
  1389. assert len(another.containers(stopped=True)) == 0
  1390. self.dispatch(['up', '-d'], None)
  1391. assert len(simple.containers()) == 1
  1392. assert len(another.containers()) == 1
  1393. self.dispatch(['rm', '-fs', 'another'], None)
  1394. assert len(simple.containers()) == 1
  1395. assert len(another.containers(stopped=True)) == 0
  1396. def test_rm_all(self):
  1397. service = self.project.get_service('simple')
  1398. service.create_container(one_off=False)
  1399. service.create_container(one_off=True)
  1400. kill_service(service)
  1401. self.assertEqual(len(service.containers(stopped=True)), 1)
  1402. self.assertEqual(len(service.containers(stopped=True, one_off=OneOffFilter.only)), 1)
  1403. self.dispatch(['rm', '-f'], None)
  1404. self.assertEqual(len(service.containers(stopped=True)), 0)
  1405. self.assertEqual(len(service.containers(stopped=True, one_off=OneOffFilter.only)), 0)
  1406. service.create_container(one_off=False)
  1407. service.create_container(one_off=True)
  1408. kill_service(service)
  1409. self.assertEqual(len(service.containers(stopped=True)), 1)
  1410. self.assertEqual(len(service.containers(stopped=True, one_off=OneOffFilter.only)), 1)
  1411. self.dispatch(['rm', '-f', '--all'], None)
  1412. self.assertEqual(len(service.containers(stopped=True)), 0)
  1413. self.assertEqual(len(service.containers(stopped=True, one_off=OneOffFilter.only)), 0)
  1414. def test_stop(self):
  1415. self.dispatch(['up', '-d'], None)
  1416. service = self.project.get_service('simple')
  1417. self.assertEqual(len(service.containers()), 1)
  1418. self.assertTrue(service.containers()[0].is_running)
  1419. self.dispatch(['stop', '-t', '1'], None)
  1420. self.assertEqual(len(service.containers(stopped=True)), 1)
  1421. self.assertFalse(service.containers(stopped=True)[0].is_running)
  1422. def test_stop_signal(self):
  1423. self.base_dir = 'tests/fixtures/stop-signal-composefile'
  1424. self.dispatch(['up', '-d'], None)
  1425. service = self.project.get_service('simple')
  1426. self.assertEqual(len(service.containers()), 1)
  1427. self.assertTrue(service.containers()[0].is_running)
  1428. self.dispatch(['stop', '-t', '1'], None)
  1429. self.assertEqual(len(service.containers(stopped=True)), 1)
  1430. self.assertFalse(service.containers(stopped=True)[0].is_running)
  1431. self.assertEqual(service.containers(stopped=True)[0].exit_code, 0)
  1432. def test_start_no_containers(self):
  1433. result = self.dispatch(['start'], returncode=1)
  1434. assert 'No containers to start' in result.stderr
  1435. @v2_only()
  1436. def test_up_logging(self):
  1437. self.base_dir = 'tests/fixtures/logging-composefile'
  1438. self.dispatch(['up', '-d'])
  1439. simple = self.project.get_service('simple').containers()[0]
  1440. log_config = simple.get('HostConfig.LogConfig')
  1441. self.assertTrue(log_config)
  1442. self.assertEqual(log_config.get('Type'), 'none')
  1443. another = self.project.get_service('another').containers()[0]
  1444. log_config = another.get('HostConfig.LogConfig')
  1445. self.assertTrue(log_config)
  1446. self.assertEqual(log_config.get('Type'), 'json-file')
  1447. self.assertEqual(log_config.get('Config')['max-size'], '10m')
  1448. def test_up_logging_legacy(self):
  1449. self.base_dir = 'tests/fixtures/logging-composefile-legacy'
  1450. self.dispatch(['up', '-d'])
  1451. simple = self.project.get_service('simple').containers()[0]
  1452. log_config = simple.get('HostConfig.LogConfig')
  1453. self.assertTrue(log_config)
  1454. self.assertEqual(log_config.get('Type'), 'none')
  1455. another = self.project.get_service('another').containers()[0]
  1456. log_config = another.get('HostConfig.LogConfig')
  1457. self.assertTrue(log_config)
  1458. self.assertEqual(log_config.get('Type'), 'json-file')
  1459. self.assertEqual(log_config.get('Config')['max-size'], '10m')
  1460. def test_pause_unpause(self):
  1461. self.dispatch(['up', '-d'], None)
  1462. service = self.project.get_service('simple')
  1463. self.assertFalse(service.containers()[0].is_paused)
  1464. self.dispatch(['pause'], None)
  1465. self.assertTrue(service.containers()[0].is_paused)
  1466. self.dispatch(['unpause'], None)
  1467. self.assertFalse(service.containers()[0].is_paused)
  1468. def test_pause_no_containers(self):
  1469. result = self.dispatch(['pause'], returncode=1)
  1470. assert 'No containers to pause' in result.stderr
  1471. def test_unpause_no_containers(self):
  1472. result = self.dispatch(['unpause'], returncode=1)
  1473. assert 'No containers to unpause' in result.stderr
  1474. def test_logs_invalid_service_name(self):
  1475. self.dispatch(['logs', 'madeupname'], returncode=1)
  1476. def test_logs_follow(self):
  1477. self.base_dir = 'tests/fixtures/echo-services'
  1478. self.dispatch(['up', '-d'])
  1479. result = self.dispatch(['logs', '-f'])
  1480. assert result.stdout.count('\n') == 5
  1481. assert 'simple' in result.stdout
  1482. assert 'another' in result.stdout
  1483. assert 'exited with code 0' in result.stdout
  1484. def test_logs_follow_logs_from_new_containers(self):
  1485. self.base_dir = 'tests/fixtures/logs-composefile'
  1486. self.dispatch(['up', '-d', 'simple'])
  1487. proc = start_process(self.base_dir, ['logs', '-f'])
  1488. self.dispatch(['up', '-d', 'another'])
  1489. wait_on_condition(ContainerStateCondition(
  1490. self.project.client,
  1491. 'logscomposefile_another_1',
  1492. 'exited'))
  1493. self.dispatch(['kill', 'simple'])
  1494. result = wait_on_process(proc)
  1495. assert 'hello' in result.stdout
  1496. assert 'test' in result.stdout
  1497. assert 'logscomposefile_another_1 exited with code 0' in result.stdout
  1498. assert 'logscomposefile_simple_1 exited with code 137' in result.stdout
  1499. def test_logs_default(self):
  1500. self.base_dir = 'tests/fixtures/logs-composefile'
  1501. self.dispatch(['up', '-d'])
  1502. result = self.dispatch(['logs'])
  1503. assert 'hello' in result.stdout
  1504. assert 'test' in result.stdout
  1505. assert 'exited with' not in result.stdout
  1506. def test_logs_on_stopped_containers_exits(self):
  1507. self.base_dir = 'tests/fixtures/echo-services'
  1508. self.dispatch(['up'])
  1509. result = self.dispatch(['logs'])
  1510. assert 'simple' in result.stdout
  1511. assert 'another' in result.stdout
  1512. assert 'exited with' not in result.stdout
  1513. def test_logs_timestamps(self):
  1514. self.base_dir = 'tests/fixtures/echo-services'
  1515. self.dispatch(['up', '-d'])
  1516. result = self.dispatch(['logs', '-f', '-t'])
  1517. self.assertRegexpMatches(result.stdout, '(\d{4})-(\d{2})-(\d{2})T(\d{2})\:(\d{2})\:(\d{2})')
  1518. def test_logs_tail(self):
  1519. self.base_dir = 'tests/fixtures/logs-tail-composefile'
  1520. self.dispatch(['up'])
  1521. result = self.dispatch(['logs', '--tail', '2'])
  1522. assert result.stdout.count('\n') == 3
  1523. def test_kill(self):
  1524. self.dispatch(['up', '-d'], None)
  1525. service = self.project.get_service('simple')
  1526. self.assertEqual(len(service.containers()), 1)
  1527. self.assertTrue(service.containers()[0].is_running)
  1528. self.dispatch(['kill'], None)
  1529. self.assertEqual(len(service.containers(stopped=True)), 1)
  1530. self.assertFalse(service.containers(stopped=True)[0].is_running)
  1531. def test_kill_signal_sigstop(self):
  1532. self.dispatch(['up', '-d'], None)
  1533. service = self.project.get_service('simple')
  1534. self.assertEqual(len(service.containers()), 1)
  1535. self.assertTrue(service.containers()[0].is_running)
  1536. self.dispatch(['kill', '-s', 'SIGSTOP'], None)
  1537. self.assertEqual(len(service.containers()), 1)
  1538. # The container is still running. It has only been paused
  1539. self.assertTrue(service.containers()[0].is_running)
  1540. def test_kill_stopped_service(self):
  1541. self.dispatch(['up', '-d'], None)
  1542. service = self.project.get_service('simple')
  1543. self.dispatch(['kill', '-s', 'SIGSTOP'], None)
  1544. self.assertTrue(service.containers()[0].is_running)
  1545. self.dispatch(['kill', '-s', 'SIGKILL'], None)
  1546. self.assertEqual(len(service.containers(stopped=True)), 1)
  1547. self.assertFalse(service.containers(stopped=True)[0].is_running)
  1548. def test_restart(self):
  1549. service = self.project.get_service('simple')
  1550. container = service.create_container()
  1551. service.start_container(container)
  1552. started_at = container.dictionary['State']['StartedAt']
  1553. self.dispatch(['restart', '-t', '1'], None)
  1554. container.inspect()
  1555. self.assertNotEqual(
  1556. container.dictionary['State']['FinishedAt'],
  1557. '0001-01-01T00:00:00Z',
  1558. )
  1559. self.assertNotEqual(
  1560. container.dictionary['State']['StartedAt'],
  1561. started_at,
  1562. )
  1563. def test_restart_stopped_container(self):
  1564. service = self.project.get_service('simple')
  1565. container = service.create_container()
  1566. container.start()
  1567. container.kill()
  1568. self.assertEqual(len(service.containers(stopped=True)), 1)
  1569. self.dispatch(['restart', '-t', '1'], None)
  1570. self.assertEqual(len(service.containers(stopped=False)), 1)
  1571. def test_restart_no_containers(self):
  1572. result = self.dispatch(['restart'], returncode=1)
  1573. assert 'No containers to restart' in result.stderr
  1574. def test_scale(self):
  1575. project = self.project
  1576. self.dispatch(['scale', 'simple=1'])
  1577. self.assertEqual(len(project.get_service('simple').containers()), 1)
  1578. self.dispatch(['scale', 'simple=3', 'another=2'])
  1579. self.assertEqual(len(project.get_service('simple').containers()), 3)
  1580. self.assertEqual(len(project.get_service('another').containers()), 2)
  1581. self.dispatch(['scale', 'simple=1', 'another=1'])
  1582. self.assertEqual(len(project.get_service('simple').containers()), 1)
  1583. self.assertEqual(len(project.get_service('another').containers()), 1)
  1584. self.dispatch(['scale', 'simple=1', 'another=1'])
  1585. self.assertEqual(len(project.get_service('simple').containers()), 1)
  1586. self.assertEqual(len(project.get_service('another').containers()), 1)
  1587. self.dispatch(['scale', 'simple=0', 'another=0'])
  1588. self.assertEqual(len(project.get_service('simple').containers()), 0)
  1589. self.assertEqual(len(project.get_service('another').containers()), 0)
  1590. def test_scale_v2_2(self):
  1591. self.base_dir = 'tests/fixtures/scale'
  1592. result = self.dispatch(['scale', 'web=1'], returncode=1)
  1593. assert 'incompatible with the v2.2 format' in result.stderr
  1594. def test_up_scale_scale_up(self):
  1595. self.base_dir = 'tests/fixtures/scale'
  1596. project = self.project
  1597. self.dispatch(['up', '-d'])
  1598. assert len(project.get_service('web').containers()) == 2
  1599. assert len(project.get_service('db').containers()) == 1
  1600. self.dispatch(['up', '-d', '--scale', 'web=3'])
  1601. assert len(project.get_service('web').containers()) == 3
  1602. assert len(project.get_service('db').containers()) == 1
  1603. def test_up_scale_scale_down(self):
  1604. self.base_dir = 'tests/fixtures/scale'
  1605. project = self.project
  1606. self.dispatch(['up', '-d'])
  1607. assert len(project.get_service('web').containers()) == 2
  1608. assert len(project.get_service('db').containers()) == 1
  1609. self.dispatch(['up', '-d', '--scale', 'web=1'])
  1610. assert len(project.get_service('web').containers()) == 1
  1611. assert len(project.get_service('db').containers()) == 1
  1612. def test_up_scale_reset(self):
  1613. self.base_dir = 'tests/fixtures/scale'
  1614. project = self.project
  1615. self.dispatch(['up', '-d', '--scale', 'web=3', '--scale', 'db=3'])
  1616. assert len(project.get_service('web').containers()) == 3
  1617. assert len(project.get_service('db').containers()) == 3
  1618. self.dispatch(['up', '-d'])
  1619. assert len(project.get_service('web').containers()) == 2
  1620. assert len(project.get_service('db').containers()) == 1
  1621. def test_up_scale_to_zero(self):
  1622. self.base_dir = 'tests/fixtures/scale'
  1623. project = self.project
  1624. self.dispatch(['up', '-d'])
  1625. assert len(project.get_service('web').containers()) == 2
  1626. assert len(project.get_service('db').containers()) == 1
  1627. self.dispatch(['up', '-d', '--scale', 'web=0', '--scale', 'db=0'])
  1628. assert len(project.get_service('web').containers()) == 0
  1629. assert len(project.get_service('db').containers()) == 0
  1630. def test_port(self):
  1631. self.base_dir = 'tests/fixtures/ports-composefile'
  1632. self.dispatch(['up', '-d'], None)
  1633. container = self.project.get_service('simple').get_container()
  1634. def get_port(number):
  1635. result = self.dispatch(['port', 'simple', str(number)])
  1636. return result.stdout.rstrip()
  1637. assert get_port(3000) == container.get_local_port(3000)
  1638. assert ':49152' in get_port(3001)
  1639. assert ':49153' in get_port(3002)
  1640. def test_expanded_port(self):
  1641. self.base_dir = 'tests/fixtures/ports-composefile'
  1642. self.dispatch(['-f', 'expanded-notation.yml', 'up', '-d'])
  1643. container = self.project.get_service('simple').get_container()
  1644. def get_port(number):
  1645. result = self.dispatch(['port', 'simple', str(number)])
  1646. return result.stdout.rstrip()
  1647. assert get_port(3000) == container.get_local_port(3000)
  1648. assert ':49152' in get_port(3001)
  1649. assert ':49153' in get_port(3002)
  1650. def test_port_with_scale(self):
  1651. self.base_dir = 'tests/fixtures/ports-composefile-scale'
  1652. self.dispatch(['scale', 'simple=2'], None)
  1653. containers = sorted(
  1654. self.project.containers(service_names=['simple']),
  1655. key=attrgetter('name'))
  1656. def get_port(number, index=None):
  1657. if index is None:
  1658. result = self.dispatch(['port', 'simple', str(number)])
  1659. else:
  1660. result = self.dispatch(['port', '--index=' + str(index), 'simple', str(number)])
  1661. return result.stdout.rstrip()
  1662. self.assertEqual(get_port(3000), containers[0].get_local_port(3000))
  1663. self.assertEqual(get_port(3000, index=1), containers[0].get_local_port(3000))
  1664. self.assertEqual(get_port(3000, index=2), containers[1].get_local_port(3000))
  1665. self.assertEqual(get_port(3002), "")
  1666. def test_events_json(self):
  1667. events_proc = start_process(self.base_dir, ['events', '--json'])
  1668. self.dispatch(['up', '-d'])
  1669. wait_on_condition(ContainerCountCondition(self.project, 2))
  1670. os.kill(events_proc.pid, signal.SIGINT)
  1671. result = wait_on_process(events_proc, returncode=1)
  1672. lines = [json.loads(line) for line in result.stdout.rstrip().split('\n')]
  1673. assert Counter(e['action'] for e in lines) == {'create': 2, 'start': 2}
  1674. def test_events_human_readable(self):
  1675. def has_timestamp(string):
  1676. str_iso_date, str_iso_time, container_info = string.split(' ', 2)
  1677. try:
  1678. return isinstance(datetime.datetime.strptime(
  1679. '%s %s' % (str_iso_date, str_iso_time),
  1680. '%Y-%m-%d %H:%M:%S.%f'),
  1681. datetime.datetime)
  1682. except ValueError:
  1683. return False
  1684. events_proc = start_process(self.base_dir, ['events'])
  1685. self.dispatch(['up', '-d', 'simple'])
  1686. wait_on_condition(ContainerCountCondition(self.project, 1))
  1687. os.kill(events_proc.pid, signal.SIGINT)
  1688. result = wait_on_process(events_proc, returncode=1)
  1689. lines = result.stdout.rstrip().split('\n')
  1690. assert len(lines) == 2
  1691. container, = self.project.containers()
  1692. expected_template = ' container {} {}'
  1693. expected_meta_info = ['image=busybox:latest', 'name=simplecomposefile_simple_1']
  1694. assert expected_template.format('create', container.id) in lines[0]
  1695. assert expected_template.format('start', container.id) in lines[1]
  1696. for line in lines:
  1697. for info in expected_meta_info:
  1698. assert info in line
  1699. assert has_timestamp(lines[0])
  1700. def test_env_file_relative_to_compose_file(self):
  1701. config_path = os.path.abspath('tests/fixtures/env-file/docker-compose.yml')
  1702. self.dispatch(['-f', config_path, 'up', '-d'], None)
  1703. self._project = get_project(self.base_dir, [config_path])
  1704. containers = self.project.containers(stopped=True)
  1705. self.assertEqual(len(containers), 1)
  1706. self.assertIn("FOO=1", containers[0].get('Config.Env'))
  1707. @mock.patch.dict(os.environ)
  1708. def test_home_and_env_var_in_volume_path(self):
  1709. os.environ['VOLUME_NAME'] = 'my-volume'
  1710. os.environ['HOME'] = '/tmp/home-dir'
  1711. self.base_dir = 'tests/fixtures/volume-path-interpolation'
  1712. self.dispatch(['up', '-d'], None)
  1713. container = self.project.containers(stopped=True)[0]
  1714. actual_host_path = container.get_mount('/container-path')['Source']
  1715. components = actual_host_path.split('/')
  1716. assert components[-2:] == ['home-dir', 'my-volume']
  1717. def test_up_with_default_override_file(self):
  1718. self.base_dir = 'tests/fixtures/override-files'
  1719. self.dispatch(['up', '-d'], None)
  1720. containers = self.project.containers()
  1721. self.assertEqual(len(containers), 2)
  1722. web, db = containers
  1723. self.assertEqual(web.human_readable_command, 'top')
  1724. self.assertEqual(db.human_readable_command, 'top')
  1725. def test_up_with_multiple_files(self):
  1726. self.base_dir = 'tests/fixtures/override-files'
  1727. config_paths = [
  1728. 'docker-compose.yml',
  1729. 'docker-compose.override.yml',
  1730. 'extra.yml',
  1731. ]
  1732. self._project = get_project(self.base_dir, config_paths)
  1733. self.dispatch(
  1734. [
  1735. '-f', config_paths[0],
  1736. '-f', config_paths[1],
  1737. '-f', config_paths[2],
  1738. 'up', '-d',
  1739. ],
  1740. None)
  1741. containers = self.project.containers()
  1742. self.assertEqual(len(containers), 3)
  1743. web, other, db = containers
  1744. self.assertEqual(web.human_readable_command, 'top')
  1745. self.assertEqual(db.human_readable_command, 'top')
  1746. self.assertEqual(other.human_readable_command, 'top')
  1747. def test_up_with_extends(self):
  1748. self.base_dir = 'tests/fixtures/extends'
  1749. self.dispatch(['up', '-d'], None)
  1750. self.assertEqual(
  1751. set([s.name for s in self.project.services]),
  1752. set(['mydb', 'myweb']),
  1753. )
  1754. # Sort by name so we get [db, web]
  1755. containers = sorted(
  1756. self.project.containers(stopped=True),
  1757. key=lambda c: c.name,
  1758. )
  1759. self.assertEqual(len(containers), 2)
  1760. web = containers[1]
  1761. self.assertEqual(
  1762. set(get_links(web)),
  1763. set(['db', 'mydb_1', 'extends_mydb_1']))
  1764. expected_env = set([
  1765. "FOO=1",
  1766. "BAR=2",
  1767. "BAZ=2",
  1768. ])
  1769. self.assertTrue(expected_env <= set(web.get('Config.Env')))
  1770. def test_top_services_not_running(self):
  1771. self.base_dir = 'tests/fixtures/top'
  1772. result = self.dispatch(['top'])
  1773. assert len(result.stdout) == 0
  1774. def test_top_services_running(self):
  1775. self.base_dir = 'tests/fixtures/top'
  1776. self.dispatch(['up', '-d'])
  1777. result = self.dispatch(['top'])
  1778. self.assertIn('top_service_a', result.stdout)
  1779. self.assertIn('top_service_b', result.stdout)
  1780. self.assertNotIn('top_not_a_service', result.stdout)
  1781. def test_top_processes_running(self):
  1782. self.base_dir = 'tests/fixtures/top'
  1783. self.dispatch(['up', '-d'])
  1784. result = self.dispatch(['top'])
  1785. assert result.stdout.count("top") == 4
  1786. def test_forward_exitval(self):
  1787. self.base_dir = 'tests/fixtures/exit-code-from'
  1788. proc = start_process(
  1789. self.base_dir,
  1790. ['up', '--abort-on-container-exit', '--exit-code-from', 'another'])
  1791. result = wait_on_process(proc, returncode=1)
  1792. assert 'exitcodefrom_another_1 exited with code 1' in result.stdout
  1793. def test_images(self):
  1794. self.project.get_service('simple').create_container()
  1795. result = self.dispatch(['images'])
  1796. assert 'busybox' in result.stdout
  1797. assert 'simplecomposefile_simple_1' in result.stdout
  1798. def test_images_default_composefile(self):
  1799. self.base_dir = 'tests/fixtures/multiple-composefiles'
  1800. self.dispatch(['up', '-d'])
  1801. result = self.dispatch(['images'])
  1802. assert 'busybox' in result.stdout
  1803. assert 'multiplecomposefiles_another_1' in result.stdout
  1804. assert 'multiplecomposefiles_simple_1' in result.stdout
  1805. def test_up_with_override_yaml(self):
  1806. self.base_dir = 'tests/fixtures/override-yaml-files'
  1807. self._project = get_project(self.base_dir, [])
  1808. self.dispatch(
  1809. [
  1810. 'up', '-d',
  1811. ],
  1812. None)
  1813. containers = self.project.containers()
  1814. self.assertEqual(len(containers), 2)
  1815. web, db = containers
  1816. self.assertEqual(web.human_readable_command, 'sleep 100')
  1817. self.assertEqual(db.human_readable_command, 'top')
  1818. def test_up_with_duplicate_override_yaml_files(self):
  1819. self.base_dir = 'tests/fixtures/duplicate-override-yaml-files'
  1820. with self.assertRaises(DuplicateOverrideFileFound):
  1821. get_project(self.base_dir, [])
  1822. self.base_dir = None