added tests for 'migrations new' command

Author: dantownsend

tests/apps/migrations/commands/test_base.py

@@ -18,11 +18,6 @@ def tearDown(self):
 
 
 class TestGetMigrationModules(TestCase):
-    """
-    This is tricky, because we need to have some example migrations on disk.
-    Just create them.
-    """
-
     @patch.object(BaseMigrationManager, "get_app_config")
     def test_get_migration_modules(self, get_app_config):
         get_app_config.return_value = AppConfig(

tests/apps/migrations/commands/test_new.py

@@ -0,0 +1,35 @@
+import os
+import shutil
+from unittest import TestCase
+
+from piccolo.apps.migrations.commands.base import AppConfig
+from piccolo.apps.migrations.commands.new import (
+    _create_new_migration,
+    BaseMigrationManager,
+)
+
+from tests.example_project.tables import Manager
+
+
+class TestNewMigrationCommand(TestCase):
+    def test_create_new_migration(self):
+        migration_folder = "/tmp/piccolo_migrations/"
+
+        if os.path.exists(migration_folder):
+            shutil.rmtree(migration_folder)
+
+        os.mkdir(migration_folder)
+
+        app_config = AppConfig(
+            app_name="music",
+            migrations_folder_path=migration_folder,
+            table_classes=[Manager],
+        )
+
+        _create_new_migration(app_config, auto=False)
+
+        migration_modules = BaseMigrationManager().get_migration_modules(
+            migration_folder
+        )
+
+        self.assertTrue(len(migration_modules.keys()) == 1)

tests/example_project/tables.py

@@ -1,31 +1,31 @@
-from piccolo import table
-from piccolo import columns
+from piccolo.table import Table
+from piccolo.columns import Varchar, ForeignKey, Integer
 
 
 ###############################################################################
 # Simple example
 
 
-class Manager(table.Table):
-    name = columns.Varchar(length=50)
+class Manager(Table):
+    name = Varchar(length=50)
 
 
-class Band(table.Table):
-    name = columns.Varchar(length=50)
-    manager = columns.ForeignKey(Manager, null=True)
-    popularity = columns.Integer(default=0)
+class Band(Table):
+    name = Varchar(length=50)
+    manager = ForeignKey(Manager, null=True)
+    popularity = Integer(default=0)
 
 
 ###############################################################################
 # More complex
 
 
-class Venue(table.Table):
-    name = columns.Varchar(length=100)
-    capacity = columns.Integer(default=0)
+class Venue(Table):
+    name = Varchar(length=100)
+    capacity = Integer(default=0)
 
 
-class Concert(table.Table):
-    band_1 = columns.ForeignKey(Band)
-    band_2 = columns.ForeignKey(Band)
-    venue = columns.ForeignKey(Venue)
+class Concert(Table):
+    band_1 = ForeignKey(Band)
+    band_2 = ForeignKey(Band)
+    venue = ForeignKey(Venue)