velocity-python 0.1.70__tar.gz → 0.1.71__tar.gz

{velocity_python-0.1.70 → velocity_python-0.1.71}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: velocity-python
-Version: 0.1.70
+Version: 0.1.71
 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.70 → velocity_python-0.1.71}/pyproject.toml

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

{velocity_python-0.1.70 → velocity_python-0.1.71}/src/velocity/__init__.py

@@ -1,4 +1,4 @@
-__version__ = version = "0.1.70"
+__version__ = version = "0.1.71"
 
 import importlib as _importlib
 

{velocity_python-0.1.70 → velocity_python-0.1.71}/src/velocity/db/core/column.py

@@ -15,6 +15,17 @@ class Column:
         self.name = name
         self.table = table
 
+    def cursor(self):
+        """
+        Return the active cursor for this column's transaction.
+
+        Required by the ``@return_default`` decorator, which brackets the wrapped
+        call in a savepoint (``create``/``rollback``/``release``) on ``self.cursor()``.
+        Delegates to the owning table so the savepoint shares the table's cursor —
+        the same one ``self.tx.execute`` runs the statement on.
+        """
+        return self.table.cursor()
+
     def __str__(self):
         return (
             f"Table: {self.table.name}\n"
@@ -113,6 +124,7 @@ class Column:
         Returns the distinct values in this column, optionally ordered and/or limited in quantity.
         """
         sql, vals = self.sql.select(
+            self.tx,
             columns=f"distinct {self.name}",
             table=self.table.name,
             orderby=f"{self.name} {order}",
@@ -127,7 +139,7 @@ class Column:
         try:
             qcol = self.sql.quote_identifier(self.name)
             sql, vals = self.sql.select(
-                columns=f"max({qcol})", table=self.table.name, where=where
+                self.tx, columns=f"max({qcol})", table=self.table.name, where=where
             )
             return self.tx.execute(sql, vals).scalar()
         except (exceptions.DbTableMissingError, exceptions.DbColumnMissingError):

{velocity_python-0.1.70 → velocity_python-0.1.71}/src/velocity_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: velocity-python
-Version: 0.1.70
+Version: 0.1.71
 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.70 → velocity_python-0.1.71}/src/velocity_python.egg-info/SOURCES.txt

@@ -156,6 +156,7 @@ tests/test_asset_references.py
 tests/test_assets_service.py
 tests/test_async_support.py
 tests/test_batch_operations.py
+tests/test_column_tx_arg.py
 tests/test_concurrency_safety.py
 tests/test_connection_pool.py
 tests/test_connection_resilience.py

velocity_python-0.1.71/tests/test_column_tx_arg.py

@@ -0,0 +1,64 @@
+"""
+Regression tests for Column query methods staying in sync with the SQL builder.
+
+Two bugs these guard against:
+- ``Column`` lacked a ``cursor()`` method, so the ``@return_default`` decorator
+  (which brackets the call in a savepoint on ``self.cursor()``) raised
+  ``AttributeError: 'Column' object has no attribute 'cursor'``.
+- ``Column.distinct``/``Column.max`` called ``self.sql.select(columns=..., ...)``
+  without the leading ``tx`` positional the current ``SQL.select(tx, ...)``
+  signature requires, raising ``TypeError: ... missing 1 required positional
+  argument: 'tx'``.
+"""
+
+import unittest
+from unittest.mock import MagicMock
+
+
+def _make_column(name="class"):
+    from velocity.db.core.column import Column
+
+    tx = MagicMock()
+    tx.create_savepoint.return_value = "sp1"
+    result = MagicMock()
+    result.as_simple_list.return_value.all.return_value = ["a", "b"]
+    result.scalar.return_value = 5
+    tx.execute.return_value = result
+
+    sql = MagicMock()
+    sql.select.return_value = ("SELECT ...", [])
+    tx.engine.sql = sql
+
+    table = MagicMock()
+    table.tx = tx
+    table.name = "contact"
+    table.cursor.return_value = "CUR"
+
+    return Column(table, name), tx, sql, table
+
+
+class TestColumnCursorDelegation(unittest.TestCase):
+    def test_cursor_delegates_to_table(self):
+        col, _tx, _sql, table = _make_column()
+        self.assertEqual(col.cursor(), "CUR")
+        table.cursor.assert_called()
+
+
+class TestColumnSelectPassesTx(unittest.TestCase):
+    def test_distinct_passes_tx_first_and_returns_rows(self):
+        col, tx, sql, _table = _make_column()
+        result = col.distinct()
+        self.assertEqual(result, ["a", "b"])
+        args, _kwargs = sql.select.call_args
+        self.assertIs(args[0], tx, "distinct() must pass tx as the first positional arg")
+
+    def test_max_passes_tx_first_and_returns_scalar(self):
+        col, tx, sql, _table = _make_column()
+        result = col.max()
+        self.assertEqual(result, 5)
+        args, _kwargs = sql.select.call_args
+        self.assertIs(args[0], tx, "max() must pass tx as the first positional arg")
+
+
+if __name__ == "__main__":
+    unittest.main()