service_test.py 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import os
  4. import shutil
  5. import tempfile
  6. from os import path
  7. from docker.errors import APIError
  8. from six import StringIO
  9. from six import text_type
  10. from .. import mock
  11. from .testcases import DockerClientTestCase
  12. from .testcases import pull_busybox
  13. from compose import __version__
  14. from compose.const import LABEL_CONTAINER_NUMBER
  15. from compose.const import LABEL_ONE_OFF
  16. from compose.const import LABEL_PROJECT
  17. from compose.const import LABEL_SERVICE
  18. from compose.const import LABEL_VERSION
  19. from compose.container import Container
  20. from compose.service import build_extra_hosts
  21. from compose.service import ConfigError
  22. from compose.service import ConvergencePlan
  23. from compose.service import Net
  24. from compose.service import Service
  25. from compose.service import VolumeFromSpec
  26. def create_and_start_container(service, **override_options):
  27. container = service.create_container(**override_options)
  28. container.start()
  29. return container
  30. class ServiceTest(DockerClientTestCase):
  31. def test_containers(self):
  32. foo = self.create_service('foo')
  33. bar = self.create_service('bar')
  34. create_and_start_container(foo)
  35. self.assertEqual(len(foo.containers()), 1)
  36. self.assertEqual(foo.containers()[0].name, 'composetest_foo_1')
  37. self.assertEqual(len(bar.containers()), 0)
  38. create_and_start_container(bar)
  39. create_and_start_container(bar)
  40. self.assertEqual(len(foo.containers()), 1)
  41. self.assertEqual(len(bar.containers()), 2)
  42. names = [c.name for c in bar.containers()]
  43. self.assertIn('composetest_bar_1', names)
  44. self.assertIn('composetest_bar_2', names)
  45. def test_containers_one_off(self):
  46. db = self.create_service('db')
  47. container = db.create_container(one_off=True)
  48. self.assertEqual(db.containers(stopped=True), [])
  49. self.assertEqual(db.containers(one_off=True, stopped=True), [container])
  50. def test_project_is_added_to_container_name(self):
  51. service = self.create_service('web')
  52. create_and_start_container(service)
  53. self.assertEqual(service.containers()[0].name, 'composetest_web_1')
  54. def test_start_stop(self):
  55. service = self.create_service('scalingtest')
  56. self.assertEqual(len(service.containers(stopped=True)), 0)
  57. service.create_container()
  58. self.assertEqual(len(service.containers()), 0)
  59. self.assertEqual(len(service.containers(stopped=True)), 1)
  60. service.start()
  61. self.assertEqual(len(service.containers()), 1)
  62. self.assertEqual(len(service.containers(stopped=True)), 1)
  63. service.stop(timeout=1)
  64. self.assertEqual(len(service.containers()), 0)
  65. self.assertEqual(len(service.containers(stopped=True)), 1)
  66. service.stop(timeout=1)
  67. self.assertEqual(len(service.containers()), 0)
  68. self.assertEqual(len(service.containers(stopped=True)), 1)
  69. def test_kill_remove(self):
  70. service = self.create_service('scalingtest')
  71. create_and_start_container(service)
  72. self.assertEqual(len(service.containers()), 1)
  73. service.remove_stopped()
  74. self.assertEqual(len(service.containers()), 1)
  75. service.kill()
  76. self.assertEqual(len(service.containers()), 0)
  77. self.assertEqual(len(service.containers(stopped=True)), 1)
  78. service.remove_stopped()
  79. self.assertEqual(len(service.containers(stopped=True)), 0)
  80. def test_create_container_with_one_off(self):
  81. db = self.create_service('db')
  82. container = db.create_container(one_off=True)
  83. self.assertEqual(container.name, 'composetest_db_run_1')
  84. def test_create_container_with_one_off_when_existing_container_is_running(self):
  85. db = self.create_service('db')
  86. db.start()
  87. container = db.create_container(one_off=True)
  88. self.assertEqual(container.name, 'composetest_db_run_1')
  89. def test_create_container_with_unspecified_volume(self):
  90. service = self.create_service('db', volumes=['/var/db'])
  91. container = service.create_container()
  92. container.start()
  93. self.assertIn('/var/db', container.get('Volumes'))
  94. def test_create_container_with_volume_driver(self):
  95. service = self.create_service('db', volume_driver='foodriver')
  96. container = service.create_container()
  97. container.start()
  98. self.assertEqual('foodriver', container.get('Config.VolumeDriver'))
  99. def test_create_container_with_cpu_shares(self):
  100. service = self.create_service('db', cpu_shares=73)
  101. container = service.create_container()
  102. container.start()
  103. self.assertEqual(container.get('HostConfig.CpuShares'), 73)
  104. def test_build_extra_hosts(self):
  105. # string
  106. self.assertRaises(ConfigError, lambda: build_extra_hosts("www.example.com: 192.168.0.17"))
  107. # list of strings
  108. self.assertEqual(build_extra_hosts(
  109. ["www.example.com:192.168.0.17"]),
  110. {'www.example.com': '192.168.0.17'})
  111. self.assertEqual(build_extra_hosts(
  112. ["www.example.com: 192.168.0.17"]),
  113. {'www.example.com': '192.168.0.17'})
  114. self.assertEqual(build_extra_hosts(
  115. ["www.example.com: 192.168.0.17",
  116. "static.example.com:192.168.0.19",
  117. "api.example.com: 192.168.0.18"]),
  118. {'www.example.com': '192.168.0.17',
  119. 'static.example.com': '192.168.0.19',
  120. 'api.example.com': '192.168.0.18'})
  121. # list of dictionaries
  122. self.assertRaises(ConfigError, lambda: build_extra_hosts(
  123. [{'www.example.com': '192.168.0.17'},
  124. {'api.example.com': '192.168.0.18'}]))
  125. # dictionaries
  126. self.assertEqual(build_extra_hosts(
  127. {'www.example.com': '192.168.0.17',
  128. 'api.example.com': '192.168.0.18'}),
  129. {'www.example.com': '192.168.0.17',
  130. 'api.example.com': '192.168.0.18'})
  131. def test_create_container_with_extra_hosts_list(self):
  132. extra_hosts = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
  133. service = self.create_service('db', extra_hosts=extra_hosts)
  134. container = service.create_container()
  135. container.start()
  136. self.assertEqual(set(container.get('HostConfig.ExtraHosts')), set(extra_hosts))
  137. def test_create_container_with_extra_hosts_dicts(self):
  138. extra_hosts = {'somehost': '162.242.195.82', 'otherhost': '50.31.209.229'}
  139. extra_hosts_list = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
  140. service = self.create_service('db', extra_hosts=extra_hosts)
  141. container = service.create_container()
  142. container.start()
  143. self.assertEqual(set(container.get('HostConfig.ExtraHosts')), set(extra_hosts_list))
  144. def test_create_container_with_cpu_set(self):
  145. service = self.create_service('db', cpuset='0')
  146. container = service.create_container()
  147. container.start()
  148. self.assertEqual(container.get('HostConfig.CpusetCpus'), '0')
  149. def test_create_container_with_read_only_root_fs(self):
  150. read_only = True
  151. service = self.create_service('db', read_only=read_only)
  152. container = service.create_container()
  153. container.start()
  154. self.assertEqual(container.get('HostConfig.ReadonlyRootfs'), read_only, container.get('HostConfig'))
  155. def test_create_container_with_security_opt(self):
  156. security_opt = ['label:disable']
  157. service = self.create_service('db', security_opt=security_opt)
  158. container = service.create_container()
  159. container.start()
  160. self.assertEqual(set(container.get('HostConfig.SecurityOpt')), set(security_opt))
  161. def test_create_container_with_mac_address(self):
  162. service = self.create_service('db', mac_address='02:42:ac:11:65:43')
  163. container = service.create_container()
  164. container.start()
  165. self.assertEqual(container.inspect()['Config']['MacAddress'], '02:42:ac:11:65:43')
  166. def test_create_container_with_specified_volume(self):
  167. host_path = '/tmp/host-path'
  168. container_path = '/container-path'
  169. service = self.create_service('db', volumes=['%s:%s' % (host_path, container_path)])
  170. container = service.create_container()
  171. container.start()
  172. volumes = container.inspect()['Volumes']
  173. self.assertIn(container_path, volumes)
  174. # Match the last component ("host-path"), because boot2docker symlinks /tmp
  175. actual_host_path = volumes[container_path]
  176. self.assertTrue(path.basename(actual_host_path) == path.basename(host_path),
  177. msg=("Last component differs: %s, %s" % (actual_host_path, host_path)))
  178. def test_recreate_preserves_volume_with_trailing_slash(self):
  179. """
  180. When the Compose file specifies a trailing slash in the container path, make
  181. sure we copy the volume over when recreating.
  182. """
  183. service = self.create_service('data', volumes=['/data/'])
  184. old_container = create_and_start_container(service)
  185. volume_path = old_container.get('Volumes')['/data']
  186. new_container = service.recreate_container(old_container)
  187. self.assertEqual(new_container.get('Volumes')['/data'], volume_path)
  188. def test_duplicate_volume_trailing_slash(self):
  189. """
  190. When an image specifies a volume, and the Compose file specifies a host path
  191. but adds a trailing slash, make sure that we don't create duplicate binds.
  192. """
  193. host_path = '/tmp/data'
  194. container_path = '/data'
  195. volumes = ['{}:{}/'.format(host_path, container_path)]
  196. tmp_container = self.client.create_container(
  197. 'busybox', 'true',
  198. volumes={container_path: {}},
  199. labels={'com.docker.compose.test_image': 'true'},
  200. )
  201. image = self.client.commit(tmp_container)['Id']
  202. service = self.create_service('db', image=image, volumes=volumes)
  203. old_container = create_and_start_container(service)
  204. self.assertEqual(
  205. old_container.get('Config.Volumes'),
  206. {container_path: {}},
  207. )
  208. service = self.create_service('db', image=image, volumes=volumes)
  209. new_container = service.recreate_container(old_container)
  210. self.assertEqual(
  211. new_container.get('Config.Volumes'),
  212. {container_path: {}},
  213. )
  214. self.assertEqual(service.containers(stopped=False), [new_container])
  215. def test_create_container_with_volumes_from(self):
  216. volume_service = self.create_service('data')
  217. volume_container_1 = volume_service.create_container()
  218. volume_container_2 = Container.create(
  219. self.client,
  220. image='busybox:latest',
  221. command=["top"],
  222. labels={LABEL_PROJECT: 'composetest'},
  223. )
  224. host_service = self.create_service(
  225. 'host',
  226. volumes_from=[
  227. VolumeFromSpec(volume_service, 'rw'),
  228. VolumeFromSpec(volume_container_2, 'rw')
  229. ]
  230. )
  231. host_container = host_service.create_container()
  232. host_container.start()
  233. self.assertIn(volume_container_1.id + ':rw',
  234. host_container.get('HostConfig.VolumesFrom'))
  235. self.assertIn(volume_container_2.id + ':rw',
  236. host_container.get('HostConfig.VolumesFrom'))
  237. def test_execute_convergence_plan_recreate(self):
  238. service = self.create_service(
  239. 'db',
  240. environment={'FOO': '1'},
  241. volumes=['/etc'],
  242. entrypoint=['top'],
  243. command=['-d', '1']
  244. )
  245. old_container = service.create_container()
  246. self.assertEqual(old_container.get('Config.Entrypoint'), ['top'])
  247. self.assertEqual(old_container.get('Config.Cmd'), ['-d', '1'])
  248. self.assertIn('FOO=1', old_container.get('Config.Env'))
  249. self.assertEqual(old_container.name, 'composetest_db_1')
  250. old_container.start()
  251. old_container.inspect() # reload volume data
  252. volume_path = old_container.get('Volumes')['/etc']
  253. num_containers_before = len(self.client.containers(all=True))
  254. service.options['environment']['FOO'] = '2'
  255. new_container, = service.execute_convergence_plan(
  256. ConvergencePlan('recreate', [old_container]))
  257. self.assertEqual(new_container.get('Config.Entrypoint'), ['top'])
  258. self.assertEqual(new_container.get('Config.Cmd'), ['-d', '1'])
  259. self.assertIn('FOO=2', new_container.get('Config.Env'))
  260. self.assertEqual(new_container.name, 'composetest_db_1')
  261. self.assertEqual(new_container.get('Volumes')['/etc'], volume_path)
  262. self.assertIn(
  263. 'affinity:container==%s' % old_container.id,
  264. new_container.get('Config.Env'))
  265. self.assertEqual(len(self.client.containers(all=True)), num_containers_before)
  266. self.assertNotEqual(old_container.id, new_container.id)
  267. self.assertRaises(APIError,
  268. self.client.inspect_container,
  269. old_container.id)
  270. def test_execute_convergence_plan_when_containers_are_stopped(self):
  271. service = self.create_service(
  272. 'db',
  273. environment={'FOO': '1'},
  274. volumes=['/var/db'],
  275. entrypoint=['top'],
  276. command=['-d', '1']
  277. )
  278. service.create_container()
  279. containers = service.containers(stopped=True)
  280. self.assertEqual(len(containers), 1)
  281. container, = containers
  282. self.assertFalse(container.is_running)
  283. service.execute_convergence_plan(ConvergencePlan('start', [container]))
  284. containers = service.containers()
  285. self.assertEqual(len(containers), 1)
  286. container.inspect()
  287. self.assertEqual(container, containers[0])
  288. self.assertTrue(container.is_running)
  289. def test_execute_convergence_plan_with_image_declared_volume(self):
  290. service = Service(
  291. project='composetest',
  292. name='db',
  293. client=self.client,
  294. build='tests/fixtures/dockerfile-with-volume',
  295. )
  296. old_container = create_and_start_container(service)
  297. self.assertEqual(list(old_container.get('Volumes').keys()), ['/data'])
  298. volume_path = old_container.get('Volumes')['/data']
  299. new_container, = service.execute_convergence_plan(
  300. ConvergencePlan('recreate', [old_container]))
  301. self.assertEqual(list(new_container.get('Volumes')), ['/data'])
  302. self.assertEqual(new_container.get('Volumes')['/data'], volume_path)
  303. def test_start_container_passes_through_options(self):
  304. db = self.create_service('db')
  305. create_and_start_container(db, environment={'FOO': 'BAR'})
  306. self.assertEqual(db.containers()[0].environment['FOO'], 'BAR')
  307. def test_start_container_inherits_options_from_constructor(self):
  308. db = self.create_service('db', environment={'FOO': 'BAR'})
  309. create_and_start_container(db)
  310. self.assertEqual(db.containers()[0].environment['FOO'], 'BAR')
  311. def test_start_container_creates_links(self):
  312. db = self.create_service('db')
  313. web = self.create_service('web', links=[(db, None)])
  314. create_and_start_container(db)
  315. create_and_start_container(db)
  316. create_and_start_container(web)
  317. self.assertEqual(
  318. set(web.containers()[0].links()),
  319. set([
  320. 'composetest_db_1', 'db_1',
  321. 'composetest_db_2', 'db_2',
  322. 'db'])
  323. )
  324. def test_start_container_creates_links_with_names(self):
  325. db = self.create_service('db')
  326. web = self.create_service('web', links=[(db, 'custom_link_name')])
  327. create_and_start_container(db)
  328. create_and_start_container(db)
  329. create_and_start_container(web)
  330. self.assertEqual(
  331. set(web.containers()[0].links()),
  332. set([
  333. 'composetest_db_1', 'db_1',
  334. 'composetest_db_2', 'db_2',
  335. 'custom_link_name'])
  336. )
  337. def test_start_container_with_external_links(self):
  338. db = self.create_service('db')
  339. web = self.create_service('web', external_links=['composetest_db_1',
  340. 'composetest_db_2',
  341. 'composetest_db_3:db_3'])
  342. for _ in range(3):
  343. create_and_start_container(db)
  344. create_and_start_container(web)
  345. self.assertEqual(
  346. set(web.containers()[0].links()),
  347. set([
  348. 'composetest_db_1',
  349. 'composetest_db_2',
  350. 'db_3']),
  351. )
  352. def test_start_normal_container_does_not_create_links_to_its_own_service(self):
  353. db = self.create_service('db')
  354. create_and_start_container(db)
  355. create_and_start_container(db)
  356. c = create_and_start_container(db)
  357. self.assertEqual(set(c.links()), set([]))
  358. def test_start_one_off_container_creates_links_to_its_own_service(self):
  359. db = self.create_service('db')
  360. create_and_start_container(db)
  361. create_and_start_container(db)
  362. c = create_and_start_container(db, one_off=True)
  363. self.assertEqual(
  364. set(c.links()),
  365. set([
  366. 'composetest_db_1', 'db_1',
  367. 'composetest_db_2', 'db_2',
  368. 'db'])
  369. )
  370. def test_start_container_builds_images(self):
  371. service = Service(
  372. name='test',
  373. client=self.client,
  374. build='tests/fixtures/simple-dockerfile',
  375. project='composetest',
  376. )
  377. container = create_and_start_container(service)
  378. container.wait()
  379. self.assertIn(b'success', container.logs())
  380. self.assertEqual(len(self.client.images(name='composetest_test')), 1)
  381. def test_start_container_uses_tagged_image_if_it_exists(self):
  382. self.check_build('tests/fixtures/simple-dockerfile', tag='composetest_test')
  383. service = Service(
  384. name='test',
  385. client=self.client,
  386. build='this/does/not/exist/and/will/throw/error',
  387. project='composetest',
  388. )
  389. container = create_and_start_container(service)
  390. container.wait()
  391. self.assertIn(b'success', container.logs())
  392. def test_start_container_creates_ports(self):
  393. service = self.create_service('web', ports=[8000])
  394. container = create_and_start_container(service).inspect()
  395. self.assertEqual(list(container['NetworkSettings']['Ports'].keys()), ['8000/tcp'])
  396. self.assertNotEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8000')
  397. def test_build(self):
  398. base_dir = tempfile.mkdtemp()
  399. self.addCleanup(shutil.rmtree, base_dir)
  400. with open(os.path.join(base_dir, 'Dockerfile'), 'w') as f:
  401. f.write("FROM busybox\n")
  402. self.create_service('web', build=base_dir).build()
  403. self.assertEqual(len(self.client.images(name='composetest_web')), 1)
  404. def test_build_non_ascii_filename(self):
  405. base_dir = tempfile.mkdtemp()
  406. self.addCleanup(shutil.rmtree, base_dir)
  407. with open(os.path.join(base_dir, 'Dockerfile'), 'w') as f:
  408. f.write("FROM busybox\n")
  409. with open(os.path.join(base_dir.encode('utf8'), b'foo\xE2bar'), 'w') as f:
  410. f.write("hello world\n")
  411. self.create_service('web', build=text_type(base_dir)).build()
  412. self.assertEqual(len(self.client.images(name='composetest_web')), 1)
  413. def test_start_container_stays_unpriviliged(self):
  414. service = self.create_service('web')
  415. container = create_and_start_container(service).inspect()
  416. self.assertEqual(container['HostConfig']['Privileged'], False)
  417. def test_start_container_becomes_priviliged(self):
  418. service = self.create_service('web', privileged=True)
  419. container = create_and_start_container(service).inspect()
  420. self.assertEqual(container['HostConfig']['Privileged'], True)
  421. def test_expose_does_not_publish_ports(self):
  422. service = self.create_service('web', expose=["8000"])
  423. container = create_and_start_container(service).inspect()
  424. self.assertEqual(container['NetworkSettings']['Ports'], {'8000/tcp': None})
  425. def test_start_container_creates_port_with_explicit_protocol(self):
  426. service = self.create_service('web', ports=['8000/udp'])
  427. container = create_and_start_container(service).inspect()
  428. self.assertEqual(list(container['NetworkSettings']['Ports'].keys()), ['8000/udp'])
  429. def test_start_container_creates_fixed_external_ports(self):
  430. service = self.create_service('web', ports=['8000:8000'])
  431. container = create_and_start_container(service).inspect()
  432. self.assertIn('8000/tcp', container['NetworkSettings']['Ports'])
  433. self.assertEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8000')
  434. def test_start_container_creates_fixed_external_ports_when_it_is_different_to_internal_port(self):
  435. service = self.create_service('web', ports=['8001:8000'])
  436. container = create_and_start_container(service).inspect()
  437. self.assertIn('8000/tcp', container['NetworkSettings']['Ports'])
  438. self.assertEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8001')
  439. def test_port_with_explicit_interface(self):
  440. service = self.create_service('web', ports=[
  441. '127.0.0.1:8001:8000',
  442. '0.0.0.0:9001:9000/udp',
  443. ])
  444. container = create_and_start_container(service).inspect()
  445. self.assertEqual(container['NetworkSettings']['Ports'], {
  446. '8000/tcp': [
  447. {
  448. 'HostIp': '127.0.0.1',
  449. 'HostPort': '8001',
  450. },
  451. ],
  452. '9000/udp': [
  453. {
  454. 'HostIp': '0.0.0.0',
  455. 'HostPort': '9001',
  456. },
  457. ],
  458. })
  459. def test_create_with_image_id(self):
  460. # Get image id for the current busybox:latest
  461. pull_busybox(self.client)
  462. image_id = self.client.inspect_image('busybox:latest')['Id'][:12]
  463. service = self.create_service('foo', image=image_id)
  464. service.create_container()
  465. def test_scale(self):
  466. service = self.create_service('web')
  467. service.scale(1)
  468. self.assertEqual(len(service.containers()), 1)
  469. # Ensure containers don't have stdout or stdin connected
  470. container = service.containers()[0]
  471. config = container.inspect()['Config']
  472. self.assertFalse(config['AttachStderr'])
  473. self.assertFalse(config['AttachStdout'])
  474. self.assertFalse(config['AttachStdin'])
  475. service.scale(3)
  476. self.assertEqual(len(service.containers()), 3)
  477. service.scale(1)
  478. self.assertEqual(len(service.containers()), 1)
  479. service.scale(0)
  480. self.assertEqual(len(service.containers()), 0)
  481. def test_scale_with_stopped_containers(self):
  482. """
  483. Given there are some stopped containers and scale is called with a
  484. desired number that is the same as the number of stopped containers,
  485. test that those containers are restarted and not removed/recreated.
  486. """
  487. service = self.create_service('web')
  488. next_number = service._next_container_number()
  489. valid_numbers = [next_number, next_number + 1]
  490. service.create_container(number=next_number)
  491. service.create_container(number=next_number + 1)
  492. with mock.patch('sys.stdout', new_callable=StringIO) as mock_stdout:
  493. service.scale(2)
  494. for container in service.containers():
  495. self.assertTrue(container.is_running)
  496. self.assertTrue(container.number in valid_numbers)
  497. captured_output = mock_stdout.getvalue()
  498. self.assertNotIn('Creating', captured_output)
  499. self.assertIn('Starting', captured_output)
  500. def test_scale_with_stopped_containers_and_needing_creation(self):
  501. """
  502. Given there are some stopped containers and scale is called with a
  503. desired number that is greater than the number of stopped containers,
  504. test that those containers are restarted and required number are created.
  505. """
  506. service = self.create_service('web')
  507. next_number = service._next_container_number()
  508. service.create_container(number=next_number, quiet=True)
  509. for container in service.containers():
  510. self.assertFalse(container.is_running)
  511. with mock.patch('sys.stdout', new_callable=StringIO) as mock_stdout:
  512. service.scale(2)
  513. self.assertEqual(len(service.containers()), 2)
  514. for container in service.containers():
  515. self.assertTrue(container.is_running)
  516. captured_output = mock_stdout.getvalue()
  517. self.assertIn('Creating', captured_output)
  518. self.assertIn('Starting', captured_output)
  519. def test_scale_with_api_returns_errors(self):
  520. """
  521. Test that when scaling if the API returns an error, that error is handled
  522. and the remaining threads continue.
  523. """
  524. service = self.create_service('web')
  525. next_number = service._next_container_number()
  526. service.create_container(number=next_number, quiet=True)
  527. with mock.patch(
  528. 'compose.container.Container.create',
  529. side_effect=APIError(message="testing", response={}, explanation="Boom")):
  530. with mock.patch('sys.stdout', new_callable=StringIO) as mock_stdout:
  531. service.scale(3)
  532. self.assertEqual(len(service.containers()), 1)
  533. self.assertTrue(service.containers()[0].is_running)
  534. self.assertIn("ERROR: for 2 Boom", mock_stdout.getvalue())
  535. def test_scale_with_api_returns_unexpected_exception(self):
  536. """
  537. Test that when scaling if the API returns an error, that is not of type
  538. APIError, that error is re-raised.
  539. """
  540. service = self.create_service('web')
  541. next_number = service._next_container_number()
  542. service.create_container(number=next_number, quiet=True)
  543. with mock.patch(
  544. 'compose.container.Container.create',
  545. side_effect=ValueError("BOOM")
  546. ):
  547. with self.assertRaises(ValueError):
  548. service.scale(3)
  549. self.assertEqual(len(service.containers()), 1)
  550. self.assertTrue(service.containers()[0].is_running)
  551. @mock.patch('compose.service.log')
  552. def test_scale_with_desired_number_already_achieved(self, mock_log):
  553. """
  554. Test that calling scale with a desired number that is equal to the
  555. number of containers already running results in no change.
  556. """
  557. service = self.create_service('web')
  558. next_number = service._next_container_number()
  559. container = service.create_container(number=next_number, quiet=True)
  560. container.start()
  561. self.assertTrue(container.is_running)
  562. self.assertEqual(len(service.containers()), 1)
  563. service.scale(1)
  564. self.assertEqual(len(service.containers()), 1)
  565. container.inspect()
  566. self.assertTrue(container.is_running)
  567. captured_output = mock_log.info.call_args[0]
  568. self.assertIn('Desired container number already achieved', captured_output)
  569. @mock.patch('compose.service.log')
  570. def test_scale_with_custom_container_name_outputs_warning(self, mock_log):
  571. """Test that calling scale on a service that has a custom container name
  572. results in warning output.
  573. """
  574. # Disable this test against earlier versions because it is flaky
  575. self.require_api_version('1.21')
  576. service = self.create_service('app', container_name='custom-container')
  577. self.assertEqual(service.custom_container_name(), 'custom-container')
  578. service.scale(3)
  579. captured_output = mock_log.warn.call_args[0][0]
  580. self.assertEqual(len(service.containers()), 1)
  581. self.assertIn(
  582. "Remove the custom name to scale the service.",
  583. captured_output
  584. )
  585. def test_scale_sets_ports(self):
  586. service = self.create_service('web', ports=['8000'])
  587. service.scale(2)
  588. containers = service.containers()
  589. self.assertEqual(len(containers), 2)
  590. for container in containers:
  591. self.assertEqual(list(container.inspect()['HostConfig']['PortBindings'].keys()), ['8000/tcp'])
  592. def test_network_mode_none(self):
  593. service = self.create_service('web', net=Net('none'))
  594. container = create_and_start_container(service)
  595. self.assertEqual(container.get('HostConfig.NetworkMode'), 'none')
  596. def test_network_mode_bridged(self):
  597. service = self.create_service('web', net=Net('bridge'))
  598. container = create_and_start_container(service)
  599. self.assertEqual(container.get('HostConfig.NetworkMode'), 'bridge')
  600. def test_network_mode_host(self):
  601. service = self.create_service('web', net=Net('host'))
  602. container = create_and_start_container(service)
  603. self.assertEqual(container.get('HostConfig.NetworkMode'), 'host')
  604. def test_pid_mode_none_defined(self):
  605. service = self.create_service('web', pid=None)
  606. container = create_and_start_container(service)
  607. self.assertEqual(container.get('HostConfig.PidMode'), '')
  608. def test_pid_mode_host(self):
  609. service = self.create_service('web', pid='host')
  610. container = create_and_start_container(service)
  611. self.assertEqual(container.get('HostConfig.PidMode'), 'host')
  612. def test_dns_no_value(self):
  613. service = self.create_service('web')
  614. container = create_and_start_container(service)
  615. self.assertIsNone(container.get('HostConfig.Dns'))
  616. def test_dns_single_value(self):
  617. service = self.create_service('web', dns='8.8.8.8')
  618. container = create_and_start_container(service)
  619. self.assertEqual(container.get('HostConfig.Dns'), ['8.8.8.8'])
  620. def test_dns_list(self):
  621. service = self.create_service('web', dns=['8.8.8.8', '9.9.9.9'])
  622. container = create_and_start_container(service)
  623. self.assertEqual(container.get('HostConfig.Dns'), ['8.8.8.8', '9.9.9.9'])
  624. def test_restart_always_value(self):
  625. service = self.create_service('web', restart='always')
  626. container = create_and_start_container(service)
  627. self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'always')
  628. def test_restart_on_failure_value(self):
  629. service = self.create_service('web', restart='on-failure:5')
  630. container = create_and_start_container(service)
  631. self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'on-failure')
  632. self.assertEqual(container.get('HostConfig.RestartPolicy.MaximumRetryCount'), 5)
  633. def test_cap_add_list(self):
  634. service = self.create_service('web', cap_add=['SYS_ADMIN', 'NET_ADMIN'])
  635. container = create_and_start_container(service)
  636. self.assertEqual(container.get('HostConfig.CapAdd'), ['SYS_ADMIN', 'NET_ADMIN'])
  637. def test_cap_drop_list(self):
  638. service = self.create_service('web', cap_drop=['SYS_ADMIN', 'NET_ADMIN'])
  639. container = create_and_start_container(service)
  640. self.assertEqual(container.get('HostConfig.CapDrop'), ['SYS_ADMIN', 'NET_ADMIN'])
  641. def test_dns_search_no_value(self):
  642. service = self.create_service('web')
  643. container = create_and_start_container(service)
  644. self.assertIsNone(container.get('HostConfig.DnsSearch'))
  645. def test_dns_search_single_value(self):
  646. service = self.create_service('web', dns_search='example.com')
  647. container = create_and_start_container(service)
  648. self.assertEqual(container.get('HostConfig.DnsSearch'), ['example.com'])
  649. def test_dns_search_list(self):
  650. service = self.create_service('web', dns_search=['dc1.example.com', 'dc2.example.com'])
  651. container = create_and_start_container(service)
  652. self.assertEqual(container.get('HostConfig.DnsSearch'), ['dc1.example.com', 'dc2.example.com'])
  653. def test_working_dir_param(self):
  654. service = self.create_service('container', working_dir='/working/dir/sample')
  655. container = service.create_container()
  656. self.assertEqual(container.get('Config.WorkingDir'), '/working/dir/sample')
  657. def test_split_env(self):
  658. service = self.create_service('web', environment=['NORMAL=F1', 'CONTAINS_EQUALS=F=2', 'TRAILING_EQUALS='])
  659. env = create_and_start_container(service).environment
  660. for k, v in {'NORMAL': 'F1', 'CONTAINS_EQUALS': 'F=2', 'TRAILING_EQUALS': ''}.items():
  661. self.assertEqual(env[k], v)
  662. def test_env_from_file_combined_with_env(self):
  663. service = self.create_service(
  664. 'web',
  665. environment=['ONE=1', 'TWO=2', 'THREE=3'],
  666. env_file=['tests/fixtures/env/one.env', 'tests/fixtures/env/two.env'])
  667. env = create_and_start_container(service).environment
  668. for k, v in {'ONE': '1', 'TWO': '2', 'THREE': '3', 'FOO': 'baz', 'DOO': 'dah'}.items():
  669. self.assertEqual(env[k], v)
  670. @mock.patch.dict(os.environ)
  671. def test_resolve_env(self):
  672. os.environ['FILE_DEF'] = 'E1'
  673. os.environ['FILE_DEF_EMPTY'] = 'E2'
  674. os.environ['ENV_DEF'] = 'E3'
  675. service = self.create_service('web', environment={'FILE_DEF': 'F1', 'FILE_DEF_EMPTY': '', 'ENV_DEF': None, 'NO_DEF': None})
  676. env = create_and_start_container(service).environment
  677. for k, v in {'FILE_DEF': 'F1', 'FILE_DEF_EMPTY': '', 'ENV_DEF': 'E3', 'NO_DEF': ''}.items():
  678. self.assertEqual(env[k], v)
  679. def test_with_high_enough_api_version_we_get_default_network_mode(self):
  680. # TODO: remove this test once minimum docker version is 1.8.x
  681. with mock.patch.object(self.client, '_version', '1.20'):
  682. service = self.create_service('web')
  683. service_config = service._get_container_host_config({})
  684. self.assertEquals(service_config['NetworkMode'], 'default')
  685. def test_labels(self):
  686. labels_dict = {
  687. 'com.example.description': "Accounting webapp",
  688. 'com.example.department': "Finance",
  689. 'com.example.label-with-empty-value': "",
  690. }
  691. compose_labels = {
  692. LABEL_CONTAINER_NUMBER: '1',
  693. LABEL_ONE_OFF: 'False',
  694. LABEL_PROJECT: 'composetest',
  695. LABEL_SERVICE: 'web',
  696. LABEL_VERSION: __version__,
  697. }
  698. expected = dict(labels_dict, **compose_labels)
  699. service = self.create_service('web', labels=labels_dict)
  700. labels = create_and_start_container(service).labels.items()
  701. for pair in expected.items():
  702. self.assertIn(pair, labels)
  703. service.kill()
  704. service.remove_stopped()
  705. labels_list = ["%s=%s" % pair for pair in labels_dict.items()]
  706. service = self.create_service('web', labels=labels_list)
  707. labels = create_and_start_container(service).labels.items()
  708. for pair in expected.items():
  709. self.assertIn(pair, labels)
  710. def test_empty_labels(self):
  711. labels_list = ['foo', 'bar']
  712. service = self.create_service('web', labels=labels_list)
  713. labels = create_and_start_container(service).labels.items()
  714. for name in labels_list:
  715. self.assertIn((name, ''), labels)
  716. def test_custom_container_name(self):
  717. service = self.create_service('web', container_name='my-web-container')
  718. self.assertEqual(service.custom_container_name(), 'my-web-container')
  719. container = create_and_start_container(service)
  720. self.assertEqual(container.name, 'my-web-container')
  721. one_off_container = service.create_container(one_off=True)
  722. self.assertNotEqual(one_off_container.name, 'my-web-container')
  723. def test_log_drive_invalid(self):
  724. service = self.create_service('web', log_driver='xxx')
  725. expected_error_msg = "logger: no log driver named 'xxx' is registered"
  726. with self.assertRaisesRegexp(APIError, expected_error_msg):
  727. create_and_start_container(service)
  728. def test_log_drive_empty_default_jsonfile(self):
  729. service = self.create_service('web')
  730. log_config = create_and_start_container(service).log_config
  731. self.assertEqual('json-file', log_config['Type'])
  732. self.assertFalse(log_config['Config'])
  733. def test_log_drive_none(self):
  734. service = self.create_service('web', log_driver='none')
  735. log_config = create_and_start_container(service).log_config
  736. self.assertEqual('none', log_config['Type'])
  737. self.assertFalse(log_config['Config'])
  738. def test_devices(self):
  739. service = self.create_service('web', devices=["/dev/random:/dev/mapped-random"])
  740. device_config = create_and_start_container(service).get('HostConfig.Devices')
  741. device_dict = {
  742. 'PathOnHost': '/dev/random',
  743. 'CgroupPermissions': 'rwm',
  744. 'PathInContainer': '/dev/mapped-random'
  745. }
  746. self.assertEqual(1, len(device_config))
  747. self.assertDictEqual(device_dict, device_config[0])
  748. def test_duplicate_containers(self):
  749. service = self.create_service('web')
  750. options = service._get_container_create_options({}, 1)
  751. original = Container.create(service.client, **options)
  752. self.assertEqual(set(service.containers(stopped=True)), set([original]))
  753. self.assertEqual(set(service.duplicate_containers()), set())
  754. options['name'] = 'temporary_container_name'
  755. duplicate = Container.create(service.client, **options)
  756. self.assertEqual(set(service.containers(stopped=True)), set([original, duplicate]))
  757. self.assertEqual(set(service.duplicate_containers()), set([duplicate]))