|
|
@@ -79,8 +79,7 @@ impl RendezvousServer {
|
|
79
|
79
|
#[cfg(test)]
|
|
80
|
80
|
mod tests {
|
|
81
|
81
|
use super::*;
|
|
82
|
|
- use hbb_common::tokio;
|
|
83
|
|
- use std::io::{Error, ErrorKind};
|
|
|
82
|
+ use hbb_common::{new_error, tokio};
|
|
84
|
83
|
|
|
85
|
84
|
#[allow(unused_must_use)]
|
|
86
|
85
|
#[tokio::main]
|
|
|
@@ -127,7 +126,7 @@ mod tests {
|
|
127
|
126
|
if let Some(Ok((bytes, addr))) = socket_b.next_timeout(1000).await {
|
|
128
|
127
|
assert_eq!(addr_server, addr);
|
|
129
|
128
|
let msg_in = parse_from_bytes::<Message>(&bytes)?;
|
|
130
|
|
- let remote_addr_a = AddrMangle::decode(&msg_in.get_punch_hole().socket_addr[..]);
|
|
|
129
|
+ let remote_addr_a = AddrMangle::decode(&msg_in.get_punch_hole().socket_addr);
|
|
131
|
130
|
assert_eq!(local_addr_a, remote_addr_a);
|
|
132
|
131
|
|
|
133
|
132
|
// B punch A
|
|
|
@@ -150,14 +149,13 @@ mod tests {
|
|
150
|
149
|
if let Some(Ok((bytes, addr))) = socket_a.next_timeout(1000).await {
|
|
151
|
150
|
assert_eq!(addr_server, addr);
|
|
152
|
151
|
let msg_in = parse_from_bytes::<Message>(&bytes)?;
|
|
153
|
|
- let remote_addr_b =
|
|
154
|
|
- AddrMangle::decode(&msg_in.get_punch_hole_response().socket_addr[..]);
|
|
|
152
|
+ let remote_addr_b = AddrMangle::decode(&msg_in.get_punch_hole_response().socket_addr);
|
|
155
|
153
|
assert_eq!(local_addr_b, remote_addr_b);
|
|
156
|
154
|
} else {
|
|
157
|
155
|
panic!("failed");
|
|
158
|
156
|
}
|
|
159
|
157
|
|
|
160
|
|
- Err(Box::new(Error::new(ErrorKind::Other, "done")))
|
|
|
158
|
+ Err(new_error("done"))
|
|
161
|
159
|
}
|
|
162
|
160
|
|
|
163
|
161
|
#[test]
|