project_test.py 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406
  1. from __future__ import unicode_literals
  2. from .testcases import DockerClientTestCase
  3. from compose.config import config
  4. from compose.const import LABEL_PROJECT
  5. from compose.container import Container
  6. from compose.project import Project
  7. from compose.service import ConvergenceStrategy
  8. def build_service_dicts(service_config):
  9. return config.load(
  10. config.ConfigDetails(
  11. 'working_dir',
  12. [config.ConfigFile(None, service_config)]))
  13. class ProjectTest(DockerClientTestCase):
  14. def test_containers(self):
  15. web = self.create_service('web')
  16. db = self.create_service('db')
  17. project = Project('composetest', [web, db], self.client)
  18. project.up()
  19. containers = project.containers()
  20. self.assertEqual(len(containers), 2)
  21. def test_containers_with_service_names(self):
  22. web = self.create_service('web')
  23. db = self.create_service('db')
  24. project = Project('composetest', [web, db], self.client)
  25. project.up()
  26. containers = project.containers(['web'])
  27. self.assertEqual(
  28. [c.name for c in containers],
  29. ['composetest_web_1'])
  30. def test_containers_with_extra_service(self):
  31. web = self.create_service('web')
  32. web_1 = web.create_container()
  33. db = self.create_service('db')
  34. db_1 = db.create_container()
  35. self.create_service('extra').create_container()
  36. project = Project('composetest', [web, db], self.client)
  37. self.assertEqual(
  38. set(project.containers(stopped=True)),
  39. set([web_1, db_1]),
  40. )
  41. def test_volumes_from_service(self):
  42. service_dicts = build_service_dicts({
  43. 'data': {
  44. 'image': 'busybox:latest',
  45. 'volumes': ['/var/data'],
  46. },
  47. 'db': {
  48. 'image': 'busybox:latest',
  49. 'volumes_from': ['data'],
  50. },
  51. })
  52. project = Project.from_dicts(
  53. name='composetest',
  54. service_dicts=service_dicts,
  55. client=self.client,
  56. )
  57. db = project.get_service('db')
  58. data = project.get_service('data')
  59. self.assertEqual(db.volumes_from, [data])
  60. def test_volumes_from_container(self):
  61. data_container = Container.create(
  62. self.client,
  63. image='busybox:latest',
  64. volumes=['/var/data'],
  65. name='composetest_data_container',
  66. labels={LABEL_PROJECT: 'composetest'},
  67. )
  68. project = Project.from_dicts(
  69. name='composetest',
  70. service_dicts=build_service_dicts({
  71. 'db': {
  72. 'image': 'busybox:latest',
  73. 'volumes_from': ['composetest_data_container'],
  74. },
  75. }),
  76. client=self.client,
  77. )
  78. db = project.get_service('db')
  79. self.assertEqual(db.volumes_from, [data_container])
  80. def test_net_from_service(self):
  81. project = Project.from_dicts(
  82. name='composetest',
  83. service_dicts=build_service_dicts({
  84. 'net': {
  85. 'image': 'busybox:latest',
  86. 'command': ["top"]
  87. },
  88. 'web': {
  89. 'image': 'busybox:latest',
  90. 'net': 'container:net',
  91. 'command': ["top"]
  92. },
  93. }),
  94. client=self.client,
  95. )
  96. project.up()
  97. web = project.get_service('web')
  98. net = project.get_service('net')
  99. self.assertEqual(web.net.mode, 'container:' + net.containers()[0].id)
  100. def test_net_from_container(self):
  101. net_container = Container.create(
  102. self.client,
  103. image='busybox:latest',
  104. name='composetest_net_container',
  105. command='top',
  106. labels={LABEL_PROJECT: 'composetest'},
  107. )
  108. net_container.start()
  109. project = Project.from_dicts(
  110. name='composetest',
  111. service_dicts=build_service_dicts({
  112. 'web': {
  113. 'image': 'busybox:latest',
  114. 'net': 'container:composetest_net_container'
  115. },
  116. }),
  117. client=self.client,
  118. )
  119. project.up()
  120. web = project.get_service('web')
  121. self.assertEqual(web.net.mode, 'container:' + net_container.id)
  122. def test_start_pause_unpause_stop_kill_remove(self):
  123. web = self.create_service('web')
  124. db = self.create_service('db')
  125. project = Project('composetest', [web, db], self.client)
  126. project.start()
  127. self.assertEqual(len(web.containers()), 0)
  128. self.assertEqual(len(db.containers()), 0)
  129. web_container_1 = web.create_container()
  130. web_container_2 = web.create_container()
  131. db_container = db.create_container()
  132. project.start(service_names=['web'])
  133. self.assertEqual(set(c.name for c in project.containers()), set([web_container_1.name, web_container_2.name]))
  134. project.start()
  135. self.assertEqual(set(c.name for c in project.containers()),
  136. set([web_container_1.name, web_container_2.name, db_container.name]))
  137. project.pause(service_names=['web'])
  138. self.assertEqual(set([c.name for c in project.containers() if c.is_paused]),
  139. set([web_container_1.name, web_container_2.name]))
  140. project.pause()
  141. self.assertEqual(set([c.name for c in project.containers() if c.is_paused]),
  142. set([web_container_1.name, web_container_2.name, db_container.name]))
  143. project.unpause(service_names=['db'])
  144. self.assertEqual(len([c.name for c in project.containers() if c.is_paused]), 2)
  145. project.unpause()
  146. self.assertEqual(len([c.name for c in project.containers() if c.is_paused]), 0)
  147. project.stop(service_names=['web'], timeout=1)
  148. self.assertEqual(set(c.name for c in project.containers()), set([db_container.name]))
  149. project.kill(service_names=['db'])
  150. self.assertEqual(len(project.containers()), 0)
  151. self.assertEqual(len(project.containers(stopped=True)), 3)
  152. project.remove_stopped(service_names=['web'])
  153. self.assertEqual(len(project.containers(stopped=True)), 1)
  154. project.remove_stopped()
  155. self.assertEqual(len(project.containers(stopped=True)), 0)
  156. def test_project_up(self):
  157. web = self.create_service('web')
  158. db = self.create_service('db', volumes=['/var/db'])
  159. project = Project('composetest', [web, db], self.client)
  160. project.start()
  161. self.assertEqual(len(project.containers()), 0)
  162. project.up(['db'])
  163. self.assertEqual(len(project.containers()), 1)
  164. self.assertEqual(len(db.containers()), 1)
  165. self.assertEqual(len(web.containers()), 0)
  166. def test_project_up_starts_uncreated_services(self):
  167. db = self.create_service('db')
  168. web = self.create_service('web', links=[(db, 'db')])
  169. project = Project('composetest', [db, web], self.client)
  170. project.up(['db'])
  171. self.assertEqual(len(project.containers()), 1)
  172. project.up()
  173. self.assertEqual(len(project.containers()), 2)
  174. self.assertEqual(len(db.containers()), 1)
  175. self.assertEqual(len(web.containers()), 1)
  176. def test_recreate_preserves_volumes(self):
  177. web = self.create_service('web')
  178. db = self.create_service('db', volumes=['/etc'])
  179. project = Project('composetest', [web, db], self.client)
  180. project.start()
  181. self.assertEqual(len(project.containers()), 0)
  182. project.up(['db'])
  183. self.assertEqual(len(project.containers()), 1)
  184. old_db_id = project.containers()[0].id
  185. db_volume_path = project.containers()[0].get('Volumes./etc')
  186. project.up(strategy=ConvergenceStrategy.always)
  187. self.assertEqual(len(project.containers()), 2)
  188. db_container = [c for c in project.containers() if 'db' in c.name][0]
  189. self.assertNotEqual(db_container.id, old_db_id)
  190. self.assertEqual(db_container.get('Volumes./etc'), db_volume_path)
  191. def test_project_up_with_no_recreate_running(self):
  192. web = self.create_service('web')
  193. db = self.create_service('db', volumes=['/var/db'])
  194. project = Project('composetest', [web, db], self.client)
  195. project.start()
  196. self.assertEqual(len(project.containers()), 0)
  197. project.up(['db'])
  198. self.assertEqual(len(project.containers()), 1)
  199. old_db_id = project.containers()[0].id
  200. db_volume_path = project.containers()[0].inspect()['Volumes']['/var/db']
  201. project.up(strategy=ConvergenceStrategy.never)
  202. self.assertEqual(len(project.containers()), 2)
  203. db_container = [c for c in project.containers() if 'db' in c.name][0]
  204. self.assertEqual(db_container.id, old_db_id)
  205. self.assertEqual(db_container.inspect()['Volumes']['/var/db'],
  206. db_volume_path)
  207. def test_project_up_with_no_recreate_stopped(self):
  208. web = self.create_service('web')
  209. db = self.create_service('db', volumes=['/var/db'])
  210. project = Project('composetest', [web, db], self.client)
  211. project.start()
  212. self.assertEqual(len(project.containers()), 0)
  213. project.up(['db'])
  214. project.kill()
  215. old_containers = project.containers(stopped=True)
  216. self.assertEqual(len(old_containers), 1)
  217. old_db_id = old_containers[0].id
  218. db_volume_path = old_containers[0].inspect()['Volumes']['/var/db']
  219. project.up(strategy=ConvergenceStrategy.never)
  220. new_containers = project.containers(stopped=True)
  221. self.assertEqual(len(new_containers), 2)
  222. self.assertEqual([c.is_running for c in new_containers], [True, True])
  223. db_container = [c for c in new_containers if 'db' in c.name][0]
  224. self.assertEqual(db_container.id, old_db_id)
  225. self.assertEqual(db_container.inspect()['Volumes']['/var/db'],
  226. db_volume_path)
  227. def test_project_up_without_all_services(self):
  228. console = self.create_service('console')
  229. db = self.create_service('db')
  230. project = Project('composetest', [console, db], self.client)
  231. project.start()
  232. self.assertEqual(len(project.containers()), 0)
  233. project.up()
  234. self.assertEqual(len(project.containers()), 2)
  235. self.assertEqual(len(db.containers()), 1)
  236. self.assertEqual(len(console.containers()), 1)
  237. def test_project_up_starts_links(self):
  238. console = self.create_service('console')
  239. db = self.create_service('db', volumes=['/var/db'])
  240. web = self.create_service('web', links=[(db, 'db')])
  241. project = Project('composetest', [web, db, console], self.client)
  242. project.start()
  243. self.assertEqual(len(project.containers()), 0)
  244. project.up(['web'])
  245. self.assertEqual(len(project.containers()), 2)
  246. self.assertEqual(len(web.containers()), 1)
  247. self.assertEqual(len(db.containers()), 1)
  248. self.assertEqual(len(console.containers()), 0)
  249. def test_project_up_starts_depends(self):
  250. project = Project.from_dicts(
  251. name='composetest',
  252. service_dicts=build_service_dicts({
  253. 'console': {
  254. 'image': 'busybox:latest',
  255. 'command': ["top"],
  256. },
  257. 'data': {
  258. 'image': 'busybox:latest',
  259. 'command': ["top"]
  260. },
  261. 'db': {
  262. 'image': 'busybox:latest',
  263. 'command': ["top"],
  264. 'volumes_from': ['data'],
  265. },
  266. 'web': {
  267. 'image': 'busybox:latest',
  268. 'command': ["top"],
  269. 'links': ['db'],
  270. },
  271. }),
  272. client=self.client,
  273. )
  274. project.start()
  275. self.assertEqual(len(project.containers()), 0)
  276. project.up(['web'])
  277. self.assertEqual(len(project.containers()), 3)
  278. self.assertEqual(len(project.get_service('web').containers()), 1)
  279. self.assertEqual(len(project.get_service('db').containers()), 1)
  280. self.assertEqual(len(project.get_service('data').containers()), 1)
  281. self.assertEqual(len(project.get_service('console').containers()), 0)
  282. def test_project_up_with_no_deps(self):
  283. project = Project.from_dicts(
  284. name='composetest',
  285. service_dicts=build_service_dicts({
  286. 'console': {
  287. 'image': 'busybox:latest',
  288. 'command': ["top"],
  289. },
  290. 'data': {
  291. 'image': 'busybox:latest',
  292. 'command': ["top"]
  293. },
  294. 'db': {
  295. 'image': 'busybox:latest',
  296. 'command': ["top"],
  297. 'volumes_from': ['data'],
  298. },
  299. 'web': {
  300. 'image': 'busybox:latest',
  301. 'command': ["top"],
  302. 'links': ['db'],
  303. },
  304. }),
  305. client=self.client,
  306. )
  307. project.start()
  308. self.assertEqual(len(project.containers()), 0)
  309. project.up(['db'], start_deps=False)
  310. self.assertEqual(len(project.containers(stopped=True)), 2)
  311. self.assertEqual(len(project.get_service('web').containers()), 0)
  312. self.assertEqual(len(project.get_service('db').containers()), 1)
  313. self.assertEqual(len(project.get_service('data').containers()), 0)
  314. self.assertEqual(len(project.get_service('data').containers(stopped=True)), 1)
  315. self.assertEqual(len(project.get_service('console').containers()), 0)
  316. def test_unscale_after_restart(self):
  317. web = self.create_service('web')
  318. project = Project('composetest', [web], self.client)
  319. project.start()
  320. service = project.get_service('web')
  321. service.scale(1)
  322. self.assertEqual(len(service.containers()), 1)
  323. service.scale(3)
  324. self.assertEqual(len(service.containers()), 3)
  325. project.up()
  326. service = project.get_service('web')
  327. self.assertEqual(len(service.containers()), 3)
  328. service.scale(1)
  329. self.assertEqual(len(service.containers()), 1)
  330. project.up()
  331. service = project.get_service('web')
  332. self.assertEqual(len(service.containers()), 1)
  333. # does scale=0 ,makes any sense? after recreating at least 1 container is running
  334. service.scale(0)
  335. project.up()
  336. service = project.get_service('web')
  337. self.assertEqual(len(service.containers()), 1)