fix: uses NamedTempFile for inline test pathing

xmtp · Apr 4, 2024 · 98e7776 · 98e7776
1 parent 84041df
commit 98e7776
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/main.rs b/src/main.rs
@@ -31,7 +31,18 @@ async fn upload_file(
 
     // Create a new UUID for the file.
     let file_id = Uuid::new_v4();
-    let file_name = format!("uploads/{}", file_id);
+    let file_name: String;
+
+    #[cfg(not(test))]
+    {
+        file_name = format!("uploads/{}", file_id)
+    }
+
+    #[cfg(test)]
+    {
+        use tempfile::NamedTempFile;
+        file_name = NamedTempFile::new().unwrap().path().to_string_lossy().to_string();
+    }
 
     // Try to create the file.
     let mut file = match File::create(&file_name) {