Add test coverage for masoniteorm make and seed console commands

fastapi_startkit/src/fastapi_startkit/masoniteorm/schema/platforms/SQLitePlatform.py

@@ -418,9 +418,7 @@ def compile_table_exists(self, table, database=None, schema=None):
         return f"SELECT name FROM sqlite_master WHERE type='table' AND name='{table}'"
 
     def compile_column_exists(self, table, column):
-        return (
-            f"SELECT column_name FROM information_schema.columns WHERE table_name='{table}' and column_name='{column}'"
-        )
+        return f"SELECT name FROM pragma_table_info('{table}') WHERE name='{column}'"
 
     def compile_get_all_tables(self, database, schema=None):
         return "SELECT name FROM sqlite_master WHERE type='table'"

fastapi_startkit/tests/masoniteorm/commands/fixtures/databases/seeders/database_seeder.py

@@ -1,8 +1,8 @@
 from fastapi_startkit.masoniteorm.seeders import Seeder
 
-from .recorder import CALLS
+from ...models import SeededUser
 
 
 class DatabaseSeeder(Seeder):
     async def run(self):
-        CALLS.append(("database", self.connection))
+        await SeededUser.create({"name": "database-seeder"})

fastapi_startkit/tests/masoniteorm/commands/fixtures/databases/seeders/sample_seeder.py

@@ -1,8 +1,8 @@
 from fastapi_startkit.masoniteorm.seeders import Seeder
 
-from .recorder import CALLS
+from ...models import SeededUser
 
 
 class SampleSeeder(Seeder):
     async def run(self):
-        CALLS.append(("sample", self.connection))
+        await SeededUser.create({"name": "sample-seeder"})

fastapi_startkit/tests/masoniteorm/commands/fixtures/databases/seeders/special_seeder.py

@@ -1,8 +1,8 @@
 from fastapi_startkit.masoniteorm.seeders import Seeder
 
-from .recorder import CALLS
+from ...models import SeededUser
 
 
 class SpecialSeeder(Seeder):
     async def run(self):
-        CALLS.append(("special", self.connection))
+        await SeededUser.create({"name": "special-seeder"})

fastapi_startkit/tests/masoniteorm/commands/fixtures/databases/seeders/user_table_seeder.py

@@ -1,8 +1,8 @@
 from fastapi_startkit.masoniteorm.seeders import Seeder
 
-from .recorder import CALLS
+from ...models import SeededUser
 
 
 class UserTableSeeder(Seeder):
     async def run(self):
-        CALLS.append(("user_table", self.connection))
+        await SeededUser.create({"name": "user-table-seeder"})

fastapi_startkit/tests/masoniteorm/commands/fixtures/models.py

@@ -0,0 +1,10 @@
+from fastapi_startkit.masoniteorm.models.model import Model
+
+
+class SeededUser(Model):
+    """Real model backing the sqlite table the fixture seeders write into."""
+
+    __table__ = "seed_users"
+    __timestamps__ = False
+
+    name: str

fastapi_startkit/tests/masoniteorm/commands/test_db_seed_command.py

@@ -0,0 +1,109 @@
+import asyncio
+import unittest
+
+from cleo.testers.command_tester import CommandTester
+
+from fastapi_startkit.masoniteorm.commands.DBSeedCommand import DBSeedCommand
+
+from .fixtures.app import create_app, DB_PATH
+from .fixtures.models import SeededUser
+
+FIXTURE_SEED_PATH = "tests.masoniteorm.commands.fixtures.databases.seeders"
+
+
+class TestDBSeedCommand(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.app = create_app()
+
+    def setUp(self):
+        asyncio.run(self._reset_table())
+
+    def tearDown(self):
+        asyncio.run(self._drop_table())
+        if DB_PATH.exists():
+            DB_PATH.unlink()
+
+    async def _reset_table(self):
+        db = self.app.make("db")
+        await db.clear()
+        schema = db.get_schema_builder()
+        await schema.drop_table_if_exists("seed_users")
+        async with await schema.create("seed_users") as table:
+            table.id()
+            table.string("name")
+
+    async def _drop_table(self):
+        db = self.app.make("db")
+        schema = db.get_schema_builder()
+        await schema.drop_table_if_exists("seed_users")
+        await db.clear()
+
+    def _run(self, args=""):
+        tester = CommandTester(DBSeedCommand())
+        tester.execute(args)
+        return tester.io.fetch_output()
+
+    def _seeded_names(self):
+        return asyncio.run(self._fetch_names())
+
+    @staticmethod
+    async def _fetch_names():
+        rows = await SeededUser.all()
+        return sorted(row.name for row in rows)
+
+    # -- behavior is proven by the rows the real Seeder + real fixture seeder
+    # classes write into a real sqlite table, not by console output --
+
+    def test_runs_database_seeder_by_default(self):
+        self._run(f"--directory {FIXTURE_SEED_PATH} --connection sqlite")
+
+        self.assertEqual(self._seeded_names(), ["database-seeder"])
+
+    def test_seeds_specific_table_from_argument(self):
+        self._run(f"user --directory {FIXTURE_SEED_PATH} --connection sqlite")
+
+        self.assertEqual(self._seeded_names(), ["user-table-seeder"])
+
+    def test_class_option_resolves_plain_class_name(self):
+        self._run(f"--directory {FIXTURE_SEED_PATH} --class SampleSeeder --connection sqlite")
+
+        self.assertEqual(self._seeded_names(), ["sample-seeder"])
+
+    def test_class_option_resolves_table_seeder_suffix(self):
+        self._run(f"--directory {FIXTURE_SEED_PATH} --class UserTableSeeder --connection sqlite")
+
+        self.assertEqual(self._seeded_names(), ["user-table-seeder"])
+
+    def test_class_option_accepts_dotted_path(self):
+        self._run(f"--directory {FIXTURE_SEED_PATH} --class special_seeder.SpecialSeeder --connection sqlite")
+
+        self.assertEqual(self._seeded_names(), ["special-seeder"])
+
+    def test_class_option_takes_precedence_over_table_argument(self):
+        self._run(f"user --directory {FIXTURE_SEED_PATH} --class SampleSeeder --connection sqlite")
+
+        self.assertEqual(self._seeded_names(), ["sample-seeder"])
+
+    def test_uses_default_connection_when_not_specified(self):
+        self._run(f"--directory {FIXTURE_SEED_PATH}")
+
+        self.assertEqual(self._seeded_names(), ["database-seeder"])
+
+    def test_success_message_names_the_seeder(self):
+        # Minimal, secondary output check -- the user-facing message contract,
+        # not a substitute for the row assertions above.
+        output = self._run(f"--directory {FIXTURE_SEED_PATH} --connection sqlite")
+
+        self.assertIn("Database Seeder seeded!", output)
+
+    # -- error paths: driven directly through CommandTester, since the console
+    # application catches command exceptions instead of propagating them --
+
+    def test_raises_when_seeder_class_cannot_be_found(self):
+        with self.assertRaises(ValueError):
+            self._run(f"--directory {FIXTURE_SEED_PATH} --class does_not_exist.NopeSeeder --connection sqlite")
+
+    def test_raises_when_database_seeder_missing_from_directory(self):
+        with self.assertRaises(ValueError):
+            self._run("--directory tests.masoniteorm.commands.fixtures.databases.migrations --connection sqlite")