From fe1d09bb8860bdc9a2ee087d3e314bacdd1182aa Mon Sep 17 00:00:00 2001
From: Tim Van Wassenhove <tim@timvw.be>
Date: Tue, 26 Mar 2024 22:29:20 +0100
Subject: [PATCH] feat: add test for listing multiple parquet files

---
 src/main.rs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index 909783b..609bcb0 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -109,4 +109,21 @@ mod tests {
             ));
         Ok(())
     }
+
+    #[tokio::test]
+    async fn run_with_local_parquet_files_in_folder() -> Result<()> {
+        let mut cmd = get_qv_cmd()?;
+        let cmd = cmd.arg(get_qv_testing_path(
+            "data/iceberg/db/COVID-19_NYT/data",
+        ));
+        cmd.assert()
+            .success()
+            .stdout(predicate::str::contains(
+                r#"| date       | county     | state    | fips  | cases | deaths |"#,
+            ))
+            .stdout(predicate::str::contains(
+                r#"| 2020-05-19 | Lawrence   | Illinois | 17101 | 4     | 0      |"#,
+            ));
+        Ok(())
+    }
 }