tvl-depot/tvix/cli/tests/repl.rs

99 lines
2.2 KiB
Rust
Raw Normal View History

use std::ffi::OsString;
use clap::Parser;
use expect_test::expect;
use tvix_cli::init_io_handle;
macro_rules! test_repl {
($name:ident() {$($send:expr => $expect:expr;)*}) => {
#[test]
fn $name() {
let tokio_runtime = tokio::runtime::Runtime::new().unwrap();
let args = tvix_cli::Args::parse_from(vec![
OsString::from("tvix"),
OsString::from("--nix-search-path"),
OsString::from("nixpkgs=/tmp"),
]);
let mut repl = tvix_cli::Repl::new(init_io_handle(&tokio_runtime, &args), &args);
$({
let result = repl.send($send.into());
$expect.assert_eq(result.output())
;
})*
}
}
}
test_repl!(simple_expr_eval() {
"1" => expect![[r#"
=> 1 :: int
"#]];
});
test_repl!(multiline_input() {
"{ x = 1; " => expect![[""]];
"y = 2; }" => expect![[r#"
=> { x = 1; y = 2; } :: set
"#]];
});
test_repl!(bind_literal() {
"x = 1" => expect![[""]];
"x" => expect![[r#"
=> 1 :: int
"#]];
});
test_repl!(bind_lazy() {
"x = { z = 1; }" => expect![[""]];
"x" => expect![[r#"
=> { z = 1; } :: set
"#]];
"x.z" => expect![[r#"
=> 1 :: int
"#]];
"x.z" => expect![[r#"
=> 1 :: int
"#]];
});
test_repl!(bind_lazy_errors() {
r#"x = (_: "x" + 1)"# => expect![[""]];
"x null" => expect![[""]];
});
test_repl!(bind_referencing_import() {
"six = import ./tests/six.nix {}" => expect![[""]];
"six.six" => expect![[r#"
=> 6 :: int
"#]];
"imported = import ./tests/import.nix" => expect![[""]];
"(imported {}).six" => expect![[r#"
=> 6 :: int
"#]];
});
test_repl!(deep_print() {
"builtins.map (x: x + 1) [ 1 2 3 ]" => expect![[r#"
=> [ <CODE> <CODE> <CODE> ] :: list
"#]];
":p builtins.map (x: x + 1) [ 1 2 3 ]" => expect![[r#"
=> [ 2 3 4 ] :: list
"#]];
});
test_repl!(explain() {
":d { x = 1; y = [ 2 3 4 ]; }" => expect![[r#"
=> a 2-item attribute set
"#]];
});
test_repl!(reference_nix_path() {
"<nixpkgs>" => expect![[r#"
=> /tmp :: path
"#]];
"<nixpkgs>" => expect![[r#"
=> /tmp :: path
"#]];
});