service_test.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658
  1. from __future__ import unicode_literals
  2. from __future__ import absolute_import
  3. import os
  4. from os import path
  5. import mock
  6. import tempfile
  7. import shutil
  8. import six
  9. from compose import Service
  10. from compose.service import (
  11. CannotBeScaledError,
  12. build_extra_hosts,
  13. ConfigError,
  14. )
  15. from compose.container import Container
  16. from docker.errors import APIError
  17. from .testcases import DockerClientTestCase
  18. def create_and_start_container(service, **override_options):
  19. container = service.create_container(**override_options)
  20. return service.start_container(container)
  21. class ServiceTest(DockerClientTestCase):
  22. def test_containers(self):
  23. foo = self.create_service('foo')
  24. bar = self.create_service('bar')
  25. create_and_start_container(foo)
  26. self.assertEqual(len(foo.containers()), 1)
  27. self.assertEqual(foo.containers()[0].name, 'composetest_foo_1')
  28. self.assertEqual(len(bar.containers()), 0)
  29. create_and_start_container(bar)
  30. create_and_start_container(bar)
  31. self.assertEqual(len(foo.containers()), 1)
  32. self.assertEqual(len(bar.containers()), 2)
  33. names = [c.name for c in bar.containers()]
  34. self.assertIn('composetest_bar_1', names)
  35. self.assertIn('composetest_bar_2', names)
  36. def test_containers_one_off(self):
  37. db = self.create_service('db')
  38. container = db.create_container(one_off=True)
  39. self.assertEqual(db.containers(stopped=True), [])
  40. self.assertEqual(db.containers(one_off=True, stopped=True), [container])
  41. def test_project_is_added_to_container_name(self):
  42. service = self.create_service('web')
  43. create_and_start_container(service)
  44. self.assertEqual(service.containers()[0].name, 'composetest_web_1')
  45. def test_start_stop(self):
  46. service = self.create_service('scalingtest')
  47. self.assertEqual(len(service.containers(stopped=True)), 0)
  48. service.create_container()
  49. self.assertEqual(len(service.containers()), 0)
  50. self.assertEqual(len(service.containers(stopped=True)), 1)
  51. service.start()
  52. self.assertEqual(len(service.containers()), 1)
  53. self.assertEqual(len(service.containers(stopped=True)), 1)
  54. service.stop(timeout=1)
  55. self.assertEqual(len(service.containers()), 0)
  56. self.assertEqual(len(service.containers(stopped=True)), 1)
  57. service.stop(timeout=1)
  58. self.assertEqual(len(service.containers()), 0)
  59. self.assertEqual(len(service.containers(stopped=True)), 1)
  60. def test_kill_remove(self):
  61. service = self.create_service('scalingtest')
  62. create_and_start_container(service)
  63. self.assertEqual(len(service.containers()), 1)
  64. service.remove_stopped()
  65. self.assertEqual(len(service.containers()), 1)
  66. service.kill()
  67. self.assertEqual(len(service.containers()), 0)
  68. self.assertEqual(len(service.containers(stopped=True)), 1)
  69. service.remove_stopped()
  70. self.assertEqual(len(service.containers(stopped=True)), 0)
  71. def test_create_container_with_one_off(self):
  72. db = self.create_service('db')
  73. container = db.create_container(one_off=True)
  74. self.assertEqual(container.name, 'composetest_db_run_1')
  75. def test_create_container_with_one_off_when_existing_container_is_running(self):
  76. db = self.create_service('db')
  77. db.start()
  78. container = db.create_container(one_off=True)
  79. self.assertEqual(container.name, 'composetest_db_run_1')
  80. def test_create_container_with_unspecified_volume(self):
  81. service = self.create_service('db', volumes=['/var/db'])
  82. container = service.create_container()
  83. service.start_container(container)
  84. self.assertIn('/var/db', container.inspect()['Volumes'])
  85. def test_create_container_with_cpu_shares(self):
  86. service = self.create_service('db', cpu_shares=73)
  87. container = service.create_container()
  88. service.start_container(container)
  89. self.assertEqual(container.inspect()['Config']['CpuShares'], 73)
  90. def test_build_extra_hosts(self):
  91. # string
  92. self.assertRaises(ConfigError, lambda: build_extra_hosts("www.example.com: 192.168.0.17"))
  93. # list of strings
  94. self.assertEqual(build_extra_hosts(
  95. ["www.example.com:192.168.0.17"]),
  96. {'www.example.com': '192.168.0.17'})
  97. self.assertEqual(build_extra_hosts(
  98. ["www.example.com: 192.168.0.17"]),
  99. {'www.example.com': '192.168.0.17'})
  100. self.assertEqual(build_extra_hosts(
  101. ["www.example.com: 192.168.0.17",
  102. "static.example.com:192.168.0.19",
  103. "api.example.com: 192.168.0.18"]),
  104. {'www.example.com': '192.168.0.17',
  105. 'static.example.com': '192.168.0.19',
  106. 'api.example.com': '192.168.0.18'})
  107. # list of dictionaries
  108. self.assertRaises(ConfigError, lambda: build_extra_hosts(
  109. [{'www.example.com': '192.168.0.17'},
  110. {'api.example.com': '192.168.0.18'}]))
  111. # dictionaries
  112. self.assertEqual(build_extra_hosts(
  113. {'www.example.com': '192.168.0.17',
  114. 'api.example.com': '192.168.0.18'}),
  115. {'www.example.com': '192.168.0.17',
  116. 'api.example.com': '192.168.0.18'})
  117. def test_create_container_with_extra_hosts_list(self):
  118. extra_hosts = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
  119. service = self.create_service('db', extra_hosts=extra_hosts)
  120. container = service.create_container()
  121. service.start_container(container)
  122. self.assertEqual(set(container.get('HostConfig.ExtraHosts')), set(extra_hosts))
  123. def test_create_container_with_extra_hosts_string(self):
  124. extra_hosts = 'somehost:162.242.195.82'
  125. service = self.create_service('db', extra_hosts=extra_hosts)
  126. self.assertRaises(ConfigError, lambda: service.create_container())
  127. def test_create_container_with_extra_hosts_list_of_dicts(self):
  128. extra_hosts = [{'somehost': '162.242.195.82'}, {'otherhost': '50.31.209.229'}]
  129. service = self.create_service('db', extra_hosts=extra_hosts)
  130. self.assertRaises(ConfigError, lambda: service.create_container())
  131. def test_create_container_with_extra_hosts_dicts(self):
  132. extra_hosts = {'somehost': '162.242.195.82', 'otherhost': '50.31.209.229'}
  133. extra_hosts_list = ['somehost:162.242.195.82', 'otherhost:50.31.209.229']
  134. service = self.create_service('db', extra_hosts=extra_hosts)
  135. container = service.create_container()
  136. service.start_container(container)
  137. self.assertEqual(set(container.get('HostConfig.ExtraHosts')), set(extra_hosts_list))
  138. def test_create_container_with_cpu_set(self):
  139. service = self.create_service('db', cpuset='0')
  140. container = service.create_container()
  141. service.start_container(container)
  142. self.assertEqual(container.inspect()['Config']['Cpuset'], '0')
  143. def test_create_container_with_read_only_root_fs(self):
  144. read_only = True
  145. service = self.create_service('db', read_only=read_only)
  146. container = service.create_container()
  147. service.start_container(container)
  148. self.assertEqual(container.get('HostConfig.ReadonlyRootfs'), read_only, container.get('HostConfig'))
  149. def test_create_container_with_specified_volume(self):
  150. host_path = '/tmp/host-path'
  151. container_path = '/container-path'
  152. service = self.create_service('db', volumes=['%s:%s' % (host_path, container_path)])
  153. container = service.create_container()
  154. service.start_container(container)
  155. volumes = container.inspect()['Volumes']
  156. self.assertIn(container_path, volumes)
  157. # Match the last component ("host-path"), because boot2docker symlinks /tmp
  158. actual_host_path = volumes[container_path]
  159. self.assertTrue(path.basename(actual_host_path) == path.basename(host_path),
  160. msg=("Last component differs: %s, %s" % (actual_host_path, host_path)))
  161. @mock.patch.dict(os.environ)
  162. def test_create_container_with_home_and_env_var_in_volume_path(self):
  163. os.environ['VOLUME_NAME'] = 'my-volume'
  164. os.environ['HOME'] = '/tmp/home-dir'
  165. expected_host_path = os.path.join(os.environ['HOME'], os.environ['VOLUME_NAME'])
  166. host_path = '~/${VOLUME_NAME}'
  167. container_path = '/container-path'
  168. service = self.create_service('db', volumes=['%s:%s' % (host_path, container_path)])
  169. container = service.create_container()
  170. service.start_container(container)
  171. actual_host_path = container.get('Volumes')[container_path]
  172. components = actual_host_path.split('/')
  173. self.assertTrue(components[-2:] == ['home-dir', 'my-volume'],
  174. msg="Last two components differ: %s, %s" % (actual_host_path, expected_host_path))
  175. def test_create_container_with_volumes_from(self):
  176. volume_service = self.create_service('data')
  177. volume_container_1 = volume_service.create_container()
  178. volume_container_2 = Container.create(self.client, image='busybox:latest', command=["/bin/sleep", "300"])
  179. host_service = self.create_service('host', volumes_from=[volume_service, volume_container_2])
  180. host_container = host_service.create_container()
  181. host_service.start_container(host_container)
  182. self.assertIn(volume_container_1.id,
  183. host_container.get('HostConfig.VolumesFrom'))
  184. self.assertIn(volume_container_2.id,
  185. host_container.get('HostConfig.VolumesFrom'))
  186. def test_recreate_containers(self):
  187. service = self.create_service(
  188. 'db',
  189. environment={'FOO': '1'},
  190. volumes=['/etc'],
  191. entrypoint=['sleep'],
  192. command=['300']
  193. )
  194. old_container = service.create_container()
  195. self.assertEqual(old_container.dictionary['Config']['Entrypoint'], ['sleep'])
  196. self.assertEqual(old_container.dictionary['Config']['Cmd'], ['300'])
  197. self.assertIn('FOO=1', old_container.dictionary['Config']['Env'])
  198. self.assertEqual(old_container.name, 'composetest_db_1')
  199. service.start_container(old_container)
  200. volume_path = old_container.inspect()['Volumes']['/etc']
  201. num_containers_before = len(self.client.containers(all=True))
  202. service.options['environment']['FOO'] = '2'
  203. tuples = service.recreate_containers()
  204. self.assertEqual(len(tuples), 1)
  205. intermediate_container = tuples[0][0]
  206. new_container = tuples[0][1]
  207. self.assertEqual(intermediate_container.dictionary['Config']['Entrypoint'], ['/bin/echo'])
  208. self.assertEqual(new_container.dictionary['Config']['Entrypoint'], ['sleep'])
  209. self.assertEqual(new_container.dictionary['Config']['Cmd'], ['300'])
  210. self.assertIn('FOO=2', new_container.dictionary['Config']['Env'])
  211. self.assertEqual(new_container.name, 'composetest_db_1')
  212. self.assertEqual(new_container.inspect()['Volumes']['/etc'], volume_path)
  213. self.assertIn(intermediate_container.id, new_container.dictionary['HostConfig']['VolumesFrom'])
  214. self.assertEqual(len(self.client.containers(all=True)), num_containers_before)
  215. self.assertNotEqual(old_container.id, new_container.id)
  216. self.assertRaises(APIError,
  217. self.client.inspect_container,
  218. intermediate_container.id)
  219. def test_recreate_containers_when_containers_are_stopped(self):
  220. service = self.create_service(
  221. 'db',
  222. environment={'FOO': '1'},
  223. volumes=['/var/db'],
  224. entrypoint=['sleep'],
  225. command=['300']
  226. )
  227. service.create_container()
  228. self.assertEqual(len(service.containers(stopped=True)), 1)
  229. service.recreate_containers()
  230. self.assertEqual(len(service.containers(stopped=True)), 1)
  231. def test_recreate_containers_with_image_declared_volume(self):
  232. service = Service(
  233. project='composetest',
  234. name='db',
  235. client=self.client,
  236. build='tests/fixtures/dockerfile-with-volume',
  237. )
  238. old_container = create_and_start_container(service)
  239. self.assertEqual(old_container.get('Volumes').keys(), ['/data'])
  240. volume_path = old_container.get('Volumes')['/data']
  241. service.recreate_containers()
  242. new_container = service.containers()[0]
  243. service.start_container(new_container)
  244. self.assertEqual(new_container.get('Volumes').keys(), ['/data'])
  245. self.assertEqual(new_container.get('Volumes')['/data'], volume_path)
  246. def test_start_container_passes_through_options(self):
  247. db = self.create_service('db')
  248. create_and_start_container(db, environment={'FOO': 'BAR'})
  249. self.assertEqual(db.containers()[0].environment['FOO'], 'BAR')
  250. def test_start_container_inherits_options_from_constructor(self):
  251. db = self.create_service('db', environment={'FOO': 'BAR'})
  252. create_and_start_container(db)
  253. self.assertEqual(db.containers()[0].environment['FOO'], 'BAR')
  254. def test_start_container_creates_links(self):
  255. db = self.create_service('db')
  256. web = self.create_service('web', links=[(db, None)])
  257. create_and_start_container(db)
  258. create_and_start_container(db)
  259. create_and_start_container(web)
  260. self.assertEqual(
  261. set(web.containers()[0].links()),
  262. set([
  263. 'composetest_db_1', 'db_1',
  264. 'composetest_db_2', 'db_2',
  265. 'db'])
  266. )
  267. def test_start_container_creates_links_with_names(self):
  268. db = self.create_service('db')
  269. web = self.create_service('web', links=[(db, 'custom_link_name')])
  270. create_and_start_container(db)
  271. create_and_start_container(db)
  272. create_and_start_container(web)
  273. self.assertEqual(
  274. set(web.containers()[0].links()),
  275. set([
  276. 'composetest_db_1', 'db_1',
  277. 'composetest_db_2', 'db_2',
  278. 'custom_link_name'])
  279. )
  280. def test_start_container_with_external_links(self):
  281. db = self.create_service('db')
  282. web = self.create_service('web', external_links=['composetest_db_1',
  283. 'composetest_db_2',
  284. 'composetest_db_3:db_3'])
  285. for _ in range(3):
  286. create_and_start_container(db)
  287. create_and_start_container(web)
  288. self.assertEqual(
  289. set(web.containers()[0].links()),
  290. set([
  291. 'composetest_db_1',
  292. 'composetest_db_2',
  293. 'db_3']),
  294. )
  295. def test_start_normal_container_does_not_create_links_to_its_own_service(self):
  296. db = self.create_service('db')
  297. create_and_start_container(db)
  298. create_and_start_container(db)
  299. c = create_and_start_container(db)
  300. self.assertEqual(set(c.links()), set([]))
  301. def test_start_one_off_container_creates_links_to_its_own_service(self):
  302. db = self.create_service('db')
  303. create_and_start_container(db)
  304. create_and_start_container(db)
  305. c = create_and_start_container(db, one_off=True)
  306. self.assertEqual(
  307. set(c.links()),
  308. set([
  309. 'composetest_db_1', 'db_1',
  310. 'composetest_db_2', 'db_2',
  311. 'db'])
  312. )
  313. def test_start_container_builds_images(self):
  314. service = Service(
  315. name='test',
  316. client=self.client,
  317. build='tests/fixtures/simple-dockerfile',
  318. project='composetest',
  319. )
  320. container = create_and_start_container(service)
  321. container.wait()
  322. self.assertIn('success', container.logs())
  323. self.assertEqual(len(self.client.images(name='composetest_test')), 1)
  324. def test_start_container_uses_tagged_image_if_it_exists(self):
  325. self.client.build('tests/fixtures/simple-dockerfile', tag='composetest_test')
  326. service = Service(
  327. name='test',
  328. client=self.client,
  329. build='this/does/not/exist/and/will/throw/error',
  330. project='composetest',
  331. )
  332. container = create_and_start_container(service)
  333. container.wait()
  334. self.assertIn('success', container.logs())
  335. def test_start_container_creates_ports(self):
  336. service = self.create_service('web', ports=[8000])
  337. container = create_and_start_container(service).inspect()
  338. self.assertEqual(list(container['NetworkSettings']['Ports'].keys()), ['8000/tcp'])
  339. self.assertNotEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8000')
  340. def test_build(self):
  341. base_dir = tempfile.mkdtemp()
  342. self.addCleanup(shutil.rmtree, base_dir)
  343. with open(os.path.join(base_dir, 'Dockerfile'), 'w') as f:
  344. f.write("FROM busybox\n")
  345. self.create_service('web', build=base_dir).build()
  346. self.assertEqual(len(self.client.images(name='composetest_web')), 1)
  347. def test_build_non_ascii_filename(self):
  348. base_dir = tempfile.mkdtemp()
  349. self.addCleanup(shutil.rmtree, base_dir)
  350. with open(os.path.join(base_dir, 'Dockerfile'), 'w') as f:
  351. f.write("FROM busybox\n")
  352. with open(os.path.join(base_dir, b'foo\xE2bar'), 'w') as f:
  353. f.write("hello world\n")
  354. self.create_service('web', build=six.text_type(base_dir)).build()
  355. self.assertEqual(len(self.client.images(name='composetest_web')), 1)
  356. def test_start_container_stays_unpriviliged(self):
  357. service = self.create_service('web')
  358. container = create_and_start_container(service).inspect()
  359. self.assertEqual(container['HostConfig']['Privileged'], False)
  360. def test_start_container_becomes_priviliged(self):
  361. service = self.create_service('web', privileged=True)
  362. container = create_and_start_container(service).inspect()
  363. self.assertEqual(container['HostConfig']['Privileged'], True)
  364. def test_expose_does_not_publish_ports(self):
  365. service = self.create_service('web', expose=[8000])
  366. container = create_and_start_container(service).inspect()
  367. self.assertEqual(container['NetworkSettings']['Ports'], {'8000/tcp': None})
  368. def test_start_container_creates_port_with_explicit_protocol(self):
  369. service = self.create_service('web', ports=['8000/udp'])
  370. container = create_and_start_container(service).inspect()
  371. self.assertEqual(list(container['NetworkSettings']['Ports'].keys()), ['8000/udp'])
  372. def test_start_container_creates_fixed_external_ports(self):
  373. service = self.create_service('web', ports=['8000:8000'])
  374. container = create_and_start_container(service).inspect()
  375. self.assertIn('8000/tcp', container['NetworkSettings']['Ports'])
  376. self.assertEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8000')
  377. def test_start_container_creates_fixed_external_ports_when_it_is_different_to_internal_port(self):
  378. service = self.create_service('web', ports=['8001:8000'])
  379. container = create_and_start_container(service).inspect()
  380. self.assertIn('8000/tcp', container['NetworkSettings']['Ports'])
  381. self.assertEqual(container['NetworkSettings']['Ports']['8000/tcp'][0]['HostPort'], '8001')
  382. def test_port_with_explicit_interface(self):
  383. service = self.create_service('web', ports=[
  384. '127.0.0.1:8001:8000',
  385. '0.0.0.0:9001:9000/udp',
  386. ])
  387. container = create_and_start_container(service).inspect()
  388. self.assertEqual(container['NetworkSettings']['Ports'], {
  389. '8000/tcp': [
  390. {
  391. 'HostIp': '127.0.0.1',
  392. 'HostPort': '8001',
  393. },
  394. ],
  395. '9000/udp': [
  396. {
  397. 'HostIp': '0.0.0.0',
  398. 'HostPort': '9001',
  399. },
  400. ],
  401. })
  402. def test_start_with_image_id(self):
  403. # Image id for the current busybox:latest
  404. service = self.create_service('foo', image='8c2e06607696')
  405. self.assertTrue(service.start_or_create_containers())
  406. def test_scale(self):
  407. service = self.create_service('web')
  408. service.scale(1)
  409. self.assertEqual(len(service.containers()), 1)
  410. # Ensure containers don't have stdout or stdin connected
  411. container = service.containers()[0]
  412. config = container.inspect()['Config']
  413. self.assertFalse(config['AttachStderr'])
  414. self.assertFalse(config['AttachStdout'])
  415. self.assertFalse(config['AttachStdin'])
  416. service.scale(3)
  417. self.assertEqual(len(service.containers()), 3)
  418. service.scale(1)
  419. self.assertEqual(len(service.containers()), 1)
  420. service.scale(0)
  421. self.assertEqual(len(service.containers()), 0)
  422. def test_scale_on_service_that_cannot_be_scaled(self):
  423. service = self.create_service('web', ports=['8000:8000'])
  424. self.assertRaises(CannotBeScaledError, lambda: service.scale(1))
  425. def test_scale_sets_ports(self):
  426. service = self.create_service('web', ports=['8000'])
  427. service.scale(2)
  428. containers = service.containers()
  429. self.assertEqual(len(containers), 2)
  430. for container in containers:
  431. self.assertEqual(list(container.inspect()['HostConfig']['PortBindings'].keys()), ['8000/tcp'])
  432. def test_network_mode_none(self):
  433. service = self.create_service('web', net='none')
  434. container = create_and_start_container(service)
  435. self.assertEqual(container.get('HostConfig.NetworkMode'), 'none')
  436. def test_network_mode_bridged(self):
  437. service = self.create_service('web', net='bridge')
  438. container = create_and_start_container(service)
  439. self.assertEqual(container.get('HostConfig.NetworkMode'), 'bridge')
  440. def test_network_mode_host(self):
  441. service = self.create_service('web', net='host')
  442. container = create_and_start_container(service)
  443. self.assertEqual(container.get('HostConfig.NetworkMode'), 'host')
  444. def test_pid_mode_none_defined(self):
  445. service = self.create_service('web', pid=None)
  446. container = create_and_start_container(service)
  447. self.assertEqual(container.get('HostConfig.PidMode'), '')
  448. def test_pid_mode_host(self):
  449. service = self.create_service('web', pid='host')
  450. container = create_and_start_container(service)
  451. self.assertEqual(container.get('HostConfig.PidMode'), 'host')
  452. def test_dns_no_value(self):
  453. service = self.create_service('web')
  454. container = create_and_start_container(service)
  455. self.assertIsNone(container.get('HostConfig.Dns'))
  456. def test_dns_single_value(self):
  457. service = self.create_service('web', dns='8.8.8.8')
  458. container = create_and_start_container(service)
  459. self.assertEqual(container.get('HostConfig.Dns'), ['8.8.8.8'])
  460. def test_dns_list(self):
  461. service = self.create_service('web', dns=['8.8.8.8', '9.9.9.9'])
  462. container = create_and_start_container(service)
  463. self.assertEqual(container.get('HostConfig.Dns'), ['8.8.8.8', '9.9.9.9'])
  464. def test_restart_always_value(self):
  465. service = self.create_service('web', restart='always')
  466. container = create_and_start_container(service)
  467. self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'always')
  468. def test_restart_on_failure_value(self):
  469. service = self.create_service('web', restart='on-failure:5')
  470. container = create_and_start_container(service)
  471. self.assertEqual(container.get('HostConfig.RestartPolicy.Name'), 'on-failure')
  472. self.assertEqual(container.get('HostConfig.RestartPolicy.MaximumRetryCount'), 5)
  473. def test_cap_add_list(self):
  474. service = self.create_service('web', cap_add=['SYS_ADMIN', 'NET_ADMIN'])
  475. container = create_and_start_container(service)
  476. self.assertEqual(container.get('HostConfig.CapAdd'), ['SYS_ADMIN', 'NET_ADMIN'])
  477. def test_cap_drop_list(self):
  478. service = self.create_service('web', cap_drop=['SYS_ADMIN', 'NET_ADMIN'])
  479. container = create_and_start_container(service)
  480. self.assertEqual(container.get('HostConfig.CapDrop'), ['SYS_ADMIN', 'NET_ADMIN'])
  481. def test_dns_search_no_value(self):
  482. service = self.create_service('web')
  483. container = create_and_start_container(service)
  484. self.assertIsNone(container.get('HostConfig.DnsSearch'))
  485. def test_dns_search_single_value(self):
  486. service = self.create_service('web', dns_search='example.com')
  487. container = create_and_start_container(service)
  488. self.assertEqual(container.get('HostConfig.DnsSearch'), ['example.com'])
  489. def test_dns_search_list(self):
  490. service = self.create_service('web', dns_search=['dc1.example.com', 'dc2.example.com'])
  491. container = create_and_start_container(service)
  492. self.assertEqual(container.get('HostConfig.DnsSearch'), ['dc1.example.com', 'dc2.example.com'])
  493. def test_working_dir_param(self):
  494. service = self.create_service('container', working_dir='/working/dir/sample')
  495. container = service.create_container()
  496. self.assertEqual(container.get('Config.WorkingDir'), '/working/dir/sample')
  497. def test_split_env(self):
  498. service = self.create_service('web', environment=['NORMAL=F1', 'CONTAINS_EQUALS=F=2', 'TRAILING_EQUALS='])
  499. env = create_and_start_container(service).environment
  500. for k, v in {'NORMAL': 'F1', 'CONTAINS_EQUALS': 'F=2', 'TRAILING_EQUALS': ''}.items():
  501. self.assertEqual(env[k], v)
  502. def test_env_from_file_combined_with_env(self):
  503. service = self.create_service('web', environment=['ONE=1', 'TWO=2', 'THREE=3'], env_file=['tests/fixtures/env/one.env', 'tests/fixtures/env/two.env'])
  504. env = create_and_start_container(service).environment
  505. for k, v in {'ONE': '1', 'TWO': '2', 'THREE': '3', 'FOO': 'baz', 'DOO': 'dah'}.items():
  506. self.assertEqual(env[k], v)
  507. @mock.patch.dict(os.environ)
  508. def test_resolve_env(self):
  509. os.environ['FILE_DEF'] = 'E1'
  510. os.environ['FILE_DEF_EMPTY'] = 'E2'
  511. os.environ['ENV_DEF'] = 'E3'
  512. service = self.create_service('web', environment={'FILE_DEF': 'F1', 'FILE_DEF_EMPTY': '', 'ENV_DEF': None, 'NO_DEF': None})
  513. env = create_and_start_container(service).environment
  514. for k, v in {'FILE_DEF': 'F1', 'FILE_DEF_EMPTY': '', 'ENV_DEF': 'E3', 'NO_DEF': ''}.items():
  515. self.assertEqual(env[k], v)
  516. def test_labels(self):
  517. labels_dict = {
  518. 'com.example.description': "Accounting webapp",
  519. 'com.example.department': "Finance",
  520. 'com.example.label-with-empty-value': "",
  521. }
  522. service = self.create_service('web', labels=labels_dict)
  523. labels = create_and_start_container(service).labels.items()
  524. for pair in labels_dict.items():
  525. self.assertIn(pair, labels)
  526. labels_list = ["%s=%s" % pair for pair in labels_dict.items()]
  527. service = self.create_service('web', labels=labels_list)
  528. labels = create_and_start_container(service).labels.items()
  529. for pair in labels_dict.items():
  530. self.assertIn(pair, labels)
  531. def test_empty_labels(self):
  532. labels_list = ['foo', 'bar']
  533. service = self.create_service('web', labels=labels_list)
  534. labels = create_and_start_container(service).labels.items()
  535. for name in labels_list:
  536. self.assertIn((name, ''), labels)