service_test.py 38 KB

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