cli_test.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656
  1. from __future__ import absolute_import
  2. import os
  3. import shlex
  4. import sys
  5. from operator import attrgetter
  6. from six import StringIO
  7. from .. import mock
  8. from .testcases import DockerClientTestCase
  9. from compose.cli.errors import UserError
  10. from compose.cli.main import TopLevelCommand
  11. from compose.project import NoSuchService
  12. class CLITestCase(DockerClientTestCase):
  13. def setUp(self):
  14. super(CLITestCase, self).setUp()
  15. self.old_sys_exit = sys.exit
  16. sys.exit = lambda code=0: None
  17. self.command = TopLevelCommand()
  18. self.command.base_dir = 'tests/fixtures/simple-composefile'
  19. def tearDown(self):
  20. sys.exit = self.old_sys_exit
  21. self.project.kill()
  22. self.project.remove_stopped()
  23. for container in self.project.containers(stopped=True, one_off=True):
  24. container.remove(force=True)
  25. super(CLITestCase, self).tearDown()
  26. @property
  27. def project(self):
  28. # Hack: allow project to be overridden. This needs refactoring so that
  29. # the project object is built exactly once, by the command object, and
  30. # accessed by the test case object.
  31. if hasattr(self, '_project'):
  32. return self._project
  33. return self.command.get_project()
  34. def test_help(self):
  35. old_base_dir = self.command.base_dir
  36. self.command.base_dir = 'tests/fixtures/no-composefile'
  37. with self.assertRaises(SystemExit) as exc_context:
  38. self.command.dispatch(['help', 'up'], None)
  39. self.assertIn('Usage: up [options] [SERVICE...]', str(exc_context.exception))
  40. # self.project.kill() fails during teardown
  41. # unless there is a composefile.
  42. self.command.base_dir = old_base_dir
  43. # TODO: address the "Inappropriate ioctl for device" warnings in test output
  44. @mock.patch('sys.stdout', new_callable=StringIO)
  45. def test_ps(self, mock_stdout):
  46. self.project.get_service('simple').create_container()
  47. self.command.dispatch(['ps'], None)
  48. self.assertIn('simplecomposefile_simple_1', mock_stdout.getvalue())
  49. @mock.patch('sys.stdout', new_callable=StringIO)
  50. def test_ps_default_composefile(self, mock_stdout):
  51. self.command.base_dir = 'tests/fixtures/multiple-composefiles'
  52. self.command.dispatch(['up', '-d'], None)
  53. self.command.dispatch(['ps'], None)
  54. output = mock_stdout.getvalue()
  55. self.assertIn('multiplecomposefiles_simple_1', output)
  56. self.assertIn('multiplecomposefiles_another_1', output)
  57. self.assertNotIn('multiplecomposefiles_yetanother_1', output)
  58. @mock.patch('sys.stdout', new_callable=StringIO)
  59. def test_ps_alternate_composefile(self, mock_stdout):
  60. config_path = os.path.abspath(
  61. 'tests/fixtures/multiple-composefiles/compose2.yml')
  62. self._project = self.command.get_project(config_path)
  63. self.command.base_dir = 'tests/fixtures/multiple-composefiles'
  64. self.command.dispatch(['-f', 'compose2.yml', 'up', '-d'], None)
  65. self.command.dispatch(['-f', 'compose2.yml', 'ps'], None)
  66. output = mock_stdout.getvalue()
  67. self.assertNotIn('multiplecomposefiles_simple_1', output)
  68. self.assertNotIn('multiplecomposefiles_another_1', output)
  69. self.assertIn('multiplecomposefiles_yetanother_1', output)
  70. @mock.patch('compose.service.log')
  71. def test_pull(self, mock_logging):
  72. self.command.dispatch(['pull'], None)
  73. mock_logging.info.assert_any_call('Pulling simple (busybox:latest)...')
  74. mock_logging.info.assert_any_call('Pulling another (busybox:latest)...')
  75. @mock.patch('compose.service.log')
  76. def test_pull_with_digest(self, mock_logging):
  77. self.command.dispatch(['-f', 'digest.yml', 'pull'], None)
  78. mock_logging.info.assert_any_call('Pulling simple (busybox:latest)...')
  79. mock_logging.info.assert_any_call(
  80. 'Pulling digest (busybox@'
  81. 'sha256:38a203e1986cf79639cfb9b2e1d6e773de84002feea2d4eb006b52004ee8502d)...')
  82. @mock.patch('sys.stdout', new_callable=StringIO)
  83. def test_build_plain(self, mock_stdout):
  84. self.command.base_dir = 'tests/fixtures/simple-dockerfile'
  85. self.command.dispatch(['build', 'simple'], None)
  86. mock_stdout.truncate(0)
  87. cache_indicator = 'Using cache'
  88. pull_indicator = 'Status: Image is up to date for busybox:latest'
  89. self.command.dispatch(['build', 'simple'], None)
  90. output = mock_stdout.getvalue()
  91. self.assertIn(cache_indicator, output)
  92. self.assertNotIn(pull_indicator, output)
  93. @mock.patch('sys.stdout', new_callable=StringIO)
  94. def test_build_no_cache(self, mock_stdout):
  95. self.command.base_dir = 'tests/fixtures/simple-dockerfile'
  96. self.command.dispatch(['build', 'simple'], None)
  97. mock_stdout.truncate(0)
  98. cache_indicator = 'Using cache'
  99. pull_indicator = 'Status: Image is up to date for busybox:latest'
  100. self.command.dispatch(['build', '--no-cache', 'simple'], None)
  101. output = mock_stdout.getvalue()
  102. self.assertNotIn(cache_indicator, output)
  103. self.assertNotIn(pull_indicator, output)
  104. @mock.patch('sys.stdout', new_callable=StringIO)
  105. def test_build_pull(self, mock_stdout):
  106. self.command.base_dir = 'tests/fixtures/simple-dockerfile'
  107. self.command.dispatch(['build', 'simple'], None)
  108. mock_stdout.truncate(0)
  109. cache_indicator = 'Using cache'
  110. pull_indicator = 'Status: Image is up to date for busybox:latest'
  111. self.command.dispatch(['build', '--pull', 'simple'], None)
  112. output = mock_stdout.getvalue()
  113. self.assertIn(cache_indicator, output)
  114. self.assertIn(pull_indicator, output)
  115. @mock.patch('sys.stdout', new_callable=StringIO)
  116. def test_build_no_cache_pull(self, mock_stdout):
  117. self.command.base_dir = 'tests/fixtures/simple-dockerfile'
  118. self.command.dispatch(['build', 'simple'], None)
  119. mock_stdout.truncate(0)
  120. cache_indicator = 'Using cache'
  121. pull_indicator = 'Status: Image is up to date for busybox:latest'
  122. self.command.dispatch(['build', '--no-cache', '--pull', 'simple'], None)
  123. output = mock_stdout.getvalue()
  124. self.assertNotIn(cache_indicator, output)
  125. self.assertIn(pull_indicator, output)
  126. def test_up_detached(self):
  127. self.command.dispatch(['up', '-d'], None)
  128. service = self.project.get_service('simple')
  129. another = self.project.get_service('another')
  130. self.assertEqual(len(service.containers()), 1)
  131. self.assertEqual(len(another.containers()), 1)
  132. # Ensure containers don't have stdin and stdout connected in -d mode
  133. container, = service.containers()
  134. self.assertFalse(container.get('Config.AttachStderr'))
  135. self.assertFalse(container.get('Config.AttachStdout'))
  136. self.assertFalse(container.get('Config.AttachStdin'))
  137. def test_up_attached(self):
  138. with mock.patch(
  139. 'compose.cli.main.attach_to_logs',
  140. autospec=True
  141. ) as mock_attach:
  142. self.command.dispatch(['up'], None)
  143. _, args, kwargs = mock_attach.mock_calls[0]
  144. _project, log_printer, _names, _timeout = args
  145. service = self.project.get_service('simple')
  146. another = self.project.get_service('another')
  147. self.assertEqual(len(service.containers()), 1)
  148. self.assertEqual(len(another.containers()), 1)
  149. self.assertEqual(
  150. set(log_printer.containers),
  151. set(self.project.containers())
  152. )
  153. def test_up_with_links(self):
  154. self.command.base_dir = 'tests/fixtures/links-composefile'
  155. self.command.dispatch(['up', '-d', 'web'], None)
  156. web = self.project.get_service('web')
  157. db = self.project.get_service('db')
  158. console = self.project.get_service('console')
  159. self.assertEqual(len(web.containers()), 1)
  160. self.assertEqual(len(db.containers()), 1)
  161. self.assertEqual(len(console.containers()), 0)
  162. def test_up_with_no_deps(self):
  163. self.command.base_dir = 'tests/fixtures/links-composefile'
  164. self.command.dispatch(['up', '-d', '--no-deps', 'web'], None)
  165. web = self.project.get_service('web')
  166. db = self.project.get_service('db')
  167. console = self.project.get_service('console')
  168. self.assertEqual(len(web.containers()), 1)
  169. self.assertEqual(len(db.containers()), 0)
  170. self.assertEqual(len(console.containers()), 0)
  171. def test_up_with_force_recreate(self):
  172. self.command.dispatch(['up', '-d'], None)
  173. service = self.project.get_service('simple')
  174. self.assertEqual(len(service.containers()), 1)
  175. old_ids = [c.id for c in service.containers()]
  176. self.command.dispatch(['up', '-d', '--force-recreate'], None)
  177. self.assertEqual(len(service.containers()), 1)
  178. new_ids = [c.id for c in service.containers()]
  179. self.assertNotEqual(old_ids, new_ids)
  180. def test_up_with_no_recreate(self):
  181. self.command.dispatch(['up', '-d'], None)
  182. service = self.project.get_service('simple')
  183. self.assertEqual(len(service.containers()), 1)
  184. old_ids = [c.id for c in service.containers()]
  185. self.command.dispatch(['up', '-d', '--no-recreate'], None)
  186. self.assertEqual(len(service.containers()), 1)
  187. new_ids = [c.id for c in service.containers()]
  188. self.assertEqual(old_ids, new_ids)
  189. def test_up_with_force_recreate_and_no_recreate(self):
  190. with self.assertRaises(UserError):
  191. self.command.dispatch(['up', '-d', '--force-recreate', '--no-recreate'], None)
  192. def test_up_with_timeout(self):
  193. self.command.dispatch(['up', '-d', '-t', '1'], None)
  194. service = self.project.get_service('simple')
  195. another = self.project.get_service('another')
  196. self.assertEqual(len(service.containers()), 1)
  197. self.assertEqual(len(another.containers()), 1)
  198. # Ensure containers don't have stdin and stdout connected in -d mode
  199. config = service.containers()[0].inspect()['Config']
  200. self.assertFalse(config['AttachStderr'])
  201. self.assertFalse(config['AttachStdout'])
  202. self.assertFalse(config['AttachStdin'])
  203. @mock.patch('dockerpty.start')
  204. def test_run_service_without_links(self, mock_stdout):
  205. self.command.base_dir = 'tests/fixtures/links-composefile'
  206. self.command.dispatch(['run', 'console', '/bin/true'], None)
  207. self.assertEqual(len(self.project.containers()), 0)
  208. # Ensure stdin/out was open
  209. container = self.project.containers(stopped=True, one_off=True)[0]
  210. config = container.inspect()['Config']
  211. self.assertTrue(config['AttachStderr'])
  212. self.assertTrue(config['AttachStdout'])
  213. self.assertTrue(config['AttachStdin'])
  214. @mock.patch('dockerpty.start')
  215. def test_run_service_with_links(self, _):
  216. self.command.base_dir = 'tests/fixtures/links-composefile'
  217. self.command.dispatch(['run', 'web', '/bin/true'], None)
  218. db = self.project.get_service('db')
  219. console = self.project.get_service('console')
  220. self.assertEqual(len(db.containers()), 1)
  221. self.assertEqual(len(console.containers()), 0)
  222. @mock.patch('dockerpty.start')
  223. def test_run_with_no_deps(self, _):
  224. self.command.base_dir = 'tests/fixtures/links-composefile'
  225. self.command.dispatch(['run', '--no-deps', 'web', '/bin/true'], None)
  226. db = self.project.get_service('db')
  227. self.assertEqual(len(db.containers()), 0)
  228. @mock.patch('dockerpty.start')
  229. def test_run_does_not_recreate_linked_containers(self, _):
  230. self.command.base_dir = 'tests/fixtures/links-composefile'
  231. self.command.dispatch(['up', '-d', 'db'], None)
  232. db = self.project.get_service('db')
  233. self.assertEqual(len(db.containers()), 1)
  234. old_ids = [c.id for c in db.containers()]
  235. self.command.dispatch(['run', 'web', '/bin/true'], None)
  236. self.assertEqual(len(db.containers()), 1)
  237. new_ids = [c.id for c in db.containers()]
  238. self.assertEqual(old_ids, new_ids)
  239. @mock.patch('dockerpty.start')
  240. def test_run_without_command(self, _):
  241. self.command.base_dir = 'tests/fixtures/commands-composefile'
  242. self.check_build('tests/fixtures/simple-dockerfile', tag='composetest_test')
  243. self.command.dispatch(['run', 'implicit'], None)
  244. service = self.project.get_service('implicit')
  245. containers = service.containers(stopped=True, one_off=True)
  246. self.assertEqual(
  247. [c.human_readable_command for c in containers],
  248. [u'/bin/sh -c echo "success"'],
  249. )
  250. self.command.dispatch(['run', 'explicit'], None)
  251. service = self.project.get_service('explicit')
  252. containers = service.containers(stopped=True, one_off=True)
  253. self.assertEqual(
  254. [c.human_readable_command for c in containers],
  255. [u'/bin/true'],
  256. )
  257. @mock.patch('dockerpty.start')
  258. def test_run_service_with_entrypoint_overridden(self, _):
  259. self.command.base_dir = 'tests/fixtures/dockerfile_with_entrypoint'
  260. name = 'service'
  261. self.command.dispatch(
  262. ['run', '--entrypoint', '/bin/echo', name, 'helloworld'],
  263. None
  264. )
  265. service = self.project.get_service(name)
  266. container = service.containers(stopped=True, one_off=True)[0]
  267. self.assertEqual(
  268. shlex.split(container.human_readable_command),
  269. [u'/bin/echo', u'helloworld'],
  270. )
  271. @mock.patch('dockerpty.start')
  272. def test_run_service_with_user_overridden(self, _):
  273. self.command.base_dir = 'tests/fixtures/user-composefile'
  274. name = 'service'
  275. user = 'sshd'
  276. args = ['run', '--user={user}'.format(user=user), name]
  277. self.command.dispatch(args, None)
  278. service = self.project.get_service(name)
  279. container = service.containers(stopped=True, one_off=True)[0]
  280. self.assertEqual(user, container.get('Config.User'))
  281. @mock.patch('dockerpty.start')
  282. def test_run_service_with_user_overridden_short_form(self, _):
  283. self.command.base_dir = 'tests/fixtures/user-composefile'
  284. name = 'service'
  285. user = 'sshd'
  286. args = ['run', '-u', user, name]
  287. self.command.dispatch(args, None)
  288. service = self.project.get_service(name)
  289. container = service.containers(stopped=True, one_off=True)[0]
  290. self.assertEqual(user, container.get('Config.User'))
  291. @mock.patch('dockerpty.start')
  292. def test_run_service_with_environement_overridden(self, _):
  293. name = 'service'
  294. self.command.base_dir = 'tests/fixtures/environment-composefile'
  295. self.command.dispatch(
  296. ['run', '-e', 'foo=notbar', '-e', 'allo=moto=bobo',
  297. '-e', 'alpha=beta', name],
  298. None
  299. )
  300. service = self.project.get_service(name)
  301. container = service.containers(stopped=True, one_off=True)[0]
  302. # env overriden
  303. self.assertEqual('notbar', container.environment['foo'])
  304. # keep environement from yaml
  305. self.assertEqual('world', container.environment['hello'])
  306. # added option from command line
  307. self.assertEqual('beta', container.environment['alpha'])
  308. # make sure a value with a = don't crash out
  309. self.assertEqual('moto=bobo', container.environment['allo'])
  310. @mock.patch('dockerpty.start')
  311. def test_run_service_without_map_ports(self, _):
  312. # create one off container
  313. self.command.base_dir = 'tests/fixtures/ports-composefile'
  314. self.command.dispatch(['run', '-d', 'simple'], None)
  315. container = self.project.get_service('simple').containers(one_off=True)[0]
  316. # get port information
  317. port_random = container.get_local_port(3000)
  318. port_assigned = container.get_local_port(3001)
  319. # close all one off containers we just created
  320. container.stop()
  321. # check the ports
  322. self.assertEqual(port_random, None)
  323. self.assertEqual(port_assigned, None)
  324. @mock.patch('dockerpty.start')
  325. def test_run_service_with_map_ports(self, _):
  326. # create one off container
  327. self.command.base_dir = 'tests/fixtures/ports-composefile'
  328. self.command.dispatch(['run', '-d', '--service-ports', 'simple'], None)
  329. container = self.project.get_service('simple').containers(one_off=True)[0]
  330. # get port information
  331. port_random = container.get_local_port(3000)
  332. port_assigned = container.get_local_port(3001)
  333. port_range = container.get_local_port(3002), container.get_local_port(3003)
  334. # close all one off containers we just created
  335. container.stop()
  336. # check the ports
  337. self.assertNotEqual(port_random, None)
  338. self.assertIn("0.0.0.0", port_random)
  339. self.assertEqual(port_assigned, "0.0.0.0:49152")
  340. self.assertEqual(port_range[0], "0.0.0.0:49153")
  341. self.assertEqual(port_range[1], "0.0.0.0:49154")
  342. @mock.patch('dockerpty.start')
  343. def test_run_service_with_explicitly_maped_ports(self, _):
  344. # create one off container
  345. self.command.base_dir = 'tests/fixtures/ports-composefile'
  346. self.command.dispatch(['run', '-d', '-p', '30000:3000', '--publish', '30001:3001', 'simple'], None)
  347. container = self.project.get_service('simple').containers(one_off=True)[0]
  348. # get port information
  349. port_short = container.get_local_port(3000)
  350. port_full = container.get_local_port(3001)
  351. # close all one off containers we just created
  352. container.stop()
  353. # check the ports
  354. self.assertEqual(port_short, "0.0.0.0:30000")
  355. self.assertEqual(port_full, "0.0.0.0:30001")
  356. @mock.patch('dockerpty.start')
  357. def test_run_service_with_explicitly_maped_ip_ports(self, _):
  358. # create one off container
  359. self.command.base_dir = 'tests/fixtures/ports-composefile'
  360. self.command.dispatch(['run', '-d', '-p', '127.0.0.1:30000:3000', '--publish', '127.0.0.1:30001:3001', 'simple'], None)
  361. container = self.project.get_service('simple').containers(one_off=True)[0]
  362. # get port information
  363. port_short = container.get_local_port(3000)
  364. port_full = container.get_local_port(3001)
  365. # close all one off containers we just created
  366. container.stop()
  367. # check the ports
  368. self.assertEqual(port_short, "127.0.0.1:30000")
  369. self.assertEqual(port_full, "127.0.0.1:30001")
  370. @mock.patch('dockerpty.start')
  371. def test_run_with_custom_name(self, _):
  372. self.command.base_dir = 'tests/fixtures/environment-composefile'
  373. name = 'the-container-name'
  374. self.command.dispatch(['run', '--name', name, 'service'], None)
  375. service = self.project.get_service('service')
  376. container, = service.containers(stopped=True, one_off=True)
  377. self.assertEqual(container.name, name)
  378. def test_rm(self):
  379. service = self.project.get_service('simple')
  380. service.create_container()
  381. service.kill()
  382. self.assertEqual(len(service.containers(stopped=True)), 1)
  383. self.command.dispatch(['rm', '--force'], None)
  384. self.assertEqual(len(service.containers(stopped=True)), 0)
  385. service = self.project.get_service('simple')
  386. service.create_container()
  387. service.kill()
  388. self.assertEqual(len(service.containers(stopped=True)), 1)
  389. self.command.dispatch(['rm', '-f'], None)
  390. self.assertEqual(len(service.containers(stopped=True)), 0)
  391. def test_stop(self):
  392. self.command.dispatch(['up', '-d'], None)
  393. service = self.project.get_service('simple')
  394. self.assertEqual(len(service.containers()), 1)
  395. self.assertTrue(service.containers()[0].is_running)
  396. self.command.dispatch(['stop', '-t', '1'], None)
  397. self.assertEqual(len(service.containers(stopped=True)), 1)
  398. self.assertFalse(service.containers(stopped=True)[0].is_running)
  399. def test_pause_unpause(self):
  400. self.command.dispatch(['up', '-d'], None)
  401. service = self.project.get_service('simple')
  402. self.assertFalse(service.containers()[0].is_paused)
  403. self.command.dispatch(['pause'], None)
  404. self.assertTrue(service.containers()[0].is_paused)
  405. self.command.dispatch(['unpause'], None)
  406. self.assertFalse(service.containers()[0].is_paused)
  407. def test_logs_invalid_service_name(self):
  408. with self.assertRaises(NoSuchService):
  409. self.command.dispatch(['logs', 'madeupname'], None)
  410. def test_kill(self):
  411. self.command.dispatch(['up', '-d'], None)
  412. service = self.project.get_service('simple')
  413. self.assertEqual(len(service.containers()), 1)
  414. self.assertTrue(service.containers()[0].is_running)
  415. self.command.dispatch(['kill'], None)
  416. self.assertEqual(len(service.containers(stopped=True)), 1)
  417. self.assertFalse(service.containers(stopped=True)[0].is_running)
  418. def test_kill_signal_sigstop(self):
  419. self.command.dispatch(['up', '-d'], None)
  420. service = self.project.get_service('simple')
  421. self.assertEqual(len(service.containers()), 1)
  422. self.assertTrue(service.containers()[0].is_running)
  423. self.command.dispatch(['kill', '-s', 'SIGSTOP'], None)
  424. self.assertEqual(len(service.containers()), 1)
  425. # The container is still running. It has only been paused
  426. self.assertTrue(service.containers()[0].is_running)
  427. def test_kill_stopped_service(self):
  428. self.command.dispatch(['up', '-d'], None)
  429. service = self.project.get_service('simple')
  430. self.command.dispatch(['kill', '-s', 'SIGSTOP'], None)
  431. self.assertTrue(service.containers()[0].is_running)
  432. self.command.dispatch(['kill', '-s', 'SIGKILL'], None)
  433. self.assertEqual(len(service.containers(stopped=True)), 1)
  434. self.assertFalse(service.containers(stopped=True)[0].is_running)
  435. def test_restart(self):
  436. service = self.project.get_service('simple')
  437. container = service.create_container()
  438. service.start_container(container)
  439. started_at = container.dictionary['State']['StartedAt']
  440. self.command.dispatch(['restart', '-t', '1'], None)
  441. container.inspect()
  442. self.assertNotEqual(
  443. container.dictionary['State']['FinishedAt'],
  444. '0001-01-01T00:00:00Z',
  445. )
  446. self.assertNotEqual(
  447. container.dictionary['State']['StartedAt'],
  448. started_at,
  449. )
  450. def test_scale(self):
  451. project = self.project
  452. self.command.scale(project, {'SERVICE=NUM': ['simple=1']})
  453. self.assertEqual(len(project.get_service('simple').containers()), 1)
  454. self.command.scale(project, {'SERVICE=NUM': ['simple=3', 'another=2']})
  455. self.assertEqual(len(project.get_service('simple').containers()), 3)
  456. self.assertEqual(len(project.get_service('another').containers()), 2)
  457. self.command.scale(project, {'SERVICE=NUM': ['simple=1', 'another=1']})
  458. self.assertEqual(len(project.get_service('simple').containers()), 1)
  459. self.assertEqual(len(project.get_service('another').containers()), 1)
  460. self.command.scale(project, {'SERVICE=NUM': ['simple=1', 'another=1']})
  461. self.assertEqual(len(project.get_service('simple').containers()), 1)
  462. self.assertEqual(len(project.get_service('another').containers()), 1)
  463. self.command.scale(project, {'SERVICE=NUM': ['simple=0', 'another=0']})
  464. self.assertEqual(len(project.get_service('simple').containers()), 0)
  465. self.assertEqual(len(project.get_service('another').containers()), 0)
  466. def test_port(self):
  467. self.command.base_dir = 'tests/fixtures/ports-composefile'
  468. self.command.dispatch(['up', '-d'], None)
  469. container = self.project.get_service('simple').get_container()
  470. @mock.patch('sys.stdout', new_callable=StringIO)
  471. def get_port(number, mock_stdout):
  472. self.command.dispatch(['port', 'simple', str(number)], None)
  473. return mock_stdout.getvalue().rstrip()
  474. self.assertEqual(get_port(3000), container.get_local_port(3000))
  475. self.assertEqual(get_port(3001), "0.0.0.0:49152")
  476. self.assertEqual(get_port(3002), "0.0.0.0:49153")
  477. def test_port_with_scale(self):
  478. self.command.base_dir = 'tests/fixtures/ports-composefile-scale'
  479. self.command.dispatch(['scale', 'simple=2'], None)
  480. containers = sorted(
  481. self.project.containers(service_names=['simple']),
  482. key=attrgetter('name'))
  483. @mock.patch('sys.stdout', new_callable=StringIO)
  484. def get_port(number, mock_stdout, index=None):
  485. if index is None:
  486. self.command.dispatch(['port', 'simple', str(number)], None)
  487. else:
  488. self.command.dispatch(['port', '--index=' + str(index), 'simple', str(number)], None)
  489. return mock_stdout.getvalue().rstrip()
  490. self.assertEqual(get_port(3000), containers[0].get_local_port(3000))
  491. self.assertEqual(get_port(3000, index=1), containers[0].get_local_port(3000))
  492. self.assertEqual(get_port(3000, index=2), containers[1].get_local_port(3000))
  493. self.assertEqual(get_port(3002), "")
  494. def test_env_file_relative_to_compose_file(self):
  495. config_path = os.path.abspath('tests/fixtures/env-file/docker-compose.yml')
  496. self.command.dispatch(['-f', config_path, 'up', '-d'], None)
  497. self._project = self.command.get_project(config_path)
  498. containers = self.project.containers(stopped=True)
  499. self.assertEqual(len(containers), 1)
  500. self.assertIn("FOO=1", containers[0].get('Config.Env'))
  501. @mock.patch.dict(os.environ)
  502. def test_home_and_env_var_in_volume_path(self):
  503. os.environ['VOLUME_NAME'] = 'my-volume'
  504. os.environ['HOME'] = '/tmp/home-dir'
  505. expected_host_path = os.path.join(os.environ['HOME'], os.environ['VOLUME_NAME'])
  506. self.command.base_dir = 'tests/fixtures/volume-path-interpolation'
  507. self.command.dispatch(['up', '-d'], None)
  508. container = self.project.containers(stopped=True)[0]
  509. actual_host_path = container.get('Volumes')['/container-path']
  510. components = actual_host_path.split('/')
  511. self.assertTrue(components[-2:] == ['home-dir', 'my-volume'],
  512. msg="Last two components differ: %s, %s" % (actual_host_path, expected_host_path))
  513. def test_up_with_extends(self):
  514. self.command.base_dir = 'tests/fixtures/extends'
  515. self.command.dispatch(['up', '-d'], None)
  516. self.assertEqual(
  517. set([s.name for s in self.project.services]),
  518. set(['mydb', 'myweb']),
  519. )
  520. # Sort by name so we get [db, web]
  521. containers = sorted(
  522. self.project.containers(stopped=True),
  523. key=lambda c: c.name,
  524. )
  525. self.assertEqual(len(containers), 2)
  526. web = containers[1]
  527. self.assertEqual(set(web.links()), set(['db', 'mydb_1', 'extends_mydb_1']))
  528. expected_env = set([
  529. "FOO=1",
  530. "BAR=2",
  531. "BAZ=2",
  532. ])
  533. self.assertTrue(expected_env <= set(web.get('Config.Env')))