Skip to content

Commit

Permalink
add: decrement_work_counter_err test
Browse files Browse the repository at this point in the history
  • Loading branch information
justRkive committed Dec 2, 2024
1 parent 6e8f06b commit 3acce4b
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions src/async_resolver/server_entry.rs
Original file line number Diff line number Diff line change
Expand Up @@ -99,4 +99,27 @@ mod tests {
}
}

#[test]
fn decrement_work_counter_error() {
let ip_addr = IpAddr::V4(Ipv4Addr::new(192, 168, 0, 1));
let port = 53;
let key = String::from("key");
let algorithm = String::from("algorithm");
let udp_connection = ClientUDPConnection::new_default(ip_addr, Duration::from_secs(100));
let tcp_connection = ClientTCPConnection::new_default(ip_addr, Duration::from_secs(100));
let info = ServerInfo::new(ip_addr, port, key, algorithm, udp_connection, tcp_connection);

let info_arc = Arc::new(info);
let work_counter = 0;
let mut server_entry = ServerEntry::new(info_arc.clone(), work_counter);

let result = server_entry.decrement_work_counter();

if let Err(ResolverError::RetriesLimitExceeded) = result {
assert!(true);
assert_eq!(server_entry.get_work_counter(), 0);
} else {
assert!(false);
}
}
}

0 comments on commit 3acce4b

Please sign in to comment.