velocity-python 0.1.11__tar.gz → 0.1.12__tar.gz

{velocity_python-0.1.11/src/velocity_python.egg-info → velocity_python-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: velocity-python
-Version: 0.1.11
+Version: 0.1.12
 Summary: A rapid application development library for interfacing with data storage
 Author-email: Velocity Team <info@codeclubs.org>
 License-Expression: MIT

{velocity_python-0.1.11 → velocity_python-0.1.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "velocity-python"
-version = "0.1.11"
+version = "0.1.12"
 authors = [
   { name="Velocity Team", email="info@codeclubs.org" },
 ]

{velocity_python-0.1.11 → velocity_python-0.1.12}/src/velocity/__init__.py

@@ -1,4 +1,4 @@
-__version__ = version = "0.1.11"
+__version__ = version = "0.1.12"
 
 import importlib as _importlib
 

{velocity_python-0.1.11 → velocity_python-0.1.12}/src/velocity/db/migrations.py

@@ -100,14 +100,17 @@ class MigrationRunner:
     """
     Discovers, applies, and rolls back schema migrations.
 
-    Migrations are tracked in a ``velocity_migrations`` table with columns:
+    Migrations are tracked in a ``velocity_migrations`` table managed by
+    velocity (with standard system columns like ``sys_id``, ``sys_created``,
+    etc.).  User columns:
 
-    - ``version`` (INT, PRIMARY KEY) — migration version number
+    - ``version`` (INT) — migration version number
     - ``description`` (TEXT) — human-readable description
-    - ``applied_at`` (TIMESTAMPTZ) — when the migration was applied
     - ``checksum`` (TEXT) — SHA-256 of the migration function source
     - ``execution_ms`` (INT) — how long the migration took in milliseconds
 
+    The ``sys_created`` system column serves as the applied-at timestamp.
+
     Args:
         engine: A velocity Engine instance.
         migrations_dir: Optional path to a directory containing migration .py files.
@@ -145,43 +148,23 @@ class MigrationRunner:
         """Create the velocity_migrations tracking table if it doesn't exist."""
         table = tx.table(TRACKING_TABLE)
         if not table.exists():
-            # Use raw SQL to avoid velocity system columns — we manage our own schema
-            server = getattr(getattr(self.engine, 'sql', None), 'server', 'PostgreSQL')
-            if server == 'MySQL':
-                ts_col = "applied_at  TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP"
-            elif server == 'SQLite':
-                ts_col = "applied_at  TEXT NOT NULL DEFAULT (datetime('now'))"
-            elif server == 'SQL Server':
-                ts_col = "applied_at  DATETIMEOFFSET NOT NULL DEFAULT SYSDATETIMEOFFSET()"
-            else:
-                ts_col = "applied_at  TIMESTAMPTZ NOT NULL DEFAULT NOW()"
-            tx.execute(
-                f"""
-                CREATE TABLE IF NOT EXISTS {TRACKING_TABLE} (
-                    version     INTEGER PRIMARY KEY,
-                    description TEXT NOT NULL DEFAULT '',
-                    {ts_col},
-                    checksum    TEXT NOT NULL DEFAULT '',
-                    execution_ms INTEGER NOT NULL DEFAULT 0
-                )
-                """,
-                cursor=tx.cursor(),
-            )
+            table.create(columns={
+                "version": int,
+                "description": str,
+                "checksum": str,
+                "execution_ms": int,
+            })
             logger.info("Created tracking table: %s", TRACKING_TABLE)
 
     def _get_applied_versions(self, tx) -> Dict[int, Dict[str, Any]]:
         """Return dict of {version: {description, applied_at, checksum, execution_ms}}."""
         self._ensure_tracking_table(tx)
-        result = tx.execute(
-            f"SELECT version, description, applied_at, checksum, execution_ms "
-            f"FROM {TRACKING_TABLE} ORDER BY version",
-            cursor=tx.cursor(),
-        )
+        table = tx.table(TRACKING_TABLE)
         applied = {}
-        for row in result.all():
+        for row in table.select(orderby="version").all():
             applied[row["version"]] = {
                 "description": row["description"],
-                "applied_at": row["applied_at"],
+                "applied_at": row["sys_created"],
                 "checksum": row["checksum"],
                 "execution_ms": row["execution_ms"],
             }
@@ -189,20 +172,18 @@ class MigrationRunner:
 
     def _record_migration(self, tx, version: int, description: str, checksum: str, execution_ms: int) -> None:
         """Insert a record into the tracking table."""
-        tx.execute(
-            f"INSERT INTO {TRACKING_TABLE} (version, description, checksum, execution_ms) "
-            f"VALUES (%s, %s, %s, %s)",
-            (version, description, checksum, execution_ms),
-            cursor=tx.cursor(),
-        )
+        table = tx.table(TRACKING_TABLE)
+        table.insert({
+            "version": version,
+            "description": description,
+            "checksum": checksum,
+            "execution_ms": execution_ms,
+        })
 
     def _remove_migration_record(self, tx, version: int) -> None:
         """Delete a record from the tracking table."""
-        tx.execute(
-            f"DELETE FROM {TRACKING_TABLE} WHERE version = %s",
-            (version,),
-            cursor=tx.cursor(),
-        )
+        table = tx.table(TRACKING_TABLE)
+        table.delete(where={"version": version})
 
     # ── Checksum ────────────────────────────────────────────────
 

{velocity_python-0.1.11 → velocity_python-0.1.12/src/velocity_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: velocity-python
-Version: 0.1.11
+Version: 0.1.12
 Summary: A rapid application development library for interfacing with data storage
 Author-email: Velocity Team <info@codeclubs.org>
 License-Expression: MIT

{velocity_python-0.1.11 → velocity_python-0.1.12}/tests/test_schema_migrations.py

@@ -57,10 +57,14 @@ def _mock_result(rows, columns=None):
     return result
 
 
-def _make_mock_table(exists=False):
-    """Return a mock Table that has exists(), create(), drop() methods."""
+def _make_mock_table(exists=False, select_rows=None):
+    """Return a mock Table with exists(), create(), select(), insert(), delete()."""
     table = MagicMock()
     table.exists.return_value = exists
+    if select_rows is not None:
+        table.select.return_value = _mock_result(select_rows)
+    else:
+        table.select.return_value = _mock_result([])
     return table
 
 
@@ -90,23 +94,18 @@ def _make_mock_engine(schema_locked=False):
 def _make_mock_tx(applied_rows=None, table_exists=True):
     """
     Build a mock Transaction that supports:
-    - tx.table(name) -> mock table
-    - tx.execute(sql, ...) -> mock result (context-dependent)
+    - tx.table(name) -> mock table with select/insert/delete
     - tx.cursor() -> mock cursor
     - tx.commit(), tx.rollback(), tx.close()
     """
     tx = MagicMock()
     tx.cursor.return_value = MagicMock()
 
-    mock_table = _make_mock_table(exists=table_exists)
-    tx.table.return_value = mock_table
-
-    # Default: return empty result for SELECT queries, no-op for others
     if applied_rows is None:
         applied_rows = []
 
-    select_result = _mock_result(applied_rows)
-    tx.execute.return_value = select_result
+    mock_table = _make_mock_table(exists=table_exists, select_rows=applied_rows)
+    tx.table.return_value = mock_table
 
     return tx
 
@@ -223,7 +222,7 @@ class TestMigrationRunnerPending:
         tx = _make_mock_tx(
             applied_rows=[{
                 "version": 1, "description": "v1",
-                "applied_at": "2024-01-01", "checksum": "abc", "execution_ms": 10,
+                "sys_created": "2024-01-01", "checksum": "abc", "execution_ms": 10,
             }],
             table_exists=True,
         )
@@ -249,7 +248,7 @@ class TestMigrationRunnerPending:
         tx = _make_mock_tx(
             applied_rows=[{
                 "version": 1, "description": "v1",
-                "applied_at": "2024-01-01", "checksum": "abc", "execution_ms": 10,
+                "sys_created": "2024-01-01", "checksum": "abc", "execution_ms": 10,
             }],
             table_exists=True,
         )
@@ -281,7 +280,7 @@ class TestMigrationRunnerStatus:
         tx = _make_mock_tx(
             applied_rows=[{
                 "version": 1, "description": "v1",
-                "applied_at": "2024-01-01", "checksum": MigrationRunner._checksum(up1),
+                "sys_created": "2024-01-01", "checksum": MigrationRunner._checksum(up1),
                 "execution_ms": 15,
             }],
             table_exists=True,
@@ -370,7 +369,7 @@ class TestMigrationRunnerMigrate:
         tx = _make_mock_tx(
             applied_rows=[{
                 "version": 1, "description": "v1",
-                "applied_at": "2024-01-01", "checksum": "abc", "execution_ms": 10,
+                "sys_created": "2024-01-01", "checksum": "abc", "execution_ms": 10,
             }],
             table_exists=True,
         )
@@ -476,8 +475,8 @@ class TestMigrationRunnerRollback:
         engine = _make_mock_engine()
         tx_check = _make_mock_tx(
             applied_rows=[
-                {"version": 1, "description": "v1", "applied_at": "2024-01-01", "checksum": "a", "execution_ms": 10},
-                {"version": 2, "description": "v2", "applied_at": "2024-01-02", "checksum": "b", "execution_ms": 20},
+                {"version": 1, "description": "v1", "sys_created": "2024-01-01", "checksum": "a", "execution_ms": 10},
+                {"version": 2, "description": "v2", "sys_created": "2024-01-02", "checksum": "b", "execution_ms": 20},
             ],
             table_exists=True,
         )
@@ -521,9 +520,9 @@ class TestMigrationRunnerRollback:
         engine = _make_mock_engine()
         tx_check = _make_mock_tx(
             applied_rows=[
-                {"version": 1, "description": "v1", "applied_at": "2024-01-01", "checksum": "a", "execution_ms": 10},
-                {"version": 2, "description": "v2", "applied_at": "2024-01-02", "checksum": "b", "execution_ms": 20},
-                {"version": 3, "description": "v3", "applied_at": "2024-01-03", "checksum": "c", "execution_ms": 30},
+                {"version": 1, "description": "v1", "sys_created": "2024-01-01", "checksum": "a", "execution_ms": 10},
+                {"version": 2, "description": "v2", "sys_created": "2024-01-02", "checksum": "b", "execution_ms": 20},
+                {"version": 3, "description": "v3", "sys_created": "2024-01-03", "checksum": "c", "execution_ms": 30},
             ],
             table_exists=True,
         )
@@ -546,7 +545,7 @@ class TestMigrationRunnerRollback:
         engine = _make_mock_engine()
         tx_check = _make_mock_tx(
             applied_rows=[
-                {"version": 1, "description": "v1", "applied_at": "2024-01-01", "checksum": "a", "execution_ms": 10},
+                {"version": 1, "description": "v1", "sys_created": "2024-01-01", "checksum": "a", "execution_ms": 10},
             ],
             table_exists=True,
         )
@@ -568,7 +567,7 @@ class TestMigrationRunnerRollback:
         engine = _make_mock_engine()
         tx_check = _make_mock_tx(
             applied_rows=[
-                {"version": 1, "description": "v1", "applied_at": "2024-01-01", "checksum": "a", "execution_ms": 10},
+                {"version": 1, "description": "v1", "sys_created": "2024-01-01", "checksum": "a", "execution_ms": 10},
             ],
             table_exists=True,
         )
@@ -593,7 +592,7 @@ class TestMigrationRunnerRollback:
         engine = _make_mock_engine(schema_locked=True)
         tx_check = _make_mock_tx(
             applied_rows=[
-                {"version": 1, "description": "v1", "applied_at": "2024-01-01", "checksum": "a", "execution_ms": 10},
+                {"version": 1, "description": "v1", "sys_created": "2024-01-01", "checksum": "a", "execution_ms": 10},
             ],
             table_exists=True,
         )
@@ -643,7 +642,7 @@ class TestMigrationRunnerDiff:
         tx = _make_mock_tx(
             applied_rows=[{
                 "version": 1, "description": "v1",
-                "applied_at": "2024-01-01",
+                "sys_created": "2024-01-01",
                 "checksum": MigrationRunner._checksum(up1),
                 "execution_ms": 10,
             }],
@@ -667,7 +666,7 @@ class TestMigrationRunnerDiff:
         tx = _make_mock_tx(
             applied_rows=[{
                 "version": 1, "description": "v1",
-                "applied_at": "2024-01-01",
+                "sys_created": "2024-01-01",
                 "checksum": "old_checksum_not_matching",
                 "execution_ms": 10,
             }],
@@ -685,7 +684,7 @@ class TestMigrationRunnerDiff:
         tx = _make_mock_tx(
             applied_rows=[{
                 "version": 99, "description": "ghost",
-                "applied_at": "2024-01-01",
+                "sys_created": "2024-01-01",
                 "checksum": "abc",
                 "execution_ms": 10,
             }],
@@ -800,15 +799,18 @@ class TestTrackingTable:
         runner = MigrationRunner(engine)
         runner.pending()
 
-        # Should have called execute with CREATE TABLE
-        create_calls = [
-            c for c in tx.execute.call_args_list
-            if "CREATE TABLE" in str(c)
-        ]
-        assert len(create_calls) >= 1
+        # Should have called table.create() with column definitions
+        mock_table = tx.table.return_value
+        mock_table.create.assert_called_once()
+        create_kwargs = mock_table.create.call_args
+        columns = create_kwargs[1].get("columns") or create_kwargs[0][0]
+        assert "version" in columns
+        assert "description" in columns
+        assert "checksum" in columns
+        assert "execution_ms" in columns
 
     def test_tracking_table_not_recreated_when_exists(self):
-        """If tracking table exists, no CREATE TABLE is issued."""
+        """If tracking table exists, no create() is issued."""
         @migration(version=1, description="v1")
         def up1(tx):
             pass
@@ -820,12 +822,9 @@ class TestTrackingTable:
         runner = MigrationRunner(engine)
         runner.pending()
 
-        # Should NOT have a CREATE TABLE call
-        create_calls = [
-            c for c in tx.execute.call_args_list
-            if "CREATE TABLE" in str(c)
-        ]
-        assert len(create_calls) == 0
+        # Should NOT have called table.create()
+        mock_table = tx.table.return_value
+        mock_table.create.assert_not_called()
 
 
 # ── CLI tests ──────────────────────────────────────────────────────