legacy_test.py 930 B

123456789101112131415161718192021222324252627282930
  1. import mock
  2. from compose import legacy
  3. from compose.project import Project
  4. from .testcases import DockerClientTestCase
  5. class ProjectTest(DockerClientTestCase):
  6. def test_migration_to_labels(self):
  7. services = [
  8. self.create_service('web'),
  9. self.create_service('db'),
  10. ]
  11. project = Project('composetest', services, self.client)
  12. for service in services:
  13. service.ensure_image_exists()
  14. self.client.create_container(
  15. name='{}_{}_1'.format(project.name, service.name),
  16. **service.options
  17. )
  18. with mock.patch.object(legacy, 'log', autospec=True) as mock_log:
  19. self.assertEqual(project.containers(stopped=True), [])
  20. self.assertEqual(mock_log.warn.call_count, 2)
  21. legacy.migrate_project_to_labels(project)
  22. self.assertEqual(len(project.containers(stopped=True)), 2)