From 2210466f78ad97c832fd581e72a209b15ec968b0 Mon Sep 17 00:00:00 2001
From: Jack Wright <ayax79@gmail.com>
Date: Fri, 20 Dec 2024 10:03:39 -0800
Subject: [PATCH] fix compilation errors

---
 crates/nu_plugin_polars/src/cache/rm.rs                |  2 +-
 crates/nu_plugin_polars/src/cloud/aws.rs               |  2 +-
 .../src/dataframe/command/core/save/mod.rs             |  2 +-
 .../src/dataframe/command/core/to_lazy.rs              |  2 +-
 crates/nu_plugin_polars/src/lib.rs                     | 10 +++++-----
 5 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/crates/nu_plugin_polars/src/cache/rm.rs b/crates/nu_plugin_polars/src/cache/rm.rs
index 537b7c6356..7b572bef50 100644
--- a/crates/nu_plugin_polars/src/cache/rm.rs
+++ b/crates/nu_plugin_polars/src/cache/rm.rs
@@ -89,7 +89,7 @@ mod test {
 
     #[test]
     fn test_remove() -> Result<(), ShellError> {
-        let plugin = PolarsPlugin::new_test_mode().into();
+        let plugin = PolarsPlugin::new_test_mode()?.into();
         let pipeline_data = PluginTest::new("polars", plugin)?
             .add_decl(Box::new(First))?
             .add_decl(Box::new(Get))?
diff --git a/crates/nu_plugin_polars/src/cloud/aws.rs b/crates/nu_plugin_polars/src/cloud/aws.rs
index 320acd820f..09bd2f609e 100644
--- a/crates/nu_plugin_polars/src/cloud/aws.rs
+++ b/crates/nu_plugin_polars/src/cloud/aws.rs
@@ -63,5 +63,5 @@ async fn build_aws_cloud_configs() -> Result<Vec<(AmazonS3ConfigKey, String)>, S
 
 pub(crate) fn build_cloud_options(plugin: &PolarsPlugin) -> Result<CloudOptions, ShellError> {
     let configs = plugin.runtime.block_on(build_aws_cloud_configs())?;
-    Ok(CloudOptions::default().with_aws(configs.into_iter()))
+    Ok(CloudOptions::default().with_aws(configs))
 }
diff --git a/crates/nu_plugin_polars/src/dataframe/command/core/save/mod.rs b/crates/nu_plugin_polars/src/dataframe/command/core/save/mod.rs
index 6d92b9bd33..8619ec9d42 100644
--- a/crates/nu_plugin_polars/src/dataframe/command/core/save/mod.rs
+++ b/crates/nu_plugin_polars/src/dataframe/command/core/save/mod.rs
@@ -248,7 +248,7 @@ pub(crate) mod test {
         let tmp_file_str = tmp_file.to_str().expect("should be able to get file path");
 
         let cmd = format!("{cmd} {tmp_file_str}");
-        let mut plugin_test = PluginTest::new("polars", PolarsPlugin::default().into())?;
+        let mut plugin_test = PluginTest::new("polars", PolarsPlugin::new()?.into())?;
         plugin_test.engine_state_mut().add_env_var(
             "PWD".to_string(),
             Value::string(
diff --git a/crates/nu_plugin_polars/src/dataframe/command/core/to_lazy.rs b/crates/nu_plugin_polars/src/dataframe/command/core/to_lazy.rs
index 1b42d310f3..4d8bf5bfcf 100644
--- a/crates/nu_plugin_polars/src/dataframe/command/core/to_lazy.rs
+++ b/crates/nu_plugin_polars/src/dataframe/command/core/to_lazy.rs
@@ -79,7 +79,7 @@ mod tests {
 
     #[test]
     fn test_to_lazy() -> Result<(), ShellError> {
-        let plugin: Arc<PolarsPlugin> = PolarsPlugin::new_test_mode().into();
+        let plugin: Arc<PolarsPlugin> = PolarsPlugin::new_test_mode()?.into();
         let mut plugin_test = PluginTest::new("polars", Arc::clone(&plugin))?;
         let pipeline_data = plugin_test.eval("[[a b]; [6 2] [1 4] [4 1]] | polars into-lazy")?;
         let value = pipeline_data.into_value(Span::test_data())?;
diff --git a/crates/nu_plugin_polars/src/lib.rs b/crates/nu_plugin_polars/src/lib.rs
index a199cd7f0e..d9c7e6d392 100644
--- a/crates/nu_plugin_polars/src/lib.rs
+++ b/crates/nu_plugin_polars/src/lib.rs
@@ -255,11 +255,11 @@ pub mod test {
 
     impl PolarsPlugin {
         /// Creates a new polars plugin in test mode
-        pub fn new_test_mode() -> Self {
-            PolarsPlugin {
+        pub fn new_test_mode() -> Result<Self, ShellError> {
+            Ok(PolarsPlugin {
                 disable_cache_drop: true,
-                ..PolarsPlugin::default()
-            }
+                ..PolarsPlugin::new()?
+            })
         }
     }
 
@@ -287,7 +287,7 @@ pub mod test {
         command: &impl PluginCommand,
         decls: Vec<Box<dyn Command>>,
     ) -> Result<(), ShellError> {
-        let plugin = PolarsPlugin::new_test_mode();
+        let plugin = PolarsPlugin::new_test_mode()?;
         let examples = command.examples();
 
         // we need to cache values in the examples