// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license. use super::definitions::TestDefinition; use super::definitions::TestModule; use super::lsp_custom; use crate::args::flags_from_vec; use crate::args::DenoSubcommand; use crate::factory::CliFactory; use crate::lsp::client::Client; use crate::lsp::client::TestingNotification; use crate::lsp::config; use crate::lsp::logging::lsp_log; use crate::tools::test; use crate::tools::test::FailFastTracker; use crate::tools::test::TestEventSender; use deno_core::anyhow::anyhow; use deno_core::error::AnyError; use deno_core::error::JsError; use deno_core::futures::future; use deno_core::futures::stream; use deno_core::futures::StreamExt; use deno_core::parking_lot::Mutex; use deno_core::parking_lot::RwLock; use deno_core::unsync::spawn; use deno_core::unsync::spawn_blocking; use deno_core::ModuleSpecifier; use deno_runtime::permissions::Permissions; use deno_runtime::tokio_util::create_and_run_current_thread; use indexmap::IndexMap; use std::collections::HashMap; use std::collections::HashSet; use std::num::NonZeroUsize; use std::sync::Arc; use std::time::Duration; use std::time::Instant; use tokio::sync::mpsc; use tokio_util::sync::CancellationToken; use tower_lsp::lsp_types as lsp; /// Logic to convert a test request into a set of test modules to be tested and /// any filters to be applied to those tests fn as_queue_and_filters( params: &lsp_custom::TestRunRequestParams, tests: &HashMap, ) -> ( HashSet, HashMap, ) { let mut queue: HashSet = HashSet::new(); let mut filters: HashMap = HashMap::new(); if let Some(include) = ¶ms.include { for item in include { if let Some(test_definitions) = tests.get(&item.text_document.uri) { queue.insert(item.text_document.uri.clone()); if let Some(id) = &item.id { if let Some(test) = test_definitions.get(id) { let filter = filters.entry(item.text_document.uri.clone()).or_default(); if let Some(include) = filter.include.as_mut() { include.insert(test.id.clone(), test.clone()); } else { let mut include = HashMap::new(); include.insert(test.id.clone(), test.clone()); filter.include = Some(include); } } } } } } else { queue.extend(tests.keys().cloned()); } for item in ¶ms.exclude { if let Some(test_definitions) = tests.get(&item.text_document.uri) { if let Some(id) = &item.id { // there is no way to exclude a test step if item.step_id.is_none() { if let Some(test) = test_definitions.get(id) { let filter = filters.entry(item.text_document.uri.clone()).or_default(); filter.exclude.insert(test.id.clone(), test.clone()); } } } else { // the entire test module is excluded queue.remove(&item.text_document.uri); } } } queue.retain(|s| !tests.get(s).unwrap().is_empty()); (queue, filters) } fn as_test_messages>( message: S, is_markdown: bool, ) -> Vec { let message = lsp::MarkupContent { kind: if is_markdown { lsp::MarkupKind::Markdown } else { lsp::MarkupKind::PlainText }, value: message.as_ref().to_string(), }; vec![lsp_custom::TestMessage { message, expected_output: None, actual_output: None, location: None, }] } #[derive(Debug, Clone, Default, PartialEq)] struct LspTestFilter { include: Option>, exclude: HashMap, } impl LspTestFilter { fn as_ids(&self, test_module: &TestModule) -> Vec { let ids: Vec = if let Some(include) = &self.include { include.keys().cloned().collect() } else { test_module .defs .iter() .filter(|(_, d)| d.parent_id.is_none()) .map(|(k, _)| k.clone()) .collect() }; ids .into_iter() .filter(|id| !self.exclude.contains_key(id)) .collect() } } #[derive(Debug, Clone)] pub struct TestRun { id: u32, kind: lsp_custom::TestRunKind, filters: HashMap, queue: HashSet, tests: Arc>>, token: CancellationToken, workspace_settings: config::WorkspaceSettings, } impl TestRun { pub fn new( params: &lsp_custom::TestRunRequestParams, tests: Arc>>, workspace_settings: config::WorkspaceSettings, ) -> Self { let (queue, filters) = { let tests = tests.lock(); as_queue_and_filters(params, &tests) }; Self { id: params.id, kind: params.kind.clone(), filters, queue, tests, token: CancellationToken::new(), workspace_settings, } } /// Provide the tests of a test run as an enqueued module which can be sent /// to the client to indicate tests are enqueued for testing. pub fn as_enqueued(&self) -> Vec { let tests = self.tests.lock(); self .queue .iter() .map(|s| { let ids = if let Some(test_module) = tests.get(s) { if let Some(filter) = self.filters.get(s) { filter.as_ids(test_module) } else { LspTestFilter::default().as_ids(test_module) } } else { Vec::new() }; lsp_custom::EnqueuedTestModule { text_document: lsp::TextDocumentIdentifier { uri: s.clone() }, ids, } }) .collect() } /// If being executed, cancel the test. pub fn cancel(&self) { self.token.cancel(); } /// Execute the tests, dispatching progress notifications to the client. pub async fn exec( &self, client: &Client, maybe_root_uri: Option<&ModuleSpecifier>, ) -> Result<(), AnyError> { let args = self.get_args(); lsp_log!("Executing test run with arguments: {}", args.join(" ")); let flags = flags_from_vec(args.into_iter().map(String::from).collect())?; let factory = CliFactory::from_flags(flags).await?; // Various test files should not share the same permissions in terms of // `PermissionsContainer` - otherwise granting/revoking permissions in one // file would have impact on other files, which is undesirable. let permissions = Permissions::from_options(&factory.cli_options().permissions_options())?; test::check_specifiers( factory.cli_options(), factory.file_fetcher()?, factory.module_load_preparer().await?, self .queue .iter() .map(|s| (s.clone(), test::TestMode::Executable)) .collect(), ) .await?; let (concurrent_jobs, fail_fast) = if let DenoSubcommand::Test(test_flags) = factory.cli_options().sub_command() { ( test_flags .concurrent_jobs .unwrap_or_else(|| NonZeroUsize::new(1).unwrap()) .into(), test_flags.fail_fast, ) } else { unreachable!("Should always be Test subcommand."); }; let (sender, mut receiver) = mpsc::unbounded_channel::(); let sender = TestEventSender::new(sender); let fail_fast_tracker = FailFastTracker::new(fail_fast); let mut queue = self.queue.iter().collect::>(); queue.sort(); let tests: Arc>> = Arc::new(RwLock::new(IndexMap::new())); let mut test_steps = IndexMap::new(); let worker_factory = Arc::new(factory.create_cli_main_worker_factory().await?); let join_handles = queue.into_iter().map(move |specifier| { let specifier = specifier.clone(); let worker_factory = worker_factory.clone(); let permissions = permissions.clone(); let mut sender = sender.clone(); let fail_fast_tracker = fail_fast_tracker.clone(); let lsp_filter = self.filters.get(&specifier); let filter = test::TestFilter { substring: None, regex: None, include: lsp_filter.and_then(|f| { f.include .as_ref() .map(|i| i.values().map(|t| t.name.clone()).collect()) }), exclude: lsp_filter .map(|f| f.exclude.values().map(|t| t.name.clone()).collect()) .unwrap_or_default(), }; let token = self.token.clone(); spawn_blocking(move || { if fail_fast_tracker.should_stop() { return Ok(()); } let origin = specifier.to_string(); let file_result = if token.is_cancelled() { Ok(()) } else { create_and_run_current_thread(test::test_specifier( worker_factory, permissions, specifier, sender.clone(), fail_fast_tracker, test::TestSpecifierOptions { filter, shuffle: None, trace_ops: false, }, )) }; if let Err(error) = file_result { if error.is::() { sender.send(test::TestEvent::UncaughtError( origin, Box::new(error.downcast::().unwrap()), ))?; } else { return Err(error); } } Ok(()) }) }); let join_stream = stream::iter(join_handles) .buffer_unordered(concurrent_jobs) .collect::, tokio::task::JoinError>>>(); let mut reporter = Box::new(LspTestReporter::new( self, client.clone(), maybe_root_uri, self.tests.clone(), )); let handler = { spawn(async move { let earlier = Instant::now(); let mut summary = test::TestSummary::new(); let mut tests_with_result = HashSet::new(); let mut used_only = false; while let Some(event) = receiver.recv().await { match event { test::TestEvent::Register(description) => { reporter.report_register(&description); tests.write().insert(description.id, description); } test::TestEvent::Plan(plan) => { summary.total += plan.total; summary.filtered_out += plan.filtered_out; if plan.used_only { used_only = true; } reporter.report_plan(&plan); } test::TestEvent::Wait(id) => { reporter.report_wait(tests.read().get(&id).unwrap()); } test::TestEvent::Output(output) => { reporter.report_output(&output); } test::TestEvent::Result(id, result, elapsed) => { if tests_with_result.insert(id) { let description = tests.read().get(&id).unwrap().clone(); match &result { test::TestResult::Ok => summary.passed += 1, test::TestResult::Ignored => summary.ignored += 1, test::TestResult::Failed(error) => { summary.failed += 1; summary.failures.push((description.clone(), error.clone())); } test::TestResult::Cancelled => { summary.failed += 1; } } reporter.report_result(&description, &result, elapsed); } } test::TestEvent::UncaughtError(origin, error) => { reporter.report_uncaught_error(&origin, &error); summary.failed += 1; summary.uncaught_errors.push((origin, error)); } test::TestEvent::StepRegister(description) => { reporter.report_step_register(&description); test_steps.insert(description.id, description); } test::TestEvent::StepWait(id) => { reporter.report_step_wait(test_steps.get(&id).unwrap()); } test::TestEvent::StepResult(id, result, duration) => { if tests_with_result.insert(id) { match &result { test::TestStepResult::Ok => { summary.passed_steps += 1; } test::TestStepResult::Ignored => { summary.ignored_steps += 1; } test::TestStepResult::Failed(_) => { summary.failed_steps += 1; } } reporter.report_step_result( test_steps.get(&id).unwrap(), &result, duration, ); } } test::TestEvent::ForceEndReport => {} test::TestEvent::Sigint => {} } } let elapsed = Instant::now().duration_since(earlier); reporter.report_summary(&summary, &elapsed); if used_only { return Err(anyhow!( "Test failed because the \"only\" option was used" )); } if summary.failed > 0 { return Err(anyhow!("Test failed")); } Ok(()) }) }; let (join_results, result) = future::join(join_stream, handler).await; // propagate any errors for join_result in join_results { join_result??; } result??; Ok(()) } fn get_args(&self) -> Vec<&str> { let mut args = vec!["deno", "test"]; args.extend( self .workspace_settings .testing .args .iter() .map(|s| s.as_str()), ); args.push("--trace-ops"); if self.workspace_settings.unstable && !args.contains(&"--unstable") { args.push("--unstable"); } if let Some(config) = &self.workspace_settings.config { if !args.contains(&"--config") && !args.contains(&"-c") { args.push("--config"); args.push(config.as_str()); } } if let Some(import_map) = &self.workspace_settings.import_map { if !args.contains(&"--import-map") { args.push("--import-map"); args.push(import_map.as_str()); } } if self.kind == lsp_custom::TestRunKind::Debug && !args.contains(&"--inspect") && !args.contains(&"--inspect-brk") { args.push("--inspect"); } args } } #[derive(Debug, PartialEq)] enum LspTestDescription { /// `(desc, static_id)` TestDescription(test::TestDescription, String), /// `(desc, static_id)` TestStepDescription(test::TestStepDescription, String), } impl LspTestDescription { fn origin(&self) -> &str { match self { LspTestDescription::TestDescription(d, _) => d.origin.as_str(), LspTestDescription::TestStepDescription(d, _) => d.origin.as_str(), } } fn location(&self) -> &test::TestLocation { match self { LspTestDescription::TestDescription(d, _) => &d.location, LspTestDescription::TestStepDescription(d, _) => &d.location, } } fn parent_id(&self) -> Option { match self { LspTestDescription::TestDescription(_, _) => None, LspTestDescription::TestStepDescription(d, _) => Some(d.parent_id), } } fn static_id(&self) -> &str { match self { LspTestDescription::TestDescription(_, i) => i, LspTestDescription::TestStepDescription(_, i) => i, } } fn as_test_identifier( &self, tests: &IndexMap, ) -> lsp_custom::TestIdentifier { let uri = ModuleSpecifier::parse(&self.location().file_name).unwrap(); let static_id = self.static_id(); let mut root_desc = self; while let Some(parent_id) = root_desc.parent_id() { root_desc = tests.get(&parent_id).unwrap(); } let root_static_id = root_desc.static_id(); lsp_custom::TestIdentifier { text_document: lsp::TextDocumentIdentifier { uri }, id: Some(root_static_id.to_string()), step_id: if static_id == root_static_id { None } else { Some(static_id.to_string()) }, } } } struct LspTestReporter { client: Client, id: u32, maybe_root_uri: Option, files: Arc>>, tests: IndexMap, current_test: Option, } impl LspTestReporter { fn new( run: &TestRun, client: Client, maybe_root_uri: Option<&ModuleSpecifier>, files: Arc>>, ) -> Self { Self { client, id: run.id, maybe_root_uri: maybe_root_uri.cloned(), files, tests: Default::default(), current_test: Default::default(), } } fn progress(&self, message: lsp_custom::TestRunProgressMessage) { self .client .send_test_notification(TestingNotification::Progress( lsp_custom::TestRunProgressParams { id: self.id, message, }, )); } fn report_plan(&mut self, _plan: &test::TestPlan) {} fn report_register(&mut self, desc: &test::TestDescription) { let mut files = self.files.lock(); let specifier = ModuleSpecifier::parse(&desc.location.file_name).unwrap(); let test_module = files .entry(specifier.clone()) .or_insert_with(|| TestModule::new(specifier, "1".to_string())); let (static_id, is_new) = test_module.register_dynamic(desc); self.tests.insert( desc.id, LspTestDescription::TestDescription(desc.clone(), static_id.clone()), ); if is_new { self .client .send_test_notification(TestingNotification::Module( lsp_custom::TestModuleNotificationParams { text_document: lsp::TextDocumentIdentifier { uri: test_module.specifier.clone(), }, kind: lsp_custom::TestModuleNotificationKind::Insert, label: test_module.label(self.maybe_root_uri.as_ref()), tests: vec![test_module.get_test_data(&static_id)], }, )); } } fn report_wait(&mut self, desc: &test::TestDescription) { self.current_test = Some(desc.id); let desc = self.tests.get(&desc.id).unwrap(); let test = desc.as_test_identifier(&self.tests); self.progress(lsp_custom::TestRunProgressMessage::Started { test }); } fn report_output(&mut self, output: &[u8]) { let test = self .current_test .as_ref() .map(|id| self.tests.get(id).unwrap().as_test_identifier(&self.tests)); let value = String::from_utf8_lossy(output).replace('\n', "\r\n"); self.progress(lsp_custom::TestRunProgressMessage::Output { value, test, // TODO(@kitsonk) test output should include a location location: None, }) } fn report_result( &mut self, desc: &test::TestDescription, result: &test::TestResult, elapsed: u64, ) { self.current_test = None; match result { test::TestResult::Ok => { let desc = self.tests.get(&desc.id).unwrap(); self.progress(lsp_custom::TestRunProgressMessage::Passed { test: desc.as_test_identifier(&self.tests), duration: Some(elapsed as u32), }) } test::TestResult::Ignored => { let desc = self.tests.get(&desc.id).unwrap(); self.progress(lsp_custom::TestRunProgressMessage::Skipped { test: desc.as_test_identifier(&self.tests), }) } test::TestResult::Failed(failure) => { let desc = self.tests.get(&desc.id).unwrap(); self.progress(lsp_custom::TestRunProgressMessage::Failed { test: desc.as_test_identifier(&self.tests), messages: as_test_messages(failure.to_string(), false), duration: Some(elapsed as u32), }) } test::TestResult::Cancelled => { let desc = self.tests.get(&desc.id).unwrap(); self.progress(lsp_custom::TestRunProgressMessage::Failed { test: desc.as_test_identifier(&self.tests), messages: vec![], duration: Some(elapsed as u32), }) } } } fn report_uncaught_error(&mut self, origin: &str, js_error: &JsError) { self.current_test = None; let err_string = format!( "Uncaught error from {}: {}\nThis error was not caught from a test and caused the test runner to fail on the referenced module.\nIt most likely originated from a dangling promise, event/timeout handler or top-level code.", origin, test::fmt::format_test_error(js_error) ); let messages = as_test_messages(err_string, false); for desc in self.tests.values().filter(|d| d.origin() == origin) { self.progress(lsp_custom::TestRunProgressMessage::Failed { test: desc.as_test_identifier(&self.tests), messages: messages.clone(), duration: None, }); } } fn report_step_register(&mut self, desc: &test::TestStepDescription) { let mut files = self.files.lock(); let specifier = ModuleSpecifier::parse(&desc.location.file_name).unwrap(); let test_module = files .entry(specifier.clone()) .or_insert_with(|| TestModule::new(specifier, "1".to_string())); let (static_id, is_new) = test_module.register_step_dynamic( desc, self.tests.get(&desc.parent_id).unwrap().static_id(), ); self.tests.insert( desc.id, LspTestDescription::TestStepDescription(desc.clone(), static_id.clone()), ); if is_new { self .client .send_test_notification(TestingNotification::Module( lsp_custom::TestModuleNotificationParams { text_document: lsp::TextDocumentIdentifier { uri: test_module.specifier.clone(), }, kind: lsp_custom::TestModuleNotificationKind::Insert, label: test_module.label(self.maybe_root_uri.as_ref()), tests: vec![test_module.get_test_data(&static_id)], }, )); } } fn report_step_wait(&mut self, desc: &test::TestStepDescription) { if self.current_test == Some(desc.parent_id) { self.current_test = Some(desc.id); } let desc = self.tests.get(&desc.id).unwrap(); let test = desc.as_test_identifier(&self.tests); self.progress(lsp_custom::TestRunProgressMessage::Started { test }); } fn report_step_result( &mut self, desc: &test::TestStepDescription, result: &test::TestStepResult, elapsed: u64, ) { if self.current_test == Some(desc.id) { self.current_test = Some(desc.parent_id); } let desc = self.tests.get(&desc.id).unwrap(); match result { test::TestStepResult::Ok => { self.progress(lsp_custom::TestRunProgressMessage::Passed { test: desc.as_test_identifier(&self.tests), duration: Some(elapsed as u32), }) } test::TestStepResult::Ignored => { self.progress(lsp_custom::TestRunProgressMessage::Skipped { test: desc.as_test_identifier(&self.tests), }) } test::TestStepResult::Failed(failure) => { self.progress(lsp_custom::TestRunProgressMessage::Failed { test: desc.as_test_identifier(&self.tests), messages: as_test_messages(failure.to_string(), false), duration: Some(elapsed as u32), }) } } } fn report_summary( &mut self, _summary: &test::TestSummary, _elapsed: &Duration, ) { // there is nothing to do on report_summary } } #[cfg(test)] mod tests { use super::*; use crate::lsp::testing::collectors::tests::new_range; use deno_core::serde_json::json; #[test] fn test_as_queue_and_filters() { let specifier = ModuleSpecifier::parse("file:///a/file.ts").unwrap(); // Regression test for https://github.com/denoland/vscode_deno/issues/890. let non_test_specifier = ModuleSpecifier::parse("file:///a/no_tests.ts").unwrap(); let params = lsp_custom::TestRunRequestParams { id: 1, kind: lsp_custom::TestRunKind::Run, include: Some(vec![ lsp_custom::TestIdentifier { text_document: lsp::TextDocumentIdentifier { uri: specifier.clone(), }, id: None, step_id: None, }, lsp_custom::TestIdentifier { text_document: lsp::TextDocumentIdentifier { uri: non_test_specifier.clone(), }, id: None, step_id: None, }, ]), exclude: vec![lsp_custom::TestIdentifier { text_document: lsp::TextDocumentIdentifier { uri: specifier.clone(), }, id: Some( "69d9fe87f64f5b66cb8b631d4fd2064e8224b8715a049be54276c42189ff8f9f" .to_string(), ), step_id: None, }], }; let mut tests = HashMap::new(); let test_def_a = TestDefinition { id: "0b7c6bf3cd617018d33a1bf982a08fe088c5bb54fcd5eb9e802e7c137ec1af94" .to_string(), name: "test a".to_string(), range: Some(new_range(1, 5, 1, 9)), is_dynamic: false, parent_id: None, step_ids: Default::default(), }; let test_def_b = TestDefinition { id: "69d9fe87f64f5b66cb8b631d4fd2064e8224b8715a049be54276c42189ff8f9f" .to_string(), name: "test b".to_string(), range: Some(new_range(2, 5, 2, 9)), is_dynamic: false, parent_id: None, step_ids: Default::default(), }; let test_module = TestModule { specifier: specifier.clone(), script_version: "1".to_string(), defs: vec![ (test_def_a.id.clone(), test_def_a.clone()), (test_def_b.id.clone(), test_def_b.clone()), ] .into_iter() .collect(), }; tests.insert(specifier.clone(), test_module.clone()); tests.insert( non_test_specifier.clone(), TestModule::new(non_test_specifier, "1".to_string()), ); let (queue, filters) = as_queue_and_filters(¶ms, &tests); assert_eq!(json!(queue), json!([specifier])); let mut exclude = HashMap::new(); exclude.insert( "69d9fe87f64f5b66cb8b631d4fd2064e8224b8715a049be54276c42189ff8f9f" .to_string(), test_def_b, ); let maybe_filter = filters.get(&specifier); assert!(maybe_filter.is_some()); let filter = maybe_filter.unwrap(); assert_eq!( filter, &LspTestFilter { include: None, exclude, } ); assert_eq!( filter.as_ids(&test_module), vec![ "0b7c6bf3cd617018d33a1bf982a08fe088c5bb54fcd5eb9e802e7c137ec1af94" .to_string() ] ); } }