riir
This commit is contained in:
atagen
parent
da4bc139eb
commit
0aa6ae9fbf
49 changed files with 10554 additions and 5482 deletions
500
tests/runtime_complete.rs
Normal file
500
tests/runtime_complete.rs
Normal file
|
|
@ -0,0 +1,500 @@
|
|||
use std::fs;
|
||||
use std::os::unix::fs::PermissionsExt;
|
||||
use std::process::Command;
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
|
||||
use inshellah::parsers::manpage::{ManpageEntry, ManpageResult, ManpageSubcommand, OwnedSwitch};
|
||||
use inshellah::store::write_result;
|
||||
|
||||
fn unique_temp_dir(name: &str) -> std::path::PathBuf {
|
||||
let nanos = SystemTime::now()
|
||||
.duration_since(UNIX_EPOCH)
|
||||
.expect("system time")
|
||||
.as_nanos();
|
||||
std::env::temp_dir().join(format!("{name}-{}-{nanos}", std::process::id()))
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn complete_scrapes_missing_subcommand_when_parent_is_cached() {
|
||||
let root = unique_temp_dir("inshellah-runtime-complete");
|
||||
let bin_dir = root.join("bin");
|
||||
let cache_dir = root.join("cache");
|
||||
fs::create_dir_all(&bin_dir).expect("bin dir");
|
||||
fs::create_dir_all(&cache_dir).expect("cache dir");
|
||||
|
||||
let fakecmd = bin_dir.join("fakecmd");
|
||||
fs::write(
|
||||
&fakecmd,
|
||||
r#"#!/bin/sh
|
||||
if [ "$1" = "clone" ]; then
|
||||
if [ "$2" = "--help" ] || [ "$2" = "-h" ]; then
|
||||
cat <<'EOF'
|
||||
Usage: fakecmd clone [OPTIONS] <repository> [directory]
|
||||
|
||||
Options:
|
||||
--depth <n> clone depth
|
||||
-v, --verbose verbose
|
||||
EOF
|
||||
exit 0
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$1" = "--help" ] || [ "$1" = "-h" ]; then
|
||||
cat <<'EOF'
|
||||
Usage: fakecmd [OPTIONS] COMMAND
|
||||
|
||||
Commands:
|
||||
clone Clone a repository
|
||||
|
||||
Options:
|
||||
-h, --help show help
|
||||
EOF
|
||||
exit 0
|
||||
fi
|
||||
|
||||
exit 2
|
||||
"#,
|
||||
)
|
||||
.expect("write fakecmd");
|
||||
let mut perms = fs::metadata(&fakecmd).expect("metadata").permissions();
|
||||
perms.set_mode(0o755);
|
||||
fs::set_permissions(&fakecmd, perms).expect("chmod");
|
||||
|
||||
let parent = ManpageResult {
|
||||
entries: Vec::new(),
|
||||
subcommands: vec![ManpageSubcommand {
|
||||
name: "clone".to_string(),
|
||||
desc: "Clone a repository".to_string(),
|
||||
}],
|
||||
positionals: Vec::new(),
|
||||
description: String::new(),
|
||||
};
|
||||
write_result(&cache_dir, "fakecmd", "help", &parent).expect("parent cache");
|
||||
|
||||
let old_path = std::env::var_os("PATH").unwrap_or_default();
|
||||
let output = Command::new(env!("CARGO_BIN_EXE_inshellah"))
|
||||
.arg("complete")
|
||||
.arg("--dir")
|
||||
.arg(&cache_dir)
|
||||
.arg("--timeout-ms")
|
||||
.arg("1000")
|
||||
.arg("fakecmd")
|
||||
.arg("clone")
|
||||
.arg("--")
|
||||
.env(
|
||||
"PATH",
|
||||
format!("{}:{}", bin_dir.display(), old_path.to_string_lossy()),
|
||||
)
|
||||
.output()
|
||||
.expect("run inshellah complete");
|
||||
|
||||
assert!(
|
||||
output.status.success(),
|
||||
"stderr = {}",
|
||||
String::from_utf8_lossy(&output.stderr)
|
||||
);
|
||||
let stdout = String::from_utf8(output.stdout).expect("stdout");
|
||||
assert!(stdout.contains("--depth"), "stdout = {stdout}");
|
||||
assert!(
|
||||
cache_dir.join("fakecmd_clone.json").is_file(),
|
||||
"subcommand cache was not written"
|
||||
);
|
||||
|
||||
let _ = fs::remove_dir_all(root);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn complete_does_not_scan_path_at_command_position() {
|
||||
let root = unique_temp_dir("inshellah-command-position-complete");
|
||||
let bin_dir = root.join("bin");
|
||||
let cache_dir = root.join("cache");
|
||||
fs::create_dir_all(&bin_dir).expect("bin dir");
|
||||
fs::create_dir_all(&cache_dir).expect("cache dir");
|
||||
|
||||
let fake_git = bin_dir.join("git");
|
||||
fs::write(&fake_git, "#!/bin/sh\nexit 0\n").expect("write fake git");
|
||||
let mut perms = fs::metadata(&fake_git).expect("metadata").permissions();
|
||||
perms.set_mode(0o755);
|
||||
fs::set_permissions(&fake_git, perms).expect("chmod");
|
||||
|
||||
let output = Command::new(env!("CARGO_BIN_EXE_inshellah"))
|
||||
.arg("complete")
|
||||
.arg("--dir")
|
||||
.arg(&cache_dir)
|
||||
.arg("gi")
|
||||
.env("PATH", &bin_dir)
|
||||
.output()
|
||||
.expect("run inshellah complete");
|
||||
|
||||
assert!(
|
||||
output.status.success(),
|
||||
"stderr = {}",
|
||||
String::from_utf8_lossy(&output.stderr)
|
||||
);
|
||||
let stdout = String::from_utf8(output.stdout).expect("stdout");
|
||||
assert_eq!(stdout.trim(), "null", "stdout = {stdout}");
|
||||
|
||||
let _ = fs::remove_dir_all(root);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn complete_uses_boundary_aware_fuzzy_ranking() {
|
||||
let root = unique_temp_dir("inshellah-fuzzy-complete");
|
||||
let cache_dir = root.join("cache");
|
||||
fs::create_dir_all(&cache_dir).expect("cache dir");
|
||||
|
||||
let result = ManpageResult {
|
||||
entries: Vec::new(),
|
||||
subcommands: vec![
|
||||
ManpageSubcommand {
|
||||
name: "load".to_string(),
|
||||
desc: "load something".to_string(),
|
||||
},
|
||||
ManpageSubcommand {
|
||||
name: "clone".to_string(),
|
||||
desc: "clone something".to_string(),
|
||||
},
|
||||
],
|
||||
positionals: Vec::new(),
|
||||
description: String::new(),
|
||||
};
|
||||
write_result(&cache_dir, "demo", "help", &result).expect("cache");
|
||||
|
||||
let output = Command::new(env!("CARGO_BIN_EXE_inshellah"))
|
||||
.arg("complete")
|
||||
.arg("--dir")
|
||||
.arg(&cache_dir)
|
||||
.arg("demo")
|
||||
.arg("lo")
|
||||
.output()
|
||||
.expect("run inshellah complete");
|
||||
|
||||
assert!(
|
||||
output.status.success(),
|
||||
"stderr = {}",
|
||||
String::from_utf8_lossy(&output.stderr)
|
||||
);
|
||||
let stdout = String::from_utf8(output.stdout).expect("stdout");
|
||||
let load_pos = stdout.find(r#""value":"load""#).unwrap_or(usize::MAX);
|
||||
let clone_pos = stdout.find(r#""value":"clone""#).unwrap_or(usize::MAX);
|
||||
assert!(
|
||||
load_pos < clone_pos,
|
||||
"expected boundary match to outrank substring match, stdout = {stdout}"
|
||||
);
|
||||
|
||||
let _ = fs::remove_dir_all(root);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn complete_returns_flags_only_after_hyphen() {
|
||||
let root = unique_temp_dir("inshellah-flag-prefix-complete");
|
||||
let cache_dir = root.join("cache");
|
||||
fs::create_dir_all(&cache_dir).expect("cache dir");
|
||||
|
||||
let result = ManpageResult {
|
||||
entries: vec![ManpageEntry {
|
||||
switch: OwnedSwitch::Long("verbose".to_string()),
|
||||
param: None,
|
||||
desc: "verbose output".to_string(),
|
||||
}],
|
||||
subcommands: Vec::new(),
|
||||
positionals: Vec::new(),
|
||||
description: String::new(),
|
||||
};
|
||||
write_result(&cache_dir, "demo", "help", &result).expect("cache");
|
||||
|
||||
let argument_output = Command::new(env!("CARGO_BIN_EXE_inshellah"))
|
||||
.arg("complete")
|
||||
.arg("--dir")
|
||||
.arg(&cache_dir)
|
||||
.arg("demo")
|
||||
.arg("")
|
||||
.output()
|
||||
.expect("run inshellah complete");
|
||||
assert!(
|
||||
argument_output.status.success(),
|
||||
"stderr = {}",
|
||||
String::from_utf8_lossy(&argument_output.stderr)
|
||||
);
|
||||
let argument_stdout = String::from_utf8(argument_output.stdout).expect("stdout");
|
||||
assert_eq!(argument_stdout.trim(), "null", "stdout = {argument_stdout}");
|
||||
|
||||
let flag_output = Command::new(env!("CARGO_BIN_EXE_inshellah"))
|
||||
.arg("complete")
|
||||
.arg("--dir")
|
||||
.arg(&cache_dir)
|
||||
.arg("demo")
|
||||
.arg("--")
|
||||
.output()
|
||||
.expect("run inshellah complete");
|
||||
assert!(
|
||||
flag_output.status.success(),
|
||||
"stderr = {}",
|
||||
String::from_utf8_lossy(&flag_output.stderr)
|
||||
);
|
||||
let flag_stdout = String::from_utf8(flag_output.stdout).expect("stdout");
|
||||
assert!(
|
||||
flag_stdout.contains(r#""value":"--verbose""#),
|
||||
"stdout = {flag_stdout}"
|
||||
);
|
||||
|
||||
let _ = fs::remove_dir_all(root);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn complete_resolves_absolute_path_after_elevation_wrapper() {
|
||||
let root = unique_temp_dir("inshellah-absolute-elevation-complete");
|
||||
let bin_dir = root.join("bin");
|
||||
let cache_dir = root.join("cache");
|
||||
fs::create_dir_all(&bin_dir).expect("bin dir");
|
||||
fs::create_dir_all(&cache_dir).expect("cache dir");
|
||||
|
||||
let fakecmd = bin_dir.join("fakecmd");
|
||||
fs::write(
|
||||
&fakecmd,
|
||||
r#"#!/bin/sh
|
||||
if [ "$1" = "--help" ] || [ "$1" = "-h" ]; then
|
||||
printf '%s\n' 'Usage: fakecmd [OPTIONS]' '' 'Options:' ' --verbose verbose output'
|
||||
exit 0
|
||||
fi
|
||||
exit 2
|
||||
"#,
|
||||
)
|
||||
.expect("write fakecmd");
|
||||
let mut perms = fs::metadata(&fakecmd).expect("metadata").permissions();
|
||||
perms.set_mode(0o755);
|
||||
fs::set_permissions(&fakecmd, perms).expect("chmod");
|
||||
|
||||
let output = Command::new(env!("CARGO_BIN_EXE_inshellah"))
|
||||
.arg("complete")
|
||||
.arg("--dir")
|
||||
.arg(&cache_dir)
|
||||
.arg("--timeout-ms")
|
||||
.arg("1000")
|
||||
.arg("sudo")
|
||||
.arg(&fakecmd)
|
||||
.arg("--")
|
||||
.env("PATH", "")
|
||||
.output()
|
||||
.expect("run inshellah complete");
|
||||
|
||||
assert!(
|
||||
output.status.success(),
|
||||
"stderr = {}",
|
||||
String::from_utf8_lossy(&output.stderr)
|
||||
);
|
||||
let stdout = String::from_utf8(output.stdout).expect("stdout");
|
||||
assert!(
|
||||
stdout.contains(r#""value":"--verbose""#),
|
||||
"stdout = {stdout}"
|
||||
);
|
||||
|
||||
let _ = fs::remove_dir_all(root);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn complete_adb_dynamic_values_use_live_devices_and_packages() {
|
||||
let root = unique_temp_dir("inshellah-adb-dynamic-complete");
|
||||
let bin_dir = root.join("bin");
|
||||
let cache_dir = root.join("cache");
|
||||
fs::create_dir_all(&bin_dir).expect("bin dir");
|
||||
fs::create_dir_all(&cache_dir).expect("cache dir");
|
||||
|
||||
let adb = bin_dir.join("adb");
|
||||
fs::write(
|
||||
&adb,
|
||||
r#"#!/bin/sh
|
||||
selector=""
|
||||
case "$1" in
|
||||
-s|--serial|--one-device)
|
||||
selector="$2"
|
||||
shift 2
|
||||
;;
|
||||
-t|--transport-id)
|
||||
selector="transport:$2"
|
||||
shift 2
|
||||
;;
|
||||
--serial=*)
|
||||
selector="${1#--serial=}"
|
||||
shift
|
||||
;;
|
||||
--one-device=*)
|
||||
selector="${1#--one-device=}"
|
||||
shift
|
||||
;;
|
||||
--transport-id=*)
|
||||
selector="transport:${1#--transport-id=}"
|
||||
shift
|
||||
;;
|
||||
esac
|
||||
|
||||
if [ "$1" = "devices" ] && [ "$2" = "-l" ]; then
|
||||
printf '%s\n' 'List of devices attached'
|
||||
printf '%s\n' 'emulator-5554 device product:sdk_gphone_x86 model:Pixel_8 device:emu transport_id:1'
|
||||
printf '%s\n' 'R58M123456 device product:oriole model:Pixel_6 device:oriole transport_id:2'
|
||||
printf '%s\n' 'offline-1 offline transport_id:3'
|
||||
exit 0
|
||||
fi
|
||||
|
||||
if [ "$1" = "shell" ] && [ "$2" = "pm" ] && [ "$3" = "list" ] && [ "$4" = "packages" ]; then
|
||||
case "$selector" in
|
||||
emulator-5554)
|
||||
printf '%s\n' 'package:com.example.emu'
|
||||
printf '%s\n' 'package:org.example.shared'
|
||||
;;
|
||||
transport:2)
|
||||
printf '%s\n' 'package:com.example.transport'
|
||||
printf '%s\n' 'package:org.example.transport'
|
||||
;;
|
||||
*)
|
||||
printf '%s\n' 'package:com.default.app'
|
||||
printf '%s\n' 'package:/data/app/org.default.path/base.apk=org.default.path'
|
||||
;;
|
||||
esac
|
||||
exit 0
|
||||
fi
|
||||
|
||||
exit 2
|
||||
"#,
|
||||
)
|
||||
.expect("write adb");
|
||||
let mut perms = fs::metadata(&adb).expect("metadata").permissions();
|
||||
perms.set_mode(0o755);
|
||||
fs::set_permissions(&adb, perms).expect("chmod");
|
||||
|
||||
let run_complete = |args: &[&str]| -> String {
|
||||
let mut cmd = Command::new(env!("CARGO_BIN_EXE_inshellah"));
|
||||
cmd.arg("complete")
|
||||
.arg("--dir")
|
||||
.arg(&cache_dir)
|
||||
.arg("--timeout-ms")
|
||||
.arg("1000");
|
||||
for arg in args {
|
||||
cmd.arg(arg);
|
||||
}
|
||||
let output = cmd
|
||||
.env("PATH", &bin_dir)
|
||||
.output()
|
||||
.expect("run inshellah complete");
|
||||
assert!(
|
||||
output.status.success(),
|
||||
"stderr = {}",
|
||||
String::from_utf8_lossy(&output.stderr)
|
||||
);
|
||||
String::from_utf8(output.stdout).expect("stdout")
|
||||
};
|
||||
|
||||
let stdout = run_complete(&["adb", "-s", ""]);
|
||||
assert!(
|
||||
stdout.contains(r#""value":"emulator-5554""#),
|
||||
"stdout = {stdout}"
|
||||
);
|
||||
assert!(
|
||||
stdout.contains(r#""description":"device sdk gphone x86 Pixel 8""#),
|
||||
"stdout = {stdout}"
|
||||
);
|
||||
assert!(
|
||||
stdout.contains(r#""value":"R58M123456""#),
|
||||
"stdout = {stdout}"
|
||||
);
|
||||
assert!(
|
||||
stdout.contains(r#""value":"offline-1""#),
|
||||
"stdout = {stdout}"
|
||||
);
|
||||
|
||||
let prefixed_stdout = run_complete(&["adb", "--serial=R5"]);
|
||||
assert!(
|
||||
prefixed_stdout.contains(r#""value":"--serial=R58M123456""#),
|
||||
"stdout = {prefixed_stdout}"
|
||||
);
|
||||
assert!(
|
||||
!prefixed_stdout.contains(r#""value":"--serial=emulator-5554""#),
|
||||
"stdout = {prefixed_stdout}"
|
||||
);
|
||||
|
||||
let one_device_stdout = run_complete(&["adb", "--one-device", ""]);
|
||||
assert!(
|
||||
one_device_stdout.contains(r#""value":"emulator-5554""#),
|
||||
"stdout = {one_device_stdout}"
|
||||
);
|
||||
|
||||
let transport_stdout = run_complete(&["adb", "-t", ""]);
|
||||
assert!(
|
||||
transport_stdout.contains(r#""value":"1""#),
|
||||
"stdout = {transport_stdout}"
|
||||
);
|
||||
assert!(
|
||||
transport_stdout.contains(r#""description":"emulator-5554 device sdk gphone x86 Pixel 8""#),
|
||||
"stdout = {transport_stdout}"
|
||||
);
|
||||
assert!(
|
||||
transport_stdout.contains(r#""value":"2""#),
|
||||
"stdout = {transport_stdout}"
|
||||
);
|
||||
|
||||
let transport_prefixed_stdout = run_complete(&["adb", "--transport-id=2"]);
|
||||
assert!(
|
||||
transport_prefixed_stdout.contains(r#""value":"--transport-id=2""#),
|
||||
"stdout = {transport_prefixed_stdout}"
|
||||
);
|
||||
assert!(
|
||||
!transport_prefixed_stdout.contains(r#""value":"--transport-id=1""#),
|
||||
"stdout = {transport_prefixed_stdout}"
|
||||
);
|
||||
|
||||
let uninstall_stdout = run_complete(&["adb", "uninstall", "org"]);
|
||||
assert!(
|
||||
uninstall_stdout.contains(r#""value":"org.default.path""#),
|
||||
"stdout = {uninstall_stdout}"
|
||||
);
|
||||
assert!(
|
||||
!uninstall_stdout.contains(r#""value":"com.default.app""#),
|
||||
"stdout = {uninstall_stdout}"
|
||||
);
|
||||
|
||||
let clear_stdout = run_complete(&["adb", "-s", "emulator-5554", "shell", "pm", "clear", ""]);
|
||||
assert!(
|
||||
clear_stdout.contains(r#""value":"com.example.emu""#),
|
||||
"stdout = {clear_stdout}"
|
||||
);
|
||||
assert!(
|
||||
!clear_stdout.contains(r#""value":"com.example.transport""#),
|
||||
"stdout = {clear_stdout}"
|
||||
);
|
||||
|
||||
let force_stop_stdout = run_complete(&[
|
||||
"adb",
|
||||
"-t",
|
||||
"2",
|
||||
"shell",
|
||||
"am",
|
||||
"force-stop",
|
||||
"--user",
|
||||
"0",
|
||||
"com.",
|
||||
]);
|
||||
assert!(
|
||||
force_stop_stdout.contains(r#""value":"com.example.transport""#),
|
||||
"stdout = {force_stop_stdout}"
|
||||
);
|
||||
assert!(
|
||||
!force_stop_stdout.contains(r#""value":"com.example.emu""#),
|
||||
"stdout = {force_stop_stdout}"
|
||||
);
|
||||
|
||||
let flag_value_stdout = run_complete(&["adb", "shell", "pm", "enable", "--user", ""]);
|
||||
assert_eq!(
|
||||
flag_value_stdout.trim(),
|
||||
"null",
|
||||
"stdout = {flag_value_stdout}"
|
||||
);
|
||||
|
||||
let shell_flag_stdout = run_complete(&["adb", "shell", "-s", ""]);
|
||||
assert_eq!(
|
||||
shell_flag_stdout.trim(),
|
||||
"null",
|
||||
"stdout = {shell_flag_stdout}"
|
||||
);
|
||||
|
||||
let _ = fs::remove_dir_all(root);
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue