service_test.py 39 KB

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