open-trade лет назад: 5
Родитель
Сommit
defac1f338
2 измененных файлов с 5 добавлено и 7 удалено
  1. 1 1
      libs/hbb_common
  2. 4 6
      src/rendezvous_server.rs

+ 1 - 1
libs/hbb_common

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

+ 4 - 6
src/rendezvous_server.rs

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