mirror of
https://github.com/l0ng-ai/openidea.git
synced 2026-11-07 00:07:29 +00:00
Projects now open with a user-owned trust decision. The prompt offers Preview in Restricted Mode, Don't Open and Trust Project, with an optional trusted parent folder; preview is the default focus so Enter never grants trust. Restricted projects keep browsing, editing, saving, search and highlighting. Project-controlled execution is denied by default in the backend: run, debug, build, tools, language servers, plugins and Git commands that could honour repository configuration. Trusting a project enables those paths without reopening it, and decisions can be revoked in Settings. Terminals stay available when the user opens one, without project shell settings, environment or virtualenv activation. A shell prompt that runs git in the folder can still honour the repository's core.fsmonitor. Fixes #462
606 lines
26 KiB
Rust
606 lines
26 KiB
Rust
#![cfg(unix)]
|
||
|
||
use ide_core::{IdeEvent, RunId, SharedSink};
|
||
use ide_run::{OutputChunk, ProcessManager, RunConfig, StartOptions};
|
||
use parking_lot::Mutex;
|
||
use std::collections::BTreeMap;
|
||
use std::path::{Path, PathBuf};
|
||
use std::sync::Arc;
|
||
use std::sync::mpsc::{self, Receiver};
|
||
use std::time::{Duration, Instant};
|
||
|
||
struct Harness {
|
||
pm: ProcessManager,
|
||
sink: SharedSink,
|
||
events: Receiver<IdeEvent>,
|
||
root: tempfile::TempDir,
|
||
}
|
||
|
||
fn channel_sink() -> (SharedSink, Receiver<IdeEvent>) {
|
||
let (tx, events) = mpsc::channel();
|
||
let tx = Mutex::new(tx);
|
||
let sink: SharedSink = Arc::new(move |e: IdeEvent| {
|
||
let _ = tx.lock().send(e);
|
||
});
|
||
(sink, events)
|
||
}
|
||
|
||
fn harness() -> Harness {
|
||
let (sink, events) = channel_sink();
|
||
Harness { pm: ProcessManager::new(), sink, events, root: tempfile::tempdir().unwrap() }
|
||
}
|
||
|
||
fn cfg(id: &str, command: &str, args: &[&str]) -> RunConfig {
|
||
RunConfig {
|
||
id: id.into(),
|
||
name: id.into(),
|
||
kind: "shell".into(),
|
||
command: command.into(),
|
||
args: args.iter().map(|s| (*s).to_owned()).collect(),
|
||
cwd: None,
|
||
env: BTreeMap::new(),
|
||
detected: false,
|
||
is_test: false,
|
||
debug_adapter: None,
|
||
allow_parallel: false,
|
||
extras: Default::default(),
|
||
}
|
||
}
|
||
|
||
type Chunks = Arc<Mutex<Vec<OutputChunk>>>;
|
||
|
||
fn start(h: &Harness, config: &RunConfig) -> (RunId, Chunks) {
|
||
let chunks: Chunks = Arc::default();
|
||
let sink = chunks.clone();
|
||
let info = h
|
||
.pm
|
||
.start(h.root.path(), config, &StartOptions::default(), h.sink.clone(), Box::new(move |c| sink.lock().push(c)))
|
||
.expect("start");
|
||
assert!(info.running);
|
||
assert_eq!(info.config_id, config.id);
|
||
(info.id, chunks)
|
||
}
|
||
|
||
fn wait_exit(h: &Harness, id: RunId, timeout: Duration) -> Option<i32> {
|
||
let deadline = Instant::now() + timeout;
|
||
loop {
|
||
let left = deadline.saturating_duration_since(Instant::now());
|
||
match h.events.recv_timeout(left).expect("RunExited event") {
|
||
IdeEvent::RunExited { run, code } if run == id => return code,
|
||
_ => {}
|
||
}
|
||
}
|
||
}
|
||
|
||
fn text(chunks: &Chunks, pick: fn(&OutputChunk) -> Option<&str>) -> String {
|
||
chunks.lock().iter().filter_map(pick).collect()
|
||
}
|
||
|
||
fn stdout(c: &OutputChunk) -> Option<&str> {
|
||
if let OutputChunk::Stdout { text } = c { Some(text) } else { None }
|
||
}
|
||
fn stderr(c: &OutputChunk) -> Option<&str> {
|
||
if let OutputChunk::Stderr { text } = c { Some(text) } else { None }
|
||
}
|
||
fn system(c: &OutputChunk) -> Option<&str> {
|
||
if let OutputChunk::System { text } = c { Some(text) } else { None }
|
||
}
|
||
|
||
fn wait_for(chunks: &Chunks, needle: &str) -> String {
|
||
let deadline = Instant::now() + Duration::from_secs(10);
|
||
loop {
|
||
let out = text(chunks, stdout);
|
||
if out.contains(needle) {
|
||
return out;
|
||
}
|
||
assert!(Instant::now() < deadline, "timed out waiting for {needle:?}; got {out:?}");
|
||
std::thread::sleep(Duration::from_millis(10));
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn output_streams_and_exit_code() {
|
||
let h = harness();
|
||
let config = cfg("t:out", "/bin/sh", &["-c", "printf 'out \\344\\270\\255'; printf 'err' >&2; exit 7"]);
|
||
let (id, chunks) = start(&h, &config);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(7));
|
||
|
||
let all = chunks.lock().clone();
|
||
match &all[0] {
|
||
OutputChunk::System { text } => assert!(text.starts_with("/bin/sh -c 'printf"), "{text}"),
|
||
other => panic!("first chunk must echo the command, got {other:?}"),
|
||
}
|
||
assert!(
|
||
matches!(all.last(), Some(OutputChunk::System { text }) if text.contains("Process finished with exit code 7"))
|
||
);
|
||
assert_eq!(text(&chunks, stdout), "out 中");
|
||
assert_eq!(text(&chunks, stderr), "err");
|
||
|
||
let info = h.pm.list().into_iter().find(|r| r.id == id).unwrap();
|
||
assert!(!info.running);
|
||
assert!(h.pm.write_stdin(id, "x").is_err());
|
||
}
|
||
|
||
#[test]
|
||
fn shell_command_string_and_env_and_cwd() {
|
||
let h = harness();
|
||
std::fs::create_dir(h.root.path().join("sub")).unwrap();
|
||
let mut config = cfg("t:shell", "echo \"$OPENIDEA_VAR\" && pwd | tr -d '\\n'", &[]);
|
||
config.env.insert("OPENIDEA_VAR".into(), "hello".into());
|
||
config.cwd = Some(PathBuf::from("sub").into());
|
||
let (id, chunks) = start(&h, &config);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(0));
|
||
let out = text(&chunks, stdout);
|
||
let mut lines = out.lines();
|
||
assert_eq!(lines.next(), Some("hello"));
|
||
let cwd = Path::new(lines.next().unwrap()).canonicalize().unwrap();
|
||
assert_eq!(cwd, h.root.path().join("sub").canonicalize().unwrap());
|
||
assert!(text(&chunks, system).starts_with("echo \"$OPENIDEA_VAR\""));
|
||
}
|
||
|
||
#[test]
|
||
fn stdin_echo_with_cat() {
|
||
let h = harness();
|
||
let (id, chunks) = start(&h, &cfg("t:cat", "cat", &[]));
|
||
h.pm.write_stdin(id, "ping é\n").unwrap();
|
||
wait_for(&chunks, "ping é\n");
|
||
h.pm.write_stdin(id, "pong\n").unwrap();
|
||
wait_for(&chunks, "pong\n");
|
||
h.pm.close_stdin(id).unwrap();
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(0));
|
||
}
|
||
|
||
#[test]
|
||
fn missing_command_and_cwd_are_errors() {
|
||
let h = harness();
|
||
let err =
|
||
h.pm.start(
|
||
h.root.path(),
|
||
&cfg("t:nope", "openidea-no-such-binary-xyz", &[]),
|
||
&StartOptions::default(),
|
||
h.sink.clone(),
|
||
Box::new(|_| {}),
|
||
)
|
||
.unwrap_err();
|
||
assert_eq!(err.kind, ide_core::error::ErrorKind::Unavailable);
|
||
let mut config = cfg("t:cwd", "true", &[]);
|
||
config.cwd = Some("missing-dir".into());
|
||
assert!(h.pm.start(h.root.path(), &config, &StartOptions::default(), h.sink.clone(), Box::new(|_| {})).is_err());
|
||
assert!(
|
||
h.pm.start(
|
||
h.root.path(),
|
||
&cfg("t:empty", " ", &[]),
|
||
&StartOptions::default(),
|
||
h.sink.clone(),
|
||
Box::new(|_| {})
|
||
)
|
||
.is_err()
|
||
);
|
||
}
|
||
|
||
#[test]
|
||
fn stop_interrupts_process() {
|
||
let h = harness();
|
||
let (id, chunks) = start(&h, &cfg("t:sleep", "sleep", &["30"]));
|
||
let started = Instant::now();
|
||
h.pm.stop(id).unwrap();
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(130));
|
||
assert!(started.elapsed() < Duration::from_secs(2));
|
||
assert!(text(&chunks, system).contains("Process finished with exit code 130 (interrupted by signal 2:SIGINT)"));
|
||
// Stopping a finished run is a no-op.
|
||
h.pm.stop(id).unwrap();
|
||
}
|
||
|
||
fn alive(pid: i32) -> bool {
|
||
unsafe { libc::kill(pid, 0) == 0 }
|
||
}
|
||
|
||
#[test]
|
||
fn stop_kills_whole_process_group() {
|
||
let h = harness();
|
||
// Background jobs of a non-interactive shell ignore SIGINT, so this needs the SIGKILL escalation.
|
||
let (id, chunks) = start(&h, &cfg("t:group", "/bin/sh", &["-c", "sleep 30 & echo $!; wait"]));
|
||
let out = wait_for(&chunks, "\n");
|
||
let grandchild: i32 = out.trim().parse().expect("pid");
|
||
assert!(alive(grandchild));
|
||
|
||
h.pm.stop(id).unwrap();
|
||
wait_exit(&h, id, Duration::from_secs(10));
|
||
let deadline = Instant::now() + Duration::from_secs(8);
|
||
while alive(grandchild) {
|
||
assert!(Instant::now() < deadline, "grandchild {grandchild} survived stop");
|
||
std::thread::sleep(Duration::from_millis(50));
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn list_prunes_finished_runs_of_same_config() {
|
||
let h = harness();
|
||
let config = cfg("t:list", "true", &[]);
|
||
let (first, _) = start(&h, &config);
|
||
wait_exit(&h, first, Duration::from_secs(10));
|
||
assert!(h.pm.list().iter().any(|r| r.id == first && !r.running));
|
||
|
||
let (second, _) = start(&h, &cfg("t:list", "sleep", &["5"]));
|
||
let list = h.pm.list();
|
||
assert!(list.iter().all(|r| r.id != first));
|
||
assert!(list.iter().any(|r| r.id == second && r.running && r.pid.is_some()));
|
||
h.pm.stop(second).unwrap();
|
||
wait_exit(&h, second, Duration::from_secs(10));
|
||
}
|
||
|
||
#[test]
|
||
fn large_output_is_complete_and_batched() {
|
||
let h = harness();
|
||
let script = "i=0; while [ $i -lt 3000 ]; do printf '%05d-\\342\\224\\200\\n' $i; i=$((i+1)); done";
|
||
let (id, chunks) = start(&h, &cfg("t:big", "/bin/sh", &["-c", script]));
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(20)), Some(0));
|
||
let out = text(&chunks, stdout);
|
||
assert_eq!(out.lines().count(), 3000);
|
||
assert!(!out.contains('\u{FFFD}'));
|
||
assert!(chunks.lock().len() < 3000);
|
||
}
|
||
|
||
#[test]
|
||
fn relative_program_resolves_against_cwd() {
|
||
let h = harness();
|
||
let script = h.root.path().join("tool.sh");
|
||
std::fs::write(&script, "#!/bin/sh\necho tool-ran\n").unwrap();
|
||
use std::os::unix::fs::PermissionsExt;
|
||
std::fs::set_permissions(&script, std::fs::Permissions::from_mode(0o755)).unwrap();
|
||
let (id, chunks) = start(&h, &cfg("t:rel", "./tool.sh", &[]));
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(0));
|
||
assert_eq!(text(&chunks, stdout), "tool-ran\n");
|
||
}
|
||
|
||
#[test]
|
||
fn exit_event_goes_to_the_starters_sink_only() {
|
||
let h = harness();
|
||
let (other_sink, other_events) = channel_sink();
|
||
let (mine, _) = start(&h, &cfg("t:mine", "true", &[]));
|
||
let info =
|
||
h.pm.start(h.root.path(), &cfg("t:other", "true", &[]), &StartOptions::default(), other_sink, Box::new(|_| {}))
|
||
.unwrap();
|
||
match h.events.recv_timeout(Duration::from_secs(10)).expect("RunExited for the starter") {
|
||
IdeEvent::RunExited { run, code } => assert_eq!((run, code), (mine, Some(0))),
|
||
e => panic!("unexpected {e:?}"),
|
||
}
|
||
match other_events.recv_timeout(Duration::from_secs(10)).expect("RunExited for the other sink") {
|
||
IdeEvent::RunExited { run, code } => assert_eq!((run, code), (info.id, Some(0))),
|
||
e => panic!("unexpected {e:?}"),
|
||
}
|
||
// Neither sink sees the other's run.
|
||
std::thread::sleep(Duration::from_millis(100));
|
||
assert!(other_events.try_recv().is_err());
|
||
assert!(h.events.try_recv().is_err());
|
||
}
|
||
|
||
#[test]
|
||
fn env_files_system_env_and_redirects() {
|
||
let h = harness();
|
||
let root = h.root.path();
|
||
std::fs::write(root.join(".env"), "FROM_FILE=file\nOVERRIDDEN=file\n").unwrap();
|
||
std::fs::write(root.join("input.txt"), "from file\n").unwrap();
|
||
let mut config = cfg(
|
||
"t:extras",
|
||
"/bin/sh",
|
||
&["-c", "printf '%s|%s|%s|' \"$FROM_FILE\" \"$OVERRIDDEN\" \"${HOME:-unset}\"; cat"],
|
||
);
|
||
config.env.insert("OVERRIDDEN".into(), "config".into());
|
||
config.extras.env_files = vec![".env".into()];
|
||
config.extras.include_system_env = false;
|
||
config.extras.stdin_file = Some("input.txt".into());
|
||
config.extras.output_file = Some("logs/out.log".into());
|
||
let prepared = ide_run::resolved_env(root, &config).unwrap();
|
||
assert_eq!(prepared.get("FROM_FILE").map(String::as_str), Some("file"));
|
||
assert_eq!(prepared.get("OVERRIDDEN").map(String::as_str), Some("config"));
|
||
assert!(!prepared.contains_key("HOME"));
|
||
let (id, chunks) = start(&h, &config);
|
||
// stdin comes from the file: writing to it is refused.
|
||
assert!(h.pm.write_stdin(id, "x").is_err());
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(0));
|
||
assert_eq!(text(&chunks, stdout), "file|config|unset|from file\n");
|
||
let saved = std::fs::read_to_string(root.join("logs/out.log")).unwrap();
|
||
assert!(saved.contains("file|config|unset|from file\n"), "{saved}");
|
||
assert!(saved.contains("Process finished with exit code 0"), "{saved}");
|
||
|
||
// With the system environment, HOME is inherited.
|
||
config.extras.include_system_env = true;
|
||
assert!(ide_run::resolved_env(root, &config).unwrap().contains_key("HOME"));
|
||
config.extras.stdin_file = None;
|
||
config.extras.output_file = None;
|
||
config.args = vec!["-c".into(), "printf '%s' \"${HOME:+set}\"".into()];
|
||
let (id, chunks) = start(&h, &config);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(0));
|
||
assert_eq!(text(&chunks, stdout), "set");
|
||
|
||
config.extras.env_files = vec!["missing.env".into()];
|
||
assert!(h.pm.start(root, &config, &StartOptions::default(), h.sink.clone(), Box::new(|_| {})).is_err());
|
||
}
|
||
|
||
#[test]
|
||
fn emulated_terminal_is_a_tty_with_input_resize_and_exit_code() {
|
||
let h = harness();
|
||
let script = "[ -t 1 ] && echo tty-yes; echo \"term=$TERM\"; read line; echo \"got:$line\"; stty size; exit 3";
|
||
let mut config = cfg("t:pty", "/bin/sh", &["-c", script]);
|
||
config.extras.emulate_terminal = true;
|
||
let (id, chunks) = start(&h, &config);
|
||
assert!(h.pm.list().iter().any(|r| r.id == id && r.pty));
|
||
wait_for(&chunks, "term=xterm-256color");
|
||
h.pm.resize(id, 100, 40).unwrap();
|
||
h.pm.write_stdin(id, "hello\r").unwrap();
|
||
let out = wait_for(&chunks, "40 100");
|
||
assert!(out.contains("tty-yes"), "{out:?}");
|
||
assert!(out.contains("got:hello"), "{out:?}");
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(3));
|
||
assert!(text(&chunks, system).contains("Process finished with exit code 3"));
|
||
// Stdout and stderr share the terminal.
|
||
assert_eq!(text(&chunks, stderr), "");
|
||
}
|
||
|
||
#[test]
|
||
fn emulated_terminal_stop_reports_the_signal() {
|
||
let h = harness();
|
||
let mut config = cfg("t:pty-stop", "/bin/sh", &["-c", "echo ready; sleep 30"]);
|
||
config.extras.emulate_terminal = true;
|
||
let (id, chunks) = start(&h, &config);
|
||
wait_for(&chunks, "ready");
|
||
h.pm.stop(id).unwrap();
|
||
let code = wait_exit(&h, id, Duration::from_secs(10));
|
||
assert_eq!(code, Some(130), "{:?}", text(&chunks, system));
|
||
assert!(text(&chunks, system).contains("SIGINT"));
|
||
}
|
||
|
||
#[test]
|
||
fn emulated_terminal_refuses_input_redirection() {
|
||
let h = harness();
|
||
let mut config = cfg("t:pty-stdin", "/bin/cat", &[]);
|
||
config.extras.emulate_terminal = true;
|
||
config.extras.stdin_file = Some("input.txt".into());
|
||
assert!(h.pm.start(h.root.path(), &config, &StartOptions::default(), h.sink.clone(), Box::new(|_| {})).is_err());
|
||
}
|
||
|
||
#[test]
|
||
fn pipe_runs_cannot_be_resized() {
|
||
let h = harness();
|
||
let (id, _chunks) = start(&h, &cfg("t:pipes", "/bin/sh", &["-c", "sleep 5"]));
|
||
assert!(!h.pm.list().iter().find(|r| r.id == id).unwrap().pty);
|
||
assert!(h.pm.resize(id, 80, 24).is_err());
|
||
h.pm.stop(id).unwrap();
|
||
wait_exit(&h, id, Duration::from_secs(10));
|
||
}
|
||
|
||
#[test]
|
||
fn stop_command_runs_first_with_the_run_environment() {
|
||
let h = harness();
|
||
let mut config = cfg("t:stop-cmd", "/bin/sh", &["-c", "echo ready; sleep 30"]);
|
||
config.env.insert("OPENIDEA_MARKER".into(), "stopped".into());
|
||
config.extras.stop_command = Some("touch \"$OPENIDEA_MARKER\"".into());
|
||
let (id, chunks) = start(&h, &config);
|
||
wait_for(&chunks, "ready");
|
||
h.pm.stop_with_grace(id, Duration::from_millis(200)).unwrap();
|
||
// `sleep` ignores nothing: the signals that follow the stop command end it.
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(130));
|
||
assert!(h.root.path().join("stopped").exists(), "stop command did not run");
|
||
assert!(text(&chunks, system).contains("touch \"$OPENIDEA_MARKER\""));
|
||
}
|
||
|
||
/// A `java` command line past the platform limit starts through a pathing jar
|
||
/// (JDK 8) or an `@argfile` (JDK 9+), which is deleted once the JVM exits.
|
||
#[test]
|
||
fn long_java_classpath_is_shortened() {
|
||
let Some(home) = std::env::var_os("JAVA_HOME").map(PathBuf::from) else { return };
|
||
let (java, javac) = (home.join("bin/java"), home.join("bin/javac"));
|
||
if !javac.is_file() {
|
||
return;
|
||
}
|
||
let h = harness();
|
||
let classes = h.root.path().join("out classes");
|
||
std::fs::create_dir_all(classes.join("p")).unwrap();
|
||
let src = h.root.path().join("Main.java");
|
||
std::fs::write(&src, "package p; public class Main { public static void main(String[] a) { System.out.println(\"hello \" + a[0]); } }").unwrap();
|
||
let ok = std::process::Command::new(&javac).arg("-d").arg(&classes).arg(&src).status().unwrap().success();
|
||
assert!(ok, "javac failed");
|
||
let mut cp: Vec<String> = (0..3_000).map(|i| format!("/no/such/repository/lib-{i}/1.0/lib-{i}-1.0.jar")).collect();
|
||
cp.push(classes.to_string_lossy().into_owned());
|
||
let cp = cp.join(":");
|
||
assert!(cp.len() > ide_run::java_cmdline::UNIX_LIMIT);
|
||
let config = cfg("t:java", &java.to_string_lossy(), &["-cp", &cp, "p.Main", "world"]);
|
||
let (id, chunks) = start(&h, &config);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(30)), Some(0), "{}", text(&chunks, stderr));
|
||
assert_eq!(text(&chunks, stdout).trim(), "hello world");
|
||
let echo = text(&chunks, system);
|
||
let first = echo.lines().next().unwrap();
|
||
assert!(first.len() < 1_000, "{first}");
|
||
let file = first
|
||
.split(' ')
|
||
.map(|a| a.trim_matches('\'').trim_start_matches('@'))
|
||
.find(|a| a.contains("openidea-classpath"))
|
||
.unwrap_or_else(|| panic!("no pathing jar / argfile in {first}"));
|
||
assert!(!Path::new(file).exists(), "{file} not deleted");
|
||
}
|
||
|
||
/// A build tool (here a fake `mvn` that forks like `spring-boot:run`) gets the console
|
||
/// encoding in `JAVA_TOOL_OPTIONS`, and the notice its JVMs print about it is hidden.
|
||
#[test]
|
||
fn build_tools_get_the_console_encoding_for_every_jvm() {
|
||
use std::os::unix::fs::PermissionsExt;
|
||
let h = harness();
|
||
let mvn = h.root.path().join("mvn");
|
||
let script = "#!/bin/sh\nnotice() { echo \"Picked up JAVA_TOOL_OPTIONS: $JAVA_TOOL_OPTIONS\" >&2; }\n\
|
||
notice; echo \"maven $JAVA_TOOL_OPTIONS\"; notice 2>&1; printf 'Consume Time\\243\\272 1 ms\\n' >&2\n";
|
||
std::fs::write(&mvn, script).unwrap();
|
||
std::fs::set_permissions(&mvn, std::fs::Permissions::from_mode(0o755)).unwrap();
|
||
let chunks: Chunks = Arc::default();
|
||
let sink = chunks.clone();
|
||
let options = StartOptions { encoding: Some("GBK".into()), ..Default::default() };
|
||
let config = cfg("t:mvn", mvn.to_str().unwrap(), &["spring-boot:run"]);
|
||
let info =
|
||
h.pm.start(h.root.path(), &config, &options, h.sink.clone(), Box::new(move |c| sink.lock().push(c))).unwrap();
|
||
wait_exit(&h, info.id, Duration::from_secs(10));
|
||
assert_eq!(
|
||
text(&chunks, stdout),
|
||
"maven -Dstdout.encoding=GBK -Dstderr.encoding=GBK -Dsun.stdout.encoding=GBK -Dsun.stderr.encoding=GBK\n"
|
||
);
|
||
assert_eq!(text(&chunks, stderr), "Consume Time: 1 ms\n");
|
||
}
|
||
|
||
#[test]
|
||
fn python_scripts_import_sibling_packages_and_preserve_env_file_paths() {
|
||
for (kind, source_dir) in [("python", ""), ("pytest", "src")] {
|
||
let h = harness();
|
||
let root = h.root.path();
|
||
let source = root.join(source_dir);
|
||
for dir in [root.join("app"), source.join("helpers"), root.join("vendor")] {
|
||
std::fs::create_dir_all(dir).unwrap();
|
||
}
|
||
std::fs::write(source.join("helpers/__init__.py"), "def hello(): return 'import OK'\n").unwrap();
|
||
std::fs::write(root.join("vendor/custom.py"), "VALUE = 'vendor OK'\n").unwrap();
|
||
std::fs::write(
|
||
root.join("app/main.py"),
|
||
"from helpers import hello\nfrom custom import VALUE\nprint(hello(), VALUE)\n",
|
||
)
|
||
.unwrap();
|
||
std::fs::write(root.join(".env"), format!("PYTHONPATH={}\n", root.join("vendor").display())).unwrap();
|
||
let mut config = cfg("python-imports", "python3", &["app/main.py"]);
|
||
config.kind = kind.into();
|
||
config.extras.env_files = vec![".env".into()];
|
||
let options = StartOptions { python_paths: vec![root.to_owned(), source], ..Default::default() };
|
||
let chunks: Chunks = Arc::default();
|
||
let sink = chunks.clone();
|
||
let info = h.pm.start(root, &config, &options, h.sink.clone(), Box::new(move |c| sink.lock().push(c))).unwrap();
|
||
assert_eq!(wait_exit(&h, info.id, Duration::from_secs(10)), Some(0), "{}", text(&chunks, stderr));
|
||
assert_eq!(text(&chunks, stdout), "import OK vendor OK\n");
|
||
}
|
||
}
|
||
|
||
#[test]
|
||
fn revoking_one_project_stops_only_its_runs() {
|
||
let h = harness();
|
||
let other = tempfile::tempdir().unwrap();
|
||
let (mine, _) = start(&h, &cfg("mine", "/bin/sh", &["-c", "sleep 30"]));
|
||
let theirs =
|
||
h.pm.start(
|
||
other.path(),
|
||
&cfg("theirs", "/bin/sh", &["-c", "sleep 30"]),
|
||
&StartOptions::default(),
|
||
h.sink.clone(),
|
||
Box::new(|_| {}),
|
||
)
|
||
.unwrap();
|
||
h.pm.kill_project(h.root.path());
|
||
wait_exit(&h, mine, Duration::from_secs(5));
|
||
assert!(h.pm.list().iter().any(|run| run.id == theirs.id && run.running));
|
||
h.pm.stop(theirs.id).unwrap();
|
||
}
|
||
|
||
#[test]
|
||
#[ignore = "requires CMake, CTest and a C++ compiler"]
|
||
fn cmake_configure_discovers_builds_and_runs_a_multifile_target() {
|
||
let h = harness();
|
||
std::fs::write(h.root.path().join("CMakeLists.txt"), "cmake_minimum_required(VERSION 3.20)\nproject(smoke LANGUAGES CXX)\nadd_executable(smoke main.cpp math.cpp)\nenable_testing()\nadd_test(NAME smoke COMMAND smoke)\n").unwrap();
|
||
std::fs::write(h.root.path().join("main.cpp"), "int add(int,int); int main() { return add(2,3)==5 ? 0 : 1; }\n")
|
||
.unwrap();
|
||
std::fs::write(h.root.path().join("math.cpp"), "int add(int a,int b) { return a+b; }\n").unwrap();
|
||
let initial = ide_run::detect(h.root.path());
|
||
assert!(!h.root.path().join("build").exists(), "detection must not modify the project");
|
||
let configure = initial.iter().find(|c| c.id.ends_with(":configure")).unwrap();
|
||
let (id, output) = start(&h, configure);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(60)), Some(0), "{}", text(&output, stderr));
|
||
assert!(h.root.path().join("build/.cmake/api/v1/query/client-openidea/codemodel-v2").exists());
|
||
let root = h.root.path().canonicalize().unwrap();
|
||
let configs = ide_run::detect(&root);
|
||
let target = configs.iter().find(|c| c.kind == "native").expect("File API executable");
|
||
assert_eq!(target.debug_adapter.as_deref(), Some("lldb-dap"));
|
||
let sources = &target.extras.params["cmakeSources"];
|
||
assert!(sources.lines().any(|s| s.ends_with("/main.cpp")));
|
||
assert!(sources.lines().any(|s| s.ends_with("/math.cpp")));
|
||
assert!(sources.lines().any(|s| s == root.join("main.cpp").to_string_lossy()));
|
||
assert_eq!(target.extras.params["cmakeCanonicalSourceRoot"], root.to_string_lossy());
|
||
let build = configs.iter().find(|c| c.id == target.extras.before_launch[0].target).unwrap();
|
||
let (id, output) = start(&h, build);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(60)), Some(0), "{}", text(&output, stderr));
|
||
let (id, _) = start(&h, target);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(0));
|
||
let ctest = configs.iter().find(|c| c.command == "ctest").unwrap();
|
||
let (id, output) = start(&h, ctest);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(10)), Some(0), "{}", text(&output, stdout));
|
||
}
|
||
|
||
#[test]
|
||
fn go_build_runs_with_separate_arguments_environment_and_working_directory() {
|
||
let h = harness();
|
||
let Some(go) = ide_run::which::find_executable("go", &std::env::var_os("PATH").unwrap_or_default()) else {
|
||
eprintln!("skipped: no Go toolchain");
|
||
return;
|
||
};
|
||
let module = h.root.path().join("module");
|
||
std::fs::create_dir_all(&module).unwrap();
|
||
std::fs::write(module.join("go.mod"), "module example.com/demo\ngo 1.22\n").unwrap();
|
||
std::fs::write(module.join("main.go"), "//go:build demo\n\npackage main\nimport (\"fmt\"; \"os\")\nfunc main(){cwd,_:=os.Getwd();fmt.Printf(\"%q %s %s\\n\",os.Args[1:],os.Getenv(\"GO_RUN_VALUE\"),cwd)}\n").unwrap();
|
||
let mut c = cfg("go-config", &go.to_string_lossy(), &["build", "-tags", "demo", "."]);
|
||
c.kind = "go".into();
|
||
c.env.insert("GO_RUN_VALUE".into(), "configured".into());
|
||
c.extras.params = BTreeMap::from([
|
||
("goKind".into(), "build".into()),
|
||
("scope".into(), "package".into()),
|
||
("target".into(), ".".into()),
|
||
("module".into(), "module".into()),
|
||
("outputDir".into(), "bin with spaces".into()),
|
||
("runAfterBuild".into(), "true".into()),
|
||
("programArgv".into(), r#"["-tags","two words"]"#.into()),
|
||
]);
|
||
let (id, out) = start(&h, &c);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(60)), Some(0), "{}", text(&out, stderr));
|
||
let output = text(&out, stdout);
|
||
assert!(output.contains(r#"["-tags" "two words"] configured"#), "{output}");
|
||
assert!(output.contains(&h.root.path().display().to_string()), "{output}");
|
||
assert!(std::fs::read_dir(h.root.path().join("bin with spaces")).unwrap().next().is_some());
|
||
|
||
c.extras.params.insert("runAfterBuild".into(), "false".into());
|
||
let (id, out) = start(&h, &c);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(60)), Some(0));
|
||
assert!(text(&out, stdout).is_empty());
|
||
|
||
c.extras.params.insert("runAfterBuild".into(), "true".into());
|
||
c.args = vec!["build".into(), "missing.go".into()];
|
||
let (id, out) = start(&h, &c);
|
||
assert_ne!(wait_exit(&h, id, Duration::from_secs(60)), Some(0));
|
||
assert!(text(&out, stdout).is_empty(), "a failed build must never run an older binary");
|
||
}
|
||
|
||
#[test]
|
||
fn go_file_tests_use_build_selected_sources_and_only_the_selected_test_file() {
|
||
let h = harness();
|
||
let Some(go) = ide_run::which::find_executable("go", &std::env::var_os("PATH").unwrap_or_default()) else {
|
||
eprintln!("skipped: no Go toolchain");
|
||
return;
|
||
};
|
||
std::fs::write(h.root.path().join("go.mod"), "module example.com/filetests\ngo 1.22\n").unwrap();
|
||
std::fs::write(h.root.path().join("main.go"), "package filetests\nfunc value() int { return 2 }\n").unwrap();
|
||
std::fs::write(h.root.path().join("generate.go"), "//go:build ignore\n\npackage main\nfunc main() {}\n").unwrap();
|
||
std::fs::write(
|
||
h.root.path().join("demo_test.go"),
|
||
"package filetests\nimport \"testing\"\nfunc TestChosen(t *testing.T) { if value()!=2 { t.Fatal(\"bad\") } }\n",
|
||
)
|
||
.unwrap();
|
||
std::fs::write(
|
||
h.root.path().join("other_test.go"),
|
||
"package filetests\nimport \"testing\"\nfunc TestOther(t *testing.T) { t.Fatal(\"must not run\") }\n",
|
||
)
|
||
.unwrap();
|
||
let mut c = cfg("go-file", &go.to_string_lossy(), &["test", "-json", "demo_test.go"]);
|
||
c.kind = "go".into();
|
||
c.extras.params = BTreeMap::from([
|
||
("goKind".into(), "test".into()),
|
||
("scope".into(), "file".into()),
|
||
("target".into(), "demo_test.go".into()),
|
||
]);
|
||
ide_run::go::prepare_test_file(h.root.path(), &mut c).unwrap();
|
||
assert!(!c.args.iter().any(|a| a.ends_with("generate.go") || a.ends_with("other_test.go")));
|
||
let (id, out) = start(&h, &c);
|
||
assert_eq!(wait_exit(&h, id, Duration::from_secs(60)), Some(0), "{}", text(&out, stderr));
|
||
assert!(text(&out, stdout).contains("TestChosen"));
|
||
assert!(!text(&out, stdout).contains("TestOther"));
|
||
}
|