Add tests

Author: utay

tests/test_execute.py

@@ -0,0 +1,48 @@
+"""
+Integration tests for client.execute() DML row count reporting.
+
+Verifies that execute() correctly returns the number of rows
+affected by INSERT, UPDATE, and DELETE statements.
+"""
+
+from altertable_flightsql import Client
+from tests.conftest import TableInfo
+
+
+class TestExecuteRowCount:
+    """Test that execute() returns the correct number of affected rows."""
+
+    def test_insert_returns_row_count(self, altertable_client: Client, test_table: TableInfo):
+        """Test that INSERT returns the number of inserted rows."""
+        rows = altertable_client.execute(
+            f"INSERT INTO {test_table.full_name} (id, name, value) VALUES (4, 'Dave', 400), (5, 'Eve', 500)"
+        )
+        assert rows == 2
+
+    def test_update_returns_row_count(self, altertable_client: Client, test_table: TableInfo):
+        """Test that UPDATE returns the number of updated rows."""
+        rows = altertable_client.execute(
+            f"UPDATE {test_table.full_name} SET value = 999 WHERE value >= 200"
+        )
+        assert rows == 2
+
+    def test_delete_returns_row_count(self, altertable_client: Client, test_table: TableInfo):
+        """Test that DELETE returns the number of deleted rows."""
+        rows = altertable_client.execute(
+            f"DELETE FROM {test_table.full_name} WHERE id IN (1, 2)"
+        )
+        assert rows == 2
+
+    def test_delete_no_match_returns_zero(self, altertable_client: Client, test_table: TableInfo):
+        """Test that DELETE with no matching rows returns 0."""
+        rows = altertable_client.execute(
+            f"DELETE FROM {test_table.full_name} WHERE id = 9999"
+        )
+        assert rows == 0
+
+    def test_update_no_match_returns_zero(self, altertable_client: Client, test_table: TableInfo):
+        """Test that UPDATE with no matching rows returns 0."""
+        rows = altertable_client.execute(
+            f"UPDATE {test_table.full_name} SET value = 0 WHERE id = 9999"
+        )
+        assert rows == 0