|  | @@ -1,218 +0,0 @@
 | 
	
		
			
				|  |  | -import unittest
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -from docker.errors import APIError
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -from .. import mock
 | 
	
		
			
				|  |  | -from .testcases import DockerClientTestCase
 | 
	
		
			
				|  |  | -from compose import legacy
 | 
	
		
			
				|  |  | -from compose.project import Project
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -class UtilitiesTestCase(unittest.TestCase):
 | 
	
		
			
				|  |  | -    def test_has_container(self):
 | 
	
		
			
				|  |  | -        self.assertTrue(
 | 
	
		
			
				|  |  | -            legacy.has_container("composetest", "web", "composetest_web_1", one_off=False),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.has_container("composetest", "web", "composetest_web_run_1", one_off=False),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_has_container_one_off(self):
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.has_container("composetest", "web", "composetest_web_1", one_off=True),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -        self.assertTrue(
 | 
	
		
			
				|  |  | -            legacy.has_container("composetest", "web", "composetest_web_run_1", one_off=True),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_has_container_different_project(self):
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.has_container("composetest", "web", "otherapp_web_1", one_off=False),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.has_container("composetest", "web", "otherapp_web_run_1", one_off=True),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_has_container_different_service(self):
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.has_container("composetest", "web", "composetest_db_1", one_off=False),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.has_container("composetest", "web", "composetest_db_run_1", one_off=True),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_is_valid_name(self):
 | 
	
		
			
				|  |  | -        self.assertTrue(
 | 
	
		
			
				|  |  | -            legacy.is_valid_name("composetest_web_1", one_off=False),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.is_valid_name("composetest_web_run_1", one_off=False),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_is_valid_name_one_off(self):
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.is_valid_name("composetest_web_1", one_off=True),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -        self.assertTrue(
 | 
	
		
			
				|  |  | -            legacy.is_valid_name("composetest_web_run_1", one_off=True),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_is_valid_name_invalid(self):
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.is_valid_name("foo"),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -        self.assertFalse(
 | 
	
		
			
				|  |  | -            legacy.is_valid_name("composetest_web_lol_1", one_off=True),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_get_legacy_containers(self):
 | 
	
		
			
				|  |  | -        client = mock.Mock()
 | 
	
		
			
				|  |  | -        client.containers.return_value = [
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                "Id": "abc123",
 | 
	
		
			
				|  |  | -                "Image": "def456",
 | 
	
		
			
				|  |  | -                "Name": "composetest_web_1",
 | 
	
		
			
				|  |  | -                "Labels": None,
 | 
	
		
			
				|  |  | -            },
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                "Id": "ghi789",
 | 
	
		
			
				|  |  | -                "Image": "def456",
 | 
	
		
			
				|  |  | -                "Name": None,
 | 
	
		
			
				|  |  | -                "Labels": None,
 | 
	
		
			
				|  |  | -            },
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                "Id": "jkl012",
 | 
	
		
			
				|  |  | -                "Image": "def456",
 | 
	
		
			
				|  |  | -                "Labels": None,
 | 
	
		
			
				|  |  | -            },
 | 
	
		
			
				|  |  | -        ]
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        containers = legacy.get_legacy_containers(client, "composetest", ["web"])
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.assertEqual(len(containers), 1)
 | 
	
		
			
				|  |  | -        self.assertEqual(containers[0].id, 'abc123')
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -class LegacyTestCase(DockerClientTestCase):
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def setUp(self):
 | 
	
		
			
				|  |  | -        super(LegacyTestCase, self).setUp()
 | 
	
		
			
				|  |  | -        self.containers = []
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        db = self.create_service('db')
 | 
	
		
			
				|  |  | -        web = self.create_service('web', links=[(db, 'db')])
 | 
	
		
			
				|  |  | -        nginx = self.create_service('nginx', links=[(web, 'web')])
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.services = [db, web, nginx]
 | 
	
		
			
				|  |  | -        self.project = Project('composetest', self.services, self.client)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Create a legacy container for each service
 | 
	
		
			
				|  |  | -        for service in self.services:
 | 
	
		
			
				|  |  | -            service.ensure_image_exists()
 | 
	
		
			
				|  |  | -            container = self.client.create_container(
 | 
	
		
			
				|  |  | -                name='{}_{}_1'.format(self.project.name, service.name),
 | 
	
		
			
				|  |  | -                **service.options
 | 
	
		
			
				|  |  | -            )
 | 
	
		
			
				|  |  | -            self.client.start(container)
 | 
	
		
			
				|  |  | -            self.containers.append(container)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Create a single one-off legacy container
 | 
	
		
			
				|  |  | -        self.containers.append(self.client.create_container(
 | 
	
		
			
				|  |  | -            name='{}_{}_run_1'.format(self.project.name, db.name),
 | 
	
		
			
				|  |  | -            **self.services[0].options
 | 
	
		
			
				|  |  | -        ))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def tearDown(self):
 | 
	
		
			
				|  |  | -        super(LegacyTestCase, self).tearDown()
 | 
	
		
			
				|  |  | -        for container in self.containers:
 | 
	
		
			
				|  |  | -            try:
 | 
	
		
			
				|  |  | -                self.client.kill(container)
 | 
	
		
			
				|  |  | -            except APIError:
 | 
	
		
			
				|  |  | -                pass
 | 
	
		
			
				|  |  | -            try:
 | 
	
		
			
				|  |  | -                self.client.remove_container(container)
 | 
	
		
			
				|  |  | -            except APIError:
 | 
	
		
			
				|  |  | -                pass
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def get_legacy_containers(self, **kwargs):
 | 
	
		
			
				|  |  | -        return legacy.get_legacy_containers(
 | 
	
		
			
				|  |  | -            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_legacy_containers()), len(self.services))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_get_legacy_container_names_one_off(self):
 | 
	
		
			
				|  |  | -        self.assertEqual(len(self.get_legacy_containers(one_off=True)), 1)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_migration_to_labels(self):
 | 
	
		
			
				|  |  | -        # Trying to get the container list raises an exception
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        with self.assertRaises(legacy.LegacyContainersError) as cm:
 | 
	
		
			
				|  |  | -            self.project.containers(stopped=True)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.assertEqual(
 | 
	
		
			
				|  |  | -            set(cm.exception.names),
 | 
	
		
			
				|  |  | -            set(['composetest_db_1', 'composetest_web_1', 'composetest_nginx_1']),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.assertEqual(
 | 
	
		
			
				|  |  | -            set(cm.exception.one_off_names),
 | 
	
		
			
				|  |  | -            set(['composetest_db_run_1']),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Migrate the containers
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        legacy.migrate_project_to_labels(self.project)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Getting the list no longer raises an exception
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        containers = self.project.containers(stopped=True)
 | 
	
		
			
				|  |  | -        self.assertEqual(len(containers), len(self.services))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    def test_migration_one_off(self):
 | 
	
		
			
				|  |  | -        # We've already migrated
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        legacy.migrate_project_to_labels(self.project)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Trying to create a one-off container results in a Docker API error
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        with self.assertRaises(APIError) as cm:
 | 
	
		
			
				|  |  | -            self.project.get_service('db').create_container(one_off=True)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Checking for legacy one-off containers raises an exception
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        with self.assertRaises(legacy.LegacyOneOffContainersError) as cm:
 | 
	
		
			
				|  |  | -            legacy.check_for_legacy_containers(
 | 
	
		
			
				|  |  | -                self.client,
 | 
	
		
			
				|  |  | -                self.project.name,
 | 
	
		
			
				|  |  | -                ['db'],
 | 
	
		
			
				|  |  | -                allow_one_off=False,
 | 
	
		
			
				|  |  | -            )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.assertEqual(
 | 
	
		
			
				|  |  | -            set(cm.exception.one_off_names),
 | 
	
		
			
				|  |  | -            set(['composetest_db_run_1']),
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Remove the old one-off container
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        c = self.client.inspect_container('composetest_db_run_1')
 | 
	
		
			
				|  |  | -        self.client.remove_container(c)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Checking no longer raises an exception
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        legacy.check_for_legacy_containers(
 | 
	
		
			
				|  |  | -            self.client,
 | 
	
		
			
				|  |  | -            self.project.name,
 | 
	
		
			
				|  |  | -            ['db'],
 | 
	
		
			
				|  |  | -            allow_one_off=False,
 | 
	
		
			
				|  |  | -        )
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        # Creating a one-off container no longer results in an API error
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        self.project.get_service('db').create_container(one_off=True)
 | 
	
		
			
				|  |  | -        self.assertIsInstance(self.client.inspect_container('composetest_db_run_1'), dict)
 |