1
0

project_test.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921
  1. import datetime
  2. import os
  3. import tempfile
  4. import docker
  5. import pytest
  6. from docker.errors import NotFound
  7. from .. import mock
  8. from .. import unittest
  9. from ..helpers import BUSYBOX_IMAGE_WITH_TAG
  10. from compose.config import ConfigurationError
  11. from compose.config.config import Config
  12. from compose.config.types import VolumeFromSpec
  13. from compose.const import COMPOSE_SPEC as VERSION
  14. from compose.const import COMPOSEFILE_V1 as V1
  15. from compose.const import DEFAULT_TIMEOUT
  16. from compose.const import LABEL_SERVICE
  17. from compose.container import Container
  18. from compose.errors import OperationFailedError
  19. from compose.project import get_secrets
  20. from compose.project import NoSuchService
  21. from compose.project import Project
  22. from compose.project import ProjectError
  23. from compose.service import ImageType
  24. from compose.service import Service
  25. def build_config(**kwargs):
  26. return Config(
  27. version=kwargs.get('version', VERSION),
  28. services=kwargs.get('services'),
  29. volumes=kwargs.get('volumes'),
  30. networks=kwargs.get('networks'),
  31. secrets=kwargs.get('secrets'),
  32. configs=kwargs.get('configs'),
  33. )
  34. class ProjectTest(unittest.TestCase):
  35. def setUp(self):
  36. self.mock_client = mock.create_autospec(docker.APIClient)
  37. self.mock_client._general_configs = {}
  38. self.mock_client.api_version = docker.constants.DEFAULT_DOCKER_API_VERSION
  39. def test_from_config_v1(self):
  40. config = build_config(
  41. version=V1,
  42. services=[
  43. {
  44. 'name': 'web',
  45. 'image': BUSYBOX_IMAGE_WITH_TAG,
  46. },
  47. {
  48. 'name': 'db',
  49. 'image': BUSYBOX_IMAGE_WITH_TAG,
  50. },
  51. ],
  52. networks=None,
  53. volumes=None,
  54. secrets=None,
  55. configs=None,
  56. )
  57. project = Project.from_config(
  58. name='composetest',
  59. config_data=config,
  60. client=None,
  61. )
  62. assert len(project.services) == 2
  63. assert project.get_service('web').name == 'web'
  64. assert project.get_service('web').options['image'] == BUSYBOX_IMAGE_WITH_TAG
  65. assert project.get_service('db').name == 'db'
  66. assert project.get_service('db').options['image'] == BUSYBOX_IMAGE_WITH_TAG
  67. assert not project.networks.use_networking
  68. @mock.patch('compose.network.Network.true_name', lambda n: n.full_name)
  69. def test_from_config_v2(self):
  70. config = build_config(
  71. services=[
  72. {
  73. 'name': 'web',
  74. 'image': BUSYBOX_IMAGE_WITH_TAG,
  75. },
  76. {
  77. 'name': 'db',
  78. 'image': BUSYBOX_IMAGE_WITH_TAG,
  79. },
  80. ],
  81. networks=None,
  82. volumes=None,
  83. secrets=None,
  84. configs=None,
  85. )
  86. project = Project.from_config('composetest', config, None)
  87. assert len(project.services) == 2
  88. assert project.networks.use_networking
  89. def test_get_service(self):
  90. web = Service(
  91. project='composetest',
  92. name='web',
  93. client=None,
  94. image=BUSYBOX_IMAGE_WITH_TAG,
  95. )
  96. project = Project('test', [web], None)
  97. assert project.get_service('web') == web
  98. def test_get_services_returns_all_services_without_args(self):
  99. web = Service(
  100. project='composetest',
  101. name='web',
  102. image='foo',
  103. )
  104. console = Service(
  105. project='composetest',
  106. name='console',
  107. image='foo',
  108. )
  109. project = Project('test', [web, console], None)
  110. assert project.get_services() == [web, console]
  111. def test_get_services_returns_listed_services_with_args(self):
  112. web = Service(
  113. project='composetest',
  114. name='web',
  115. image='foo',
  116. )
  117. console = Service(
  118. project='composetest',
  119. name='console',
  120. image='foo',
  121. )
  122. project = Project('test', [web, console], None)
  123. assert project.get_services(['console']) == [console]
  124. def test_get_services_with_include_links(self):
  125. db = Service(
  126. project='composetest',
  127. name='db',
  128. image='foo',
  129. )
  130. web = Service(
  131. project='composetest',
  132. name='web',
  133. image='foo',
  134. links=[(db, 'database')]
  135. )
  136. cache = Service(
  137. project='composetest',
  138. name='cache',
  139. image='foo'
  140. )
  141. console = Service(
  142. project='composetest',
  143. name='console',
  144. image='foo',
  145. links=[(web, 'web')]
  146. )
  147. project = Project('test', [web, db, cache, console], None)
  148. assert project.get_services(['console'], include_deps=True) == [db, web, console]
  149. def test_get_services_removes_duplicates_following_links(self):
  150. db = Service(
  151. project='composetest',
  152. name='db',
  153. image='foo',
  154. )
  155. web = Service(
  156. project='composetest',
  157. name='web',
  158. image='foo',
  159. links=[(db, 'database')]
  160. )
  161. project = Project('test', [web, db], None)
  162. assert project.get_services(['web', 'db'], include_deps=True) == [db, web]
  163. def test_use_volumes_from_container(self):
  164. container_id = 'aabbccddee'
  165. container_dict = dict(Name='aaa', Id=container_id)
  166. self.mock_client.inspect_container.return_value = container_dict
  167. project = Project.from_config(
  168. name='test',
  169. client=self.mock_client,
  170. config_data=build_config(
  171. services=[{
  172. 'name': 'test',
  173. 'image': BUSYBOX_IMAGE_WITH_TAG,
  174. 'volumes_from': [VolumeFromSpec('aaa', 'rw', 'container')]
  175. }],
  176. networks=None,
  177. volumes=None,
  178. secrets=None,
  179. configs=None,
  180. ),
  181. )
  182. assert project.get_service('test')._get_volumes_from() == [container_id + ":rw"]
  183. def test_use_volumes_from_service_no_container(self):
  184. container_name = 'test_vol_1'
  185. self.mock_client.containers.return_value = [
  186. {
  187. "Name": container_name,
  188. "Names": [container_name],
  189. "Id": container_name,
  190. "Image": BUSYBOX_IMAGE_WITH_TAG
  191. }
  192. ]
  193. project = Project.from_config(
  194. name='test',
  195. client=self.mock_client,
  196. config_data=build_config(
  197. services=[
  198. {
  199. 'name': 'vol',
  200. 'image': BUSYBOX_IMAGE_WITH_TAG
  201. },
  202. {
  203. 'name': 'test',
  204. 'image': BUSYBOX_IMAGE_WITH_TAG,
  205. 'volumes_from': [VolumeFromSpec('vol', 'rw', 'service')]
  206. }
  207. ],
  208. networks=None,
  209. volumes=None,
  210. secrets=None,
  211. configs=None,
  212. ),
  213. )
  214. assert project.get_service('test')._get_volumes_from() == [container_name + ":rw"]
  215. @mock.patch('compose.network.Network.true_name', lambda n: n.full_name)
  216. def test_use_volumes_from_service_container(self):
  217. container_ids = ['aabbccddee', '12345']
  218. project = Project.from_config(
  219. name='test',
  220. client=None,
  221. config_data=build_config(
  222. services=[
  223. {
  224. 'name': 'vol',
  225. 'image': BUSYBOX_IMAGE_WITH_TAG
  226. },
  227. {
  228. 'name': 'test',
  229. 'image': BUSYBOX_IMAGE_WITH_TAG,
  230. 'volumes_from': [VolumeFromSpec('vol', 'rw', 'service')]
  231. }
  232. ],
  233. networks=None,
  234. volumes=None,
  235. secrets=None,
  236. configs=None,
  237. ),
  238. )
  239. with mock.patch.object(Service, 'containers') as mock_return:
  240. mock_return.return_value = [
  241. mock.Mock(id=container_id, spec=Container)
  242. for container_id in container_ids]
  243. assert (
  244. project.get_service('test')._get_volumes_from() ==
  245. [container_ids[0] + ':rw']
  246. )
  247. def test_events_legacy(self):
  248. services = [Service(name='web'), Service(name='db')]
  249. project = Project('test', services, self.mock_client)
  250. self.mock_client.api_version = '1.21'
  251. self.mock_client.events.return_value = iter([
  252. {
  253. 'status': 'create',
  254. 'from': 'example/image',
  255. 'id': 'abcde',
  256. 'time': 1420092061,
  257. 'timeNano': 14200920610000002000,
  258. },
  259. {
  260. 'status': 'attach',
  261. 'from': 'example/image',
  262. 'id': 'abcde',
  263. 'time': 1420092061,
  264. 'timeNano': 14200920610000003000,
  265. },
  266. {
  267. 'status': 'create',
  268. 'from': 'example/other',
  269. 'id': 'bdbdbd',
  270. 'time': 1420092061,
  271. 'timeNano': 14200920610000005000,
  272. },
  273. {
  274. 'status': 'create',
  275. 'from': 'example/db',
  276. 'id': 'ababa',
  277. 'time': 1420092061,
  278. 'timeNano': 14200920610000004000,
  279. },
  280. {
  281. 'status': 'destroy',
  282. 'from': 'example/db',
  283. 'id': 'eeeee',
  284. 'time': 1420092061,
  285. 'timeNano': 14200920610000004000,
  286. },
  287. ])
  288. def dt_with_microseconds(dt, us):
  289. return datetime.datetime.fromtimestamp(dt).replace(microsecond=us)
  290. def get_container(cid):
  291. if cid == 'eeeee':
  292. raise NotFound(None, None, "oops")
  293. if cid == 'abcde':
  294. name = 'web'
  295. labels = {LABEL_SERVICE: name}
  296. elif cid == 'ababa':
  297. name = 'db'
  298. labels = {LABEL_SERVICE: name}
  299. else:
  300. labels = {}
  301. name = ''
  302. return {
  303. 'Id': cid,
  304. 'Config': {'Labels': labels},
  305. 'Name': '/project_%s_1' % name,
  306. }
  307. self.mock_client.inspect_container.side_effect = get_container
  308. events = project.events()
  309. events_list = list(events)
  310. # Assert the return value is a generator
  311. assert not list(events)
  312. assert events_list == [
  313. {
  314. 'type': 'container',
  315. 'service': 'web',
  316. 'action': 'create',
  317. 'id': 'abcde',
  318. 'attributes': {
  319. 'name': 'project_web_1',
  320. 'image': 'example/image',
  321. },
  322. 'time': dt_with_microseconds(1420092061, 2),
  323. 'container': Container(None, {'Id': 'abcde'}),
  324. },
  325. {
  326. 'type': 'container',
  327. 'service': 'web',
  328. 'action': 'attach',
  329. 'id': 'abcde',
  330. 'attributes': {
  331. 'name': 'project_web_1',
  332. 'image': 'example/image',
  333. },
  334. 'time': dt_with_microseconds(1420092061, 3),
  335. 'container': Container(None, {'Id': 'abcde'}),
  336. },
  337. {
  338. 'type': 'container',
  339. 'service': 'db',
  340. 'action': 'create',
  341. 'id': 'ababa',
  342. 'attributes': {
  343. 'name': 'project_db_1',
  344. 'image': 'example/db',
  345. },
  346. 'time': dt_with_microseconds(1420092061, 4),
  347. 'container': Container(None, {'Id': 'ababa'}),
  348. },
  349. ]
  350. def test_events(self):
  351. services = [Service(name='web'), Service(name='db')]
  352. project = Project('test', services, self.mock_client)
  353. self.mock_client.api_version = '1.35'
  354. self.mock_client.events.return_value = iter([
  355. {
  356. 'status': 'create',
  357. 'from': 'example/image',
  358. 'Type': 'container',
  359. 'Actor': {
  360. 'ID': 'abcde',
  361. 'Attributes': {
  362. 'com.docker.compose.project': 'test',
  363. 'com.docker.compose.service': 'web',
  364. 'image': 'example/image',
  365. 'name': 'test_web_1',
  366. }
  367. },
  368. 'id': 'abcde',
  369. 'time': 1420092061,
  370. 'timeNano': 14200920610000002000,
  371. },
  372. {
  373. 'status': 'attach',
  374. 'from': 'example/image',
  375. 'Type': 'container',
  376. 'Actor': {
  377. 'ID': 'abcde',
  378. 'Attributes': {
  379. 'com.docker.compose.project': 'test',
  380. 'com.docker.compose.service': 'web',
  381. 'image': 'example/image',
  382. 'name': 'test_web_1',
  383. }
  384. },
  385. 'id': 'abcde',
  386. 'time': 1420092061,
  387. 'timeNano': 14200920610000003000,
  388. },
  389. {
  390. 'status': 'create',
  391. 'from': 'example/other',
  392. 'Type': 'container',
  393. 'Actor': {
  394. 'ID': 'bdbdbd',
  395. 'Attributes': {
  396. 'image': 'example/other',
  397. 'name': 'shrewd_einstein',
  398. }
  399. },
  400. 'id': 'bdbdbd',
  401. 'time': 1420092061,
  402. 'timeNano': 14200920610000005000,
  403. },
  404. {
  405. 'status': 'create',
  406. 'from': 'example/db',
  407. 'Type': 'container',
  408. 'Actor': {
  409. 'ID': 'ababa',
  410. 'Attributes': {
  411. 'com.docker.compose.project': 'test',
  412. 'com.docker.compose.service': 'db',
  413. 'image': 'example/db',
  414. 'name': 'test_db_1',
  415. }
  416. },
  417. 'id': 'ababa',
  418. 'time': 1420092061,
  419. 'timeNano': 14200920610000004000,
  420. },
  421. {
  422. 'status': 'destroy',
  423. 'from': 'example/db',
  424. 'Type': 'container',
  425. 'Actor': {
  426. 'ID': 'eeeee',
  427. 'Attributes': {
  428. 'com.docker.compose.project': 'test',
  429. 'com.docker.compose.service': 'db',
  430. 'image': 'example/db',
  431. 'name': 'test_db_1',
  432. }
  433. },
  434. 'id': 'eeeee',
  435. 'time': 1420092061,
  436. 'timeNano': 14200920610000004000,
  437. },
  438. ])
  439. def dt_with_microseconds(dt, us):
  440. return datetime.datetime.fromtimestamp(dt).replace(microsecond=us)
  441. def get_container(cid):
  442. if cid == 'eeeee':
  443. raise NotFound(None, None, "oops")
  444. if cid == 'abcde':
  445. name = 'web'
  446. labels = {LABEL_SERVICE: name}
  447. elif cid == 'ababa':
  448. name = 'db'
  449. labels = {LABEL_SERVICE: name}
  450. else:
  451. labels = {}
  452. name = ''
  453. return {
  454. 'Id': cid,
  455. 'Config': {'Labels': labels},
  456. 'Name': '/project_%s_1' % name,
  457. }
  458. self.mock_client.inspect_container.side_effect = get_container
  459. events = project.events()
  460. events_list = list(events)
  461. # Assert the return value is a generator
  462. assert not list(events)
  463. assert events_list == [
  464. {
  465. 'type': 'container',
  466. 'service': 'web',
  467. 'action': 'create',
  468. 'id': 'abcde',
  469. 'attributes': {
  470. 'name': 'test_web_1',
  471. 'image': 'example/image',
  472. },
  473. 'time': dt_with_microseconds(1420092061, 2),
  474. 'container': Container(None, get_container('abcde')),
  475. },
  476. {
  477. 'type': 'container',
  478. 'service': 'web',
  479. 'action': 'attach',
  480. 'id': 'abcde',
  481. 'attributes': {
  482. 'name': 'test_web_1',
  483. 'image': 'example/image',
  484. },
  485. 'time': dt_with_microseconds(1420092061, 3),
  486. 'container': Container(None, get_container('abcde')),
  487. },
  488. {
  489. 'type': 'container',
  490. 'service': 'db',
  491. 'action': 'create',
  492. 'id': 'ababa',
  493. 'attributes': {
  494. 'name': 'test_db_1',
  495. 'image': 'example/db',
  496. },
  497. 'time': dt_with_microseconds(1420092061, 4),
  498. 'container': Container(None, get_container('ababa')),
  499. },
  500. {
  501. 'type': 'container',
  502. 'service': 'db',
  503. 'action': 'destroy',
  504. 'id': 'eeeee',
  505. 'attributes': {
  506. 'name': 'test_db_1',
  507. 'image': 'example/db',
  508. },
  509. 'time': dt_with_microseconds(1420092061, 4),
  510. 'container': None,
  511. },
  512. ]
  513. def test_net_unset(self):
  514. project = Project.from_config(
  515. name='test',
  516. client=self.mock_client,
  517. config_data=build_config(
  518. version=V1,
  519. services=[
  520. {
  521. 'name': 'test',
  522. 'image': BUSYBOX_IMAGE_WITH_TAG,
  523. }
  524. ],
  525. networks=None,
  526. volumes=None,
  527. secrets=None,
  528. configs=None,
  529. ),
  530. )
  531. service = project.get_service('test')
  532. assert service.network_mode.id is None
  533. assert 'NetworkMode' not in service._get_container_host_config({})
  534. def test_use_net_from_container(self):
  535. container_id = 'aabbccddee'
  536. container_dict = dict(Name='aaa', Id=container_id)
  537. self.mock_client.inspect_container.return_value = container_dict
  538. project = Project.from_config(
  539. name='test',
  540. client=self.mock_client,
  541. config_data=build_config(
  542. services=[
  543. {
  544. 'name': 'test',
  545. 'image': BUSYBOX_IMAGE_WITH_TAG,
  546. 'network_mode': 'container:aaa'
  547. },
  548. ],
  549. networks=None,
  550. volumes=None,
  551. secrets=None,
  552. configs=None,
  553. ),
  554. )
  555. service = project.get_service('test')
  556. assert service.network_mode.mode == 'container:' + container_id
  557. def test_use_net_from_service(self):
  558. container_name = 'test_aaa_1'
  559. self.mock_client.containers.return_value = [
  560. {
  561. "Name": container_name,
  562. "Names": [container_name],
  563. "Id": container_name,
  564. "Image": BUSYBOX_IMAGE_WITH_TAG
  565. }
  566. ]
  567. project = Project.from_config(
  568. name='test',
  569. client=self.mock_client,
  570. config_data=build_config(
  571. services=[
  572. {
  573. 'name': 'aaa',
  574. 'image': BUSYBOX_IMAGE_WITH_TAG
  575. },
  576. {
  577. 'name': 'test',
  578. 'image': BUSYBOX_IMAGE_WITH_TAG,
  579. 'network_mode': 'service:aaa'
  580. },
  581. ],
  582. networks=None,
  583. volumes=None,
  584. secrets=None,
  585. configs=None,
  586. ),
  587. )
  588. service = project.get_service('test')
  589. assert service.network_mode.mode == 'container:' + container_name
  590. def test_uses_default_network_true(self):
  591. project = Project.from_config(
  592. name='test',
  593. client=self.mock_client,
  594. config_data=build_config(
  595. services=[
  596. {
  597. 'name': 'foo',
  598. 'image': BUSYBOX_IMAGE_WITH_TAG
  599. },
  600. ],
  601. networks=None,
  602. volumes=None,
  603. secrets=None,
  604. configs=None,
  605. ),
  606. )
  607. assert 'default' in project.networks.networks
  608. def test_uses_default_network_false(self):
  609. project = Project.from_config(
  610. name='test',
  611. client=self.mock_client,
  612. config_data=build_config(
  613. services=[
  614. {
  615. 'name': 'foo',
  616. 'image': BUSYBOX_IMAGE_WITH_TAG,
  617. 'networks': {'custom': None}
  618. },
  619. ],
  620. networks={'custom': {}},
  621. volumes=None,
  622. secrets=None,
  623. configs=None,
  624. ),
  625. )
  626. assert 'default' not in project.networks.networks
  627. def test_container_without_name(self):
  628. self.mock_client.containers.return_value = [
  629. {'Image': BUSYBOX_IMAGE_WITH_TAG, 'Id': '1', 'Name': '1'},
  630. {'Image': BUSYBOX_IMAGE_WITH_TAG, 'Id': '2', 'Name': None},
  631. {'Image': BUSYBOX_IMAGE_WITH_TAG, 'Id': '3'},
  632. ]
  633. self.mock_client.inspect_container.return_value = {
  634. 'Id': '1',
  635. 'Config': {
  636. 'Labels': {
  637. LABEL_SERVICE: 'web',
  638. },
  639. },
  640. }
  641. project = Project.from_config(
  642. name='test',
  643. client=self.mock_client,
  644. config_data=build_config(
  645. services=[{
  646. 'name': 'web',
  647. 'image': BUSYBOX_IMAGE_WITH_TAG,
  648. }],
  649. networks=None,
  650. volumes=None,
  651. secrets=None,
  652. configs=None,
  653. ),
  654. )
  655. assert [c.id for c in project.containers()] == ['1']
  656. def test_down_with_no_resources(self):
  657. project = Project.from_config(
  658. name='test',
  659. client=self.mock_client,
  660. config_data=build_config(
  661. services=[{
  662. 'name': 'web',
  663. 'image': BUSYBOX_IMAGE_WITH_TAG,
  664. }],
  665. networks={'default': {}},
  666. volumes={'data': {}},
  667. secrets=None,
  668. configs=None,
  669. ),
  670. )
  671. self.mock_client.remove_network.side_effect = NotFound(None, None, 'oops')
  672. self.mock_client.remove_volume.side_effect = NotFound(None, None, 'oops')
  673. project.down(ImageType.all, True)
  674. self.mock_client.remove_image.assert_called_once_with(BUSYBOX_IMAGE_WITH_TAG)
  675. def test_no_warning_on_stop(self):
  676. self.mock_client.info.return_value = {'Swarm': {'LocalNodeState': 'active'}}
  677. project = Project('composetest', [], self.mock_client)
  678. with mock.patch('compose.project.log') as fake_log:
  679. project.stop()
  680. assert fake_log.warn.call_count == 0
  681. def test_no_warning_in_normal_mode(self):
  682. self.mock_client.info.return_value = {'Swarm': {'LocalNodeState': 'inactive'}}
  683. project = Project('composetest', [], self.mock_client)
  684. with mock.patch('compose.project.log') as fake_log:
  685. project.up()
  686. assert fake_log.warn.call_count == 0
  687. def test_no_warning_with_no_swarm_info(self):
  688. self.mock_client.info.return_value = {}
  689. project = Project('composetest', [], self.mock_client)
  690. with mock.patch('compose.project.log') as fake_log:
  691. project.up()
  692. assert fake_log.warn.call_count == 0
  693. def test_no_such_service_unicode(self):
  694. assert NoSuchService('十六夜 咲夜'.encode()).msg == 'No such service: 十六夜 咲夜'
  695. assert NoSuchService('十六夜 咲夜').msg == 'No such service: 十六夜 咲夜'
  696. def test_project_platform_value(self):
  697. service_config = {
  698. 'name': 'web',
  699. 'image': BUSYBOX_IMAGE_WITH_TAG,
  700. }
  701. config_data = build_config(
  702. services=[service_config], networks={}, volumes={}, secrets=None, configs=None
  703. )
  704. project = Project.from_config(name='test', client=self.mock_client, config_data=config_data)
  705. assert project.get_service('web').platform is None
  706. project = Project.from_config(
  707. name='test', client=self.mock_client, config_data=config_data, default_platform='windows'
  708. )
  709. assert project.get_service('web').platform == 'windows'
  710. service_config['platform'] = 'linux/s390x'
  711. project = Project.from_config(name='test', client=self.mock_client, config_data=config_data)
  712. assert project.get_service('web').platform == 'linux/s390x'
  713. project = Project.from_config(
  714. name='test', client=self.mock_client, config_data=config_data, default_platform='windows'
  715. )
  716. assert project.get_service('web').platform == 'linux/s390x'
  717. def test_build_container_operation_with_timeout_func_does_not_mutate_options_with_timeout(self):
  718. config_data = build_config(
  719. services=[
  720. {'name': 'web', 'image': BUSYBOX_IMAGE_WITH_TAG},
  721. {'name': 'db', 'image': BUSYBOX_IMAGE_WITH_TAG, 'stop_grace_period': '1s'},
  722. ],
  723. networks={}, volumes={}, secrets=None, configs=None,
  724. )
  725. project = Project.from_config(name='test', client=self.mock_client, config_data=config_data)
  726. stop_op = project.build_container_operation_with_timeout_func('stop', options={})
  727. web_container = mock.create_autospec(Container, service='web')
  728. db_container = mock.create_autospec(Container, service='db')
  729. # `stop_grace_period` is not set to 'web' service,
  730. # then it is stopped with the default timeout.
  731. stop_op(web_container)
  732. web_container.stop.assert_called_once_with(timeout=DEFAULT_TIMEOUT)
  733. # `stop_grace_period` is set to 'db' service,
  734. # then it is stopped with the specified timeout and
  735. # the value is not overridden by the previous function call.
  736. stop_op(db_container)
  737. db_container.stop.assert_called_once_with(timeout=1)
  738. @mock.patch('compose.parallel.ParallelStreamWriter._write_noansi')
  739. def test_error_parallel_pull(self, mock_write):
  740. project = Project.from_config(
  741. name='test',
  742. client=self.mock_client,
  743. config_data=build_config(
  744. services=[{
  745. 'name': 'web',
  746. 'image': BUSYBOX_IMAGE_WITH_TAG,
  747. }],
  748. networks=None,
  749. volumes=None,
  750. secrets=None,
  751. configs=None,
  752. ),
  753. )
  754. self.mock_client.pull.side_effect = OperationFailedError('pull error')
  755. with pytest.raises(ProjectError):
  756. project.pull(parallel_pull=True)
  757. self.mock_client.pull.side_effect = OperationFailedError(b'pull error')
  758. with pytest.raises(ProjectError):
  759. project.pull(parallel_pull=True)
  760. def test_avoid_multiple_push(self):
  761. service_config_latest = {'image': 'busybox:latest', 'build': '.'}
  762. service_config_default = {'image': 'busybox', 'build': '.'}
  763. service_config_sha = {
  764. 'image': 'busybox@sha256:38a203e1986cf79639cfb9b2e1d6e773de84002feea2d4eb006b52004ee8502d',
  765. 'build': '.'
  766. }
  767. svc1 = Service('busy1', **service_config_latest)
  768. svc1_1 = Service('busy11', **service_config_latest)
  769. svc2 = Service('busy2', **service_config_default)
  770. svc2_1 = Service('busy21', **service_config_default)
  771. svc3 = Service('busy3', **service_config_sha)
  772. svc3_1 = Service('busy31', **service_config_sha)
  773. project = Project(
  774. 'composetest', [svc1, svc1_1, svc2, svc2_1, svc3, svc3_1], self.mock_client
  775. )
  776. with mock.patch('compose.service.Service.push') as fake_push:
  777. project.push()
  778. assert fake_push.call_count == 2
  779. def test_get_secrets_no_secret_def(self):
  780. service = 'foo'
  781. secret_source = 'bar'
  782. secret_defs = mock.Mock()
  783. secret_defs.get.return_value = None
  784. secret = mock.Mock(source=secret_source)
  785. with self.assertRaises(ConfigurationError):
  786. get_secrets(service, [secret], secret_defs)
  787. def test_get_secrets_external_warning(self):
  788. service = 'foo'
  789. secret_source = 'bar'
  790. secret_def = mock.Mock()
  791. secret_def.get.return_value = True
  792. secret_defs = mock.Mock()
  793. secret_defs.get.side_effect = secret_def
  794. secret = mock.Mock(source=secret_source)
  795. with mock.patch('compose.project.log') as mock_log:
  796. get_secrets(service, [secret], secret_defs)
  797. mock_log.warning.assert_called_with("Service \"{service}\" uses secret \"{secret}\" "
  798. "which is external. External secrets are not available"
  799. " to containers created by docker-compose."
  800. .format(service=service, secret=secret_source))
  801. def test_get_secrets_uid_gid_mode_warning(self):
  802. service = 'foo'
  803. secret_source = 'bar'
  804. fd, filename_path = tempfile.mkstemp()
  805. os.close(fd)
  806. self.addCleanup(os.remove, filename_path)
  807. def mock_get(key):
  808. return {'external': False, 'file': filename_path}[key]
  809. secret_def = mock.MagicMock()
  810. secret_def.get = mock.MagicMock(side_effect=mock_get)
  811. secret_defs = mock.Mock()
  812. secret_defs.get.return_value = secret_def
  813. secret = mock.Mock(uid=True, gid=True, mode=True, source=secret_source)
  814. with mock.patch('compose.project.log') as mock_log:
  815. get_secrets(service, [secret], secret_defs)
  816. mock_log.warning.assert_called_with("Service \"{service}\" uses secret \"{secret}\" with uid, "
  817. "gid, or mode. These fields are not supported by this "
  818. "implementation of the Compose file"
  819. .format(service=service, secret=secret_source))
  820. def test_get_secrets_secret_file_warning(self):
  821. service = 'foo'
  822. secret_source = 'bar'
  823. not_a_path = 'NOT_A_PATH'
  824. def mock_get(key):
  825. return {'external': False, 'file': not_a_path}[key]
  826. secret_def = mock.MagicMock()
  827. secret_def.get = mock.MagicMock(side_effect=mock_get)
  828. secret_defs = mock.Mock()
  829. secret_defs.get.return_value = secret_def
  830. secret = mock.Mock(uid=False, gid=False, mode=False, source=secret_source)
  831. with mock.patch('compose.project.log') as mock_log:
  832. get_secrets(service, [secret], secret_defs)
  833. mock_log.warning.assert_called_with("Service \"{service}\" uses an undefined secret file "
  834. "\"{secret_file}\", the following file should be created "
  835. "\"{secret_file}\""
  836. .format(service=service, secret_file=not_a_path))