mirror of
https://github.com/Tencent/BrowserSkill.git
synced 2026-09-10 13:01:23 +08:00
Add `bsk network` as a sibling of the `console` command from #7, reading the buffered network responses / failures for a tab. Fills the remaining half of the request in #2 (console landed in #7; network did not). Mirrors the console command's conventions exactly: bsk network --session <id> [--tab-id N] [--since C] [--limit N] [--max-text-chars N] The extension enables `Network` on attach (best-effort) and buffers `responseReceived` / `loadingFailed` events per tab with a monotonic sequence, correlating them to `requestWillBeSent` for method/URL. Reads are cursor-paginated (`since` -> `next_since`) and bounded (`limit`, `max_text_chars`) for agent-context safety, same as console. Entries carry the real HTTP status code (response) or the CDP failure reason (failure) — things `performance.getEntriesByType('resource')` can't give. Classified read-only (pass-through under the pending-interrupt gate); defaults to the Agent Window's active tab. Wires bsk-protocol (Method + params/result/entry structs + schema), bsk-cli (network subcommand + daemon dispatch allow-list) and the extension (handler + dispatcher route), with unit / parse / IPC tests paralleling the console command. Refs #2.
195 lines
5.3 KiB
Rust
195 lines
5.3 KiB
Rust
//! Smoke tests for clap-derive command parsing.
|
|
|
|
use std::time::Duration;
|
|
|
|
use bsk::cli::daemon::{DaemonCmd, parse_duration};
|
|
use bsk::cli::navigate::NavigateCmd;
|
|
use bsk::{Cli, Command};
|
|
use clap::Parser;
|
|
|
|
fn parse(args: &[&str]) -> Cli {
|
|
Cli::try_parse_from(args).expect("clap parse should succeed")
|
|
}
|
|
|
|
#[test]
|
|
fn parses_daemon_start_with_defaults() {
|
|
let cli = parse(&["bsk", "daemon", "start"]);
|
|
let Command::Daemon(DaemonCmd::Start(args)) = cli.command else {
|
|
panic!("expected daemon start subcommand");
|
|
};
|
|
assert!(args.port.is_none());
|
|
assert!(!args.foreground);
|
|
assert_eq!(args.resolved_port(), 52800);
|
|
assert_eq!(args.resolved_daemon_idle(), Duration::from_secs(600));
|
|
}
|
|
|
|
#[test]
|
|
fn parses_daemon_start_with_flags() {
|
|
let cli = parse(&[
|
|
"bsk",
|
|
"daemon",
|
|
"start",
|
|
"--foreground",
|
|
"--port",
|
|
"52900",
|
|
"--daemon-idle",
|
|
"2s",
|
|
"--session-idle",
|
|
"30s",
|
|
]);
|
|
let Command::Daemon(DaemonCmd::Start(args)) = cli.command else {
|
|
panic!("expected daemon start subcommand");
|
|
};
|
|
assert!(args.foreground);
|
|
assert_eq!(args.resolved_port(), 52900);
|
|
assert_eq!(args.resolved_daemon_idle(), Duration::from_secs(2));
|
|
assert_eq!(args.resolved_session_idle(), Duration::from_secs(30));
|
|
}
|
|
|
|
#[test]
|
|
fn parses_daemon_stop_and_restart() {
|
|
let cli = parse(&["bsk", "daemon", "stop"]);
|
|
assert!(matches!(cli.command, Command::Daemon(DaemonCmd::Stop)));
|
|
|
|
let cli = parse(&["bsk", "daemon", "restart", "--foreground"]);
|
|
let Command::Daemon(DaemonCmd::Restart(args)) = cli.command else {
|
|
panic!("expected daemon restart subcommand");
|
|
};
|
|
assert!(args.foreground);
|
|
}
|
|
|
|
#[test]
|
|
fn parses_top_level_status_and_doctor() {
|
|
let cli = parse(&["bsk", "status"]);
|
|
assert!(matches!(cli.command, Command::Status));
|
|
|
|
let cli = parse(&["bsk", "doctor"]);
|
|
assert!(matches!(cli.command, Command::Doctor));
|
|
}
|
|
|
|
#[test]
|
|
fn parses_console_command_with_context_safety_flags() {
|
|
let cli = parse(&[
|
|
"bsk",
|
|
"console",
|
|
"--session",
|
|
"s1",
|
|
"--tab-id",
|
|
"9",
|
|
"--since",
|
|
"12",
|
|
"--limit",
|
|
"75",
|
|
"--max-text-chars",
|
|
"2048",
|
|
"--include-stack",
|
|
]);
|
|
let Command::Console(args) = cli.command else {
|
|
panic!("expected console command");
|
|
};
|
|
assert_eq!(args.session, "s1");
|
|
assert_eq!(args.tab_id, Some(9));
|
|
assert_eq!(args.since, Some(12));
|
|
assert_eq!(args.limit, Some(75));
|
|
assert_eq!(args.max_text_chars, Some(2048));
|
|
assert!(args.include_stack);
|
|
}
|
|
|
|
#[test]
|
|
fn rejects_zero_console_bounds() {
|
|
assert!(Cli::try_parse_from(["bsk", "console", "--session", "s1", "--limit", "0"]).is_err());
|
|
assert!(
|
|
Cli::try_parse_from(["bsk", "console", "--session", "s1", "--max-text-chars", "0"])
|
|
.is_err()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parses_network_command_with_context_safety_flags() {
|
|
let cli = parse(&[
|
|
"bsk",
|
|
"network",
|
|
"--session",
|
|
"s1",
|
|
"--tab-id",
|
|
"9",
|
|
"--since",
|
|
"12",
|
|
"--limit",
|
|
"75",
|
|
"--max-text-chars",
|
|
"2048",
|
|
]);
|
|
let Command::Network(args) = cli.command else {
|
|
panic!("expected network command");
|
|
};
|
|
assert_eq!(args.session, "s1");
|
|
assert_eq!(args.tab_id, Some(9));
|
|
assert_eq!(args.since, Some(12));
|
|
assert_eq!(args.limit, Some(75));
|
|
assert_eq!(args.max_text_chars, Some(2048));
|
|
}
|
|
|
|
#[test]
|
|
fn rejects_zero_network_bounds() {
|
|
assert!(Cli::try_parse_from(["bsk", "network", "--session", "s1", "--limit", "0"]).is_err());
|
|
assert!(
|
|
Cli::try_parse_from(["bsk", "network", "--session", "s1", "--max-text-chars", "0"])
|
|
.is_err()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn parses_install_skill_subcommand() {
|
|
let cli = parse(&["bsk", "install-skill", "--list"]);
|
|
assert!(matches!(cli.command, Command::InstallSkill(_)));
|
|
}
|
|
|
|
#[test]
|
|
fn parses_update_subcommand_with_flags() {
|
|
let cli = parse(&["bsk", "update", "--check", "--yes", "--no-restart-daemon"]);
|
|
let Command::Update(args) = cli.command else {
|
|
panic!("expected update subcommand");
|
|
};
|
|
assert!(args.check);
|
|
assert!(args.yes);
|
|
assert!(!args.restart_daemon);
|
|
}
|
|
|
|
#[test]
|
|
fn duration_parser_accepts_units() {
|
|
assert_eq!(parse_duration("750ms").unwrap(), Duration::from_millis(750));
|
|
assert_eq!(parse_duration("2m").unwrap(), Duration::from_secs(120));
|
|
}
|
|
|
|
#[test]
|
|
fn parses_nested_navigate_back_and_forward() {
|
|
let cli = parse(&["bsk", "navigate", "back", "--session", "s1"]);
|
|
let Command::Navigate(cmd) = cli.command else {
|
|
panic!("expected navigate command");
|
|
};
|
|
assert!(matches!(cmd.command, Some(NavigateCmd::Back(_))));
|
|
|
|
let cli = parse(&["bsk", "navigate", "forward", "--session", "s1"]);
|
|
let Command::Navigate(cmd) = cli.command else {
|
|
panic!("expected navigate command");
|
|
};
|
|
assert!(matches!(cmd.command, Some(NavigateCmd::Forward(_))));
|
|
}
|
|
|
|
#[test]
|
|
fn parses_click_count_alias() {
|
|
let cli = parse(&["bsk", "click", "@e1", "--session", "s1", "--count", "2"]);
|
|
let Command::Click(args) = cli.command else {
|
|
panic!("expected click command");
|
|
};
|
|
assert_eq!(args.click_count, 2);
|
|
}
|
|
|
|
#[test]
|
|
fn rejects_zero_click_count() {
|
|
assert!(
|
|
Cli::try_parse_from(["bsk", "click", "@e1", "--session", "s1", "--count", "0"]).is_err()
|
|
);
|
|
}
|