fix: 修正Core故障退避与熔断计数
This commit is contained in:
@@ -2,9 +2,29 @@
|
||||
use notesagent_host::core::CoreSupervisor;
|
||||
use notesagent_host::credentials::{CredentialBroker, CredentialId, Scope};
|
||||
use std::path::Path;
|
||||
#[cfg(windows)]
|
||||
use std::process::Command;
|
||||
use std::sync::{Arc, Mutex};
|
||||
#[cfg(windows)]
|
||||
use std::time::{Duration, Instant};
|
||||
use zeroize::Zeroizing;
|
||||
|
||||
#[cfg(windows)]
|
||||
fn file_sha256(path: &Path) -> String {
|
||||
use sha2::{Digest, Sha256};
|
||||
let bytes = std::fs::read(path).unwrap();
|
||||
format!("{:x}", Sha256::digest(bytes))
|
||||
}
|
||||
|
||||
#[cfg(windows)]
|
||||
fn terminate_process_tree(pid: u32) {
|
||||
let status = Command::new("taskkill")
|
||||
.args(["/PID", &pid.to_string(), "/T", "/F"])
|
||||
.status()
|
||||
.unwrap();
|
||||
assert!(status.success(), "无法终止 Core 进程树 {pid}");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn real_python_core_authenticates_and_rotates_generation() {
|
||||
let backend = Path::new(env!("CARGO_MANIFEST_DIR"))
|
||||
@@ -48,6 +68,64 @@ fn real_python_core_authenticates_and_rotates_generation() {
|
||||
assert!(std::net::TcpStream::connect(endpoint).is_err());
|
||||
}
|
||||
|
||||
#[cfg(windows)]
|
||||
#[test]
|
||||
fn six_real_core_crashes_back_off_then_open_the_circuit_without_touching_local_edits() {
|
||||
let backend = Path::new(env!("CARGO_MANIFEST_DIR"))
|
||||
.join("../../backend")
|
||||
.canonicalize()
|
||||
.unwrap();
|
||||
let python = backend.join(".venv/Scripts/python.exe");
|
||||
assert!(python.is_file(), "需要已锁定的后端虚拟环境");
|
||||
let temp = tempfile::tempdir().unwrap();
|
||||
let note = temp.path().join("local-edit.md");
|
||||
std::fs::write(¬e, "Core 故障期间仍由 Host 保存的本地修改。\n").unwrap();
|
||||
let expected_hash = file_sha256(¬e);
|
||||
let mut core = CoreSupervisor::new(
|
||||
python,
|
||||
vec!["-m".into(), "app.sidecar".into()],
|
||||
backend,
|
||||
temp.path().join("core"),
|
||||
);
|
||||
|
||||
let first = core.request_session("/health").unwrap();
|
||||
let first_endpoint = first
|
||||
.url
|
||||
.trim_start_matches("http://")
|
||||
.trim_end_matches("/health")
|
||||
.to_string();
|
||||
for crash in 0..6 {
|
||||
terminate_process_tree(core.process_id().unwrap());
|
||||
let stopped = Instant::now();
|
||||
while core.available() {
|
||||
assert!(stopped.elapsed() < Duration::from_secs(2));
|
||||
std::thread::sleep(Duration::from_millis(10));
|
||||
}
|
||||
assert_eq!(file_sha256(¬e), expected_hash);
|
||||
if crash == 5 {
|
||||
assert_eq!(
|
||||
core.request_session("/health").err().unwrap(),
|
||||
"CORE_RESTART_LIMIT"
|
||||
);
|
||||
break;
|
||||
}
|
||||
let expected_delay = Duration::from_secs(1 << crash);
|
||||
assert_eq!(
|
||||
core.request_session("/health").err().unwrap(),
|
||||
"CORE_RESTART_BACKOFF"
|
||||
);
|
||||
while stopped.elapsed() < expected_delay {
|
||||
std::thread::sleep(Duration::from_millis(20));
|
||||
}
|
||||
core.request_session("/health").unwrap();
|
||||
}
|
||||
|
||||
let shutdown = Instant::now();
|
||||
drop(core);
|
||||
assert!(shutdown.elapsed() < Duration::from_secs(10));
|
||||
assert!(std::net::TcpStream::connect(first_endpoint).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn real_core_credential_api_uses_host_stronghold_without_plaintext_response() {
|
||||
use std::io::{Read, Write};
|
||||
|
||||
Reference in New Issue
Block a user