chore(derive): test channel reader flushing (#661)

anton-rs · Oct 8, 2024 · 0a1ca04 · 0a1ca04
1 parent 01cb17c
commit 0a1ca04
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/crates/derive/src/stages/channel_reader.rs b/crates/derive/src/stages/channel_reader.rs
@@ -264,6 +264,15 @@ mod test {
         data.into()
     }
 
+    #[tokio::test]
+    async fn test_flush_channel_reader() {
+        let mock = MockChannelReaderProvider::new(vec![Ok(Some(new_compressed_batch_data()))]);
+        let mut reader = ChannelReader::new(mock, Arc::new(RollupConfig::default()));
+        reader.next_batch = Some(BatchReader::from(new_compressed_batch_data()));
+        reader.flush_channel().await.unwrap();
+        assert!(reader.next_batch.is_none());
+    }
+
     #[tokio::test]
     async fn test_reset_channel_reader() {
         let mock = MockChannelReaderProvider::new(vec![Ok(None)]);