From 8bef159582c5217deb72637c8af87e9b40cef659 Mon Sep 17 00:00:00 2001
From: PThorpe92 <preston@turso.tech>
Date: Sun, 9 Nov 2025 10:36:59 -0500
Subject: [PATCH] adjust testing extension to add some test assertions for new
 features

---
 extensions/tests/src/lib.rs | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/extensions/tests/src/lib.rs b/extensions/tests/src/lib.rs
index 639465088..8517155ef 100644
--- a/extensions/tests/src/lib.rs
+++ b/extensions/tests/src/lib.rs
@@ -53,6 +53,7 @@ impl VTabModule for KVStoreVTabModule {
             schema,
             KVStoreTable {
                 store: Rc::new(RefCell::new(BTreeMap::new())),
+                in_tx: false,
             },
         ))
     }
@@ -152,6 +153,7 @@ impl VTabCursor for KVStoreCursor {
 
 pub struct KVStoreTable {
     store: Store,
+    in_tx: bool,
 }
 
 impl VTable for KVStoreTable {
@@ -167,6 +169,31 @@ impl VTable for KVStoreTable {
         })
     }
 
+    fn begin(&mut self) -> Result<(), Self::Error> {
+        assert!(!self.in_tx, "Already in a transaction");
+        self.in_tx = true;
+        Ok(())
+    }
+
+    fn commit(&mut self) -> Result<(), Self::Error> {
+        assert!(self.in_tx, "Not in a transaction");
+        self.in_tx = false;
+        Ok(())
+    }
+
+    fn rollback(&mut self) -> Result<(), Self::Error> {
+        assert!(self.in_tx, "Not in a transaction");
+        self.in_tx = false;
+        Ok(())
+    }
+
+    fn rename(&mut self, new_name: &str) -> Result<(), Self::Error> {
+        // not a real extension of course, just asserting test in
+        // testing/cli_tests/extensions.py
+        assert_eq!(new_name, "renamed");
+        Ok(())
+    }
+
     fn best_index(
         constraints: &[ConstraintInfo],
         _order_by: &[OrderByInfo],