cli_test.py 39 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import datetime
  4. import json
  5. import os
  6. import shlex
  7. import signal
  8. import subprocess
  9. import time
  10. from collections import namedtuple
  11. from operator import attrgetter
  12. from textwrap import dedent
  13. from docker import errors
  14. from .. import mock
  15. from compose.cli.command import get_project
  16. from compose.cli.docker_client import docker_client
  17. from compose.container import Container
  18. from tests.integration.testcases import DockerClientTestCase
  19. from tests.integration.testcases import get_links
  20. from tests.integration.testcases import pull_busybox
  21. ProcessResult = namedtuple('ProcessResult', 'stdout stderr')
  22. BUILD_CACHE_TEXT = 'Using cache'
  23. BUILD_PULL_TEXT = 'Status: Image is up to date for busybox:latest'
  24. def start_process(base_dir, options):
  25. proc = subprocess.Popen(
  26. ['docker-compose'] + options,
  27. stdout=subprocess.PIPE,
  28. stderr=subprocess.PIPE,
  29. cwd=base_dir)
  30. print("Running process: %s" % proc.pid)
  31. return proc
  32. def wait_on_process(proc, returncode=0):
  33. stdout, stderr = proc.communicate()
  34. if proc.returncode != returncode:
  35. print(stderr.decode('utf-8'))
  36. assert proc.returncode == returncode
  37. return ProcessResult(stdout.decode('utf-8'), stderr.decode('utf-8'))
  38. def wait_on_condition(condition, delay=0.1, timeout=20):
  39. start_time = time.time()
  40. while not condition():
  41. if time.time() - start_time > timeout:
  42. raise AssertionError("Timeout: %s" % condition)
  43. time.sleep(delay)
  44. def kill_service(service):
  45. for container in service.containers():
  46. container.kill()
  47. class ContainerCountCondition(object):
  48. def __init__(self, project, expected):
  49. self.project = project
  50. self.expected = expected
  51. def __call__(self):
  52. return len(self.project.containers()) == self.expected
  53. def __str__(self):
  54. return "waiting for counter count == %s" % self.expected
  55. class ContainerStateCondition(object):
  56. def __init__(self, client, name, running):
  57. self.client = client
  58. self.name = name
  59. self.running = running
  60. # State.Running == true
  61. def __call__(self):
  62. try:
  63. container = self.client.inspect_container(self.name)
  64. return container['State']['Running'] == self.running
  65. except errors.APIError:
  66. return False
  67. def __str__(self):
  68. state = 'running' if self.running else 'stopped'
  69. return "waiting for container to be %s" % state
  70. class CLITestCase(DockerClientTestCase):
  71. def setUp(self):
  72. super(CLITestCase, self).setUp()
  73. self.base_dir = 'tests/fixtures/simple-composefile'
  74. def tearDown(self):
  75. if self.base_dir:
  76. self.project.kill()
  77. self.project.remove_stopped()
  78. for container in self.project.containers(stopped=True, one_off=True):
  79. container.remove(force=True)
  80. super(CLITestCase, self).tearDown()
  81. @property
  82. def project(self):
  83. # Hack: allow project to be overridden
  84. if not hasattr(self, '_project'):
  85. self._project = get_project(self.base_dir)
  86. return self._project
  87. def dispatch(self, options, project_options=None, returncode=0):
  88. project_options = project_options or []
  89. proc = start_process(self.base_dir, project_options + options)
  90. return wait_on_process(proc, returncode=returncode)
  91. def test_help(self):
  92. self.base_dir = 'tests/fixtures/no-composefile'
  93. result = self.dispatch(['help', 'up'], returncode=1)
  94. assert 'Usage: up [options] [SERVICE...]' in result.stderr
  95. # Prevent tearDown from trying to create a project
  96. self.base_dir = None
  97. def test_config_list_services(self):
  98. result = self.dispatch(['config', '--services'])
  99. assert set(result.stdout.rstrip().split('\n')) == {'simple', 'another'}
  100. def test_config_quiet_with_error(self):
  101. self.base_dir = None
  102. result = self.dispatch([
  103. '-f', 'tests/fixtures/invalid-composefile/invalid.yml',
  104. 'config', '-q'
  105. ], returncode=1)
  106. assert "'notaservice' doesn't have any configuration" in result.stderr
  107. def test_config_quiet(self):
  108. assert self.dispatch(['config', '-q']).stdout == ''
  109. def test_config_default(self):
  110. result = self.dispatch(['config'])
  111. assert dedent("""
  112. simple:
  113. command: top
  114. image: busybox:latest
  115. """).lstrip() in result.stdout
  116. assert dedent("""
  117. another:
  118. command: top
  119. image: busybox:latest
  120. """).lstrip() in result.stdout
  121. def test_ps(self):
  122. self.project.get_service('simple').create_container()
  123. result = self.dispatch(['ps'])
  124. assert 'simplecomposefile_simple_1' in result.stdout
  125. def test_ps_default_composefile(self):
  126. self.base_dir = 'tests/fixtures/multiple-composefiles'
  127. self.dispatch(['up', '-d'])
  128. result = self.dispatch(['ps'])
  129. self.assertIn('multiplecomposefiles_simple_1', result.stdout)
  130. self.assertIn('multiplecomposefiles_another_1', result.stdout)
  131. self.assertNotIn('multiplecomposefiles_yetanother_1', result.stdout)
  132. def test_ps_alternate_composefile(self):
  133. config_path = os.path.abspath(
  134. 'tests/fixtures/multiple-composefiles/compose2.yml')
  135. self._project = get_project(self.base_dir, [config_path])
  136. self.base_dir = 'tests/fixtures/multiple-composefiles'
  137. self.dispatch(['-f', 'compose2.yml', 'up', '-d'])
  138. result = self.dispatch(['-f', 'compose2.yml', 'ps'])
  139. self.assertNotIn('multiplecomposefiles_simple_1', result.stdout)
  140. self.assertNotIn('multiplecomposefiles_another_1', result.stdout)
  141. self.assertIn('multiplecomposefiles_yetanother_1', result.stdout)
  142. def test_pull(self):
  143. result = self.dispatch(['pull'])
  144. assert sorted(result.stderr.split('\n'))[1:] == [
  145. 'Pulling another (busybox:latest)...',
  146. 'Pulling simple (busybox:latest)...',
  147. ]
  148. def test_pull_with_digest(self):
  149. result = self.dispatch(['-f', 'digest.yml', 'pull'])
  150. assert 'Pulling simple (busybox:latest)...' in result.stderr
  151. assert ('Pulling digest (busybox@'
  152. 'sha256:38a203e1986cf79639cfb9b2e1d6e773de84002feea2d4eb006b520'
  153. '04ee8502d)...') in result.stderr
  154. def test_pull_with_ignore_pull_failures(self):
  155. result = self.dispatch([
  156. '-f', 'ignore-pull-failures.yml',
  157. 'pull', '--ignore-pull-failures'])
  158. assert 'Pulling simple (busybox:latest)...' in result.stderr
  159. assert 'Pulling another (nonexisting-image:latest)...' in result.stderr
  160. assert 'Error: image library/nonexisting-image:latest not found' in result.stderr
  161. def test_build_plain(self):
  162. self.base_dir = 'tests/fixtures/simple-dockerfile'
  163. self.dispatch(['build', 'simple'])
  164. result = self.dispatch(['build', 'simple'])
  165. assert BUILD_CACHE_TEXT in result.stdout
  166. assert BUILD_PULL_TEXT not in result.stdout
  167. def test_build_no_cache(self):
  168. self.base_dir = 'tests/fixtures/simple-dockerfile'
  169. self.dispatch(['build', 'simple'])
  170. result = self.dispatch(['build', '--no-cache', 'simple'])
  171. assert BUILD_CACHE_TEXT not in result.stdout
  172. assert BUILD_PULL_TEXT not in result.stdout
  173. def test_build_pull(self):
  174. # Make sure we have the latest busybox already
  175. pull_busybox(self.client)
  176. self.base_dir = 'tests/fixtures/simple-dockerfile'
  177. self.dispatch(['build', 'simple'], None)
  178. result = self.dispatch(['build', '--pull', 'simple'])
  179. assert BUILD_CACHE_TEXT in result.stdout
  180. assert BUILD_PULL_TEXT in result.stdout
  181. def test_build_no_cache_pull(self):
  182. # Make sure we have the latest busybox already
  183. pull_busybox(self.client)
  184. self.base_dir = 'tests/fixtures/simple-dockerfile'
  185. self.dispatch(['build', 'simple'])
  186. result = self.dispatch(['build', '--no-cache', '--pull', 'simple'])
  187. assert BUILD_CACHE_TEXT not in result.stdout
  188. assert BUILD_PULL_TEXT in result.stdout
  189. def test_build_failed(self):
  190. self.base_dir = 'tests/fixtures/simple-failing-dockerfile'
  191. self.dispatch(['build', 'simple'], returncode=1)
  192. labels = ["com.docker.compose.test_failing_image=true"]
  193. containers = [
  194. Container.from_ps(self.project.client, c)
  195. for c in self.project.client.containers(
  196. all=True,
  197. filters={"label": labels})
  198. ]
  199. assert len(containers) == 1
  200. def test_build_failed_forcerm(self):
  201. self.base_dir = 'tests/fixtures/simple-failing-dockerfile'
  202. self.dispatch(['build', '--force-rm', 'simple'], returncode=1)
  203. labels = ["com.docker.compose.test_failing_image=true"]
  204. containers = [
  205. Container.from_ps(self.project.client, c)
  206. for c in self.project.client.containers(
  207. all=True,
  208. filters={"label": labels})
  209. ]
  210. assert not containers
  211. def test_create(self):
  212. self.dispatch(['create'])
  213. service = self.project.get_service('simple')
  214. another = self.project.get_service('another')
  215. self.assertEqual(len(service.containers()), 0)
  216. self.assertEqual(len(another.containers()), 0)
  217. self.assertEqual(len(service.containers(stopped=True)), 1)
  218. self.assertEqual(len(another.containers(stopped=True)), 1)
  219. def test_create_with_force_recreate(self):
  220. self.dispatch(['create'], None)
  221. service = self.project.get_service('simple')
  222. self.assertEqual(len(service.containers()), 0)
  223. self.assertEqual(len(service.containers(stopped=True)), 1)
  224. old_ids = [c.id for c in service.containers(stopped=True)]
  225. self.dispatch(['create', '--force-recreate'], None)
  226. self.assertEqual(len(service.containers()), 0)
  227. self.assertEqual(len(service.containers(stopped=True)), 1)
  228. new_ids = [c.id for c in service.containers(stopped=True)]
  229. self.assertNotEqual(old_ids, new_ids)
  230. def test_create_with_no_recreate(self):
  231. self.dispatch(['create'], None)
  232. service = self.project.get_service('simple')
  233. self.assertEqual(len(service.containers()), 0)
  234. self.assertEqual(len(service.containers(stopped=True)), 1)
  235. old_ids = [c.id for c in service.containers(stopped=True)]
  236. self.dispatch(['create', '--no-recreate'], None)
  237. self.assertEqual(len(service.containers()), 0)
  238. self.assertEqual(len(service.containers(stopped=True)), 1)
  239. new_ids = [c.id for c in service.containers(stopped=True)]
  240. self.assertEqual(old_ids, new_ids)
  241. def test_create_with_force_recreate_and_no_recreate(self):
  242. self.dispatch(
  243. ['create', '--force-recreate', '--no-recreate'],
  244. returncode=1)
  245. def test_up_detached(self):
  246. self.dispatch(['up', '-d'])
  247. service = self.project.get_service('simple')
  248. another = self.project.get_service('another')
  249. self.assertEqual(len(service.containers()), 1)
  250. self.assertEqual(len(another.containers()), 1)
  251. # Ensure containers don't have stdin and stdout connected in -d mode
  252. container, = service.containers()
  253. self.assertFalse(container.get('Config.AttachStderr'))
  254. self.assertFalse(container.get('Config.AttachStdout'))
  255. self.assertFalse(container.get('Config.AttachStdin'))
  256. def test_up_attached(self):
  257. self.base_dir = 'tests/fixtures/echo-services'
  258. result = self.dispatch(['up', '--no-color'])
  259. assert 'simple_1 | simple' in result.stdout
  260. assert 'another_1 | another' in result.stdout
  261. def test_up_without_networking(self):
  262. self.require_api_version('1.21')
  263. self.base_dir = 'tests/fixtures/links-composefile'
  264. self.dispatch(['up', '-d'], None)
  265. client = docker_client(version='1.21')
  266. networks = client.networks(names=[self.project.name])
  267. self.assertEqual(len(networks), 0)
  268. for service in self.project.get_services():
  269. containers = service.containers()
  270. self.assertEqual(len(containers), 1)
  271. self.assertNotEqual(containers[0].get('Config.Hostname'), service.name)
  272. web_container = self.project.get_service('web').containers()[0]
  273. self.assertTrue(web_container.get('HostConfig.Links'))
  274. def test_up_with_networking(self):
  275. self.require_api_version('1.21')
  276. self.base_dir = 'tests/fixtures/links-composefile'
  277. self.dispatch(['--x-networking', 'up', '-d'], None)
  278. client = docker_client(version='1.21')
  279. services = self.project.get_services()
  280. networks = client.networks(names=[self.project.name])
  281. for n in networks:
  282. self.addCleanup(client.remove_network, n['Id'])
  283. self.assertEqual(len(networks), 1)
  284. self.assertEqual(networks[0]['Driver'], 'bridge')
  285. network = client.inspect_network(networks[0]['Id'])
  286. self.assertEqual(len(network['Containers']), len(services))
  287. for service in services:
  288. containers = service.containers()
  289. self.assertEqual(len(containers), 1)
  290. self.assertIn(containers[0].id, network['Containers'])
  291. web_container = self.project.get_service('web').containers()[0]
  292. self.assertFalse(web_container.get('HostConfig.Links'))
  293. def test_up_with_links(self):
  294. self.base_dir = 'tests/fixtures/links-composefile'
  295. self.dispatch(['up', '-d', 'web'], None)
  296. web = self.project.get_service('web')
  297. db = self.project.get_service('db')
  298. console = self.project.get_service('console')
  299. self.assertEqual(len(web.containers()), 1)
  300. self.assertEqual(len(db.containers()), 1)
  301. self.assertEqual(len(console.containers()), 0)
  302. def test_up_with_no_deps(self):
  303. self.base_dir = 'tests/fixtures/links-composefile'
  304. self.dispatch(['up', '-d', '--no-deps', 'web'], None)
  305. web = self.project.get_service('web')
  306. db = self.project.get_service('db')
  307. console = self.project.get_service('console')
  308. self.assertEqual(len(web.containers()), 1)
  309. self.assertEqual(len(db.containers()), 0)
  310. self.assertEqual(len(console.containers()), 0)
  311. def test_up_with_force_recreate(self):
  312. self.dispatch(['up', '-d'], None)
  313. service = self.project.get_service('simple')
  314. self.assertEqual(len(service.containers()), 1)
  315. old_ids = [c.id for c in service.containers()]
  316. self.dispatch(['up', '-d', '--force-recreate'], None)
  317. self.assertEqual(len(service.containers()), 1)
  318. new_ids = [c.id for c in service.containers()]
  319. self.assertNotEqual(old_ids, new_ids)
  320. def test_up_with_no_recreate(self):
  321. self.dispatch(['up', '-d'], None)
  322. service = self.project.get_service('simple')
  323. self.assertEqual(len(service.containers()), 1)
  324. old_ids = [c.id for c in service.containers()]
  325. self.dispatch(['up', '-d', '--no-recreate'], None)
  326. self.assertEqual(len(service.containers()), 1)
  327. new_ids = [c.id for c in service.containers()]
  328. self.assertEqual(old_ids, new_ids)
  329. def test_up_with_force_recreate_and_no_recreate(self):
  330. self.dispatch(
  331. ['up', '-d', '--force-recreate', '--no-recreate'],
  332. returncode=1)
  333. def test_up_with_timeout(self):
  334. self.dispatch(['up', '-d', '-t', '1'])
  335. service = self.project.get_service('simple')
  336. another = self.project.get_service('another')
  337. self.assertEqual(len(service.containers()), 1)
  338. self.assertEqual(len(another.containers()), 1)
  339. # Ensure containers don't have stdin and stdout connected in -d mode
  340. config = service.containers()[0].inspect()['Config']
  341. self.assertFalse(config['AttachStderr'])
  342. self.assertFalse(config['AttachStdout'])
  343. self.assertFalse(config['AttachStdin'])
  344. def test_up_handles_sigint(self):
  345. proc = start_process(self.base_dir, ['up', '-t', '2'])
  346. wait_on_condition(ContainerCountCondition(self.project, 2))
  347. os.kill(proc.pid, signal.SIGINT)
  348. wait_on_condition(ContainerCountCondition(self.project, 0))
  349. def test_up_handles_sigterm(self):
  350. proc = start_process(self.base_dir, ['up', '-t', '2'])
  351. wait_on_condition(ContainerCountCondition(self.project, 2))
  352. os.kill(proc.pid, signal.SIGTERM)
  353. wait_on_condition(ContainerCountCondition(self.project, 0))
  354. def test_run_service_without_links(self):
  355. self.base_dir = 'tests/fixtures/links-composefile'
  356. self.dispatch(['run', 'console', '/bin/true'])
  357. self.assertEqual(len(self.project.containers()), 0)
  358. # Ensure stdin/out was open
  359. container = self.project.containers(stopped=True, one_off=True)[0]
  360. config = container.inspect()['Config']
  361. self.assertTrue(config['AttachStderr'])
  362. self.assertTrue(config['AttachStdout'])
  363. self.assertTrue(config['AttachStdin'])
  364. def test_run_service_with_links(self):
  365. self.base_dir = 'tests/fixtures/links-composefile'
  366. self.dispatch(['run', 'web', '/bin/true'], None)
  367. db = self.project.get_service('db')
  368. console = self.project.get_service('console')
  369. self.assertEqual(len(db.containers()), 1)
  370. self.assertEqual(len(console.containers()), 0)
  371. def test_run_with_no_deps(self):
  372. self.base_dir = 'tests/fixtures/links-composefile'
  373. self.dispatch(['run', '--no-deps', 'web', '/bin/true'])
  374. db = self.project.get_service('db')
  375. self.assertEqual(len(db.containers()), 0)
  376. def test_run_does_not_recreate_linked_containers(self):
  377. self.base_dir = 'tests/fixtures/links-composefile'
  378. self.dispatch(['up', '-d', 'db'])
  379. db = self.project.get_service('db')
  380. self.assertEqual(len(db.containers()), 1)
  381. old_ids = [c.id for c in db.containers()]
  382. self.dispatch(['run', 'web', '/bin/true'], None)
  383. self.assertEqual(len(db.containers()), 1)
  384. new_ids = [c.id for c in db.containers()]
  385. self.assertEqual(old_ids, new_ids)
  386. def test_run_without_command(self):
  387. self.base_dir = 'tests/fixtures/commands-composefile'
  388. self.check_build('tests/fixtures/simple-dockerfile', tag='composetest_test')
  389. self.dispatch(['run', 'implicit'])
  390. service = self.project.get_service('implicit')
  391. containers = service.containers(stopped=True, one_off=True)
  392. self.assertEqual(
  393. [c.human_readable_command for c in containers],
  394. [u'/bin/sh -c echo "success"'],
  395. )
  396. self.dispatch(['run', 'explicit'])
  397. service = self.project.get_service('explicit')
  398. containers = service.containers(stopped=True, one_off=True)
  399. self.assertEqual(
  400. [c.human_readable_command for c in containers],
  401. [u'/bin/true'],
  402. )
  403. def test_run_service_with_entrypoint_overridden(self):
  404. self.base_dir = 'tests/fixtures/dockerfile_with_entrypoint'
  405. name = 'service'
  406. self.dispatch(['run', '--entrypoint', '/bin/echo', name, 'helloworld'])
  407. service = self.project.get_service(name)
  408. container = service.containers(stopped=True, one_off=True)[0]
  409. self.assertEqual(
  410. shlex.split(container.human_readable_command),
  411. [u'/bin/echo', u'helloworld'],
  412. )
  413. def test_run_service_with_user_overridden(self):
  414. self.base_dir = 'tests/fixtures/user-composefile'
  415. name = 'service'
  416. user = 'sshd'
  417. self.dispatch(['run', '--user={user}'.format(user=user), name], returncode=1)
  418. service = self.project.get_service(name)
  419. container = service.containers(stopped=True, one_off=True)[0]
  420. self.assertEqual(user, container.get('Config.User'))
  421. def test_run_service_with_user_overridden_short_form(self):
  422. self.base_dir = 'tests/fixtures/user-composefile'
  423. name = 'service'
  424. user = 'sshd'
  425. self.dispatch(['run', '-u', user, name], returncode=1)
  426. service = self.project.get_service(name)
  427. container = service.containers(stopped=True, one_off=True)[0]
  428. self.assertEqual(user, container.get('Config.User'))
  429. def test_run_service_with_environement_overridden(self):
  430. name = 'service'
  431. self.base_dir = 'tests/fixtures/environment-composefile'
  432. self.dispatch([
  433. 'run', '-e', 'foo=notbar',
  434. '-e', 'allo=moto=bobo',
  435. '-e', 'alpha=beta',
  436. name,
  437. '/bin/true',
  438. ])
  439. service = self.project.get_service(name)
  440. container = service.containers(stopped=True, one_off=True)[0]
  441. # env overriden
  442. self.assertEqual('notbar', container.environment['foo'])
  443. # keep environement from yaml
  444. self.assertEqual('world', container.environment['hello'])
  445. # added option from command line
  446. self.assertEqual('beta', container.environment['alpha'])
  447. # make sure a value with a = don't crash out
  448. self.assertEqual('moto=bobo', container.environment['allo'])
  449. def test_run_service_without_map_ports(self):
  450. # create one off container
  451. self.base_dir = 'tests/fixtures/ports-composefile'
  452. self.dispatch(['run', '-d', 'simple'])
  453. container = self.project.get_service('simple').containers(one_off=True)[0]
  454. # get port information
  455. port_random = container.get_local_port(3000)
  456. port_assigned = container.get_local_port(3001)
  457. # close all one off containers we just created
  458. container.stop()
  459. # check the ports
  460. self.assertEqual(port_random, None)
  461. self.assertEqual(port_assigned, None)
  462. def test_run_service_with_map_ports(self):
  463. # create one off container
  464. self.base_dir = 'tests/fixtures/ports-composefile'
  465. self.dispatch(['run', '-d', '--service-ports', 'simple'])
  466. container = self.project.get_service('simple').containers(one_off=True)[0]
  467. # get port information
  468. port_random = container.get_local_port(3000)
  469. port_assigned = container.get_local_port(3001)
  470. port_range = container.get_local_port(3002), container.get_local_port(3003)
  471. # close all one off containers we just created
  472. container.stop()
  473. # check the ports
  474. self.assertNotEqual(port_random, None)
  475. self.assertIn("0.0.0.0", port_random)
  476. self.assertEqual(port_assigned, "0.0.0.0:49152")
  477. self.assertEqual(port_range[0], "0.0.0.0:49153")
  478. self.assertEqual(port_range[1], "0.0.0.0:49154")
  479. def test_run_service_with_explicitly_maped_ports(self):
  480. # create one off container
  481. self.base_dir = 'tests/fixtures/ports-composefile'
  482. self.dispatch(['run', '-d', '-p', '30000:3000', '--publish', '30001:3001', 'simple'])
  483. container = self.project.get_service('simple').containers(one_off=True)[0]
  484. # get port information
  485. port_short = container.get_local_port(3000)
  486. port_full = container.get_local_port(3001)
  487. # close all one off containers we just created
  488. container.stop()
  489. # check the ports
  490. self.assertEqual(port_short, "0.0.0.0:30000")
  491. self.assertEqual(port_full, "0.0.0.0:30001")
  492. def test_run_service_with_explicitly_maped_ip_ports(self):
  493. # create one off container
  494. self.base_dir = 'tests/fixtures/ports-composefile'
  495. self.dispatch(['run', '-d', '-p', '127.0.0.1:30000:3000', '--publish', '127.0.0.1:30001:3001', 'simple'], None)
  496. container = self.project.get_service('simple').containers(one_off=True)[0]
  497. # get port information
  498. port_short = container.get_local_port(3000)
  499. port_full = container.get_local_port(3001)
  500. # close all one off containers we just created
  501. container.stop()
  502. # check the ports
  503. self.assertEqual(port_short, "127.0.0.1:30000")
  504. self.assertEqual(port_full, "127.0.0.1:30001")
  505. def test_run_with_expose_ports(self):
  506. # create one off container
  507. self.base_dir = 'tests/fixtures/expose-composefile'
  508. self.dispatch(['run', '-d', '--service-ports', 'simple'])
  509. container = self.project.get_service('simple').containers(one_off=True)[0]
  510. ports = container.ports
  511. self.assertEqual(len(ports), 9)
  512. # exposed ports are not mapped to host ports
  513. assert ports['3000/tcp'] is None
  514. assert ports['3001/tcp'] is None
  515. assert ports['3001/udp'] is None
  516. assert ports['3002/tcp'] is None
  517. assert ports['3003/tcp'] is None
  518. assert ports['3004/tcp'] is None
  519. assert ports['3005/tcp'] is None
  520. assert ports['3006/udp'] is None
  521. assert ports['3007/udp'] is None
  522. # close all one off containers we just created
  523. container.stop()
  524. def test_run_with_custom_name(self):
  525. self.base_dir = 'tests/fixtures/environment-composefile'
  526. name = 'the-container-name'
  527. self.dispatch(['run', '--name', name, 'service', '/bin/true'])
  528. service = self.project.get_service('service')
  529. container, = service.containers(stopped=True, one_off=True)
  530. self.assertEqual(container.name, name)
  531. def test_run_with_networking(self):
  532. self.require_api_version('1.21')
  533. client = docker_client(version='1.21')
  534. self.base_dir = 'tests/fixtures/simple-dockerfile'
  535. self.dispatch(['--x-networking', 'run', 'simple', 'true'], None)
  536. service = self.project.get_service('simple')
  537. container, = service.containers(stopped=True, one_off=True)
  538. networks = client.networks(names=[self.project.name])
  539. for n in networks:
  540. self.addCleanup(client.remove_network, n['Id'])
  541. self.assertEqual(len(networks), 1)
  542. self.assertEqual(container.human_readable_command, u'true')
  543. def test_run_handles_sigint(self):
  544. proc = start_process(self.base_dir, ['run', '-T', 'simple', 'top'])
  545. wait_on_condition(ContainerStateCondition(
  546. self.project.client,
  547. 'simplecomposefile_simple_run_1',
  548. running=True))
  549. os.kill(proc.pid, signal.SIGINT)
  550. wait_on_condition(ContainerStateCondition(
  551. self.project.client,
  552. 'simplecomposefile_simple_run_1',
  553. running=False))
  554. def test_run_handles_sigterm(self):
  555. proc = start_process(self.base_dir, ['run', '-T', 'simple', 'top'])
  556. wait_on_condition(ContainerStateCondition(
  557. self.project.client,
  558. 'simplecomposefile_simple_run_1',
  559. running=True))
  560. os.kill(proc.pid, signal.SIGTERM)
  561. wait_on_condition(ContainerStateCondition(
  562. self.project.client,
  563. 'simplecomposefile_simple_run_1',
  564. running=False))
  565. def test_rm(self):
  566. service = self.project.get_service('simple')
  567. service.create_container()
  568. kill_service(service)
  569. self.assertEqual(len(service.containers(stopped=True)), 1)
  570. self.dispatch(['rm', '--force'], None)
  571. self.assertEqual(len(service.containers(stopped=True)), 0)
  572. service = self.project.get_service('simple')
  573. service.create_container()
  574. kill_service(service)
  575. self.assertEqual(len(service.containers(stopped=True)), 1)
  576. self.dispatch(['rm', '-f'], None)
  577. self.assertEqual(len(service.containers(stopped=True)), 0)
  578. def test_stop(self):
  579. self.dispatch(['up', '-d'], None)
  580. service = self.project.get_service('simple')
  581. self.assertEqual(len(service.containers()), 1)
  582. self.assertTrue(service.containers()[0].is_running)
  583. self.dispatch(['stop', '-t', '1'], None)
  584. self.assertEqual(len(service.containers(stopped=True)), 1)
  585. self.assertFalse(service.containers(stopped=True)[0].is_running)
  586. def test_start_no_containers(self):
  587. result = self.dispatch(['start'], returncode=1)
  588. assert 'No containers to start' in result.stderr
  589. def test_up_logging(self):
  590. self.base_dir = 'tests/fixtures/logging-composefile'
  591. self.dispatch(['up', '-d'])
  592. simple = self.project.get_service('simple').containers()[0]
  593. log_config = simple.get('HostConfig.LogConfig')
  594. self.assertTrue(log_config)
  595. self.assertEqual(log_config.get('Type'), 'none')
  596. another = self.project.get_service('another').containers()[0]
  597. log_config = another.get('HostConfig.LogConfig')
  598. self.assertTrue(log_config)
  599. self.assertEqual(log_config.get('Type'), 'json-file')
  600. self.assertEqual(log_config.get('Config')['max-size'], '10m')
  601. def test_up_logging_legacy(self):
  602. self.base_dir = 'tests/fixtures/logging-composefile-legacy'
  603. self.dispatch(['up', '-d'])
  604. simple = self.project.get_service('simple').containers()[0]
  605. log_config = simple.get('HostConfig.LogConfig')
  606. self.assertTrue(log_config)
  607. self.assertEqual(log_config.get('Type'), 'none')
  608. another = self.project.get_service('another').containers()[0]
  609. log_config = another.get('HostConfig.LogConfig')
  610. self.assertTrue(log_config)
  611. self.assertEqual(log_config.get('Type'), 'json-file')
  612. self.assertEqual(log_config.get('Config')['max-size'], '10m')
  613. def test_pause_unpause(self):
  614. self.dispatch(['up', '-d'], None)
  615. service = self.project.get_service('simple')
  616. self.assertFalse(service.containers()[0].is_paused)
  617. self.dispatch(['pause'], None)
  618. self.assertTrue(service.containers()[0].is_paused)
  619. self.dispatch(['unpause'], None)
  620. self.assertFalse(service.containers()[0].is_paused)
  621. def test_pause_no_containers(self):
  622. result = self.dispatch(['pause'], returncode=1)
  623. assert 'No containers to pause' in result.stderr
  624. def test_unpause_no_containers(self):
  625. result = self.dispatch(['unpause'], returncode=1)
  626. assert 'No containers to unpause' in result.stderr
  627. def test_logs_invalid_service_name(self):
  628. self.dispatch(['logs', 'madeupname'], returncode=1)
  629. def test_kill(self):
  630. self.dispatch(['up', '-d'], None)
  631. service = self.project.get_service('simple')
  632. self.assertEqual(len(service.containers()), 1)
  633. self.assertTrue(service.containers()[0].is_running)
  634. self.dispatch(['kill'], None)
  635. self.assertEqual(len(service.containers(stopped=True)), 1)
  636. self.assertFalse(service.containers(stopped=True)[0].is_running)
  637. def test_kill_signal_sigstop(self):
  638. self.dispatch(['up', '-d'], None)
  639. service = self.project.get_service('simple')
  640. self.assertEqual(len(service.containers()), 1)
  641. self.assertTrue(service.containers()[0].is_running)
  642. self.dispatch(['kill', '-s', 'SIGSTOP'], None)
  643. self.assertEqual(len(service.containers()), 1)
  644. # The container is still running. It has only been paused
  645. self.assertTrue(service.containers()[0].is_running)
  646. def test_kill_stopped_service(self):
  647. self.dispatch(['up', '-d'], None)
  648. service = self.project.get_service('simple')
  649. self.dispatch(['kill', '-s', 'SIGSTOP'], None)
  650. self.assertTrue(service.containers()[0].is_running)
  651. self.dispatch(['kill', '-s', 'SIGKILL'], None)
  652. self.assertEqual(len(service.containers(stopped=True)), 1)
  653. self.assertFalse(service.containers(stopped=True)[0].is_running)
  654. def test_restart(self):
  655. service = self.project.get_service('simple')
  656. container = service.create_container()
  657. container.start()
  658. started_at = container.dictionary['State']['StartedAt']
  659. self.dispatch(['restart', '-t', '1'], None)
  660. container.inspect()
  661. self.assertNotEqual(
  662. container.dictionary['State']['FinishedAt'],
  663. '0001-01-01T00:00:00Z',
  664. )
  665. self.assertNotEqual(
  666. container.dictionary['State']['StartedAt'],
  667. started_at,
  668. )
  669. def test_restart_stopped_container(self):
  670. service = self.project.get_service('simple')
  671. container = service.create_container()
  672. container.start()
  673. container.kill()
  674. self.assertEqual(len(service.containers(stopped=True)), 1)
  675. self.dispatch(['restart', '-t', '1'], None)
  676. self.assertEqual(len(service.containers(stopped=False)), 1)
  677. def test_restart_no_containers(self):
  678. result = self.dispatch(['restart'], returncode=1)
  679. assert 'No containers to restart' in result.stderr
  680. def test_scale(self):
  681. project = self.project
  682. self.dispatch(['scale', 'simple=1'])
  683. self.assertEqual(len(project.get_service('simple').containers()), 1)
  684. self.dispatch(['scale', 'simple=3', 'another=2'])
  685. self.assertEqual(len(project.get_service('simple').containers()), 3)
  686. self.assertEqual(len(project.get_service('another').containers()), 2)
  687. self.dispatch(['scale', 'simple=1', 'another=1'])
  688. self.assertEqual(len(project.get_service('simple').containers()), 1)
  689. self.assertEqual(len(project.get_service('another').containers()), 1)
  690. self.dispatch(['scale', 'simple=1', 'another=1'])
  691. self.assertEqual(len(project.get_service('simple').containers()), 1)
  692. self.assertEqual(len(project.get_service('another').containers()), 1)
  693. self.dispatch(['scale', 'simple=0', 'another=0'])
  694. self.assertEqual(len(project.get_service('simple').containers()), 0)
  695. self.assertEqual(len(project.get_service('another').containers()), 0)
  696. def test_port(self):
  697. self.base_dir = 'tests/fixtures/ports-composefile'
  698. self.dispatch(['up', '-d'], None)
  699. container = self.project.get_service('simple').get_container()
  700. def get_port(number):
  701. result = self.dispatch(['port', 'simple', str(number)])
  702. return result.stdout.rstrip()
  703. self.assertEqual(get_port(3000), container.get_local_port(3000))
  704. self.assertEqual(get_port(3001), "0.0.0.0:49152")
  705. self.assertEqual(get_port(3002), "0.0.0.0:49153")
  706. def test_port_with_scale(self):
  707. self.base_dir = 'tests/fixtures/ports-composefile-scale'
  708. self.dispatch(['scale', 'simple=2'], None)
  709. containers = sorted(
  710. self.project.containers(service_names=['simple']),
  711. key=attrgetter('name'))
  712. def get_port(number, index=None):
  713. if index is None:
  714. result = self.dispatch(['port', 'simple', str(number)])
  715. else:
  716. result = self.dispatch(['port', '--index=' + str(index), 'simple', str(number)])
  717. return result.stdout.rstrip()
  718. self.assertEqual(get_port(3000), containers[0].get_local_port(3000))
  719. self.assertEqual(get_port(3000, index=1), containers[0].get_local_port(3000))
  720. self.assertEqual(get_port(3000, index=2), containers[1].get_local_port(3000))
  721. self.assertEqual(get_port(3002), "")
  722. def test_events_json(self):
  723. events_proc = start_process(self.base_dir, ['events', '--json'])
  724. self.dispatch(['up', '-d'])
  725. wait_on_condition(ContainerCountCondition(self.project, 2))
  726. os.kill(events_proc.pid, signal.SIGINT)
  727. result = wait_on_process(events_proc, returncode=1)
  728. lines = [json.loads(line) for line in result.stdout.rstrip().split('\n')]
  729. assert [e['action'] for e in lines] == ['create', 'start', 'create', 'start']
  730. def test_events_human_readable(self):
  731. events_proc = start_process(self.base_dir, ['events'])
  732. self.dispatch(['up', '-d', 'simple'])
  733. wait_on_condition(ContainerCountCondition(self.project, 1))
  734. os.kill(events_proc.pid, signal.SIGINT)
  735. result = wait_on_process(events_proc, returncode=1)
  736. lines = result.stdout.rstrip().split('\n')
  737. assert len(lines) == 2
  738. container, = self.project.containers()
  739. expected_template = (
  740. ' container {} {} (image=busybox:latest, '
  741. 'name=simplecomposefile_simple_1)')
  742. assert expected_template.format('create', container.id) in lines[0]
  743. assert expected_template.format('start', container.id) in lines[1]
  744. assert lines[0].startswith(datetime.date.today().isoformat())
  745. def test_env_file_relative_to_compose_file(self):
  746. config_path = os.path.abspath('tests/fixtures/env-file/docker-compose.yml')
  747. self.dispatch(['-f', config_path, 'up', '-d'], None)
  748. self._project = get_project(self.base_dir, [config_path])
  749. containers = self.project.containers(stopped=True)
  750. self.assertEqual(len(containers), 1)
  751. self.assertIn("FOO=1", containers[0].get('Config.Env'))
  752. @mock.patch.dict(os.environ)
  753. def test_home_and_env_var_in_volume_path(self):
  754. os.environ['VOLUME_NAME'] = 'my-volume'
  755. os.environ['HOME'] = '/tmp/home-dir'
  756. self.base_dir = 'tests/fixtures/volume-path-interpolation'
  757. self.dispatch(['up', '-d'], None)
  758. container = self.project.containers(stopped=True)[0]
  759. actual_host_path = container.get_mount('/container-path')['Source']
  760. components = actual_host_path.split('/')
  761. assert components[-2:] == ['home-dir', 'my-volume']
  762. def test_up_with_default_override_file(self):
  763. self.base_dir = 'tests/fixtures/override-files'
  764. self.dispatch(['up', '-d'], None)
  765. containers = self.project.containers()
  766. self.assertEqual(len(containers), 2)
  767. web, db = containers
  768. self.assertEqual(web.human_readable_command, 'top')
  769. self.assertEqual(db.human_readable_command, 'top')
  770. def test_up_with_multiple_files(self):
  771. self.base_dir = 'tests/fixtures/override-files'
  772. config_paths = [
  773. 'docker-compose.yml',
  774. 'docker-compose.override.yml',
  775. 'extra.yml',
  776. ]
  777. self._project = get_project(self.base_dir, config_paths)
  778. self.dispatch(
  779. [
  780. '-f', config_paths[0],
  781. '-f', config_paths[1],
  782. '-f', config_paths[2],
  783. 'up', '-d',
  784. ],
  785. None)
  786. containers = self.project.containers()
  787. self.assertEqual(len(containers), 3)
  788. web, other, db = containers
  789. self.assertEqual(web.human_readable_command, 'top')
  790. self.assertTrue({'db', 'other'} <= set(get_links(web)))
  791. self.assertEqual(db.human_readable_command, 'top')
  792. self.assertEqual(other.human_readable_command, 'top')
  793. def test_up_with_extends(self):
  794. self.base_dir = 'tests/fixtures/extends'
  795. self.dispatch(['up', '-d'], None)
  796. self.assertEqual(
  797. set([s.name for s in self.project.services]),
  798. set(['mydb', 'myweb']),
  799. )
  800. # Sort by name so we get [db, web]
  801. containers = sorted(
  802. self.project.containers(stopped=True),
  803. key=lambda c: c.name,
  804. )
  805. self.assertEqual(len(containers), 2)
  806. web = containers[1]
  807. self.assertEqual(
  808. set(get_links(web)),
  809. set(['db', 'mydb_1', 'extends_mydb_1']))
  810. expected_env = set([
  811. "FOO=1",
  812. "BAR=2",
  813. "BAZ=2",
  814. ])
  815. self.assertTrue(expected_env <= set(web.get('Config.Env')))