mirror of
https://github.com/docker/compose.git
synced 2025-04-08 17:05:13 +02:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
import mock
|
|
|
|
from compose import legacy
|
|
from compose.project import Project
|
|
from .testcases import DockerClientTestCase
|
|
|
|
|
|
class ProjectTest(DockerClientTestCase):
|
|
|
|
def setUp(self):
|
|
super(ProjectTest, self).setUp()
|
|
|
|
self.services = [
|
|
self.create_service('web'),
|
|
self.create_service('db'),
|
|
]
|
|
|
|
self.project = Project('composetest', self.services, self.client)
|
|
|
|
for service in self.services:
|
|
service.ensure_image_exists()
|
|
self.client.create_container(
|
|
name='{}_{}_1'.format(self.project.name, service.name),
|
|
**service.options
|
|
)
|
|
|
|
def get_names(self, **kwargs):
|
|
if 'stopped' not in kwargs:
|
|
kwargs['stopped'] = True
|
|
|
|
return list(legacy.get_legacy_container_names(
|
|
self.client,
|
|
self.project.name,
|
|
[s.name for s in self.services],
|
|
**kwargs
|
|
))
|
|
|
|
def test_get_legacy_container_names(self):
|
|
self.assertEqual(len(self.get_names()), len(self.services))
|
|
|
|
def test_migration_to_labels(self):
|
|
with mock.patch.object(legacy, 'log', autospec=True) as mock_log:
|
|
self.assertEqual(self.project.containers(stopped=True), [])
|
|
self.assertEqual(mock_log.warn.call_count, len(self.services))
|
|
|
|
legacy.migrate_project_to_labels(self.project)
|
|
self.assertEqual(len(self.project.containers(stopped=True)), len(self.services))
|