service_test.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import docker
  4. from docker.errors import APIError
  5. from .. import mock
  6. from .. import unittest
  7. from compose.config.types import VolumeFromSpec
  8. from compose.config.types import VolumeSpec
  9. from compose.const import LABEL_CONFIG_HASH
  10. from compose.const import LABEL_ONE_OFF
  11. from compose.const import LABEL_PROJECT
  12. from compose.const import LABEL_SERVICE
  13. from compose.container import Container
  14. from compose.service import build_ulimits
  15. from compose.service import build_volume_binding
  16. from compose.service import ContainerNet
  17. from compose.service import get_container_data_volumes
  18. from compose.service import ImageType
  19. from compose.service import merge_volume_bindings
  20. from compose.service import NeedsBuildError
  21. from compose.service import Net
  22. from compose.service import NoSuchImageError
  23. from compose.service import parse_repository_tag
  24. from compose.service import Service
  25. from compose.service import ServiceNet
  26. from compose.service import warn_on_masked_volume
  27. class ServiceTest(unittest.TestCase):
  28. def setUp(self):
  29. self.mock_client = mock.create_autospec(docker.Client)
  30. def test_containers(self):
  31. service = Service('db', self.mock_client, 'myproject', image='foo')
  32. self.mock_client.containers.return_value = []
  33. self.assertEqual(list(service.containers()), [])
  34. def test_containers_with_containers(self):
  35. self.mock_client.containers.return_value = [
  36. dict(Name=str(i), Image='foo', Id=i) for i in range(3)
  37. ]
  38. service = Service('db', self.mock_client, 'myproject', image='foo')
  39. self.assertEqual([c.id for c in service.containers()], list(range(3)))
  40. expected_labels = [
  41. '{0}=myproject'.format(LABEL_PROJECT),
  42. '{0}=db'.format(LABEL_SERVICE),
  43. '{0}=False'.format(LABEL_ONE_OFF),
  44. ]
  45. self.mock_client.containers.assert_called_once_with(
  46. all=False,
  47. filters={'label': expected_labels})
  48. def test_container_without_name(self):
  49. self.mock_client.containers.return_value = [
  50. {'Image': 'foo', 'Id': '1', 'Name': '1'},
  51. {'Image': 'foo', 'Id': '2', 'Name': None},
  52. {'Image': 'foo', 'Id': '3'},
  53. ]
  54. service = Service('db', self.mock_client, 'myproject', image='foo')
  55. self.assertEqual([c.id for c in service.containers()], ['1'])
  56. self.assertEqual(service._next_container_number(), 2)
  57. self.assertEqual(service.get_container(1).id, '1')
  58. def test_get_volumes_from_container(self):
  59. container_id = 'aabbccddee'
  60. service = Service(
  61. 'test',
  62. image='foo',
  63. volumes_from=[
  64. VolumeFromSpec(
  65. mock.Mock(id=container_id, spec=Container),
  66. 'rw',
  67. 'container')])
  68. self.assertEqual(service._get_volumes_from(), [container_id + ':rw'])
  69. def test_get_volumes_from_container_read_only(self):
  70. container_id = 'aabbccddee'
  71. service = Service(
  72. 'test',
  73. image='foo',
  74. volumes_from=[
  75. VolumeFromSpec(
  76. mock.Mock(id=container_id, spec=Container),
  77. 'ro',
  78. 'container')])
  79. self.assertEqual(service._get_volumes_from(), [container_id + ':ro'])
  80. def test_get_volumes_from_service_container_exists(self):
  81. container_ids = ['aabbccddee', '12345']
  82. from_service = mock.create_autospec(Service)
  83. from_service.containers.return_value = [
  84. mock.Mock(id=container_id, spec=Container)
  85. for container_id in container_ids
  86. ]
  87. service = Service(
  88. 'test',
  89. volumes_from=[VolumeFromSpec(from_service, 'rw', 'service')],
  90. image='foo')
  91. self.assertEqual(service._get_volumes_from(), [container_ids[0] + ":rw"])
  92. def test_get_volumes_from_service_container_exists_with_flags(self):
  93. for mode in ['ro', 'rw', 'z', 'rw,z', 'z,rw']:
  94. container_ids = ['aabbccddee:' + mode, '12345:' + mode]
  95. from_service = mock.create_autospec(Service)
  96. from_service.containers.return_value = [
  97. mock.Mock(id=container_id.split(':')[0], spec=Container)
  98. for container_id in container_ids
  99. ]
  100. service = Service(
  101. 'test',
  102. volumes_from=[VolumeFromSpec(from_service, mode, 'service')],
  103. image='foo')
  104. self.assertEqual(service._get_volumes_from(), [container_ids[0]])
  105. def test_get_volumes_from_service_no_container(self):
  106. container_id = 'abababab'
  107. from_service = mock.create_autospec(Service)
  108. from_service.containers.return_value = []
  109. from_service.create_container.return_value = mock.Mock(
  110. id=container_id,
  111. spec=Container)
  112. service = Service(
  113. 'test',
  114. image='foo',
  115. volumes_from=[VolumeFromSpec(from_service, 'rw', 'service')])
  116. self.assertEqual(service._get_volumes_from(), [container_id + ':rw'])
  117. from_service.create_container.assert_called_once_with()
  118. def test_split_domainname_none(self):
  119. service = Service('foo', image='foo', hostname='name', client=self.mock_client)
  120. opts = service._get_container_create_options({'image': 'foo'}, 1)
  121. self.assertEqual(opts['hostname'], 'name', 'hostname')
  122. self.assertFalse('domainname' in opts, 'domainname')
  123. def test_memory_swap_limit(self):
  124. self.mock_client.create_host_config.return_value = {}
  125. service = Service(name='foo', image='foo', hostname='name', client=self.mock_client, mem_limit=1000000000, memswap_limit=2000000000)
  126. service._get_container_create_options({'some': 'overrides'}, 1)
  127. self.assertTrue(self.mock_client.create_host_config.called)
  128. self.assertEqual(
  129. self.mock_client.create_host_config.call_args[1]['mem_limit'],
  130. 1000000000
  131. )
  132. self.assertEqual(
  133. self.mock_client.create_host_config.call_args[1]['memswap_limit'],
  134. 2000000000
  135. )
  136. def test_cgroup_parent(self):
  137. self.mock_client.create_host_config.return_value = {}
  138. service = Service(name='foo', image='foo', hostname='name', client=self.mock_client, cgroup_parent='test')
  139. service._get_container_create_options({'some': 'overrides'}, 1)
  140. self.assertTrue(self.mock_client.create_host_config.called)
  141. self.assertEqual(
  142. self.mock_client.create_host_config.call_args[1]['cgroup_parent'],
  143. 'test'
  144. )
  145. def test_log_opt(self):
  146. self.mock_client.create_host_config.return_value = {}
  147. log_opt = {'syslog-address': 'tcp://192.168.0.42:123'}
  148. logging = {'driver': 'syslog', 'options': log_opt}
  149. service = Service(name='foo', image='foo', hostname='name', client=self.mock_client, logging=logging)
  150. service._get_container_create_options({'some': 'overrides'}, 1)
  151. self.assertTrue(self.mock_client.create_host_config.called)
  152. self.assertEqual(
  153. self.mock_client.create_host_config.call_args[1]['log_config'],
  154. {'Type': 'syslog', 'Config': {'syslog-address': 'tcp://192.168.0.42:123'}}
  155. )
  156. def test_split_domainname_fqdn(self):
  157. service = Service(
  158. 'foo',
  159. hostname='name.domain.tld',
  160. image='foo',
  161. client=self.mock_client)
  162. opts = service._get_container_create_options({'image': 'foo'}, 1)
  163. self.assertEqual(opts['hostname'], 'name', 'hostname')
  164. self.assertEqual(opts['domainname'], 'domain.tld', 'domainname')
  165. def test_split_domainname_both(self):
  166. service = Service(
  167. 'foo',
  168. hostname='name',
  169. image='foo',
  170. domainname='domain.tld',
  171. client=self.mock_client)
  172. opts = service._get_container_create_options({'image': 'foo'}, 1)
  173. self.assertEqual(opts['hostname'], 'name', 'hostname')
  174. self.assertEqual(opts['domainname'], 'domain.tld', 'domainname')
  175. def test_split_domainname_weird(self):
  176. service = Service(
  177. 'foo',
  178. hostname='name.sub',
  179. domainname='domain.tld',
  180. image='foo',
  181. client=self.mock_client)
  182. opts = service._get_container_create_options({'image': 'foo'}, 1)
  183. self.assertEqual(opts['hostname'], 'name.sub', 'hostname')
  184. self.assertEqual(opts['domainname'], 'domain.tld', 'domainname')
  185. def test_no_default_hostname_when_not_using_networking(self):
  186. service = Service(
  187. 'foo',
  188. image='foo',
  189. use_networking=False,
  190. client=self.mock_client,
  191. )
  192. opts = service._get_container_create_options({'image': 'foo'}, 1)
  193. self.assertIsNone(opts.get('hostname'))
  194. def test_get_container_create_options_with_name_option(self):
  195. service = Service(
  196. 'foo',
  197. image='foo',
  198. client=self.mock_client,
  199. container_name='foo1')
  200. name = 'the_new_name'
  201. opts = service._get_container_create_options(
  202. {'name': name},
  203. 1,
  204. one_off=True)
  205. self.assertEqual(opts['name'], name)
  206. def test_get_container_create_options_does_not_mutate_options(self):
  207. labels = {'thing': 'real'}
  208. environment = {'also': 'real'}
  209. service = Service(
  210. 'foo',
  211. image='foo',
  212. labels=dict(labels),
  213. client=self.mock_client,
  214. environment=dict(environment),
  215. )
  216. self.mock_client.inspect_image.return_value = {'Id': 'abcd'}
  217. prev_container = mock.Mock(
  218. id='ababab',
  219. image_config={'ContainerConfig': {}})
  220. prev_container.get.return_value = None
  221. opts = service._get_container_create_options(
  222. {},
  223. 1,
  224. previous_container=prev_container)
  225. self.assertEqual(service.options['labels'], labels)
  226. self.assertEqual(service.options['environment'], environment)
  227. self.assertEqual(
  228. opts['labels'][LABEL_CONFIG_HASH],
  229. '3c85881a8903b9d73a06c41860c8be08acce1494ab4cf8408375966dccd714de')
  230. self.assertEqual(
  231. opts['environment'],
  232. {
  233. 'affinity:container': '=ababab',
  234. 'also': 'real',
  235. }
  236. )
  237. def test_get_container_not_found(self):
  238. self.mock_client.containers.return_value = []
  239. service = Service('foo', client=self.mock_client, image='foo')
  240. self.assertRaises(ValueError, service.get_container)
  241. @mock.patch('compose.service.Container', autospec=True)
  242. def test_get_container(self, mock_container_class):
  243. container_dict = dict(Name='default_foo_2')
  244. self.mock_client.containers.return_value = [container_dict]
  245. service = Service('foo', image='foo', client=self.mock_client)
  246. container = service.get_container(number=2)
  247. self.assertEqual(container, mock_container_class.from_ps.return_value)
  248. mock_container_class.from_ps.assert_called_once_with(
  249. self.mock_client, container_dict)
  250. @mock.patch('compose.service.log', autospec=True)
  251. def test_pull_image(self, mock_log):
  252. service = Service('foo', client=self.mock_client, image='someimage:sometag')
  253. service.pull()
  254. self.mock_client.pull.assert_called_once_with(
  255. 'someimage',
  256. tag='sometag',
  257. stream=True)
  258. mock_log.info.assert_called_once_with('Pulling foo (someimage:sometag)...')
  259. def test_pull_image_no_tag(self):
  260. service = Service('foo', client=self.mock_client, image='ababab')
  261. service.pull()
  262. self.mock_client.pull.assert_called_once_with(
  263. 'ababab',
  264. tag='latest',
  265. stream=True)
  266. @mock.patch('compose.service.log', autospec=True)
  267. def test_pull_image_digest(self, mock_log):
  268. service = Service('foo', client=self.mock_client, image='someimage@sha256:1234')
  269. service.pull()
  270. self.mock_client.pull.assert_called_once_with(
  271. 'someimage',
  272. tag='sha256:1234',
  273. stream=True)
  274. mock_log.info.assert_called_once_with('Pulling foo (someimage@sha256:1234)...')
  275. @mock.patch('compose.service.Container', autospec=True)
  276. def test_recreate_container(self, _):
  277. mock_container = mock.create_autospec(Container)
  278. service = Service('foo', client=self.mock_client, image='someimage')
  279. service.image = lambda: {'Id': 'abc123'}
  280. new_container = service.recreate_container(mock_container)
  281. mock_container.stop.assert_called_once_with(timeout=10)
  282. mock_container.rename_to_tmp_name.assert_called_once_with()
  283. new_container.start.assert_called_once_with()
  284. mock_container.remove.assert_called_once_with()
  285. @mock.patch('compose.service.Container', autospec=True)
  286. def test_recreate_container_with_timeout(self, _):
  287. mock_container = mock.create_autospec(Container)
  288. self.mock_client.inspect_image.return_value = {'Id': 'abc123'}
  289. service = Service('foo', client=self.mock_client, image='someimage')
  290. service.recreate_container(mock_container, timeout=1)
  291. mock_container.stop.assert_called_once_with(timeout=1)
  292. def test_parse_repository_tag(self):
  293. self.assertEqual(parse_repository_tag("root"), ("root", "", ":"))
  294. self.assertEqual(parse_repository_tag("root:tag"), ("root", "tag", ":"))
  295. self.assertEqual(parse_repository_tag("user/repo"), ("user/repo", "", ":"))
  296. self.assertEqual(parse_repository_tag("user/repo:tag"), ("user/repo", "tag", ":"))
  297. self.assertEqual(parse_repository_tag("url:5000/repo"), ("url:5000/repo", "", ":"))
  298. self.assertEqual(parse_repository_tag("url:5000/repo:tag"), ("url:5000/repo", "tag", ":"))
  299. self.assertEqual(parse_repository_tag("root@sha256:digest"), ("root", "sha256:digest", "@"))
  300. self.assertEqual(parse_repository_tag("user/repo@sha256:digest"), ("user/repo", "sha256:digest", "@"))
  301. self.assertEqual(parse_repository_tag("url:5000/repo@sha256:digest"), ("url:5000/repo", "sha256:digest", "@"))
  302. def test_create_container_with_build(self):
  303. service = Service('foo', client=self.mock_client, build={'context': '.'})
  304. self.mock_client.inspect_image.side_effect = [
  305. NoSuchImageError,
  306. {'Id': 'abc123'},
  307. ]
  308. self.mock_client.build.return_value = [
  309. '{"stream": "Successfully built abcd"}',
  310. ]
  311. service.create_container(do_build=True)
  312. self.mock_client.build.assert_called_once_with(
  313. tag='default_foo',
  314. dockerfile=None,
  315. stream=True,
  316. path='.',
  317. pull=False,
  318. forcerm=False,
  319. nocache=False,
  320. rm=True,
  321. buildargs=None,
  322. )
  323. def test_create_container_no_build(self):
  324. service = Service('foo', client=self.mock_client, build={'context': '.'})
  325. self.mock_client.inspect_image.return_value = {'Id': 'abc123'}
  326. service.create_container(do_build=False)
  327. self.assertFalse(self.mock_client.build.called)
  328. def test_create_container_no_build_but_needs_build(self):
  329. service = Service('foo', client=self.mock_client, build={'context': '.'})
  330. self.mock_client.inspect_image.side_effect = NoSuchImageError
  331. with self.assertRaises(NeedsBuildError):
  332. service.create_container(do_build=False)
  333. def test_build_does_not_pull(self):
  334. self.mock_client.build.return_value = [
  335. b'{"stream": "Successfully built 12345"}',
  336. ]
  337. service = Service('foo', client=self.mock_client, build={'context': '.'})
  338. service.build()
  339. self.assertEqual(self.mock_client.build.call_count, 1)
  340. self.assertFalse(self.mock_client.build.call_args[1]['pull'])
  341. def test_config_dict(self):
  342. self.mock_client.inspect_image.return_value = {'Id': 'abcd'}
  343. service = Service(
  344. 'foo',
  345. image='example.com/foo',
  346. client=self.mock_client,
  347. net=ServiceNet(Service('other')),
  348. links=[(Service('one'), 'one')],
  349. volumes_from=[VolumeFromSpec(Service('two'), 'rw', 'service')])
  350. config_dict = service.config_dict()
  351. expected = {
  352. 'image_id': 'abcd',
  353. 'options': {'image': 'example.com/foo'},
  354. 'links': [('one', 'one')],
  355. 'net': 'other',
  356. 'volumes_from': [('two', 'rw')],
  357. }
  358. self.assertEqual(config_dict, expected)
  359. def test_config_dict_with_net_from_container(self):
  360. self.mock_client.inspect_image.return_value = {'Id': 'abcd'}
  361. container = Container(
  362. self.mock_client,
  363. {'Id': 'aaabbb', 'Name': '/foo_1'})
  364. service = Service(
  365. 'foo',
  366. image='example.com/foo',
  367. client=self.mock_client,
  368. net=container)
  369. config_dict = service.config_dict()
  370. expected = {
  371. 'image_id': 'abcd',
  372. 'options': {'image': 'example.com/foo'},
  373. 'links': [],
  374. 'net': 'aaabbb',
  375. 'volumes_from': [],
  376. }
  377. self.assertEqual(config_dict, expected)
  378. def test_remove_image_none(self):
  379. web = Service('web', image='example', client=self.mock_client)
  380. assert not web.remove_image(ImageType.none)
  381. assert not self.mock_client.remove_image.called
  382. def test_remove_image_local_with_image_name_doesnt_remove(self):
  383. web = Service('web', image='example', client=self.mock_client)
  384. assert not web.remove_image(ImageType.local)
  385. assert not self.mock_client.remove_image.called
  386. def test_remove_image_local_without_image_name_does_remove(self):
  387. web = Service('web', build='.', client=self.mock_client)
  388. assert web.remove_image(ImageType.local)
  389. self.mock_client.remove_image.assert_called_once_with(web.image_name)
  390. def test_remove_image_all_does_remove(self):
  391. web = Service('web', image='example', client=self.mock_client)
  392. assert web.remove_image(ImageType.all)
  393. self.mock_client.remove_image.assert_called_once_with(web.image_name)
  394. def test_remove_image_with_error(self):
  395. self.mock_client.remove_image.side_effect = error = APIError(
  396. message="testing",
  397. response={},
  398. explanation="Boom")
  399. web = Service('web', image='example', client=self.mock_client)
  400. with mock.patch('compose.service.log', autospec=True) as mock_log:
  401. assert not web.remove_image(ImageType.all)
  402. mock_log.error.assert_called_once_with(
  403. "Failed to remove image for service %s: %s", web.name, error)
  404. def test_specifies_host_port_with_no_ports(self):
  405. service = Service(
  406. 'foo',
  407. image='foo')
  408. self.assertEqual(service.specifies_host_port(), False)
  409. def test_specifies_host_port_with_container_port(self):
  410. service = Service(
  411. 'foo',
  412. image='foo',
  413. ports=["2000"])
  414. self.assertEqual(service.specifies_host_port(), False)
  415. def test_specifies_host_port_with_host_port(self):
  416. service = Service(
  417. 'foo',
  418. image='foo',
  419. ports=["1000:2000"])
  420. self.assertEqual(service.specifies_host_port(), True)
  421. def test_specifies_host_port_with_host_ip_no_port(self):
  422. service = Service(
  423. 'foo',
  424. image='foo',
  425. ports=["127.0.0.1::2000"])
  426. self.assertEqual(service.specifies_host_port(), False)
  427. def test_specifies_host_port_with_host_ip_and_port(self):
  428. service = Service(
  429. 'foo',
  430. image='foo',
  431. ports=["127.0.0.1:1000:2000"])
  432. self.assertEqual(service.specifies_host_port(), True)
  433. def test_specifies_host_port_with_container_port_range(self):
  434. service = Service(
  435. 'foo',
  436. image='foo',
  437. ports=["2000-3000"])
  438. self.assertEqual(service.specifies_host_port(), False)
  439. def test_specifies_host_port_with_host_port_range(self):
  440. service = Service(
  441. 'foo',
  442. image='foo',
  443. ports=["1000-2000:2000-3000"])
  444. self.assertEqual(service.specifies_host_port(), True)
  445. def test_specifies_host_port_with_host_ip_no_port_range(self):
  446. service = Service(
  447. 'foo',
  448. image='foo',
  449. ports=["127.0.0.1::2000-3000"])
  450. self.assertEqual(service.specifies_host_port(), False)
  451. def test_specifies_host_port_with_host_ip_and_port_range(self):
  452. service = Service(
  453. 'foo',
  454. image='foo',
  455. ports=["127.0.0.1:1000-2000:2000-3000"])
  456. self.assertEqual(service.specifies_host_port(), True)
  457. def test_image_name_from_config(self):
  458. image_name = 'example/web:latest'
  459. service = Service('foo', image=image_name)
  460. assert service.image_name == image_name
  461. def test_image_name_default(self):
  462. service = Service('foo', project='testing')
  463. assert service.image_name == 'testing_foo'
  464. def sort_by_name(dictionary_list):
  465. return sorted(dictionary_list, key=lambda k: k['name'])
  466. class BuildUlimitsTestCase(unittest.TestCase):
  467. def test_build_ulimits_with_dict(self):
  468. ulimits = build_ulimits(
  469. {
  470. 'nofile': {'soft': 10000, 'hard': 20000},
  471. 'nproc': {'soft': 65535, 'hard': 65535}
  472. }
  473. )
  474. expected = [
  475. {'name': 'nofile', 'soft': 10000, 'hard': 20000},
  476. {'name': 'nproc', 'soft': 65535, 'hard': 65535}
  477. ]
  478. assert sort_by_name(ulimits) == sort_by_name(expected)
  479. def test_build_ulimits_with_ints(self):
  480. ulimits = build_ulimits({'nofile': 20000, 'nproc': 65535})
  481. expected = [
  482. {'name': 'nofile', 'soft': 20000, 'hard': 20000},
  483. {'name': 'nproc', 'soft': 65535, 'hard': 65535}
  484. ]
  485. assert sort_by_name(ulimits) == sort_by_name(expected)
  486. def test_build_ulimits_with_integers_and_dicts(self):
  487. ulimits = build_ulimits(
  488. {
  489. 'nproc': 65535,
  490. 'nofile': {'soft': 10000, 'hard': 20000}
  491. }
  492. )
  493. expected = [
  494. {'name': 'nofile', 'soft': 10000, 'hard': 20000},
  495. {'name': 'nproc', 'soft': 65535, 'hard': 65535}
  496. ]
  497. assert sort_by_name(ulimits) == sort_by_name(expected)
  498. class NetTestCase(unittest.TestCase):
  499. def test_net(self):
  500. net = Net('host')
  501. self.assertEqual(net.id, 'host')
  502. self.assertEqual(net.mode, 'host')
  503. self.assertEqual(net.service_name, None)
  504. def test_net_container(self):
  505. container_id = 'abcd'
  506. net = ContainerNet(Container(None, {'Id': container_id}))
  507. self.assertEqual(net.id, container_id)
  508. self.assertEqual(net.mode, 'container:' + container_id)
  509. self.assertEqual(net.service_name, None)
  510. def test_net_service(self):
  511. container_id = 'bbbb'
  512. service_name = 'web'
  513. mock_client = mock.create_autospec(docker.Client)
  514. mock_client.containers.return_value = [
  515. {'Id': container_id, 'Name': container_id, 'Image': 'abcd'},
  516. ]
  517. service = Service(name=service_name, client=mock_client)
  518. net = ServiceNet(service)
  519. self.assertEqual(net.id, service_name)
  520. self.assertEqual(net.mode, 'container:' + container_id)
  521. self.assertEqual(net.service_name, service_name)
  522. def test_net_service_no_containers(self):
  523. service_name = 'web'
  524. mock_client = mock.create_autospec(docker.Client)
  525. mock_client.containers.return_value = []
  526. service = Service(name=service_name, client=mock_client)
  527. net = ServiceNet(service)
  528. self.assertEqual(net.id, service_name)
  529. self.assertEqual(net.mode, None)
  530. self.assertEqual(net.service_name, service_name)
  531. def build_mount(destination, source, mode='rw'):
  532. return {'Source': source, 'Destination': destination, 'Mode': mode}
  533. class ServiceVolumesTest(unittest.TestCase):
  534. def setUp(self):
  535. self.mock_client = mock.create_autospec(docker.Client)
  536. def test_build_volume_binding(self):
  537. binding = build_volume_binding(VolumeSpec.parse('/outside:/inside'))
  538. assert binding == ('/inside', '/outside:/inside:rw')
  539. def test_get_container_data_volumes(self):
  540. options = [VolumeSpec.parse(v) for v in [
  541. '/host/volume:/host/volume:ro',
  542. '/new/volume',
  543. '/existing/volume',
  544. ]]
  545. self.mock_client.inspect_image.return_value = {
  546. 'ContainerConfig': {
  547. 'Volumes': {
  548. '/mnt/image/data': {},
  549. }
  550. }
  551. }
  552. container = Container(self.mock_client, {
  553. 'Image': 'ababab',
  554. 'Mounts': [
  555. {
  556. 'Source': '/host/volume',
  557. 'Destination': '/host/volume',
  558. 'Mode': '',
  559. 'RW': True,
  560. 'Name': 'hostvolume',
  561. }, {
  562. 'Source': '/var/lib/docker/aaaaaaaa',
  563. 'Destination': '/existing/volume',
  564. 'Mode': '',
  565. 'RW': True,
  566. 'Name': 'existingvolume',
  567. }, {
  568. 'Source': '/var/lib/docker/bbbbbbbb',
  569. 'Destination': '/removed/volume',
  570. 'Mode': '',
  571. 'RW': True,
  572. 'Name': 'removedvolume',
  573. }, {
  574. 'Source': '/var/lib/docker/cccccccc',
  575. 'Destination': '/mnt/image/data',
  576. 'Mode': '',
  577. 'RW': True,
  578. 'Name': 'imagedata',
  579. },
  580. ]
  581. }, has_been_inspected=True)
  582. expected = [
  583. VolumeSpec.parse('/var/lib/docker/aaaaaaaa:/existing/volume:rw'),
  584. VolumeSpec.parse('/var/lib/docker/cccccccc:/mnt/image/data:rw'),
  585. ]
  586. volumes = get_container_data_volumes(container, options)
  587. assert sorted(volumes) == sorted(expected)
  588. def test_merge_volume_bindings(self):
  589. options = [
  590. VolumeSpec.parse('/host/volume:/host/volume:ro'),
  591. VolumeSpec.parse('/host/rw/volume:/host/rw/volume'),
  592. VolumeSpec.parse('/new/volume'),
  593. VolumeSpec.parse('/existing/volume'),
  594. ]
  595. self.mock_client.inspect_image.return_value = {
  596. 'ContainerConfig': {'Volumes': {}}
  597. }
  598. intermediate_container = Container(self.mock_client, {
  599. 'Image': 'ababab',
  600. 'Mounts': [{
  601. 'Source': '/var/lib/docker/aaaaaaaa',
  602. 'Destination': '/existing/volume',
  603. 'Mode': '',
  604. 'RW': True,
  605. 'Name': 'existingvolume',
  606. }],
  607. }, has_been_inspected=True)
  608. expected = [
  609. '/host/volume:/host/volume:ro',
  610. '/host/rw/volume:/host/rw/volume:rw',
  611. '/var/lib/docker/aaaaaaaa:/existing/volume:rw',
  612. ]
  613. binds = merge_volume_bindings(options, intermediate_container)
  614. self.assertEqual(set(binds), set(expected))
  615. def test_mount_same_host_path_to_two_volumes(self):
  616. service = Service(
  617. 'web',
  618. image='busybox',
  619. volumes=[
  620. VolumeSpec.parse('/host/path:/data1'),
  621. VolumeSpec.parse('/host/path:/data2'),
  622. ],
  623. client=self.mock_client,
  624. )
  625. self.mock_client.inspect_image.return_value = {
  626. 'Id': 'ababab',
  627. 'ContainerConfig': {
  628. 'Volumes': {}
  629. }
  630. }
  631. service._get_container_create_options(
  632. override_options={},
  633. number=1,
  634. )
  635. self.assertEqual(
  636. set(self.mock_client.create_host_config.call_args[1]['binds']),
  637. set([
  638. '/host/path:/data1:rw',
  639. '/host/path:/data2:rw',
  640. ]),
  641. )
  642. def test_different_host_path_in_container_json(self):
  643. service = Service(
  644. 'web',
  645. image='busybox',
  646. volumes=[VolumeSpec.parse('/host/path:/data')],
  647. client=self.mock_client,
  648. )
  649. self.mock_client.inspect_image.return_value = {
  650. 'Id': 'ababab',
  651. 'ContainerConfig': {
  652. 'Volumes': {
  653. '/data': {},
  654. }
  655. }
  656. }
  657. self.mock_client.inspect_container.return_value = {
  658. 'Id': '123123123',
  659. 'Image': 'ababab',
  660. 'Mounts': [
  661. {
  662. 'Destination': '/data',
  663. 'Source': '/mnt/sda1/host/path',
  664. 'Mode': '',
  665. 'RW': True,
  666. 'Driver': 'local',
  667. 'Name': 'abcdefff1234'
  668. },
  669. ]
  670. }
  671. service._get_container_create_options(
  672. override_options={},
  673. number=1,
  674. previous_container=Container(self.mock_client, {'Id': '123123123'}),
  675. )
  676. self.assertEqual(
  677. self.mock_client.create_host_config.call_args[1]['binds'],
  678. ['/mnt/sda1/host/path:/data:rw'],
  679. )
  680. def test_warn_on_masked_volume_no_warning_when_no_container_volumes(self):
  681. volumes_option = [VolumeSpec('/home/user', '/path', 'rw')]
  682. container_volumes = []
  683. service = 'service_name'
  684. with mock.patch('compose.service.log', autospec=True) as mock_log:
  685. warn_on_masked_volume(volumes_option, container_volumes, service)
  686. assert not mock_log.warn.called
  687. def test_warn_on_masked_volume_when_masked(self):
  688. volumes_option = [VolumeSpec('/home/user', '/path', 'rw')]
  689. container_volumes = [
  690. VolumeSpec('/var/lib/docker/path', '/path', 'rw'),
  691. VolumeSpec('/var/lib/docker/path', '/other', 'rw'),
  692. ]
  693. service = 'service_name'
  694. with mock.patch('compose.service.log', autospec=True) as mock_log:
  695. warn_on_masked_volume(volumes_option, container_volumes, service)
  696. mock_log.warn.assert_called_once_with(mock.ANY)
  697. def test_warn_on_masked_no_warning_with_same_path(self):
  698. volumes_option = [VolumeSpec('/home/user', '/path', 'rw')]
  699. container_volumes = [VolumeSpec('/home/user', '/path', 'rw')]
  700. service = 'service_name'
  701. with mock.patch('compose.service.log', autospec=True) as mock_log:
  702. warn_on_masked_volume(volumes_option, container_volumes, service)
  703. assert not mock_log.warn.called
  704. def test_warn_on_masked_no_warning_with_container_only_option(self):
  705. volumes_option = [VolumeSpec(None, '/path', 'rw')]
  706. container_volumes = [
  707. VolumeSpec('/var/lib/docker/volume/path', '/path', 'rw')
  708. ]
  709. service = 'service_name'
  710. with mock.patch('compose.service.log', autospec=True) as mock_log:
  711. warn_on_masked_volume(volumes_option, container_volumes, service)
  712. assert not mock_log.warn.called
  713. def test_create_with_special_volume_mode(self):
  714. self.mock_client.inspect_image.return_value = {'Id': 'imageid'}
  715. self.mock_client.create_container.return_value = {'Id': 'containerid'}
  716. volume = '/tmp:/foo:z'
  717. Service(
  718. 'web',
  719. client=self.mock_client,
  720. image='busybox',
  721. volumes=[VolumeSpec.parse(volume)],
  722. ).create_container()
  723. assert self.mock_client.create_container.call_count == 1
  724. self.assertEqual(
  725. self.mock_client.create_host_config.call_args[1]['binds'],
  726. [volume])