gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Anatoly Yakovenko
2018-03-14 11:02:38 -07:00
parent 65185c0011
commit 75ef0f0329
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/streamer.rs
View File

@ -1,5 +1,6 @@
use std::sync::{Arc, Mutex, RwLock}; use std::sync::{Arc, Mutex, RwLock};
use std::sync::mpsc; use std::sync::mpsc;
use std::fmt;
use std::time::Duration; use std::time::Duration;
use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr, UdpSocket}; use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr, UdpSocket};
use std::thread::{spawn, JoinHandle}; use std::thread::{spawn, JoinHandle};
@ -16,7 +17,11 @@ pub struct Packet {
pub port: u16, pub port: u16,
pub v6: bool, pub v6: bool,
} }
impl fmt::Debug for Packet {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "Packet {{ size: {:?}, addr: {:?} }}", self.size, self.get_addr())
}
}
impl Default for Packet { impl Default for Packet {
fn default() -> Packet { fn default() -> Packet {
Packet { Packet {
@ -72,6 +77,7 @@ impl Packet {
} }
} }
#[derive(Clone, Debug)]
pub struct PacketData { pub struct PacketData {
pub packets: Vec<Packet>, pub packets: Vec<Packet>,
} }
@ -292,11 +298,11 @@ mod test {
} }
fn get_msgs(r: Receiver, num: &mut usize) { fn get_msgs(r: Receiver, num: &mut usize) {
for _ in [0..5].iter() { for _t in 0..5 {
let timer = Duration::new(1, 0); let timer = Duration::new(1, 0);
match r.recv_timeout(timer) { match r.recv_timeout(timer) {
Ok(m) => *num += m.read().unwrap().packets.len(), Ok(m) => *num += m.read().unwrap().packets.len(),
_ => (), e => println!("error {:?}", e),
} }
if *num == 10 { if *num == 10 {
break; break;