project_test.py 38 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import random
  4. import py
  5. import pytest
  6. from docker.errors import NotFound
  7. from .. import mock
  8. from ..helpers import build_config
  9. from .testcases import DockerClientTestCase
  10. from compose.config import config
  11. from compose.config import ConfigurationError
  12. from compose.config.config import V2_0
  13. from compose.config.types import VolumeFromSpec
  14. from compose.config.types import VolumeSpec
  15. from compose.const import LABEL_PROJECT
  16. from compose.const import LABEL_SERVICE
  17. from compose.container import Container
  18. from compose.project import Project
  19. from compose.service import ConvergenceStrategy
  20. from tests.integration.testcases import v2_only
  21. class ProjectTest(DockerClientTestCase):
  22. def test_containers(self):
  23. web = self.create_service('web')
  24. db = self.create_service('db')
  25. project = Project('composetest', [web, db], self.client)
  26. project.up()
  27. containers = project.containers()
  28. self.assertEqual(len(containers), 2)
  29. def test_containers_with_service_names(self):
  30. web = self.create_service('web')
  31. db = self.create_service('db')
  32. project = Project('composetest', [web, db], self.client)
  33. project.up()
  34. containers = project.containers(['web'])
  35. self.assertEqual(
  36. [c.name for c in containers],
  37. ['composetest_web_1'])
  38. def test_containers_with_extra_service(self):
  39. web = self.create_service('web')
  40. web_1 = web.create_container()
  41. db = self.create_service('db')
  42. db_1 = db.create_container()
  43. self.create_service('extra').create_container()
  44. project = Project('composetest', [web, db], self.client)
  45. self.assertEqual(
  46. set(project.containers(stopped=True)),
  47. set([web_1, db_1]),
  48. )
  49. def test_volumes_from_service(self):
  50. project = Project.from_config(
  51. name='composetest',
  52. config_data=build_config({
  53. 'data': {
  54. 'image': 'busybox:latest',
  55. 'volumes': ['/var/data'],
  56. },
  57. 'db': {
  58. 'image': 'busybox:latest',
  59. 'volumes_from': ['data'],
  60. },
  61. }),
  62. client=self.client,
  63. )
  64. db = project.get_service('db')
  65. data = project.get_service('data')
  66. self.assertEqual(db.volumes_from, [VolumeFromSpec(data, 'rw', 'service')])
  67. def test_volumes_from_container(self):
  68. data_container = Container.create(
  69. self.client,
  70. image='busybox:latest',
  71. volumes=['/var/data'],
  72. name='composetest_data_container',
  73. labels={LABEL_PROJECT: 'composetest'},
  74. )
  75. project = Project.from_config(
  76. name='composetest',
  77. config_data=build_config({
  78. 'db': {
  79. 'image': 'busybox:latest',
  80. 'volumes_from': ['composetest_data_container'],
  81. },
  82. }),
  83. client=self.client,
  84. )
  85. db = project.get_service('db')
  86. self.assertEqual(db._get_volumes_from(), [data_container.id + ':rw'])
  87. @v2_only()
  88. def test_network_mode_from_service(self):
  89. project = Project.from_config(
  90. name='composetest',
  91. client=self.client,
  92. config_data=build_config({
  93. 'version': V2_0,
  94. 'services': {
  95. 'net': {
  96. 'image': 'busybox:latest',
  97. 'command': ["top"]
  98. },
  99. 'web': {
  100. 'image': 'busybox:latest',
  101. 'network_mode': 'service:net',
  102. 'command': ["top"]
  103. },
  104. },
  105. }),
  106. )
  107. project.up()
  108. web = project.get_service('web')
  109. net = project.get_service('net')
  110. self.assertEqual(web.network_mode.mode, 'container:' + net.containers()[0].id)
  111. @v2_only()
  112. def test_network_mode_from_container(self):
  113. def get_project():
  114. return Project.from_config(
  115. name='composetest',
  116. config_data=build_config({
  117. 'version': V2_0,
  118. 'services': {
  119. 'web': {
  120. 'image': 'busybox:latest',
  121. 'network_mode': 'container:composetest_net_container'
  122. },
  123. },
  124. }),
  125. client=self.client,
  126. )
  127. with pytest.raises(ConfigurationError) as excinfo:
  128. get_project()
  129. assert "container 'composetest_net_container' which does not exist" in excinfo.exconly()
  130. net_container = Container.create(
  131. self.client,
  132. image='busybox:latest',
  133. name='composetest_net_container',
  134. command='top',
  135. labels={LABEL_PROJECT: 'composetest'},
  136. )
  137. net_container.start()
  138. project = get_project()
  139. project.up()
  140. web = project.get_service('web')
  141. self.assertEqual(web.network_mode.mode, 'container:' + net_container.id)
  142. def test_net_from_service_v1(self):
  143. project = Project.from_config(
  144. name='composetest',
  145. config_data=build_config({
  146. 'net': {
  147. 'image': 'busybox:latest',
  148. 'command': ["top"]
  149. },
  150. 'web': {
  151. 'image': 'busybox:latest',
  152. 'net': 'container:net',
  153. 'command': ["top"]
  154. },
  155. }),
  156. client=self.client,
  157. )
  158. project.up()
  159. web = project.get_service('web')
  160. net = project.get_service('net')
  161. self.assertEqual(web.network_mode.mode, 'container:' + net.containers()[0].id)
  162. def test_net_from_container_v1(self):
  163. def get_project():
  164. return Project.from_config(
  165. name='composetest',
  166. config_data=build_config({
  167. 'web': {
  168. 'image': 'busybox:latest',
  169. 'net': 'container:composetest_net_container'
  170. },
  171. }),
  172. client=self.client,
  173. )
  174. with pytest.raises(ConfigurationError) as excinfo:
  175. get_project()
  176. assert "container 'composetest_net_container' which does not exist" in excinfo.exconly()
  177. net_container = Container.create(
  178. self.client,
  179. image='busybox:latest',
  180. name='composetest_net_container',
  181. command='top',
  182. labels={LABEL_PROJECT: 'composetest'},
  183. )
  184. net_container.start()
  185. project = get_project()
  186. project.up()
  187. web = project.get_service('web')
  188. self.assertEqual(web.network_mode.mode, 'container:' + net_container.id)
  189. def test_start_pause_unpause_stop_kill_remove(self):
  190. web = self.create_service('web')
  191. db = self.create_service('db')
  192. project = Project('composetest', [web, db], self.client)
  193. project.start()
  194. self.assertEqual(len(web.containers()), 0)
  195. self.assertEqual(len(db.containers()), 0)
  196. web_container_1 = web.create_container()
  197. web_container_2 = web.create_container()
  198. db_container = db.create_container()
  199. project.start(service_names=['web'])
  200. self.assertEqual(
  201. set(c.name for c in project.containers()),
  202. set([web_container_1.name, web_container_2.name]))
  203. project.start()
  204. self.assertEqual(
  205. set(c.name for c in project.containers()),
  206. set([web_container_1.name, web_container_2.name, db_container.name]))
  207. project.pause(service_names=['web'])
  208. self.assertEqual(
  209. set([c.name for c in project.containers() if c.is_paused]),
  210. set([web_container_1.name, web_container_2.name]))
  211. project.pause()
  212. self.assertEqual(
  213. set([c.name for c in project.containers() if c.is_paused]),
  214. set([web_container_1.name, web_container_2.name, db_container.name]))
  215. project.unpause(service_names=['db'])
  216. self.assertEqual(len([c.name for c in project.containers() if c.is_paused]), 2)
  217. project.unpause()
  218. self.assertEqual(len([c.name for c in project.containers() if c.is_paused]), 0)
  219. project.stop(service_names=['web'], timeout=1)
  220. self.assertEqual(set(c.name for c in project.containers()), set([db_container.name]))
  221. project.kill(service_names=['db'])
  222. self.assertEqual(len(project.containers()), 0)
  223. self.assertEqual(len(project.containers(stopped=True)), 3)
  224. project.remove_stopped(service_names=['web'])
  225. self.assertEqual(len(project.containers(stopped=True)), 1)
  226. project.remove_stopped()
  227. self.assertEqual(len(project.containers(stopped=True)), 0)
  228. def test_create(self):
  229. web = self.create_service('web')
  230. db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
  231. project = Project('composetest', [web, db], self.client)
  232. project.create(['db'])
  233. self.assertEqual(len(project.containers()), 0)
  234. self.assertEqual(len(project.containers(stopped=True)), 1)
  235. self.assertEqual(len(db.containers()), 0)
  236. self.assertEqual(len(db.containers(stopped=True)), 1)
  237. self.assertEqual(len(web.containers(stopped=True)), 0)
  238. def test_create_twice(self):
  239. web = self.create_service('web')
  240. db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
  241. project = Project('composetest', [web, db], self.client)
  242. project.create(['db', 'web'])
  243. project.create(['db', 'web'])
  244. self.assertEqual(len(project.containers()), 0)
  245. self.assertEqual(len(project.containers(stopped=True)), 2)
  246. self.assertEqual(len(db.containers()), 0)
  247. self.assertEqual(len(db.containers(stopped=True)), 1)
  248. self.assertEqual(len(web.containers()), 0)
  249. self.assertEqual(len(web.containers(stopped=True)), 1)
  250. def test_create_with_links(self):
  251. db = self.create_service('db')
  252. web = self.create_service('web', links=[(db, 'db')])
  253. project = Project('composetest', [db, web], self.client)
  254. project.create(['web'])
  255. self.assertEqual(len(project.containers()), 0)
  256. self.assertEqual(len(project.containers(stopped=True)), 2)
  257. self.assertEqual(len(db.containers()), 0)
  258. self.assertEqual(len(db.containers(stopped=True)), 1)
  259. self.assertEqual(len(web.containers()), 0)
  260. self.assertEqual(len(web.containers(stopped=True)), 1)
  261. def test_create_strategy_always(self):
  262. db = self.create_service('db')
  263. project = Project('composetest', [db], self.client)
  264. project.create(['db'])
  265. old_id = project.containers(stopped=True)[0].id
  266. project.create(['db'], strategy=ConvergenceStrategy.always)
  267. self.assertEqual(len(project.containers()), 0)
  268. self.assertEqual(len(project.containers(stopped=True)), 1)
  269. db_container = project.containers(stopped=True)[0]
  270. self.assertNotEqual(db_container.id, old_id)
  271. def test_create_strategy_never(self):
  272. db = self.create_service('db')
  273. project = Project('composetest', [db], self.client)
  274. project.create(['db'])
  275. old_id = project.containers(stopped=True)[0].id
  276. project.create(['db'], strategy=ConvergenceStrategy.never)
  277. self.assertEqual(len(project.containers()), 0)
  278. self.assertEqual(len(project.containers(stopped=True)), 1)
  279. db_container = project.containers(stopped=True)[0]
  280. self.assertEqual(db_container.id, old_id)
  281. def test_project_up(self):
  282. web = self.create_service('web')
  283. db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
  284. project = Project('composetest', [web, db], self.client)
  285. project.start()
  286. self.assertEqual(len(project.containers()), 0)
  287. project.up(['db'])
  288. self.assertEqual(len(project.containers()), 1)
  289. self.assertEqual(len(db.containers()), 1)
  290. self.assertEqual(len(web.containers()), 0)
  291. def test_project_up_starts_uncreated_services(self):
  292. db = self.create_service('db')
  293. web = self.create_service('web', links=[(db, 'db')])
  294. project = Project('composetest', [db, web], self.client)
  295. project.up(['db'])
  296. self.assertEqual(len(project.containers()), 1)
  297. project.up()
  298. self.assertEqual(len(project.containers()), 2)
  299. self.assertEqual(len(db.containers()), 1)
  300. self.assertEqual(len(web.containers()), 1)
  301. def test_recreate_preserves_volumes(self):
  302. web = self.create_service('web')
  303. db = self.create_service('db', volumes=[VolumeSpec.parse('/etc')])
  304. project = Project('composetest', [web, db], self.client)
  305. project.start()
  306. self.assertEqual(len(project.containers()), 0)
  307. project.up(['db'])
  308. self.assertEqual(len(project.containers()), 1)
  309. old_db_id = project.containers()[0].id
  310. db_volume_path = project.containers()[0].get('Volumes./etc')
  311. project.up(strategy=ConvergenceStrategy.always)
  312. self.assertEqual(len(project.containers()), 2)
  313. db_container = [c for c in project.containers() if 'db' in c.name][0]
  314. self.assertNotEqual(db_container.id, old_db_id)
  315. self.assertEqual(db_container.get('Volumes./etc'), db_volume_path)
  316. def test_project_up_with_no_recreate_running(self):
  317. web = self.create_service('web')
  318. db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
  319. project = Project('composetest', [web, db], self.client)
  320. project.start()
  321. self.assertEqual(len(project.containers()), 0)
  322. project.up(['db'])
  323. self.assertEqual(len(project.containers()), 1)
  324. old_db_id = project.containers()[0].id
  325. container, = project.containers()
  326. db_volume_path = container.get_mount('/var/db')['Source']
  327. project.up(strategy=ConvergenceStrategy.never)
  328. self.assertEqual(len(project.containers()), 2)
  329. db_container = [c for c in project.containers() if 'db' in c.name][0]
  330. self.assertEqual(db_container.id, old_db_id)
  331. self.assertEqual(
  332. db_container.get_mount('/var/db')['Source'],
  333. db_volume_path)
  334. def test_project_up_with_no_recreate_stopped(self):
  335. web = self.create_service('web')
  336. db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
  337. project = Project('composetest', [web, db], self.client)
  338. project.start()
  339. self.assertEqual(len(project.containers()), 0)
  340. project.up(['db'])
  341. project.kill()
  342. old_containers = project.containers(stopped=True)
  343. self.assertEqual(len(old_containers), 1)
  344. old_container, = old_containers
  345. old_db_id = old_container.id
  346. db_volume_path = old_container.get_mount('/var/db')['Source']
  347. project.up(strategy=ConvergenceStrategy.never)
  348. new_containers = project.containers(stopped=True)
  349. self.assertEqual(len(new_containers), 2)
  350. self.assertEqual([c.is_running for c in new_containers], [True, True])
  351. db_container = [c for c in new_containers if 'db' in c.name][0]
  352. self.assertEqual(db_container.id, old_db_id)
  353. self.assertEqual(
  354. db_container.get_mount('/var/db')['Source'],
  355. db_volume_path)
  356. def test_project_up_without_all_services(self):
  357. console = self.create_service('console')
  358. db = self.create_service('db')
  359. project = Project('composetest', [console, db], self.client)
  360. project.start()
  361. self.assertEqual(len(project.containers()), 0)
  362. project.up()
  363. self.assertEqual(len(project.containers()), 2)
  364. self.assertEqual(len(db.containers()), 1)
  365. self.assertEqual(len(console.containers()), 1)
  366. def test_project_up_starts_links(self):
  367. console = self.create_service('console')
  368. db = self.create_service('db', volumes=[VolumeSpec.parse('/var/db')])
  369. web = self.create_service('web', links=[(db, 'db')])
  370. project = Project('composetest', [web, db, console], self.client)
  371. project.start()
  372. self.assertEqual(len(project.containers()), 0)
  373. project.up(['web'])
  374. self.assertEqual(len(project.containers()), 2)
  375. self.assertEqual(len(web.containers()), 1)
  376. self.assertEqual(len(db.containers()), 1)
  377. self.assertEqual(len(console.containers()), 0)
  378. def test_project_up_starts_depends(self):
  379. project = Project.from_config(
  380. name='composetest',
  381. config_data=build_config({
  382. 'console': {
  383. 'image': 'busybox:latest',
  384. 'command': ["top"],
  385. },
  386. 'data': {
  387. 'image': 'busybox:latest',
  388. 'command': ["top"]
  389. },
  390. 'db': {
  391. 'image': 'busybox:latest',
  392. 'command': ["top"],
  393. 'volumes_from': ['data'],
  394. },
  395. 'web': {
  396. 'image': 'busybox:latest',
  397. 'command': ["top"],
  398. 'links': ['db'],
  399. },
  400. }),
  401. client=self.client,
  402. )
  403. project.start()
  404. self.assertEqual(len(project.containers()), 0)
  405. project.up(['web'])
  406. self.assertEqual(len(project.containers()), 3)
  407. self.assertEqual(len(project.get_service('web').containers()), 1)
  408. self.assertEqual(len(project.get_service('db').containers()), 1)
  409. self.assertEqual(len(project.get_service('data').containers()), 1)
  410. self.assertEqual(len(project.get_service('console').containers()), 0)
  411. def test_project_up_with_no_deps(self):
  412. project = Project.from_config(
  413. name='composetest',
  414. config_data=build_config({
  415. 'console': {
  416. 'image': 'busybox:latest',
  417. 'command': ["top"],
  418. },
  419. 'data': {
  420. 'image': 'busybox:latest',
  421. 'command': ["top"]
  422. },
  423. 'db': {
  424. 'image': 'busybox:latest',
  425. 'command': ["top"],
  426. 'volumes_from': ['data'],
  427. },
  428. 'web': {
  429. 'image': 'busybox:latest',
  430. 'command': ["top"],
  431. 'links': ['db'],
  432. },
  433. }),
  434. client=self.client,
  435. )
  436. project.start()
  437. self.assertEqual(len(project.containers()), 0)
  438. project.up(['db'], start_deps=False)
  439. self.assertEqual(len(project.containers(stopped=True)), 2)
  440. self.assertEqual(len(project.get_service('web').containers()), 0)
  441. self.assertEqual(len(project.get_service('db').containers()), 1)
  442. self.assertEqual(len(project.get_service('data').containers()), 0)
  443. self.assertEqual(len(project.get_service('data').containers(stopped=True)), 1)
  444. self.assertEqual(len(project.get_service('console').containers()), 0)
  445. def test_unscale_after_restart(self):
  446. web = self.create_service('web')
  447. project = Project('composetest', [web], self.client)
  448. project.start()
  449. service = project.get_service('web')
  450. service.scale(1)
  451. self.assertEqual(len(service.containers()), 1)
  452. service.scale(3)
  453. self.assertEqual(len(service.containers()), 3)
  454. project.up()
  455. service = project.get_service('web')
  456. self.assertEqual(len(service.containers()), 3)
  457. service.scale(1)
  458. self.assertEqual(len(service.containers()), 1)
  459. project.up()
  460. service = project.get_service('web')
  461. self.assertEqual(len(service.containers()), 1)
  462. # does scale=0 ,makes any sense? after recreating at least 1 container is running
  463. service.scale(0)
  464. project.up()
  465. service = project.get_service('web')
  466. self.assertEqual(len(service.containers()), 1)
  467. @v2_only()
  468. def test_project_up_networks(self):
  469. config_data = config.Config(
  470. version=V2_0,
  471. services=[{
  472. 'name': 'web',
  473. 'image': 'busybox:latest',
  474. 'command': 'top',
  475. 'networks': {'foo': None, 'bar': None, 'baz': None},
  476. }],
  477. volumes={},
  478. networks={
  479. 'foo': {'driver': 'bridge'},
  480. 'bar': {'driver': None},
  481. 'baz': {},
  482. },
  483. )
  484. project = Project.from_config(
  485. client=self.client,
  486. name='composetest',
  487. config_data=config_data,
  488. )
  489. project.up()
  490. self.assertEqual(len(project.containers()), 1)
  491. for net_name in ['foo', 'bar', 'baz']:
  492. full_net_name = 'composetest_{}'.format(net_name)
  493. network_data = self.client.inspect_network(full_net_name)
  494. self.assertEqual(network_data['Name'], full_net_name)
  495. foo_data = self.client.inspect_network('composetest_foo')
  496. self.assertEqual(foo_data['Driver'], 'bridge')
  497. @v2_only()
  498. def test_up_with_ipam_config(self):
  499. config_data = config.Config(
  500. version=V2_0,
  501. services=[{
  502. 'name': 'web',
  503. 'image': 'busybox:latest',
  504. 'networks': {'front': None},
  505. }],
  506. volumes={},
  507. networks={
  508. 'front': {
  509. 'driver': 'bridge',
  510. 'driver_opts': {
  511. "com.docker.network.bridge.enable_icc": "false",
  512. },
  513. 'ipam': {
  514. 'driver': 'default',
  515. 'config': [{
  516. "subnet": "172.28.0.0/16",
  517. "ip_range": "172.28.5.0/24",
  518. "gateway": "172.28.5.254",
  519. "aux_addresses": {
  520. "a": "172.28.1.5",
  521. "b": "172.28.1.6",
  522. "c": "172.28.1.7",
  523. },
  524. }],
  525. },
  526. },
  527. },
  528. )
  529. project = Project.from_config(
  530. client=self.client,
  531. name='composetest',
  532. config_data=config_data,
  533. )
  534. project.up()
  535. network = self.client.networks(names=['composetest_front'])[0]
  536. assert network['Options'] == {
  537. "com.docker.network.bridge.enable_icc": "false"
  538. }
  539. assert network['IPAM'] == {
  540. 'Driver': 'default',
  541. 'Options': None,
  542. 'Config': [{
  543. 'Subnet': "172.28.0.0/16",
  544. 'IPRange': "172.28.5.0/24",
  545. 'Gateway': "172.28.5.254",
  546. 'AuxiliaryAddresses': {
  547. 'a': '172.28.1.5',
  548. 'b': '172.28.1.6',
  549. 'c': '172.28.1.7',
  550. },
  551. }],
  552. }
  553. @v2_only()
  554. def test_up_with_network_static_addresses(self):
  555. config_data = config.Config(
  556. version=V2_0,
  557. services=[{
  558. 'name': 'web',
  559. 'image': 'busybox:latest',
  560. 'networks': {
  561. 'static_test': {
  562. 'ipv4_address': '172.16.100.100',
  563. 'ipv6_address': 'fe80::1001:102'
  564. }
  565. },
  566. }],
  567. volumes={},
  568. networks={
  569. 'static_test': {
  570. 'driver': 'bridge',
  571. 'driver_opts': {
  572. "com.docker.network.enable_ipv6": "true",
  573. },
  574. 'ipam': {
  575. 'driver': 'default',
  576. 'config': [
  577. {"subnet": "172.16.100.0/24",
  578. "gateway": "172.16.100.1"},
  579. {"subnet": "fe80::/64",
  580. "gateway": "fe80::1001:1"}
  581. ]
  582. }
  583. }
  584. }
  585. )
  586. project = Project.from_config(
  587. client=self.client,
  588. name='composetest',
  589. config_data=config_data,
  590. )
  591. project.up()
  592. network = self.client.networks(names=['static_test'])[0]
  593. service_container = project.get_service('web').containers()[0]
  594. assert network['Options'] == {
  595. "com.docker.network.enable_ipv6": "true"
  596. }
  597. IPAMConfig = (service_container.inspect().get('NetworkSettings', {}).
  598. get('Networks', {}).get('composetest_static_test', {}).
  599. get('IPAMConfig', {}))
  600. assert IPAMConfig.get('IPv4Address') == '172.16.100.100'
  601. assert IPAMConfig.get('IPv6Address') == 'fe80::1001:102'
  602. @v2_only()
  603. def test_up_with_network_static_addresses_missing_subnet(self):
  604. config_data = config.Config(
  605. version=V2_0,
  606. services=[{
  607. 'name': 'web',
  608. 'image': 'busybox:latest',
  609. 'networks': {
  610. 'static_test': {
  611. 'ipv4_address': '172.16.100.100',
  612. 'ipv6_address': 'fe80::1001:101'
  613. }
  614. },
  615. }],
  616. volumes={},
  617. networks={
  618. 'static_test': {
  619. 'driver': 'bridge',
  620. 'driver_opts': {
  621. "com.docker.network.enable_ipv6": "true",
  622. },
  623. 'ipam': {
  624. 'driver': 'default',
  625. },
  626. },
  627. },
  628. )
  629. project = Project.from_config(
  630. client=self.client,
  631. name='composetest',
  632. config_data=config_data,
  633. )
  634. assert len(project.up()) == 0
  635. @v2_only()
  636. def test_project_up_volumes(self):
  637. vol_name = '{0:x}'.format(random.getrandbits(32))
  638. full_vol_name = 'composetest_{0}'.format(vol_name)
  639. config_data = config.Config(
  640. version=V2_0,
  641. services=[{
  642. 'name': 'web',
  643. 'image': 'busybox:latest',
  644. 'command': 'top'
  645. }],
  646. volumes={vol_name: {'driver': 'local'}},
  647. networks={},
  648. )
  649. project = Project.from_config(
  650. name='composetest',
  651. config_data=config_data, client=self.client
  652. )
  653. project.up()
  654. self.assertEqual(len(project.containers()), 1)
  655. volume_data = self.client.inspect_volume(full_vol_name)
  656. self.assertEqual(volume_data['Name'], full_vol_name)
  657. self.assertEqual(volume_data['Driver'], 'local')
  658. @v2_only()
  659. def test_project_up_logging_with_multiple_files(self):
  660. base_file = config.ConfigFile(
  661. 'base.yml',
  662. {
  663. 'version': V2_0,
  664. 'services': {
  665. 'simple': {'image': 'busybox:latest', 'command': 'top'},
  666. 'another': {
  667. 'image': 'busybox:latest',
  668. 'command': 'top',
  669. 'logging': {
  670. 'driver': "json-file",
  671. 'options': {
  672. 'max-size': "10m"
  673. }
  674. }
  675. }
  676. }
  677. })
  678. override_file = config.ConfigFile(
  679. 'override.yml',
  680. {
  681. 'version': V2_0,
  682. 'services': {
  683. 'another': {
  684. 'logging': {
  685. 'driver': "none"
  686. }
  687. }
  688. }
  689. })
  690. details = config.ConfigDetails('.', [base_file, override_file])
  691. tmpdir = py.test.ensuretemp('logging_test')
  692. self.addCleanup(tmpdir.remove)
  693. with tmpdir.as_cwd():
  694. config_data = config.load(details)
  695. project = Project.from_config(
  696. name='composetest', config_data=config_data, client=self.client
  697. )
  698. project.up()
  699. containers = project.containers()
  700. self.assertEqual(len(containers), 2)
  701. another = project.get_service('another').containers()[0]
  702. log_config = another.get('HostConfig.LogConfig')
  703. self.assertTrue(log_config)
  704. self.assertEqual(log_config.get('Type'), 'none')
  705. @v2_only()
  706. def test_initialize_volumes(self):
  707. vol_name = '{0:x}'.format(random.getrandbits(32))
  708. full_vol_name = 'composetest_{0}'.format(vol_name)
  709. config_data = config.Config(
  710. version=V2_0,
  711. services=[{
  712. 'name': 'web',
  713. 'image': 'busybox:latest',
  714. 'command': 'top'
  715. }],
  716. volumes={vol_name: {}},
  717. networks={},
  718. )
  719. project = Project.from_config(
  720. name='composetest',
  721. config_data=config_data, client=self.client
  722. )
  723. project.volumes.initialize()
  724. volume_data = self.client.inspect_volume(full_vol_name)
  725. self.assertEqual(volume_data['Name'], full_vol_name)
  726. self.assertEqual(volume_data['Driver'], 'local')
  727. @v2_only()
  728. def test_project_up_implicit_volume_driver(self):
  729. vol_name = '{0:x}'.format(random.getrandbits(32))
  730. full_vol_name = 'composetest_{0}'.format(vol_name)
  731. config_data = config.Config(
  732. version=V2_0,
  733. services=[{
  734. 'name': 'web',
  735. 'image': 'busybox:latest',
  736. 'command': 'top'
  737. }],
  738. volumes={vol_name: {}},
  739. networks={},
  740. )
  741. project = Project.from_config(
  742. name='composetest',
  743. config_data=config_data, client=self.client
  744. )
  745. project.up()
  746. volume_data = self.client.inspect_volume(full_vol_name)
  747. self.assertEqual(volume_data['Name'], full_vol_name)
  748. self.assertEqual(volume_data['Driver'], 'local')
  749. @v2_only()
  750. def test_initialize_volumes_invalid_volume_driver(self):
  751. vol_name = '{0:x}'.format(random.getrandbits(32))
  752. config_data = config.Config(
  753. version=V2_0,
  754. services=[{
  755. 'name': 'web',
  756. 'image': 'busybox:latest',
  757. 'command': 'top'
  758. }],
  759. volumes={vol_name: {'driver': 'foobar'}},
  760. networks={},
  761. )
  762. project = Project.from_config(
  763. name='composetest',
  764. config_data=config_data, client=self.client
  765. )
  766. with self.assertRaises(config.ConfigurationError):
  767. project.volumes.initialize()
  768. @v2_only()
  769. def test_initialize_volumes_updated_driver(self):
  770. vol_name = '{0:x}'.format(random.getrandbits(32))
  771. full_vol_name = 'composetest_{0}'.format(vol_name)
  772. config_data = config.Config(
  773. version=V2_0,
  774. services=[{
  775. 'name': 'web',
  776. 'image': 'busybox:latest',
  777. 'command': 'top'
  778. }],
  779. volumes={vol_name: {'driver': 'local'}},
  780. networks={},
  781. )
  782. project = Project.from_config(
  783. name='composetest',
  784. config_data=config_data, client=self.client
  785. )
  786. project.volumes.initialize()
  787. volume_data = self.client.inspect_volume(full_vol_name)
  788. self.assertEqual(volume_data['Name'], full_vol_name)
  789. self.assertEqual(volume_data['Driver'], 'local')
  790. config_data = config_data._replace(
  791. volumes={vol_name: {'driver': 'smb'}}
  792. )
  793. project = Project.from_config(
  794. name='composetest',
  795. config_data=config_data,
  796. client=self.client
  797. )
  798. with self.assertRaises(config.ConfigurationError) as e:
  799. project.volumes.initialize()
  800. assert 'Configuration for volume {0} specifies driver smb'.format(
  801. vol_name
  802. ) in str(e.exception)
  803. @v2_only()
  804. def test_initialize_volumes_updated_blank_driver(self):
  805. vol_name = '{0:x}'.format(random.getrandbits(32))
  806. full_vol_name = 'composetest_{0}'.format(vol_name)
  807. config_data = config.Config(
  808. version=V2_0,
  809. services=[{
  810. 'name': 'web',
  811. 'image': 'busybox:latest',
  812. 'command': 'top'
  813. }],
  814. volumes={vol_name: {'driver': 'local'}},
  815. networks={},
  816. )
  817. project = Project.from_config(
  818. name='composetest',
  819. config_data=config_data, client=self.client
  820. )
  821. project.volumes.initialize()
  822. volume_data = self.client.inspect_volume(full_vol_name)
  823. self.assertEqual(volume_data['Name'], full_vol_name)
  824. self.assertEqual(volume_data['Driver'], 'local')
  825. config_data = config_data._replace(
  826. volumes={vol_name: {}}
  827. )
  828. project = Project.from_config(
  829. name='composetest',
  830. config_data=config_data,
  831. client=self.client
  832. )
  833. project.volumes.initialize()
  834. volume_data = self.client.inspect_volume(full_vol_name)
  835. self.assertEqual(volume_data['Name'], full_vol_name)
  836. self.assertEqual(volume_data['Driver'], 'local')
  837. @v2_only()
  838. def test_initialize_volumes_external_volumes(self):
  839. # Use composetest_ prefix so it gets garbage-collected in tearDown()
  840. vol_name = 'composetest_{0:x}'.format(random.getrandbits(32))
  841. full_vol_name = 'composetest_{0}'.format(vol_name)
  842. self.client.create_volume(vol_name)
  843. config_data = config.Config(
  844. version=V2_0,
  845. services=[{
  846. 'name': 'web',
  847. 'image': 'busybox:latest',
  848. 'command': 'top'
  849. }],
  850. volumes={
  851. vol_name: {'external': True, 'external_name': vol_name}
  852. },
  853. networks=None,
  854. )
  855. project = Project.from_config(
  856. name='composetest',
  857. config_data=config_data, client=self.client
  858. )
  859. project.volumes.initialize()
  860. with self.assertRaises(NotFound):
  861. self.client.inspect_volume(full_vol_name)
  862. @v2_only()
  863. def test_initialize_volumes_inexistent_external_volume(self):
  864. vol_name = '{0:x}'.format(random.getrandbits(32))
  865. config_data = config.Config(
  866. version=V2_0,
  867. services=[{
  868. 'name': 'web',
  869. 'image': 'busybox:latest',
  870. 'command': 'top'
  871. }],
  872. volumes={
  873. vol_name: {'external': True, 'external_name': vol_name}
  874. },
  875. networks=None,
  876. )
  877. project = Project.from_config(
  878. name='composetest',
  879. config_data=config_data, client=self.client
  880. )
  881. with self.assertRaises(config.ConfigurationError) as e:
  882. project.volumes.initialize()
  883. assert 'Volume {0} declared as external'.format(
  884. vol_name
  885. ) in str(e.exception)
  886. @v2_only()
  887. def test_project_up_named_volumes_in_binds(self):
  888. vol_name = '{0:x}'.format(random.getrandbits(32))
  889. full_vol_name = 'composetest_{0}'.format(vol_name)
  890. base_file = config.ConfigFile(
  891. 'base.yml',
  892. {
  893. 'version': V2_0,
  894. 'services': {
  895. 'simple': {
  896. 'image': 'busybox:latest',
  897. 'command': 'top',
  898. 'volumes': ['{0}:/data'.format(vol_name)]
  899. },
  900. },
  901. 'volumes': {
  902. vol_name: {'driver': 'local'}
  903. }
  904. })
  905. config_details = config.ConfigDetails('.', [base_file])
  906. config_data = config.load(config_details)
  907. project = Project.from_config(
  908. name='composetest', config_data=config_data, client=self.client
  909. )
  910. service = project.services[0]
  911. self.assertEqual(service.name, 'simple')
  912. volumes = service.options.get('volumes')
  913. self.assertEqual(len(volumes), 1)
  914. self.assertEqual(volumes[0].external, full_vol_name)
  915. project.up()
  916. engine_volumes = self.client.volumes()['Volumes']
  917. container = service.get_container()
  918. assert [mount['Name'] for mount in container.get('Mounts')] == [full_vol_name]
  919. assert next((v for v in engine_volumes if v['Name'] == vol_name), None) is None
  920. def test_project_up_orphans(self):
  921. config_dict = {
  922. 'service1': {
  923. 'image': 'busybox:latest',
  924. 'command': 'top',
  925. }
  926. }
  927. config_data = build_config(config_dict)
  928. project = Project.from_config(
  929. name='composetest', config_data=config_data, client=self.client
  930. )
  931. project.up()
  932. config_dict['service2'] = config_dict['service1']
  933. del config_dict['service1']
  934. config_data = build_config(config_dict)
  935. project = Project.from_config(
  936. name='composetest', config_data=config_data, client=self.client
  937. )
  938. with mock.patch('compose.project.log') as mock_log:
  939. project.up()
  940. mock_log.warning.assert_called_once_with(mock.ANY)
  941. assert len([
  942. ctnr for ctnr in project._labeled_containers()
  943. if ctnr.labels.get(LABEL_SERVICE) == 'service1'
  944. ]) == 1
  945. project.up(remove_orphans=True)
  946. assert len([
  947. ctnr for ctnr in project._labeled_containers()
  948. if ctnr.labels.get(LABEL_SERVICE) == 'service1'
  949. ]) == 0