fix tests

Author: Maxxen

crates/duckdb/src/r2d2.rs

@@ -40,13 +40,21 @@
 //!         .unwrap()
 //! }
 //! ```
-use crate::{vscalar::VScalar, vtab::VTab, Config, Connection, Error, Result};
+use crate::{Config, Connection, Error, Result};
 use std::{
-    fmt::Debug,
     path::Path,
     sync::{Arc, Mutex},
 };
 
+#[cfg(feature = "vscalar")]
+use crate::vscalar::VScalar;
+#[cfg(feature = "vscalar")]
+use std::fmt::Debug;
+
+
+#[cfg(feature = "vtab")]
+use crate::vtab::VTab;
+
 /// An `r2d2::ManageConnection` for `duckdb::Connection`s.
 pub struct DuckdbConnectionManager {
     connection: Arc<Mutex<Connection>>,
@@ -81,13 +89,15 @@ impl DuckdbConnectionManager {
     }
 
     /// Register a table function.
+    #[cfg(feature = "vtab")]
     pub fn register_table_function<T: VTab>(&self, name: &str) -> Result<()> {
         let conn = self.connection.lock().unwrap();
         conn.register_table_function::<T>(name)
     }
 
     /// Register a scalar function.
-    pub fn register_scalar<S: VScalar>(&self, name: &str) -> Result<()>
+    #[cfg(feature = "vscalar")]
+    pub fn register_scalar_function<S: VScalar>(&self, name: &str) -> Result<()>
     where
         S::State: Debug,
     {

crates/duckdb/src/vtab/arrow.rs

@@ -539,6 +539,15 @@ pub fn write_arrow_array_to_vector(
                 &mut chunk.flat_vector(),
             );
         }
+        DataType::Utf8View => {
+            string_view_array_to_vector(
+                col.as_ref()
+                    .as_any()
+                    .downcast_ref::<StringViewArray>()
+                    .ok_or_else(|| Box::<dyn std::error::Error>::from("Unable to downcast to StringViewArray"))?,
+                &mut chunk.flat_vector(),
+            );
+        }
         DataType::Binary => {
             binary_array_to_vector(as_generic_binary_array(col.as_ref()), &mut chunk.flat_vector());
         }
@@ -554,6 +563,15 @@ pub fn write_arrow_array_to_vector(
                 &mut chunk.flat_vector(),
             );
         }
+        DataType::BinaryView => {
+            binary_view_array_to_vector(
+                col.as_ref()
+                    .as_any()
+                    .downcast_ref::<BinaryViewArray>()
+                    .ok_or_else(|| Box::<dyn std::error::Error>::from("Unable to downcast to BinaryViewArray"))?,
+                &mut chunk.flat_vector(),
+            );
+        }
         DataType::List(_) => {
             list_array_to_vector(as_list_array(col.as_ref()), &mut chunk.list_vector())?;
         }