mirror of
https://github.com/Cuprate/cuprate.git
synced 2024-12-23 03:59:31 +00:00
4b93dbec4c
Some checks failed
CI / fmt (push) Waiting to run
CI / typo (push) Waiting to run
CI / ci (macos-latest, stable, bash) (push) Waiting to run
CI / ci (ubuntu-latest, stable, bash) (push) Waiting to run
CI / ci (windows-latest, stable-x86_64-pc-windows-gnu, msys2 {0}) (push) Waiting to run
Audit / audit (push) Has been cancelled
Deny / audit (push) Has been cancelled
* rename all directories and crates * fix all `use` * fix doc link * `dandelion/` -> `dandelion-tower/` * fix epee-encoding test * fix `json-rpc` * fix pruning * crate import fixes * fix leftover merge conflicts * fix `epee-encoding`
64 lines
1.8 KiB
Rust
64 lines
1.8 KiB
Rust
use std::{sync::Arc, time::Duration};
|
|
|
|
use tokio::sync::Semaphore;
|
|
|
|
use cuprate_p2p_core::handles::HandleBuilder;
|
|
|
|
#[test]
|
|
fn send_ban_signal() {
|
|
let semaphore = Arc::new(Semaphore::new(5));
|
|
let (guard, mut connection_handle) = HandleBuilder::default()
|
|
.with_permit(semaphore.try_acquire_owned().unwrap())
|
|
.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 semaphore = Arc::new(Semaphore::new(5));
|
|
let (guard, mut connection_handle) = HandleBuilder::default()
|
|
.with_permit(semaphore.try_acquire_owned().unwrap())
|
|
.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(semaphore.try_acquire_owned().unwrap())
|
|
.build();
|
|
|
|
assert!(!connection_handle.is_closed());
|
|
drop(guard);
|
|
assert!(connection_handle.is_closed());
|
|
}
|