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#" => [ ] :: 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() { "" => expect![[r#" => /tmp :: path "#]]; "" => expect![[r#" => /tmp :: path "#]]; });