mirror of
https://github.com/Cuprate/cuprate.git
synced 2024-12-23 12:09:52 +00:00
848a6a71c4
Some checks failed
Audit / audit (push) Has been cancelled
CI / fmt (push) Has been cancelled
CI / typo (push) Has been cancelled
CI / ci (macos-latest, stable, bash) (push) Has been cancelled
CI / ci (ubuntu-latest, stable, bash) (push) Has been cancelled
CI / ci (windows-latest, stable-x86_64-pc-windows-gnu, msys2 {0}) (push) Has been cancelled
Deny / audit (push) Has been cancelled
Doc / build (push) Has been cancelled
Doc / deploy (push) Has been cancelled
* p2p-core: enable workspace lints * fmt * fix tests * fixes * fixes * fixes * expect reason
60 lines
1.7 KiB
Rust
60 lines
1.7 KiB
Rust
#![expect(unused_crate_dependencies, reason = "external test module")]
|
|
|
|
use std::{sync::Arc, time::Duration};
|
|
|
|
use tokio::sync::Semaphore;
|
|
|
|
use cuprate_p2p_core::handles::HandleBuilder;
|
|
|
|
#[test]
|
|
fn send_ban_signal() {
|
|
let (guard, mut connection_handle) = HandleBuilder::default().build();
|
|
|
|
connection_handle.ban_peer(Duration::from_secs(300));
|
|
|
|
let Some(ban_time) = connection_handle.check_should_ban() else {
|
|
panic!("ban signal not received!");
|
|
};
|
|
|
|
assert_eq!(ban_time.0, Duration::from_secs(300));
|
|
|
|
connection_handle.send_close_signal();
|
|
assert!(guard.should_shutdown());
|
|
|
|
guard.connection_closed();
|
|
assert!(connection_handle.is_closed());
|
|
}
|
|
|
|
#[test]
|
|
fn multiple_ban_signals() {
|
|
let (guard, mut connection_handle) = HandleBuilder::default().build();
|
|
|
|
connection_handle.ban_peer(Duration::from_secs(300));
|
|
connection_handle.ban_peer(Duration::from_secs(301));
|
|
connection_handle.ban_peer(Duration::from_secs(302));
|
|
|
|
let Some(ban_time) = connection_handle.check_should_ban() else {
|
|
panic!("ban signal not received!");
|
|
};
|
|
|
|
// only the first will be seen
|
|
assert_eq!(ban_time.0, Duration::from_secs(300));
|
|
|
|
connection_handle.send_close_signal();
|
|
assert!(guard.should_shutdown());
|
|
|
|
guard.connection_closed();
|
|
assert!(connection_handle.is_closed());
|
|
}
|
|
|
|
#[test]
|
|
fn dropped_guard_sends_disconnect_signal() {
|
|
let semaphore = Arc::new(Semaphore::new(5));
|
|
let (guard, connection_handle) = HandleBuilder::default()
|
|
.with_permit(Some(semaphore.try_acquire_owned().unwrap()))
|
|
.build();
|
|
|
|
assert!(!connection_handle.is_closed());
|
|
drop(guard);
|
|
assert!(connection_handle.is_closed());
|
|
}
|