service_test.py 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971
  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 get_links
  13. from .testcases import pull_busybox
  14. from compose import __version__
  15. from compose.config.types import VolumeFromSpec
  16. from compose.config.types import VolumeSpec
  17. from compose.const import LABEL_CONFIG_HASH
  18. from compose.const import LABEL_CONTAINER_NUMBER
  19. from compose.const import LABEL_ONE_OFF
  20. from compose.const import LABEL_PROJECT
  21. from compose.const import LABEL_SERVICE
  22. from compose.const import LABEL_VERSION
  23. from compose.container import Container
  24. from compose.service import ConvergencePlan
  25. from compose.service import ConvergenceStrategy
  26. from compose.service import Net
  27. from compose.service import Service
  28. def create_and_start_container(service, **override_options):
  29. container = service.create_container(**override_options)
  30. container.start()
  31. return container
  32. def remove_stopped(service):
  33. containers = [c for c in service.containers(stopped=True) if not c.is_running]
  34. for container in containers:
  35. container.remove()
  36. class ServiceTest(DockerClientTestCase):
  37. def test_containers(self):
  38. foo = self.create_service('foo')
  39. bar = self.create_service('bar')
  40. create_and_start_container(foo)
  41. self.assertEqual(len(foo.containers()), 1)
  42. self.assertEqual(foo.containers()[0].name, 'composetest_foo_1')
  43. self.assertEqual(len(bar.containers()), 0)
  44. create_and_start_container(bar)
  45. create_and_start_container(bar)
  46. self.assertEqual(len(foo.containers()), 1)
  47. self.assertEqual(len(bar.containers()), 2)
  48. names = [c.name for c in bar.containers()]
  49. self.assertIn('composetest_bar_1', names)
  50. self.assertIn('composetest_bar_2', names)
  51. def test_containers_one_off(self):
  52. db = self.create_service('db')
  53. container = db.create_container(one_off=True)
  54. self.assertEqual(db.containers(stopped=True), [])
  55. self.assertEqual(db.containers(one_off=True, stopped=True), [container])
  56. def test_project_is_added_to_container_name(self):
  57. service = self.create_service('web')
  58. create_and_start_container(service)
  59. self.assertEqual(service.containers()[0].name, 'composetest_web_1')
  60. def test_create_container_with_one_off(self):
  61. db = self.create_service('db')
  62. container = db.create_container(one_off=True)
  63. self.assertEqual(container.name, 'composetest_db_run_1')
  64. def test_create_container_with_one_off_when_existing_container_is_running(self):
  65. db = self.create_service('db')
  66. db.start()
  67. container = db.create_container(one_off=True)
  68. self.assertEqual(container.name, 'composetest_db_run_1')
  69. def test_create_container_with_unspecified_volume(self):
  70. service = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
  71. container = service.create_container()
  72. container.start()
  73. assert container.get_mount('/var/db')
  74. def test_create_container_with_volume_driver(self):
  75. service = self.create_service('db', volume_driver='foodriver')
  76. container = service.create_container()
  77. container.start()
  78. self.assertEqual('foodriver', container.get('HostConfig.VolumeDriver'))
  79. def test_create_container_with_cpu_shares(self):
  80. service = self.create_service('db', cpu_shares=73)
  81. container = service.create_container()
  82. container.start()
  83. self.assertEqual(container.get('HostConfig.CpuShares'), 73)
  84. def test_create_container_with_cpu_quota(self):
  85. service = self.create_service('db', cpu_quota=40000)
  86. container = service.create_container()
  87. container.start()
  88. self.assertEqual(container.get('HostConfig.CpuQuota'), 40000)
  89. def test_create_container_with_extra_hosts_list(self):
  90. extra_hosts = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
  91. service = self.create_service('db', extra_hosts=extra_hosts)
  92. container = service.create_container()
  93. container.start()
  94. self.assertEqual(set(container.get('HostConfig.ExtraHosts')), set(extra_hosts))
  95. def test_create_container_with_extra_hosts_dicts(self):
  96. extra_hosts = {'somehost': '162.242.195.82', 'otherhost': '50.31.209.229'}
  97. extra_hosts_list = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
  98. service = self.create_service('db', extra_hosts=extra_hosts)
  99. container = service.create_container()
  100. container.start()
  101. self.assertEqual(set(container.get('HostConfig.ExtraHosts')), set(extra_hosts_list))
  102. def test_create_container_with_cpu_set(self):
  103. service = self.create_service('db', cpuset='0')
  104. container = service.create_container()
  105. container.start()
  106. self.assertEqual(container.get('HostConfig.CpusetCpus'), '0')
  107. def test_create_container_with_read_only_root_fs(self):
  108. read_only = True
  109. service = self.create_service('db', read_only=read_only)
  110. container = service.create_container()
  111. container.start()
  112. self.assertEqual(container.get('HostConfig.ReadonlyRootfs'), read_only, container.get('HostConfig'))
  113. def test_create_container_with_security_opt(self):
  114. security_opt = ['label:disable']
  115. service = self.create_service('db', security_opt=security_opt)
  116. container = service.create_container()
  117. container.start()
  118. self.assertEqual(set(container.get('HostConfig.SecurityOpt')), set(security_opt))
  119. def test_create_container_with_mac_address(self):
  120. service = self.create_service('db', mac_address='02:42:ac:11:65:43')
  121. container = service.create_container()
  122. container.start()
  123. self.assertEqual(container.inspect()['Config']['MacAddress'], '02:42:ac:11:65:43')
  124. def test_create_container_with_specified_volume(self):
  125. host_path = '/tmp/host-path'
  126. container_path = '/container-path'
  127. service = self.create_service(
  128. 'db',
  129. volumes=[VolumeSpec(host_path, container_path, 'rw')])
  130. container = service.create_container()
  131. container.start()
  132. assert container.get_mount(container_path)
  133. # Match the last component ("host-path"), because boot2docker symlinks /tmp
  134. actual_host_path = container.get_mount(container_path)['Source']
  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_mount('/data')['Source']
  144. new_container = service.recreate_container(old_container)
  145. self.assertEqual(new_container.get_mount('/data')['Source'], 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_mount('/etc')['Source']
  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_mount('/etc')['Source'], 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(
  256. [mount['Destination'] for mount in old_container.get('Mounts')], ['/data']
  257. )
  258. volume_path = old_container.get_mount('/data')['Source']
  259. new_container, = service.execute_convergence_plan(
  260. ConvergencePlan('recreate', [old_container]))
  261. self.assertEqual(
  262. [mount['Destination'] for mount in new_container.get('Mounts')],
  263. ['/data']
  264. )
  265. self.assertEqual(new_container.get_mount('/data')['Source'], volume_path)
  266. def test_execute_convergence_plan_when_image_volume_masks_config(self):
  267. service = self.create_service(
  268. 'db',
  269. build='tests/fixtures/dockerfile-with-volume',
  270. )
  271. old_container = create_and_start_container(service)
  272. self.assertEqual(
  273. [mount['Destination'] for mount in old_container.get('Mounts')],
  274. ['/data']
  275. )
  276. volume_path = old_container.get_mount('/data')['Source']
  277. service.options['volumes'] = [VolumeSpec.parse('/tmp:/data')]
  278. with mock.patch('compose.service.log') as mock_log:
  279. new_container, = service.execute_convergence_plan(
  280. ConvergencePlan('recreate', [old_container]))
  281. mock_log.warn.assert_called_once_with(mock.ANY)
  282. _, args, kwargs = mock_log.warn.mock_calls[0]
  283. self.assertIn(
  284. "Service \"db\" is using volume \"/data\" from the previous container",
  285. args[0])
  286. self.assertEqual(
  287. [mount['Destination'] for mount in new_container.get('Mounts')],
  288. ['/data']
  289. )
  290. self.assertEqual(new_container.get_mount('/data')['Source'], volume_path)
  291. def test_execute_convergence_plan_without_start(self):
  292. service = self.create_service(
  293. 'db',
  294. build='tests/fixtures/dockerfile-with-volume'
  295. )
  296. containers = service.execute_convergence_plan(ConvergencePlan('create', []), start=False)
  297. self.assertEqual(len(service.containers()), 0)
  298. self.assertEqual(len(service.containers(stopped=True)), 1)
  299. containers = service.execute_convergence_plan(ConvergencePlan('recreate', containers), start=False)
  300. self.assertEqual(len(service.containers()), 0)
  301. self.assertEqual(len(service.containers(stopped=True)), 1)
  302. service.execute_convergence_plan(ConvergencePlan('start', containers), start=False)
  303. self.assertEqual(len(service.containers()), 0)
  304. self.assertEqual(len(service.containers(stopped=True)), 1)
  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(get_links(web.containers()[0])),
  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(get_links(web.containers()[0])),
  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(get_links(web.containers()[0])),
  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(get_links(c)), 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(get_links(c)),
  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_build_with_git_url(self):
  416. build_url = "https://github.com/dnephin/docker-build-from-url.git"
  417. service = self.create_service('buildwithurl', build=build_url)
  418. self.addCleanup(self.client.remove_image, service.image_name)
  419. service.build()
  420. assert service.image()
  421. def test_start_container_stays_unpriviliged(self):
  422. service = self.create_service('web')
  423. container = create_and_start_container(service).inspect()
  424. self.assertEqual(container['HostConfig']['Privileged'], False)
  425. def test_start_container_becomes_priviliged(self):
  426. service = self.create_service('web', privileged=True)
  427. container = create_and_start_container(service).inspect()
  428. self.assertEqual(container['HostConfig']['Privileged'], True)
  429. def test_expose_does_not_publish_ports(self):
  430. service = self.create_service('web', expose=["8000"])
  431. container = create_and_start_container(service).inspect()
  432. self.assertEqual(container['NetworkSettings']['Ports'], {'8000/tcp': None})
  433. def test_start_container_creates_port_with_explicit_protocol(self):
  434. service = self.create_service('web', ports=['8000/udp'])
  435. container = create_and_start_container(service).inspect()
  436. self.assertEqual(list(container['NetworkSettings']['Ports'].keys()), ['8000/udp'])
  437. def test_start_container_creates_fixed_external_ports(self):
  438. service = self.create_service('web', ports=['8000:8000'])
  439. container = create_and_start_container(service).inspect()
  440. self.assertIn('8000/tcp', container['NetworkSettings']['Ports'])
  441. self.assertEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8000')
  442. def test_start_container_creates_fixed_external_ports_when_it_is_different_to_internal_port(self):
  443. service = self.create_service('web', ports=['8001:8000'])
  444. container = create_and_start_container(service).inspect()
  445. self.assertIn('8000/tcp', container['NetworkSettings']['Ports'])
  446. self.assertEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8001')
  447. def test_port_with_explicit_interface(self):
  448. service = self.create_service('web', ports=[
  449. '127.0.0.1:8001:8000',
  450. '0.0.0.0:9001:9000/udp',
  451. ])
  452. container = create_and_start_container(service).inspect()
  453. self.assertEqual(container['NetworkSettings']['Ports'], {
  454. '8000/tcp': [
  455. {
  456. 'HostIp': '127.0.0.1',
  457. 'HostPort': '8001',
  458. },
  459. ],
  460. '9000/udp': [
  461. {
  462. 'HostIp': '0.0.0.0',
  463. 'HostPort': '9001',
  464. },
  465. ],
  466. })
  467. def test_create_with_image_id(self):
  468. # Get image id for the current busybox:latest
  469. pull_busybox(self.client)
  470. image_id = self.client.inspect_image('busybox:latest')['Id'][:12]
  471. service = self.create_service('foo', image=image_id)
  472. service.create_container()
  473. def test_scale(self):
  474. service = self.create_service('web')
  475. service.scale(1)
  476. self.assertEqual(len(service.containers()), 1)
  477. # Ensure containers don't have stdout or stdin connected
  478. container = service.containers()[0]
  479. config = container.inspect()['Config']
  480. self.assertFalse(config['AttachStderr'])
  481. self.assertFalse(config['AttachStdout'])
  482. self.assertFalse(config['AttachStdin'])
  483. service.scale(3)
  484. self.assertEqual(len(service.containers()), 3)
  485. service.scale(1)
  486. self.assertEqual(len(service.containers()), 1)
  487. service.scale(0)
  488. self.assertEqual(len(service.containers()), 0)
  489. def test_scale_with_stopped_containers(self):
  490. """
  491. Given there are some stopped containers and scale is called with a
  492. desired number that is the same as the number of stopped containers,
  493. test that those containers are restarted and not removed/recreated.
  494. """
  495. service = self.create_service('web')
  496. next_number = service._next_container_number()
  497. valid_numbers = [next_number, next_number + 1]
  498. service.create_container(number=next_number)
  499. service.create_container(number=next_number + 1)
  500. with mock.patch('sys.stdout', new_callable=StringIO) as mock_stdout:
  501. service.scale(2)
  502. for container in service.containers():
  503. self.assertTrue(container.is_running)
  504. self.assertTrue(container.number in valid_numbers)
  505. captured_output = mock_stdout.getvalue()
  506. self.assertNotIn('Creating', captured_output)
  507. self.assertIn('Starting', captured_output)
  508. def test_scale_with_stopped_containers_and_needing_creation(self):
  509. """
  510. Given there are some stopped containers and scale is called with a
  511. desired number that is greater than the number of stopped containers,
  512. test that those containers are restarted and required number are created.
  513. """
  514. service = self.create_service('web')
  515. next_number = service._next_container_number()
  516. service.create_container(number=next_number, quiet=True)
  517. for container in service.containers():
  518. self.assertFalse(container.is_running)
  519. with mock.patch('sys.stdout', new_callable=StringIO) as mock_stdout:
  520. service.scale(2)
  521. self.assertEqual(len(service.containers()), 2)
  522. for container in service.containers():
  523. self.assertTrue(container.is_running)
  524. captured_output = mock_stdout.getvalue()
  525. self.assertIn('Creating', captured_output)
  526. self.assertIn('Starting', captured_output)
  527. def test_scale_with_api_error(self):
  528. """Test that when scaling if the API returns an error, that error is handled
  529. and the remaining threads continue.
  530. """
  531. service = self.create_service('web')
  532. next_number = service._next_container_number()
  533. service.create_container(number=next_number, quiet=True)
  534. with mock.patch(
  535. 'compose.container.Container.create',
  536. side_effect=APIError(
  537. message="testing",
  538. response={},
  539. explanation="Boom")):
  540. with mock.patch('sys.stdout', new_callable=StringIO) as mock_stdout:
  541. service.scale(3)
  542. self.assertEqual(len(service.containers()), 1)
  543. self.assertTrue(service.containers()[0].is_running)
  544. self.assertIn("ERROR: for 2 Boom", mock_stdout.getvalue())
  545. def test_scale_with_unexpected_exception(self):
  546. """Test that when scaling if the API returns an error, that is not of type
  547. APIError, that error is re-raised.
  548. """
  549. service = self.create_service('web')
  550. next_number = service._next_container_number()
  551. service.create_container(number=next_number, quiet=True)
  552. with mock.patch(
  553. 'compose.container.Container.create',
  554. side_effect=ValueError("BOOM")
  555. ):
  556. with self.assertRaises(ValueError):
  557. service.scale(3)
  558. self.assertEqual(len(service.containers()), 1)
  559. self.assertTrue(service.containers()[0].is_running)
  560. @mock.patch('compose.service.log')
  561. def test_scale_with_desired_number_already_achieved(self, mock_log):
  562. """
  563. Test that calling scale with a desired number that is equal to the
  564. number of containers already running results in no change.
  565. """
  566. service = self.create_service('web')
  567. next_number = service._next_container_number()
  568. container = service.create_container(number=next_number, quiet=True)
  569. container.start()
  570. self.assertTrue(container.is_running)
  571. self.assertEqual(len(service.containers()), 1)
  572. service.scale(1)
  573. self.assertEqual(len(service.containers()), 1)
  574. container.inspect()
  575. self.assertTrue(container.is_running)
  576. captured_output = mock_log.info.call_args[0]
  577. self.assertIn('Desired container number already achieved', captured_output)
  578. @mock.patch('compose.service.log')
  579. def test_scale_with_custom_container_name_outputs_warning(self, mock_log):
  580. """Test that calling scale on a service that has a custom container name
  581. results in warning output.
  582. """
  583. # Disable this test against earlier versions because it is flaky
  584. self.require_api_version('1.21')
  585. service = self.create_service('app', container_name='custom-container')
  586. self.assertEqual(service.custom_container_name(), 'custom-container')
  587. service.scale(3)
  588. captured_output = mock_log.warn.call_args[0][0]
  589. self.assertEqual(len(service.containers()), 1)
  590. self.assertIn(
  591. "Remove the custom name to scale the service.",
  592. captured_output
  593. )
  594. def test_scale_sets_ports(self):
  595. service = self.create_service('web', ports=['8000'])
  596. service.scale(2)
  597. containers = service.containers()
  598. self.assertEqual(len(containers), 2)
  599. for container in containers:
  600. self.assertEqual(list(container.inspect()['HostConfig']['PortBindings'].keys()), ['8000/tcp'])
  601. def test_network_mode_none(self):
  602. service = self.create_service('web', net=Net('none'))
  603. container = create_and_start_container(service)
  604. self.assertEqual(container.get('HostConfig.NetworkMode'), 'none')
  605. def test_network_mode_bridged(self):
  606. service = self.create_service('web', net=Net('bridge'))
  607. container = create_and_start_container(service)
  608. self.assertEqual(container.get('HostConfig.NetworkMode'), 'bridge')
  609. def test_network_mode_host(self):
  610. service = self.create_service('web', net=Net('host'))
  611. container = create_and_start_container(service)
  612. self.assertEqual(container.get('HostConfig.NetworkMode'), 'host')
  613. def test_pid_mode_none_defined(self):
  614. service = self.create_service('web', pid=None)
  615. container = create_and_start_container(service)
  616. self.assertEqual(container.get('HostConfig.PidMode'), '')
  617. def test_pid_mode_host(self):
  618. service = self.create_service('web', pid='host')
  619. container = create_and_start_container(service)
  620. self.assertEqual(container.get('HostConfig.PidMode'), 'host')
  621. def test_dns_no_value(self):
  622. service = self.create_service('web')
  623. container = create_and_start_container(service)
  624. self.assertIsNone(container.get('HostConfig.Dns'))
  625. def test_dns_list(self):
  626. service = self.create_service('web', dns=['8.8.8.8', '9.9.9.9'])
  627. container = create_and_start_container(service)
  628. self.assertEqual(container.get('HostConfig.Dns'), ['8.8.8.8', '9.9.9.9'])
  629. def test_restart_always_value(self):
  630. service = self.create_service('web', restart={'Name': 'always'})
  631. container = create_and_start_container(service)
  632. self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'always')
  633. def test_restart_on_failure_value(self):
  634. service = self.create_service('web', restart={
  635. 'Name': 'on-failure',
  636. 'MaximumRetryCount': 5
  637. })
  638. container = create_and_start_container(service)
  639. self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'on-failure')
  640. self.assertEqual(container.get('HostConfig.RestartPolicy.MaximumRetryCount'), 5)
  641. def test_cap_add_list(self):
  642. service = self.create_service('web', cap_add=['SYS_ADMIN', 'NET_ADMIN'])
  643. container = create_and_start_container(service)
  644. self.assertEqual(container.get('HostConfig.CapAdd'), ['SYS_ADMIN', 'NET_ADMIN'])
  645. def test_cap_drop_list(self):
  646. service = self.create_service('web', cap_drop=['SYS_ADMIN', 'NET_ADMIN'])
  647. container = create_and_start_container(service)
  648. self.assertEqual(container.get('HostConfig.CapDrop'), ['SYS_ADMIN', 'NET_ADMIN'])
  649. def test_dns_search(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 {
  669. 'ONE': '1',
  670. 'TWO': '2',
  671. 'THREE': '3',
  672. 'FOO': 'baz',
  673. 'DOO': 'dah'
  674. }.items():
  675. self.assertEqual(env[k], v)
  676. @mock.patch.dict(os.environ)
  677. def test_resolve_env(self):
  678. os.environ['FILE_DEF'] = 'E1'
  679. os.environ['FILE_DEF_EMPTY'] = 'E2'
  680. os.environ['ENV_DEF'] = 'E3'
  681. service = self.create_service(
  682. 'web',
  683. environment={
  684. 'FILE_DEF': 'F1',
  685. 'FILE_DEF_EMPTY': '',
  686. 'ENV_DEF': None,
  687. 'NO_DEF': None
  688. }
  689. )
  690. env = create_and_start_container(service).environment
  691. for k, v in {
  692. 'FILE_DEF': 'F1',
  693. 'FILE_DEF_EMPTY': '',
  694. 'ENV_DEF': 'E3',
  695. 'NO_DEF': ''
  696. }.items():
  697. self.assertEqual(env[k], v)
  698. def test_with_high_enough_api_version_we_get_default_network_mode(self):
  699. # TODO: remove this test once minimum docker version is 1.8.x
  700. with mock.patch.object(self.client, '_version', '1.20'):
  701. service = self.create_service('web')
  702. service_config = service._get_container_host_config({})
  703. self.assertEquals(service_config['NetworkMode'], 'default')
  704. def test_labels(self):
  705. labels_dict = {
  706. 'com.example.description': "Accounting webapp",
  707. 'com.example.department': "Finance",
  708. 'com.example.label-with-empty-value': "",
  709. }
  710. compose_labels = {
  711. LABEL_CONTAINER_NUMBER: '1',
  712. LABEL_ONE_OFF: 'False',
  713. LABEL_PROJECT: 'composetest',
  714. LABEL_SERVICE: 'web',
  715. LABEL_VERSION: __version__,
  716. }
  717. expected = dict(labels_dict, **compose_labels)
  718. service = self.create_service('web', labels=labels_dict)
  719. labels = create_and_start_container(service).labels.items()
  720. for pair in expected.items():
  721. self.assertIn(pair, labels)
  722. def test_empty_labels(self):
  723. labels_dict = {'foo': '', 'bar': ''}
  724. service = self.create_service('web', labels=labels_dict)
  725. labels = create_and_start_container(service).labels.items()
  726. for name in labels_dict:
  727. self.assertIn((name, ''), labels)
  728. def test_custom_container_name(self):
  729. service = self.create_service('web', container_name='my-web-container')
  730. self.assertEqual(service.custom_container_name(), 'my-web-container')
  731. container = create_and_start_container(service)
  732. self.assertEqual(container.name, 'my-web-container')
  733. one_off_container = service.create_container(one_off=True)
  734. self.assertNotEqual(one_off_container.name, 'my-web-container')
  735. def test_log_drive_invalid(self):
  736. service = self.create_service('web', log_driver='xxx')
  737. expected_error_msg = "logger: no log driver named 'xxx' is registered"
  738. with self.assertRaisesRegexp(APIError, expected_error_msg):
  739. create_and_start_container(service)
  740. def test_log_drive_empty_default_jsonfile(self):
  741. service = self.create_service('web')
  742. log_config = create_and_start_container(service).log_config
  743. self.assertEqual('json-file', log_config['Type'])
  744. self.assertFalse(log_config['Config'])
  745. def test_log_drive_none(self):
  746. service = self.create_service('web', log_driver='none')
  747. log_config = create_and_start_container(service).log_config
  748. self.assertEqual('none', log_config['Type'])
  749. self.assertFalse(log_config['Config'])
  750. def test_devices(self):
  751. service = self.create_service('web', devices=["/dev/random:/dev/mapped-random"])
  752. device_config = create_and_start_container(service).get('HostConfig.Devices')
  753. device_dict = {
  754. 'PathOnHost': '/dev/random',
  755. 'CgroupPermissions': 'rwm',
  756. 'PathInContainer': '/dev/mapped-random'
  757. }
  758. self.assertEqual(1, len(device_config))
  759. self.assertDictEqual(device_dict, device_config[0])
  760. def test_duplicate_containers(self):
  761. service = self.create_service('web')
  762. options = service._get_container_create_options({}, 1)
  763. original = Container.create(service.client, **options)
  764. self.assertEqual(set(service.containers(stopped=True)), set([original]))
  765. self.assertEqual(set(service.duplicate_containers()), set())
  766. options['name'] = 'temporary_container_name'
  767. duplicate = Container.create(service.client, **options)
  768. self.assertEqual(set(service.containers(stopped=True)), set([original, duplicate]))
  769. self.assertEqual(set(service.duplicate_containers()), set([duplicate]))
  770. def converge(service,
  771. strategy=ConvergenceStrategy.changed,
  772. do_build=True):
  773. """Create a converge plan from a strategy and execute the plan."""
  774. plan = service.convergence_plan(strategy)
  775. return service.execute_convergence_plan(plan, do_build=do_build, timeout=1)
  776. class ConfigHashTest(DockerClientTestCase):
  777. def test_no_config_hash_when_one_off(self):
  778. web = self.create_service('web')
  779. container = web.create_container(one_off=True)
  780. self.assertNotIn(LABEL_CONFIG_HASH, container.labels)
  781. def test_no_config_hash_when_overriding_options(self):
  782. web = self.create_service('web')
  783. container = web.create_container(environment={'FOO': '1'})
  784. self.assertNotIn(LABEL_CONFIG_HASH, container.labels)
  785. def test_config_hash_with_custom_labels(self):
  786. web = self.create_service('web', labels={'foo': '1'})
  787. container = converge(web)[0]
  788. self.assertIn(LABEL_CONFIG_HASH, container.labels)
  789. self.assertIn('foo', container.labels)
  790. def test_config_hash_sticks_around(self):
  791. web = self.create_service('web', command=["top"])
  792. container = converge(web)[0]
  793. self.assertIn(LABEL_CONFIG_HASH, container.labels)
  794. web = self.create_service('web', command=["top", "-d", "1"])
  795. container = converge(web)[0]
  796. self.assertIn(LABEL_CONFIG_HASH, container.labels)