Cover some more cases

This commit is contained in:
Dominik Werder
2021-09-01 16:20:30 +02:00
parent 370b33c6a5
commit 09b671b8f0
3 changed files with 60 additions and 18 deletions

View File

@@ -147,14 +147,9 @@ impl Stream for EventChunkerMultifile {
} }
} }
#[test] fn read_expanded_for_range(range: netpod::NanoRange) -> Result<(usize, usize), Error> {
fn read_expanded() {
use netpod::timeunits::*; use netpod::timeunits::*;
use netpod::Nanos; use netpod::Nanos;
let range = netpod::NanoRange {
beg: DAY + HOUR * 5,
end: DAY + HOUR * 8,
};
let chn = netpod::Channel { let chn = netpod::Channel {
backend: "testbackend".into(), backend: "testbackend".into(),
name: "scalar-i32-be".into(), name: "scalar-i32-be".into(),
@@ -178,7 +173,6 @@ fn read_expanded() {
disk_stats_every: ByteSize::kb(1024), disk_stats_every: ByteSize::kb(1024),
}; };
let task = async move { let task = async move {
let mut c1 = 0;
let mut event_count = 0; let mut event_count = 0;
let mut events = let mut events =
EventChunkerMultifile::new(range, channel_config, node, node_ix, buffer_size, event_chunker_conf); EventChunkerMultifile::new(range, channel_config, node, node_ix, buffer_size, event_chunker_conf);
@@ -195,10 +189,6 @@ fn read_expanded() {
}, },
Err(e) => return Err(e.into()), Err(e) => return Err(e.into()),
} }
c1 += 1;
if c1 >= 6 {
break;
}
} }
events.close(); events.close();
if events.seen_before_range_count() != 1 { if events.seen_before_range_count() != 1 {
@@ -207,10 +197,63 @@ fn read_expanded() {
events.seen_before_range_count(), events.seen_before_range_count(),
))); )));
} }
if event_count != 49 { Ok((event_count, events.seen_before_range_count()))
return Err(Error::with_msg(format!("unexpected number of events: {}", event_count)));
}
Ok(())
}; };
taskrun::run(task).unwrap(); Ok(taskrun::run(task).unwrap())
}
#[test]
fn read_expanded_0() -> Result<(), Error> {
use netpod::timeunits::*;
let range = netpod::NanoRange {
beg: DAY + MS * 0,
end: DAY + MS * 0 + MS * 1500,
};
let res = read_expanded_for_range(range)?;
if res.0 != 2 {
Err(Error::with_msg(format!("unexpected number of events: {}", res.0)))?;
}
Ok(())
}
#[test]
fn read_expanded_1() -> Result<(), Error> {
use netpod::timeunits::*;
let range = netpod::NanoRange {
beg: DAY + MS * 0,
end: DAY + MS * 0 + MS * 1501,
};
let res = read_expanded_for_range(range)?;
if res.0 != 3 {
Err(Error::with_msg(format!("unexpected number of events: {}", res.0)))?;
}
Ok(())
}
#[test]
fn read_expanded_2() -> Result<(), Error> {
use netpod::timeunits::*;
let range = netpod::NanoRange {
beg: DAY - MS * 100,
end: DAY + MS * 0 + MS * 1501,
};
let res = read_expanded_for_range(range)?;
if res.0 != 3 {
Err(Error::with_msg(format!("unexpected number of events: {}", res.0)))?;
}
Ok(())
}
#[test]
fn read_expanded_3() -> Result<(), Error> {
use netpod::timeunits::*;
let range = netpod::NanoRange {
beg: DAY - MS * 1500,
end: DAY + MS * 0 + MS * 1501,
};
let res = read_expanded_for_range(range)?;
if res.0 != 4 {
Err(Error::with_msg(format!("unexpected number of events: {}", res.0)))?;
}
Ok(())
} }

View File

@@ -319,7 +319,7 @@ pub fn file_content_stream(
let n1 = file.read_buf(&mut buf).await?; let n1 = file.read_buf(&mut buf).await?;
let ts2 = Instant::now(); let ts2 = Instant::now();
if n1 == 0 { if n1 == 0 {
info!("file EOF"); trace!("file EOF");
break; break;
} }
else { else {

View File

@@ -1,4 +1,3 @@
use crate::eventvalues::EventValues;
use crate::minmaxavgdim1bins::MinMaxAvgDim1Bins; use crate::minmaxavgdim1bins::MinMaxAvgDim1Bins;
use crate::numops::NumOps; use crate::numops::NumOps;
use crate::xbinnedscalarevents::XBinnedScalarEvents; use crate::xbinnedscalarevents::XBinnedScalarEvents;