b41c75a578
Changes: - Detect the herdr multiplexer - Mark /gboom as non-production code - Bound peak memory when loading a large session - Add a subagent lifecycle soak bounding threads, fds, and heap - Stream inherited replay to bound fork memory - Copy full plan from plan approval with y - Stop armed signature verification from deleting the managed-deny smoke policy - Add source-tagged terminal version telemetry - Show the UI instantly and fetch models and settings in the background - Session test helpers - computer_reason on the ConversationHistoryDone trailer
73 lines
3.1 KiB
Rust
73 lines
3.1 KiB
Rust
#![allow(
|
|
unused_imports,
|
|
unused_variables,
|
|
unused_mut,
|
|
unreachable_code,
|
|
dead_code
|
|
)]
|
|
//! Shared test utilities for grok-build crates: mock inference server, SSE
|
|
//! generators, ACP stdio client, headless runner, env/process sandbox.
|
|
//!
|
|
//! Provides:
|
|
//! - [`MockInferenceServer`] — Mock /v1/chat/completions + /v1/responses with request logging
|
|
//! - [`GrokStdioClient`] — ACP client that drives `grok agent stdio` as a subprocess
|
|
//! - [`RawStdioClient`] — raw-wire ACP driver for bytes the typed client can't
|
|
//! produce (Foundation `\/` methods, string UUID ids)
|
|
//! - [`leader::LeaderStdioClient`] — ACP client that drives `grok agent --leader stdio` (unix)
|
|
//! - [`TestSandbox`] — Own isolated paths, hermetic child env, optional git setup, diagnostics
|
|
//! - [`TestProcess`] — Own detached child lifecycle, process-tree teardown, bounded output tails
|
|
//! - [`run_headless`] — Run `grok -p` against the mock server and capture output
|
|
//! - [`git_workdir`] — Create a git-initialized [`TestSandbox`]
|
|
//! - [`grok_binary`] — Resolve the grok binary path (GROK_BINARY env or cargo_bin)
|
|
//! - [`spawn_counting_server`] — Connection-counting HTTP/1.1 server for wire/pooling tests
|
|
//! - [`uds_proxy::UdsProxy`] — Frame-aware fault-injection proxy for leader IPC sockets (unix)
|
|
//! - [`ResourceSnapshot`] — RSS/threads/fds sampling for soak tests
|
|
/// Multiply a harness timeout by `GROK_TEST_TIMEOUT_SCALE` (positive integer,
|
|
/// default 1). CI lanes on shared runner pools raise it so pool load slows
|
|
/// tests instead of failing them (see the Grok Build merge CI workflow).
|
|
pub fn scaled(base: std::time::Duration) -> std::time::Duration {
|
|
let scale = std::env::var("GROK_TEST_TIMEOUT_SCALE")
|
|
.ok()
|
|
.and_then(|v| v.parse::<u32>().ok())
|
|
.filter(|&v| v > 0)
|
|
.unwrap_or(1);
|
|
base * scale
|
|
}
|
|
pub mod acp_client;
|
|
pub mod counting_server;
|
|
pub mod env;
|
|
pub mod headless;
|
|
mod inference_override;
|
|
#[cfg(unix)]
|
|
pub mod leader;
|
|
pub mod mock_server;
|
|
pub mod process;
|
|
pub mod resources;
|
|
pub mod sandbox;
|
|
pub mod scripted;
|
|
pub mod sse;
|
|
#[cfg(unix)]
|
|
pub mod uds_proxy;
|
|
pub use acp_client::{GrokStdioClient, RawStdioClient};
|
|
pub use counting_server::spawn_counting_server;
|
|
pub use env::{EnvGuard, git_workdir, grok_binary};
|
|
pub use headless::{
|
|
HeadlessResult, assert_headless_success, assert_no_crashes, run_headless,
|
|
run_headless_in_sandbox, run_headless_in_sandbox_borrowed,
|
|
run_headless_in_sandbox_borrowed_with_env, run_headless_in_sandbox_with_env,
|
|
run_headless_with_env, stderr_tail,
|
|
};
|
|
pub use inference_override::{InferenceEndpoint, InferenceExpectation, InferenceRequestMatcher};
|
|
#[cfg(unix)]
|
|
pub use leader::LeaderFixture;
|
|
pub use mock_server::{
|
|
MockInferenceServer, MockModelEntry, ScriptedResponse, SseEvent, StorageUpload,
|
|
};
|
|
#[cfg(unix)]
|
|
pub use process::process_has_exited_without_reap;
|
|
pub use process::{
|
|
TestOutput, TestOutputSnapshot, TestProcess, TestProcessConfig, TestProcessState,
|
|
TestProcessStderr, TestProcessStdout, TestProcessTermination, TestProcessTree, TestStdin,
|
|
};
|
|
pub use resources::{ResourceGrowth, ResourceSnapshot};
|
|
pub use sandbox::{TestSandbox, TestSandboxBuilder};
|