added an extra test for the new migration command

Author: dantownsend

tests/apps/migrations/commands/test_forwards_backwards.py

@@ -6,6 +6,7 @@
 from piccolo.apps.migrations.tables import Migration
 from piccolo.apps.migrations.commands.backwards import backwards
 from piccolo.apps.migrations.commands.forwards import forwards
+from tests.base import postgres_only
 from tests.example_app.tables import (
     Band,
     Concert,
@@ -29,6 +30,7 @@
 ]
 
 
+@postgres_only
 class TestForwardsBackwards(TestCase):
     """
     How to test this? Add migrations for the example app, and just run them

tests/apps/migrations/commands/test_new.py

@@ -6,13 +6,18 @@
 from piccolo.apps.migrations.commands.new import (
     _create_new_migration,
     BaseMigrationManager,
+    new,
 )
 
+from tests.base import postgres_only
 from tests.example_app.tables import Manager
 
 
 class TestNewMigrationCommand(TestCase):
     def test_create_new_migration(self):
+        """
+        Create a manual migration (i.e. non-auto).
+        """
         migration_folder = "/tmp/piccolo_migrations/"
 
         if os.path.exists(migration_folder):
@@ -33,3 +38,15 @@ def test_create_new_migration(self):
         )
 
         self.assertTrue(len(migration_modules.keys()) == 1)
+
+    @postgres_only
+    def test_new_command(self):
+        """
+        Call the command, when no migration changes are needed.
+        """
+        with self.assertRaises(SystemExit) as manager:
+            new(app_name="example_app", auto=True)
+
+        self.assertEqual(
+            manager.exception.__str__(), "No changes detected - exiting."
+        )