Add tests

Author: alamb

datafusion-cli/tests/cli_integration.rs

@@ -19,7 +19,7 @@ use std::process::Command;
 
 use rstest::rstest;
 
-use insta::{glob, Settings};
+use insta::{assert_snapshot, glob, Settings};
 use insta_cmd::{assert_cmd_snapshot, get_cargo_bin};
 use std::{env, fs};
 
@@ -59,6 +59,7 @@ fn init() {
     "batch_size",
     ["--command", "show datafusion.execution.batch_size", "-q", "-b", "1"],
 )]
+
 #[test]
 fn cli_quick_test<'a>(
     #[case] snapshot_name: &'a str,
@@ -74,6 +75,31 @@ fn cli_quick_test<'a>(
     assert_cmd_snapshot!(cmd);
 }
 
+#[rstest]
+#[case::default_explain_plan(
+    "batch_size",
+    // default explain format should be tree
+    ["--command", "EXPLAIN SELECT 123"],
+)]
+#[case::can_see_indent_format(
+    "batch_size",
+    // can choose the old explain format too
+    ["--command", "EXPLAIN FORMAT indent SELECT 123"],
+)]
+#[test]
+fn cli_quick_test_explain<'a>(
+    #[case] snapshot_name: &'a str,
+    #[case] args: impl IntoIterator<Item = &'a str>,
+) {
+    let mut settings = make_settings();
+    settings.set_snapshot_suffix(snapshot_name);
+    let _bound = settings.bind_to_scope();
+
+    let mut cmd = cli();
+    cmd.args(args);
+
+    assert_cmd_snapshot!(cmd);
+}
 #[rstest]
 #[case("csv")]
 #[case("tsv")]

datafusion-cli/tests/snapshots/cli_quick_test@backslash.snap

@@ -8,6 +8,7 @@ info:
     - "--format"
     - json
     - "-q"
+snapshot_kind: text
 ---
 success: true
 exit_code: 0

datafusion-cli/tests/snapshots/cli_quick_test@batch_size.snap

@@ -1,5 +1,5 @@
 ---
-source: tests/cli_integration.rs
+source: datafusion-cli/tests/cli_integration.rs
 info:
   program: datafusion-cli
   args:
@@ -8,6 +8,7 @@ info:
     - "-q"
     - "-b"
     - "1"
+snapshot_kind: text
 ---
 success: true
 exit_code: 0

datafusion-cli/tests/snapshots/cli_quick_test@files.snap

@@ -1,11 +1,12 @@
 ---
-source: tests/cli_integration.rs
+source: datafusion-cli/tests/cli_integration.rs
 info:
   program: datafusion-cli
   args:
     - "--file"
     - tests/sql/select.sql
     - "-q"
+snapshot_kind: text
 ---
 success: true
 exit_code: 0

datafusion-cli/tests/snapshots/cli_quick_test@statements.snap

@@ -1,11 +1,12 @@
 ---
-source: tests/cli_integration.rs
+source: datafusion-cli/tests/cli_integration.rs
 info:
   program: datafusion-cli
   args:
     - "--command"
     - select 1; select 2;
     - "-q"
+snapshot_kind: text
 ---
 success: true
 exit_code: 0

datafusion-cli/tests/snapshots/cli_quick_test_explain@batch_size-2.snap

@@ -0,0 +1,31 @@
+---
+source: datafusion-cli/tests/cli_integration.rs
+info:
+  program: datafusion-cli
+  args:
+    - "--command"
+    - EXPLAIN SELECT 123
+snapshot_kind: text
+---
+success: true
+exit_code: 0
+----- stdout -----
+[CLI_VERSION]
++---------------+-------------------------------+
+| plan_type     | plan                          |
++---------------+-------------------------------+
+| physical_plan | ┌───────────────────────────┐ |
+|               | │       ProjectionExec      │ |
+|               | │    --------------------   │ |
+|               | │      Int64(123): 123      │ |
+|               | └─────────────┬─────────────┘ |
+|               | ┌─────────────┴─────────────┐ |
+|               | │     PlaceholderRowExec    │ |
+|               | └───────────────────────────┘ |
+|               |                               |
++---------------+-------------------------------+
+1 row(s) fetched. 
+[ELAPSED]
+
+
+----- stderr -----

datafusion-cli/tests/snapshots/cli_quick_test_explain@batch_size.snap

@@ -0,0 +1,27 @@
+---
+source: datafusion-cli/tests/cli_integration.rs
+info:
+  program: datafusion-cli
+  args:
+    - "--command"
+    - EXPLAIN FORMAT indent SELECT 123
+snapshot_kind: text
+---
+success: true
+exit_code: 0
+----- stdout -----
+[CLI_VERSION]
++---------------+------------------------------------------+
+| plan_type     | plan                                     |
++---------------+------------------------------------------+
+| logical_plan  | Projection: Int64(123)                   |
+|               |   EmptyRelation                          |
+| physical_plan | ProjectionExec: expr=[123 as Int64(123)] |
+|               |   PlaceholderRowExec                     |
+|               |                                          |
++---------------+------------------------------------------+
+2 row(s) fetched. 
+[ELAPSED]
+
+
+----- stderr -----