From ebf7920f993416d63e73f9b1e77635489142a16d Mon Sep 17 00:00:00 2001
From: yohplala <pierre.juillard@gmail.com>
Date: Tue, 18 Jun 2024 18:34:08 +0200
Subject: [PATCH] Testing retrieval of 'pre_buffer'.

---
 tests/test_aggstream/test_aggstream_advanced.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/tests/test_aggstream/test_aggstream_advanced.py b/tests/test_aggstream/test_aggstream_advanced.py
index 02b167d..fc63e58 100644
--- a/tests/test_aggstream/test_aggstream_advanced.py
+++ b/tests/test_aggstream/test_aggstream_advanced.py
@@ -406,6 +406,22 @@ def check(on, buffer):
     assert streamagg_md_key1[KEY_PRE_BUFFER] == pre_buffer_ref
     streamagg_md_key2 = store[key2]._oups_metadata[KEY_AGGSTREAM]
     assert streamagg_md_key2[KEY_PRE_BUFFER] == pre_buffer_ref
+    # Testing retrieval of 'pre_buffer'.
+    as_2 = AggStream(
+        ordered_on=ordered_on,
+        store=store,
+        keys={
+            filter1: {key1: deepcopy(key1_cf)},
+            filter2: {key2: deepcopy(key2_cf)},
+        },
+        filters={
+            filter1: [(filter_on, "==", True)],
+            filter2: [(filter_on, "==", False)],
+        },
+        max_row_group_size=max_row_group_size,
+        pre=check,
+    )
+    assert as_2.seed_config[KEY_PRE_BUFFER] == pre_buffer_ref
 
 
 def post(buffer: dict, bin_res: DataFrame, snap_res: DataFrame):