refactored

This commit is contained in:
open-trade 2020-03-10 14:38:10 +08:00
parent af2ccf86c6
commit defac1f338
2 changed files with 5 additions and 7 deletions

@ -1 +1 @@
Subproject commit 7814ff3a395bf063300317f28644b92712d501e4
Subproject commit 2cc3edf0f634adcb1e290c5c8b11bc21de38e763

View File

@ -79,8 +79,7 @@ impl RendezvousServer {
#[cfg(test)]
mod tests {
use super::*;
use hbb_common::tokio;
use std::io::{Error, ErrorKind};
use hbb_common::{new_error, tokio};
#[allow(unused_must_use)]
#[tokio::main]
@ -127,7 +126,7 @@ mod tests {
if let Some(Ok((bytes, addr))) = socket_b.next_timeout(1000).await {
assert_eq!(addr_server, addr);
let msg_in = parse_from_bytes::<Message>(&bytes)?;
let remote_addr_a = AddrMangle::decode(&msg_in.get_punch_hole().socket_addr[..]);
let remote_addr_a = AddrMangle::decode(&msg_in.get_punch_hole().socket_addr);
assert_eq!(local_addr_a, remote_addr_a);
// B punch A
@ -150,14 +149,13 @@ mod tests {
if let Some(Ok((bytes, addr))) = socket_a.next_timeout(1000).await {
assert_eq!(addr_server, addr);
let msg_in = parse_from_bytes::<Message>(&bytes)?;
let remote_addr_b =
AddrMangle::decode(&msg_in.get_punch_hole_response().socket_addr[..]);
let remote_addr_b = AddrMangle::decode(&msg_in.get_punch_hole_response().socket_addr);
assert_eq!(local_addr_b, remote_addr_b);
} else {
panic!("failed");
}
Err(Box::new(Error::new(ErrorKind::Other, "done")))
Err(new_error("done"))
}
#[test]