Fix test

quambene · May 2, 2024 · 88cdeee · 88cdeee
1 parent f9a183e
commit 88cdeee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/cmd/query.rs b/src/cmd/query.rs
@@ -27,7 +27,7 @@ pub fn query(matches: &ArgMatches) -> Result<(), anyhow::Error> {
                     println!("Display query result: {}", df_query);
                 }
 
-                if matches.contains_id(arg::SAVE) {
+                if matches.get_flag(arg::SAVE) {
                     let save_dir = Path::new(arg::value(arg::SAVE_DIR, matches)?);
 
                     // If argument 'FILE_TYPE' is not present the default value 'csv' will be used

diff --git a/tests/cmd/test_query.rs b/tests/cmd/test_query.rs
@@ -80,6 +80,7 @@ fn test_query_save_png() {
     cmd.args([
         "query",
         test_query,
+        "--display",
         "--save",
         "--save-dir",
         save_dir,