Merge pull request #5487 from gchq/5480-Query-CLI-tests

Issue 5480 - DataFusion CLI tests

Author: patchwork01

rust/apps/tests/compactor_cli.rs renamed to rust/apps/tests/compact_cli.rs

@@ -0,0 +1,143 @@
+/*
+ * Copyright 2022-2025 Crown Copyright
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+use assert_cmd::prelude::*;
+use color_eyre::eyre::Result;
+use predicates::prelude::*; // Used for writing assertions
+use std::process::Command; // Run programs
+
+#[test]
+fn invalid_region_maxs() -> Result<()> {
+    let mut cmd = Command::cargo_bin("query")?;
+    cmd.args([
+        "/tmp/output.parquet",
+        "/tmp/input.parquet",
+        "--row-keys",
+        "col1",
+        "--row-keys",
+        "col2",
+        "--region-mins",
+        "a",
+        "--region-mins",
+        "a",
+        "--region-maxs",
+        "z",
+        "--query-mins",
+        "b",
+        "--query-mins",
+        "b",
+        "--query-maxs",
+        "g",
+        "--query-maxs",
+        "g",
+    ]);
+    cmd.assert().failure().stderr(predicate::str::contains(
+        "quantity of region maximums != quantity of row key columns",
+    ));
+    Ok(())
+}
+
+#[test]
+fn invalid_region_mins() -> Result<()> {
+    let mut cmd = Command::cargo_bin("query")?;
+    cmd.args([
+        "/tmp/output.parquet",
+        "/tmp/input.parquet",
+        "--row-keys",
+        "col1",
+        "--row-keys",
+        "col2",
+        "--region-mins",
+        "a",
+        "--region-maxs",
+        "z",
+        "--region-maxs",
+        "z",
+        "--query-mins",
+        "b",
+        "--query-mins",
+        "b",
+        "--query-maxs",
+        "g",
+        "--query-maxs",
+        "g",
+    ]);
+    cmd.assert().failure().stderr(predicate::str::contains(
+        "quantity of region minimums != quantity of row key columns",
+    ));
+    Ok(())
+}
+
+#[test]
+fn invalid_query_mins() -> Result<()> {
+    let mut cmd = Command::cargo_bin("query")?;
+    cmd.args([
+        "/tmp/output.parquet",
+        "/tmp/input.parquet",
+        "--row-keys",
+        "col1",
+        "--row-keys",
+        "col2",
+        "--region-mins",
+        "a",
+        "--region-mins",
+        "a",
+        "--region-maxs",
+        "z",
+        "--region-maxs",
+        "z",
+        "--query-mins",
+        "b",
+        "--query-maxs",
+        "g",
+        "--query-maxs",
+        "g",
+    ]);
+    cmd.assert().failure().stderr(predicate::str::contains(
+        "quantity of query region minimums != quantity of row key columns",
+    ));
+    Ok(())
+}
+
+#[test]
+fn invalid_query_maxs() -> Result<()> {
+    let mut cmd = Command::cargo_bin("query")?;
+    cmd.args([
+        "/tmp/output.parquet",
+        "/tmp/input.parquet",
+        "--row-keys",
+        "col1",
+        "--row-keys",
+        "col2",
+        "--region-mins",
+        "a",
+        "--region-mins",
+        "a",
+        "--region-maxs",
+        "z",
+        "--region-maxs",
+        "z",
+        "--query-mins",
+        "b",
+        "--query-mins",
+        "b",
+        "--query-maxs",
+        "g",
+    ]);
+    cmd.assert().failure().stderr(predicate::str::contains(
+        "quantity of query region maximums != quantity of row key columns",
+    ));
+    Ok(())
+}