use std::path::PathBuf; use nu_engine::eval_block; use nu_parser::parse; use nu_protocol::{ engine::{EngineState, Stack, StateWorkingSet}, PipelineData, ShellError, Span, Value, }; use nu_test_support::fs; use reedline::Suggestion; const SEP: char = std::path::MAIN_SEPARATOR; fn create_default_context() -> EngineState { nu_command::add_shell_command_context(nu_cmd_lang::create_default_context()) } // creates a new engine with the current path into the completions fixtures folder pub fn new_engine() -> (PathBuf, String, EngineState, Stack) { // Target folder inside assets let dir = fs::fixtures().join("completions"); let mut dir_str = dir .clone() .into_os_string() .into_string() .unwrap_or_default(); dir_str.push(SEP); // Create a new engine with default context let mut engine_state = create_default_context(); // New stack let mut stack = Stack::new(); // Add pwd as env var stack.add_env_var( "PWD".to_string(), Value::String { val: dir_str.clone(), span: nu_protocol::Span::new(0, dir_str.len()), }, ); stack.add_env_var( "TEST".to_string(), Value::String { val: "NUSHELL".to_string(), span: nu_protocol::Span::new(0, dir_str.len()), }, ); #[cfg(windows)] stack.add_env_var( "Path".to_string(), Value::String { val: "c:\\some\\path;c:\\some\\other\\path".to_string(), span: nu_protocol::Span::new(0, dir_str.len()), }, ); #[cfg(not(windows))] stack.add_env_var( "PATH".to_string(), Value::String { val: "/some/path:/some/other/path".to_string(), span: nu_protocol::Span::new(0, dir_str.len()), }, ); // Merge environment into the permanent state let merge_result = engine_state.merge_env(&mut stack, &dir); assert!(merge_result.is_ok()); (dir, dir_str, engine_state, stack) } pub fn new_quote_engine() -> (PathBuf, String, EngineState, Stack) { // Target folder inside assets let dir = fs::fixtures().join("quoted_completions"); let mut dir_str = dir .clone() .into_os_string() .into_string() .unwrap_or_default(); dir_str.push(SEP); // Create a new engine with default context let mut engine_state = create_default_context(); // New stack let mut stack = Stack::new(); // Add pwd as env var stack.add_env_var( "PWD".to_string(), Value::String { val: dir_str.clone(), span: nu_protocol::Span::new(0, dir_str.len()), }, ); stack.add_env_var( "TEST".to_string(), Value::String { val: "NUSHELL".to_string(), span: nu_protocol::Span::new(0, dir_str.len()), }, ); // Merge environment into the permanent state let merge_result = engine_state.merge_env(&mut stack, &dir); assert!(merge_result.is_ok()); (dir, dir_str, engine_state, stack) } // match a list of suggestions with the expected values pub fn match_suggestions(expected: Vec, suggestions: Vec) { let expected_len = expected.len(); let suggestions_len = suggestions.len(); if expected_len != suggestions_len { panic!( "\nexpected {expected_len} suggestions but got {suggestions_len}: \n\ Suggestions: {suggestions:#?} \n\ Expected: {expected:#?}\n" ) } expected.iter().zip(suggestions).for_each(|it| { assert_eq!(it.0, &it.1.value); }); } // append the separator to the converted path pub fn folder(path: PathBuf) -> String { let mut converted_path = file(path); converted_path.push(SEP); converted_path } // convert a given path to string pub fn file(path: PathBuf) -> String { path.into_os_string().into_string().unwrap_or_default() } // merge_input executes the given input into the engine // and merges the state pub fn merge_input( input: &[u8], engine_state: &mut EngineState, stack: &mut Stack, dir: PathBuf, ) -> Result<(), ShellError> { let (block, delta) = { let mut working_set = StateWorkingSet::new(engine_state); let block = parse(&mut working_set, None, input, false); assert!(working_set.parse_errors.is_empty()); (block, working_set.render()) }; engine_state.merge_delta(delta)?; assert!(eval_block( engine_state, stack, &block, PipelineData::Value( Value::Nothing { span: Span::unknown(), }, None ), false, false ) .is_ok()); // Merge environment into the permanent state engine_state.merge_env(stack, &dir) }