tpchgen-cli 4.0.0

Blazing fast pure Rust TPC-H data generator command line tool.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use assert_cmd::cargo::cargo_bin_cmd;
use tempfile::tempdir;

/// Help text refers to `tpchgen-cli`, not `tpcgen-cli`.
#[test]
fn test_tpchgen_cli_help_uses_binary_name() {
    let output = cargo_bin_cmd!("tpchgen-cli")
        .arg("--help")
        .assert()
        .success()
        .get_output()
        .stdout
        .clone();
    let help = String::from_utf8(output).unwrap();
    assert!(
        help.contains("tpchgen-cli -s 1 --output-dir=/tmp/tpch"),
        "{help}"
    );
    assert!(!help.contains("tpcgen-cli"), "{help}");
}

/// `-V`/`--version` reports this package's name and version, not `tpcgen-cli`'s.
#[test]
fn test_tpchgen_cli_version() {
    let expected = format!("tpchgen-cli {}\n", env!("CARGO_PKG_VERSION"));
    for flag in ["-V", "--version"] {
        let output = cargo_bin_cmd!("tpchgen-cli")
            .arg(flag)
            .assert()
            .success()
            .get_output()
            .stdout
            .clone();
        assert_eq!(String::from_utf8(output).unwrap(), expected);
    }
}

#[test]
fn test_tpchgen_cli_invalid_inputs_before_output() {
    let temp = tempdir().unwrap();
    let output = temp.path().join("output");
    cargo_bin_cmd!("tpchgen-cli")
        .args(["--tables=region", "--parts=0"])
        .arg("--output-dir")
        .arg(&output)
        .assert()
        .failure()
        .stdout("");
    assert!(!output.exists());
}

/// Smoke test for `tpchgen-cli` binary.
#[test]
fn test_tpchgen_cli_command_forms() {
    let temp_dir = tempdir().expect("Failed to create temporary directory");

    cargo_bin_cmd!("tpchgen-cli")
        .arg("tbl")
        .arg("--scale-factor")
        .arg("0.001")
        .arg("--tables")
        .arg("part")
        .arg("--output-dir")
        .arg(temp_dir.path())
        .arg("--no-progress")
        .assert()
        .success();

    let expected_file = temp_dir.path().join("part.tbl");
    assert!(
        expected_file.exists(),
        "Expected file {expected_file:?} to exist with `tpchgen-cli`",
    );
}