mirror of https://github.com/telemt/telemt.git
72 lines
2.1 KiB
Rust
72 lines
2.1 KiB
Rust
use super::*;
|
|
use std::time::{Duration, Instant};
|
|
|
|
fn auth_probe_test_guard() -> std::sync::MutexGuard<'static, ()> {
|
|
auth_probe_test_lock()
|
|
.lock()
|
|
.unwrap_or_else(|poisoned| poisoned.into_inner())
|
|
}
|
|
|
|
fn poison_saturation_mutex() {
|
|
let saturation = auth_probe_saturation_state();
|
|
let poison_thread = std::thread::spawn(move || {
|
|
let _guard = saturation
|
|
.lock()
|
|
.expect("saturation mutex must be lockable for poison setup");
|
|
panic!("intentional poison for saturation mutex resilience test");
|
|
});
|
|
let _ = poison_thread.join();
|
|
}
|
|
|
|
#[test]
|
|
fn auth_probe_saturation_note_recovers_after_mutex_poison() {
|
|
let _guard = auth_probe_test_guard();
|
|
clear_auth_probe_state_for_testing();
|
|
poison_saturation_mutex();
|
|
|
|
let now = Instant::now();
|
|
auth_probe_note_saturation(now);
|
|
|
|
assert!(
|
|
auth_probe_saturation_is_throttled_at_for_testing(now),
|
|
"poisoned saturation mutex must not disable saturation throttling"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn auth_probe_saturation_check_recovers_after_mutex_poison() {
|
|
let _guard = auth_probe_test_guard();
|
|
clear_auth_probe_state_for_testing();
|
|
poison_saturation_mutex();
|
|
|
|
{
|
|
let mut guard = auth_probe_saturation_state_lock();
|
|
*guard = Some(AuthProbeSaturationState {
|
|
fail_streak: AUTH_PROBE_BACKOFF_START_FAILS,
|
|
blocked_until: Instant::now() + Duration::from_millis(10),
|
|
last_seen: Instant::now(),
|
|
});
|
|
}
|
|
|
|
assert!(
|
|
auth_probe_saturation_is_throttled_for_testing(),
|
|
"throttle check must recover poisoned saturation mutex and stay fail-closed"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn clear_auth_probe_state_clears_saturation_even_if_poisoned() {
|
|
let _guard = auth_probe_test_guard();
|
|
clear_auth_probe_state_for_testing();
|
|
poison_saturation_mutex();
|
|
|
|
auth_probe_note_saturation(Instant::now());
|
|
assert!(auth_probe_saturation_is_throttled_for_testing());
|
|
|
|
clear_auth_probe_state_for_testing();
|
|
assert!(
|
|
!auth_probe_saturation_is_throttled_for_testing(),
|
|
"clear helper must clear saturation state even after poison"
|
|
);
|
|
}
|