From 31eb4403ad8a1df569a43b8f000cc805789319aa Mon Sep 17 00:00:00 2001
From: Diego Reis <reisdiego144@gmail.com>
Date: Fri, 1 Aug 2025 11:50:51 -0300
Subject: [PATCH] Add integration tests for query_row and get

---
 bindings/rust/tests/integration_tests.rs | 88 +++++++++++++++++++++++-
 1 file changed, 87 insertions(+), 1 deletion(-)

diff --git a/bindings/rust/tests/integration_tests.rs b/bindings/rust/tests/integration_tests.rs
index 7e79ff34e..332e86665 100644
--- a/bindings/rust/tests/integration_tests.rs
+++ b/bindings/rust/tests/integration_tests.rs
@@ -1,5 +1,5 @@
 use tokio::fs;
-use turso::{Builder, Value};
+use turso::{Builder, Error, Value};
 
 #[tokio::test]
 async fn test_rows_next() {
@@ -211,3 +211,89 @@ pub async fn test_execute_batch() {
         assert_eq!(row.get_value(0).unwrap(), Value::Integer(2));
     }
 }
+
+#[tokio::test]
+async fn test_query_row_returns_first_row() {
+    let db = Builder::new_local(":memory:").build().await.unwrap();
+    let conn = db.connect().unwrap();
+
+    conn.execute("CREATE TABLE users (id INTEGER, name TEXT)", ())
+        .await
+        .unwrap();
+
+    conn.execute("INSERT INTO users VALUES (1, 'Frodo')", ())
+        .await
+        .unwrap();
+
+    let row = conn
+        .prepare("SELECT id FROM users WHERE name = ?")
+        .await
+        .unwrap()
+        .query_row(&["Frodo"])
+        .await
+        .unwrap();
+
+    let id: i64 = row.get(0).unwrap();
+    assert_eq!(id, 1);
+}
+
+#[tokio::test]
+async fn test_query_row_returns_no_rows_error() {
+    let db = Builder::new_local(":memory:").build().await.unwrap();
+    let conn = db.connect().unwrap();
+
+    conn.execute("CREATE TABLE users (id INTEGER, name TEXT)", ())
+        .await
+        .unwrap();
+
+    let result = conn
+        .prepare("SELECT id FROM users WHERE name = ?")
+        .await
+        .unwrap()
+        .query_row(&["Ghost"])
+        .await;
+
+    assert!(matches!(result, Err(Error::QueryReturnedNoRows)));
+}
+
+#[tokio::test]
+async fn test_row_get_column_typed() {
+    let db = Builder::new_local(":memory:").build().await.unwrap();
+    let conn = db.connect().unwrap();
+
+    conn.execute("CREATE TABLE v (n INTEGER, label TEXT)", ())
+        .await
+        .unwrap();
+
+    conn.execute("INSERT INTO v VALUES (42, 'answer')", ())
+        .await
+        .unwrap();
+
+    let mut rows = conn.query("SELECT * FROM v", ()).await.unwrap();
+    let row = rows.next().await.unwrap().unwrap();
+
+    let n: i64 = row.get(0).unwrap();
+    let label: String = row.get(1).unwrap();
+
+    assert_eq!(n, 42);
+    assert_eq!(label, "answer");
+}
+
+#[tokio::test]
+async fn test_row_get_conversion_error() {
+    let db = Builder::new_local(":memory:").build().await.unwrap();
+    let conn = db.connect().unwrap();
+
+    conn.execute("CREATE TABLE t (x TEXT)", ()).await.unwrap();
+
+    conn.execute("INSERT INTO t VALUES (NULL)", ())
+        .await
+        .unwrap();
+
+    let mut rows = conn.query("SELECT x FROM t", ()).await.unwrap();
+    let row = rows.next().await.unwrap().unwrap();
+
+    // Attempt to convert TEXT into integer (should fail)
+    let result: Result<u32, _> = row.get(0);
+    assert!(matches!(result, Err(Error::ConversionFailure(_))));
+}