test for manually loading an extension

Author: asg017

tests/test_manual_load.rs

@@ -0,0 +1,40 @@
+use sqlite_loadable::prelude::*;
+use sqlite_loadable::{api, define_scalar_function, Result};
+
+pub fn add(context: *mut sqlite3_context, values: &[*mut sqlite3_value]) -> Result<()> {
+    let a = api::value_int(&values[0]);
+    let b = api::value_int(&values[1]);
+    api::result_int(context, a + b);
+    Ok(())
+}
+
+#[sqlite_entrypoint]
+pub fn sqlite3_manual_init(db: *mut sqlite3) -> Result<()> {
+    define_scalar_function(db, "addx", 2, add, FunctionFlags::empty())?;
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    use rusqlite::Connection;
+
+    #[test]
+    fn test_manual_load() {
+        let db = Connection::open_in_memory().unwrap();
+        unsafe {
+            sqlite3_manual_init(
+                std::mem::transmute(db.handle()),
+                std::ptr::null_mut(),
+                std::ptr::null_mut(),
+            );
+        }
+
+        let result: i32 = db
+            .query_row("select addx(?1, ?2)", [1, 2], |x| x.get(0))
+            .unwrap();
+
+        assert_eq!(result, 3);
+    }
+}