Extend test_simple_delete to do one more delete

tursodatabase · Jan 26, 2025 · 14486ab · 14486ab
1 parent d62395a
commit 14486ab
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/test/src/lib.rs b/test/src/lib.rs
@@ -529,6 +529,23 @@ mod tests {
             }
         };
 
+        let insert_query = "INSERT INTO test VALUES (2)";
+        match conn.query(insert_query) {
+            Ok(Some(ref mut rows)) => loop {
+                match rows.next_row()? {
+                    StepResult::IO => {
+                        tmp_db.io.run_once()?;
+                    }
+                    StepResult::Done => break,
+                    _ => unreachable!(),
+                }
+            },
+            Ok(None) => {}
+            Err(err) => {
+                eprintln!("{}", err);
+            }
+        };
+
         let delete_query = "DELETE FROM test WHERE x = 1;";
         match conn.query(delete_query) {
             Ok(Some(ref mut rows)) => loop {
@@ -546,6 +563,23 @@ mod tests {
             }
         };
 
+        let delete_query = "DELETE FROM test WHERE x = 2;";
+        match conn.query(delete_query) {
+            Ok(Some(ref mut rows)) => loop {
+                match rows.next_row()? {
+                    StepResult::IO => {
+                        tmp_db.io.run_once()?;
+                    }
+                    StepResult::Done => break,
+                    _ => unreachable!(),
+                }
+            },
+            Ok(None) => {}
+            Err(err) => {
+                eprintln!("{}", err);
+            }
+        };
+
         let count_query = "SELECT COUNT(*) FROM test;";
         match conn.query(count_query) {
             Ok(Some(ref mut rows)) => loop {