1
0

project_test.py 32 KB

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