migration_test.py 838 B

1234567891011121314151617181920212223
  1. import mock
  2. from compose import service, migration
  3. from compose.project import Project
  4. from .testcases import DockerClientTestCase
  5. class ProjectTest(DockerClientTestCase):
  6. def test_migration_to_labels(self):
  7. web = self.create_service('web')
  8. db = self.create_service('db')
  9. project = Project('composetest', [web, db], self.client)
  10. self.client.create_container(name='composetest_web_1', **web.options)
  11. self.client.create_container(name='composetest_db_1', **db.options)
  12. with mock.patch.object(service, 'log', autospec=True) as mock_log:
  13. self.assertEqual(project.containers(stopped=True), [])
  14. self.assertEqual(mock_log.warn.call_count, 2)
  15. migration.migrate_project_to_labels(project)
  16. self.assertEqual(len(project.containers(stopped=True)), 2)