Added migration support

src/django_deploy/tests/test_program.py

@@ -1,4 +1,6 @@
 import os
+import django
+import pytest
 
 from ..config import DJANGO_SETTINGS_DIR
 from ..program import Program
@@ -6,11 +8,17 @@ from ..program import Program
 from .generic import DjangoDeployTestCase
 
 
-class ProgramTestCase(DjangoDeployTestCase):
+class AbstractProgramTestCase(DjangoDeployTestCase):
+    @pytest.fixture(autouse=True)
+    def capsys(self, capsys):  # pylint: disable=method-hidden
+        self.capsys = capsys
+
     def setUp(self):
         self._program = Program()
         self._tmp_dir = str(self.tmpdir)
 
+
+class ProgramCreateTestCase(AbstractProgramTestCase):
     def test_create(self):
         project_dir = os.path.join(self._tmp_dir, 'new')
         exitval = self._program(argv=['--create', project_dir])
@@ -66,10 +74,11 @@ class ProgramTestCase(DjangoDeployTestCase):
         exitval = self._program(argv=['--install-hooks', project_dir])
         self.assertEqual(os.EX_SOFTWARE, exitval)
 
+
+class ProgramInstallTestCase(AbstractProgramTestCase):
     def test_install_apps(self):
         project_dir = self._tmp_dir
         argv = [
-            '-c',
             '--install-app', 'django_deploy.tests.fake_app1',
             '--install-app', 'django_deploy.tests.fake_app1',
             '-a', 'django_deploy.tests.fake_app2',
@@ -87,10 +96,19 @@ class ProgramTestCase(DjangoDeployTestCase):
         self.assert_installed_apps(project_dir, installed_apps)
         self.assert_urlpatterns(project_dir, [])
 
-    def test_install_apps_with_error(self):
+    def test_install_with_explicit_create(self):
         project_dir = self._tmp_dir
         argv = [
             '-c',
+            '--install-app', 'django_deploy.tests.fake_app1',
+            project_dir
+        ]
+        exitval = self._program(argv=argv)
+        self.assertEqual(os.EX_OK, exitval)
+
+    def test_install_apps_with_error(self):
+        project_dir = self._tmp_dir
+        argv = [
             '-a', 'django_deploy.tests.fake_app1',
             '-a', 'django_deploy.tests.fake_app0',
             '-a', 'django_deploy.tests.fake_app2',
@@ -106,10 +124,11 @@ class ProgramTestCase(DjangoDeployTestCase):
 
         self.assert_installed_apps(project_dir, installed_apps)
 
+
+class ProgramMountTestCase(AbstractProgramTestCase):
     def test_mount_apps(self):
         project_dir = self._tmp_dir
         argv = [
-            '-c',
             '--mount-app', 'django_deploy.tests.fake_app1', 'app1',
             '--mount-app', 'django_deploy.tests.fake_app1', 'app1',
             '-m', 'django_deploy.tests.fake_app2', 'app2/',
@@ -136,7 +155,6 @@ class ProgramTestCase(DjangoDeployTestCase):
 
     def test_mount_apps_with_errors(self):
         project_dir = self._tmp_dir
-        self._program(argv=['-c', project_dir])
         argv = [
             '-m', 'django_deploy.tests.fake_app1', '/',
             '-m', 'django_deploy.tests.fake_app0', 'app0',
@@ -153,3 +171,102 @@ class ProgramTestCase(DjangoDeployTestCase):
         ]
 
         self.assert_urlpatterns(project_dir, expected_urlpatterns)
+
+
+class ProgramMigrateTestCase(AbstractProgramTestCase):
+    def test_migrate_all(self):
+        project_dir = self._tmp_dir
+        argv = [
+            '--install-app', 'django_deploy.tests.fake_app1',
+            '--install-app', 'django_deploy.tests.fake_app2',
+            '--migrate',
+            project_dir
+        ]
+        exitval = self._program(argv=argv)
+        self.assertEqual(os.EX_OK, exitval)
+        self.assert_django_database_migration(project_dir)
+
+    def test_migrate_app_by_app_name(self):
+        project_dir = self._tmp_dir
+        argv = [
+            '--install-app', 'django_deploy.tests.fake_app1',
+            '--migrate-app', 'django_deploy.tests.fake_app1',
+            project_dir
+        ]
+        exitval = self._program(argv=argv)
+        self.assertEqual(os.EX_OK, exitval)
+        self.assert_django_database_migration(project_dir, 'fake_app1')
+
+    def test_migrate_app_by_app_label(self):
+        project_dir = self._tmp_dir
+        argv = [
+            '--install-app', 'django_deploy.tests.fake_app1',
+            '--migrate-app', 'fake_app1',
+            project_dir
+        ]
+        exitval = self._program(argv=argv)
+        self.assertEqual(os.EX_OK, exitval)
+        self.assert_django_database_migration(project_dir, 'fake_app1')
+
+    def test_migrate_app_not_installed(self):
+        project_dir = self._tmp_dir
+        argv = [
+            '-c',
+            '--migrate-app', 'django_deploy.tests.fake_app1',
+            project_dir
+        ]
+        exitval = self._program(argv=argv)
+        self.assertNotEqual(os.EX_OK, exitval)
+
+        captured = self.capsys.readouterr()
+        if django.VERSION[0] < 2:  # pragma: no cover
+            expected_msg = "CommandError: App 'fake_app1' does not have migrations.\n"
+        else:  # pragma: no cover
+            expected_msg = "CommandError: No installed app with label 'fake_app1'.\n"
+        self.assertEqual(expected_msg, captured.err)
+
+    def test_migrate_app_without_create(self):
+        project_dir = self._tmp_dir
+        argv = [
+            '--migrate-app', 'django_deploy.tests.fake_app1',
+            project_dir
+        ]
+        exitval = self._program(argv=argv)
+        self.assertNotEqual(os.EX_OK, exitval)
+        captured = self.capsys.readouterr()
+        self.assertTrue(captured.err.endswith('Have you created the django project?\n'))
+
+    def test_migrate_app_without_migrations(self):
+        project_dir = self._tmp_dir
+        argv = [
+            '--install-app', 'django_deploy.tests.fake_app1',
+            '--install-app', 'django_deploy.tests.fake_app2',
+            '--migrate-app', 'fake_app2',
+            '--migrate-app', 'fake_app1',
+            project_dir
+        ]
+        exitval = self._program(argv=argv)
+        self.assertNotEqual(os.EX_OK, exitval)
+        captured = self.capsys.readouterr()
+        self.assertEqual("CommandError: App 'fake_app2' does not have migrations.\n", captured.err)
+
+    def test_migrate_app_stop_on_first_error(self):
+        project_dir = self._tmp_dir
+        argv = [
+            '--install-app', 'django_deploy.tests.fake_app1',
+            '--install-app', 'django_deploy.tests.fake_app2',
+            '--migrate-app', 'fake_app2',
+            '--migrate-app', 'fake_app1',
+            '--migrate-app', 'django_deploy',
+            project_dir
+        ]
+        exitval = self._program(argv=argv)
+        self.assertNotEqual(os.EX_OK, exitval)
+        captured = self.capsys.readouterr()
+        self.assertEqual("CommandError: App 'fake_app2' does not have migrations.\n", captured.err)
+        try:
+            self.assert_django_database_migration(project_dir, 'fake_app1')
+        except AssertionError:
+            pass
+        else:  # pragma: no cover
+            self.fail('Unexcepted migrations done')