WIP

streams/src/test/collect.rs

@@ -1,20 +1,30 @@
+use crate::collect::Collect;
 use crate::test::runfut;
+use crate::transform::EventsToTimeBinnable;
+use crate::transform::TimeBinnableToCollectable;
 use err::Error;
 use futures_util::stream;
 use items_0::streamitem::sitem_data;
+use items_0::streamitem::RangeCompletableItem;
+use items_0::streamitem::StreamItem;
 use items_0::WithLen;
 use items_2::eventsdim0::EventsDim0CollectorOutput;
+use items_2::streams::PlainEventStream;
 use items_2::testgen::make_some_boxed_d0_f32;
 use netpod::timeunits::SEC;
 use std::time::Duration;
 use std::time::Instant;
 
 #[test]
-fn collect_channel_events() -> Result<(), Error> {
+fn collect_channel_events_00() -> Result<(), Error> {
     let fut = async {
         let evs0 = make_some_boxed_d0_f32(20, SEC * 10, SEC * 1, 0, 28736487);
         let evs1 = make_some_boxed_d0_f32(20, SEC * 30, SEC * 1, 0, 882716583);
-        let stream = stream::iter(vec![sitem_data(evs0), sitem_data(evs1)]);
+        let stream = stream::iter(vec![
+            sitem_data(evs0),
+            sitem_data(evs1),
+            Ok(StreamItem::DataItem(RangeCompletableItem::RangeComplete)),
+        ]);
         let deadline = Instant::now() + Duration::from_millis(4000);
         let events_max = 10000;
         let res = crate::collect::collect(stream, deadline, events_max, None, None).await?;
@@ -30,3 +40,32 @@ fn collect_channel_events() -> Result<(), Error> {
     };
     runfut(fut)
 }
+
+#[test]
+fn collect_channel_events_01() -> Result<(), Error> {
+    let fut = async {
+        let evs0 = make_some_boxed_d0_f32(20, SEC * 10, SEC * 1, 0, 28736487);
+        let evs1 = make_some_boxed_d0_f32(20, SEC * 30, SEC * 1, 0, 882716583);
+        let stream = stream::iter(vec![
+            sitem_data(evs0),
+            sitem_data(evs1),
+            Ok(StreamItem::DataItem(RangeCompletableItem::RangeComplete)),
+        ]);
+        // TODO build like in request code
+        let deadline = Instant::now() + Duration::from_millis(4000);
+        let events_max = 10000;
+        let stream = PlainEventStream::new(stream);
+        let stream = EventsToTimeBinnable::new(stream);
+        let stream = TimeBinnableToCollectable::new(stream);
+        let res = Collect::new(stream, deadline, events_max, None, None).await?;
+        if let Some(res) = res.as_any_ref().downcast_ref::<EventsDim0CollectorOutput<f32>>() {
+            eprintln!("Great, a match");
+            eprintln!("{res:?}");
+            assert_eq!(res.len(), 40);
+        } else {
+            return Err(Error::with_msg(format!("bad type of collected result")));
+        }
+        Ok(())
+    };
+    runfut(fut)
+}