cli_test.py 100 KB

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