1
0

project_test.py 38 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097
  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. 'command': 'top',
  561. 'networks': {
  562. 'static_test': {
  563. 'ipv4_address': '172.16.100.100',
  564. 'ipv6_address': 'fe80::1001:102'
  565. }
  566. },
  567. }],
  568. volumes={},
  569. networks={
  570. 'static_test': {
  571. 'driver': 'bridge',
  572. 'driver_opts': {
  573. "com.docker.network.enable_ipv6": "true",
  574. },
  575. 'ipam': {
  576. 'driver': 'default',
  577. 'config': [
  578. {"subnet": "172.16.100.0/24",
  579. "gateway": "172.16.100.1"},
  580. {"subnet": "fe80::/64",
  581. "gateway": "fe80::1001:1"}
  582. ]
  583. }
  584. }
  585. }
  586. )
  587. project = Project.from_config(
  588. client=self.client,
  589. name='composetest',
  590. config_data=config_data,
  591. )
  592. project.up(detached=True)
  593. network = self.client.networks(names=['static_test'])[0]
  594. service_container = project.get_service('web').containers()[0]
  595. assert network['Options'] == {
  596. "com.docker.network.enable_ipv6": "true"
  597. }
  598. IPAMConfig = (service_container.inspect().get('NetworkSettings', {}).
  599. get('Networks', {}).get('composetest_static_test', {}).
  600. get('IPAMConfig', {}))
  601. assert IPAMConfig.get('IPv4Address') == '172.16.100.100'
  602. assert IPAMConfig.get('IPv6Address') == 'fe80::1001:102'
  603. @v2_only()
  604. def test_up_with_network_static_addresses_missing_subnet(self):
  605. config_data = config.Config(
  606. version=V2_0,
  607. services=[{
  608. 'name': 'web',
  609. 'image': 'busybox:latest',
  610. 'networks': {
  611. 'static_test': {
  612. 'ipv4_address': '172.16.100.100',
  613. 'ipv6_address': 'fe80::1001:101'
  614. }
  615. },
  616. }],
  617. volumes={},
  618. networks={
  619. 'static_test': {
  620. 'driver': 'bridge',
  621. 'driver_opts': {
  622. "com.docker.network.enable_ipv6": "true",
  623. },
  624. 'ipam': {
  625. 'driver': 'default',
  626. },
  627. },
  628. },
  629. )
  630. project = Project.from_config(
  631. client=self.client,
  632. name='composetest',
  633. config_data=config_data,
  634. )
  635. assert len(project.up()) == 0
  636. @v2_only()
  637. def test_project_up_volumes(self):
  638. vol_name = '{0:x}'.format(random.getrandbits(32))
  639. full_vol_name = 'composetest_{0}'.format(vol_name)
  640. config_data = config.Config(
  641. version=V2_0,
  642. services=[{
  643. 'name': 'web',
  644. 'image': 'busybox:latest',
  645. 'command': 'top'
  646. }],
  647. volumes={vol_name: {'driver': 'local'}},
  648. networks={},
  649. )
  650. project = Project.from_config(
  651. name='composetest',
  652. config_data=config_data, client=self.client
  653. )
  654. project.up()
  655. self.assertEqual(len(project.containers()), 1)
  656. volume_data = self.client.inspect_volume(full_vol_name)
  657. self.assertEqual(volume_data['Name'], full_vol_name)
  658. self.assertEqual(volume_data['Driver'], 'local')
  659. @v2_only()
  660. def test_project_up_logging_with_multiple_files(self):
  661. base_file = config.ConfigFile(
  662. 'base.yml',
  663. {
  664. 'version': V2_0,
  665. 'services': {
  666. 'simple': {'image': 'busybox:latest', 'command': 'top'},
  667. 'another': {
  668. 'image': 'busybox:latest',
  669. 'command': 'top',
  670. 'logging': {
  671. 'driver': "json-file",
  672. 'options': {
  673. 'max-size': "10m"
  674. }
  675. }
  676. }
  677. }
  678. })
  679. override_file = config.ConfigFile(
  680. 'override.yml',
  681. {
  682. 'version': V2_0,
  683. 'services': {
  684. 'another': {
  685. 'logging': {
  686. 'driver': "none"
  687. }
  688. }
  689. }
  690. })
  691. details = config.ConfigDetails('.', [base_file, override_file])
  692. tmpdir = py.test.ensuretemp('logging_test')
  693. self.addCleanup(tmpdir.remove)
  694. with tmpdir.as_cwd():
  695. config_data = config.load(details)
  696. project = Project.from_config(
  697. name='composetest', config_data=config_data, client=self.client
  698. )
  699. project.up()
  700. containers = project.containers()
  701. self.assertEqual(len(containers), 2)
  702. another = project.get_service('another').containers()[0]
  703. log_config = another.get('HostConfig.LogConfig')
  704. self.assertTrue(log_config)
  705. self.assertEqual(log_config.get('Type'), 'none')
  706. @v2_only()
  707. def test_initialize_volumes(self):
  708. vol_name = '{0:x}'.format(random.getrandbits(32))
  709. full_vol_name = 'composetest_{0}'.format(vol_name)
  710. config_data = config.Config(
  711. version=V2_0,
  712. services=[{
  713. 'name': 'web',
  714. 'image': 'busybox:latest',
  715. 'command': 'top'
  716. }],
  717. volumes={vol_name: {}},
  718. networks={},
  719. )
  720. project = Project.from_config(
  721. name='composetest',
  722. config_data=config_data, client=self.client
  723. )
  724. project.volumes.initialize()
  725. volume_data = self.client.inspect_volume(full_vol_name)
  726. self.assertEqual(volume_data['Name'], full_vol_name)
  727. self.assertEqual(volume_data['Driver'], 'local')
  728. @v2_only()
  729. def test_project_up_implicit_volume_driver(self):
  730. vol_name = '{0:x}'.format(random.getrandbits(32))
  731. full_vol_name = 'composetest_{0}'.format(vol_name)
  732. config_data = config.Config(
  733. version=V2_0,
  734. services=[{
  735. 'name': 'web',
  736. 'image': 'busybox:latest',
  737. 'command': 'top'
  738. }],
  739. volumes={vol_name: {}},
  740. networks={},
  741. )
  742. project = Project.from_config(
  743. name='composetest',
  744. config_data=config_data, client=self.client
  745. )
  746. project.up()
  747. volume_data = self.client.inspect_volume(full_vol_name)
  748. self.assertEqual(volume_data['Name'], full_vol_name)
  749. self.assertEqual(volume_data['Driver'], 'local')
  750. @v2_only()
  751. def test_initialize_volumes_invalid_volume_driver(self):
  752. vol_name = '{0:x}'.format(random.getrandbits(32))
  753. config_data = config.Config(
  754. version=V2_0,
  755. services=[{
  756. 'name': 'web',
  757. 'image': 'busybox:latest',
  758. 'command': 'top'
  759. }],
  760. volumes={vol_name: {'driver': 'foobar'}},
  761. networks={},
  762. )
  763. project = Project.from_config(
  764. name='composetest',
  765. config_data=config_data, client=self.client
  766. )
  767. with self.assertRaises(config.ConfigurationError):
  768. project.volumes.initialize()
  769. @v2_only()
  770. def test_initialize_volumes_updated_driver(self):
  771. vol_name = '{0:x}'.format(random.getrandbits(32))
  772. full_vol_name = 'composetest_{0}'.format(vol_name)
  773. config_data = config.Config(
  774. version=V2_0,
  775. services=[{
  776. 'name': 'web',
  777. 'image': 'busybox:latest',
  778. 'command': 'top'
  779. }],
  780. volumes={vol_name: {'driver': 'local'}},
  781. networks={},
  782. )
  783. project = Project.from_config(
  784. name='composetest',
  785. config_data=config_data, client=self.client
  786. )
  787. project.volumes.initialize()
  788. volume_data = self.client.inspect_volume(full_vol_name)
  789. self.assertEqual(volume_data['Name'], full_vol_name)
  790. self.assertEqual(volume_data['Driver'], 'local')
  791. config_data = config_data._replace(
  792. volumes={vol_name: {'driver': 'smb'}}
  793. )
  794. project = Project.from_config(
  795. name='composetest',
  796. config_data=config_data,
  797. client=self.client
  798. )
  799. with self.assertRaises(config.ConfigurationError) as e:
  800. project.volumes.initialize()
  801. assert 'Configuration for volume {0} specifies driver smb'.format(
  802. vol_name
  803. ) in str(e.exception)
  804. @v2_only()
  805. def test_initialize_volumes_updated_blank_driver(self):
  806. vol_name = '{0:x}'.format(random.getrandbits(32))
  807. full_vol_name = 'composetest_{0}'.format(vol_name)
  808. config_data = config.Config(
  809. version=V2_0,
  810. services=[{
  811. 'name': 'web',
  812. 'image': 'busybox:latest',
  813. 'command': 'top'
  814. }],
  815. volumes={vol_name: {'driver': 'local'}},
  816. networks={},
  817. )
  818. project = Project.from_config(
  819. name='composetest',
  820. config_data=config_data, client=self.client
  821. )
  822. project.volumes.initialize()
  823. volume_data = self.client.inspect_volume(full_vol_name)
  824. self.assertEqual(volume_data['Name'], full_vol_name)
  825. self.assertEqual(volume_data['Driver'], 'local')
  826. config_data = config_data._replace(
  827. volumes={vol_name: {}}
  828. )
  829. project = Project.from_config(
  830. name='composetest',
  831. config_data=config_data,
  832. client=self.client
  833. )
  834. project.volumes.initialize()
  835. volume_data = self.client.inspect_volume(full_vol_name)
  836. self.assertEqual(volume_data['Name'], full_vol_name)
  837. self.assertEqual(volume_data['Driver'], 'local')
  838. @v2_only()
  839. def test_initialize_volumes_external_volumes(self):
  840. # Use composetest_ prefix so it gets garbage-collected in tearDown()
  841. vol_name = 'composetest_{0:x}'.format(random.getrandbits(32))
  842. full_vol_name = 'composetest_{0}'.format(vol_name)
  843. self.client.create_volume(vol_name)
  844. config_data = config.Config(
  845. version=V2_0,
  846. services=[{
  847. 'name': 'web',
  848. 'image': 'busybox:latest',
  849. 'command': 'top'
  850. }],
  851. volumes={
  852. vol_name: {'external': True, 'external_name': vol_name}
  853. },
  854. networks=None,
  855. )
  856. project = Project.from_config(
  857. name='composetest',
  858. config_data=config_data, client=self.client
  859. )
  860. project.volumes.initialize()
  861. with self.assertRaises(NotFound):
  862. self.client.inspect_volume(full_vol_name)
  863. @v2_only()
  864. def test_initialize_volumes_inexistent_external_volume(self):
  865. vol_name = '{0:x}'.format(random.getrandbits(32))
  866. config_data = config.Config(
  867. version=V2_0,
  868. services=[{
  869. 'name': 'web',
  870. 'image': 'busybox:latest',
  871. 'command': 'top'
  872. }],
  873. volumes={
  874. vol_name: {'external': True, 'external_name': vol_name}
  875. },
  876. networks=None,
  877. )
  878. project = Project.from_config(
  879. name='composetest',
  880. config_data=config_data, client=self.client
  881. )
  882. with self.assertRaises(config.ConfigurationError) as e:
  883. project.volumes.initialize()
  884. assert 'Volume {0} declared as external'.format(
  885. vol_name
  886. ) in str(e.exception)
  887. @v2_only()
  888. def test_project_up_named_volumes_in_binds(self):
  889. vol_name = '{0:x}'.format(random.getrandbits(32))
  890. full_vol_name = 'composetest_{0}'.format(vol_name)
  891. base_file = config.ConfigFile(
  892. 'base.yml',
  893. {
  894. 'version': V2_0,
  895. 'services': {
  896. 'simple': {
  897. 'image': 'busybox:latest',
  898. 'command': 'top',
  899. 'volumes': ['{0}:/data'.format(vol_name)]
  900. },
  901. },
  902. 'volumes': {
  903. vol_name: {'driver': 'local'}
  904. }
  905. })
  906. config_details = config.ConfigDetails('.', [base_file])
  907. config_data = config.load(config_details)
  908. project = Project.from_config(
  909. name='composetest', config_data=config_data, client=self.client
  910. )
  911. service = project.services[0]
  912. self.assertEqual(service.name, 'simple')
  913. volumes = service.options.get('volumes')
  914. self.assertEqual(len(volumes), 1)
  915. self.assertEqual(volumes[0].external, full_vol_name)
  916. project.up()
  917. engine_volumes = self.client.volumes()['Volumes']
  918. container = service.get_container()
  919. assert [mount['Name'] for mount in container.get('Mounts')] == [full_vol_name]
  920. assert next((v for v in engine_volumes if v['Name'] == vol_name), None) is None
  921. def test_project_up_orphans(self):
  922. config_dict = {
  923. 'service1': {
  924. 'image': 'busybox:latest',
  925. 'command': 'top',
  926. }
  927. }
  928. config_data = build_config(config_dict)
  929. project = Project.from_config(
  930. name='composetest', config_data=config_data, client=self.client
  931. )
  932. project.up()
  933. config_dict['service2'] = config_dict['service1']
  934. del config_dict['service1']
  935. config_data = build_config(config_dict)
  936. project = Project.from_config(
  937. name='composetest', config_data=config_data, client=self.client
  938. )
  939. with mock.patch('compose.project.log') as mock_log:
  940. project.up()
  941. mock_log.warning.assert_called_once_with(mock.ANY)
  942. assert len([
  943. ctnr for ctnr in project._labeled_containers()
  944. if ctnr.labels.get(LABEL_SERVICE) == 'service1'
  945. ]) == 1
  946. project.up(remove_orphans=True)
  947. assert len([
  948. ctnr for ctnr in project._labeled_containers()
  949. if ctnr.labels.get(LABEL_SERVICE) == 'service1'
  950. ]) == 0