Function setup_test_data

nodedb/src/engine/timeseries/query.rs:177–230

(dir: &TempDir)

    use tempfile::TempDir;

    fn setup_test_data(dir: &TempDir) -> (SegmentIndex, DictionaryRegistry) {
        let config = BucketConfig {
            l1_dir: dir.path().join("l1"),
            l2_dir: dir.path().join("l2"),
            ..Default::default()
        };
        let mut mgr = BucketManager::new(config);

        // Create metric segments for series 1.
        let mut enc1 = GorillaEncoder::new();
        for i in 0..50 {
            enc1.encode(1000 + i * 100, 10.0 + i as f64 * 0.5);
        }

        let mut enc2 = GorillaEncoder::new();
        for i in 0..50 {
            enc2.encode(6000 + i * 100, 50.0 + i as f64 * 0.3);
        }

        let flushed = vec![
            FlushedSeries {
                series_id: 1,
                kind: FlushedKind::Metric {
                    gorilla_block: enc1.finish(),
                    sample_count: 50,
                },
                min_ts: 1000,
                max_ts: 5900,
            },
            FlushedSeries {
                series_id: 1,
                kind: FlushedKind::Metric {
                    gorilla_block: enc2.finish(),
                    sample_count: 50,
                },
                min_ts: 6000,
                max_ts: 10900,
            },
        ];

        mgr.flush_to_l1(flushed, None).unwrap();

        // Clone the segment index (we need it after mgr is consumed).
        let idx = {
            let mut idx = SegmentIndex::new();
            // Re-read from disk to get accurate segment refs.
            for seg in mgr.segment_index().query(1, &TimeRange::new(0, i64::MAX)) {
                idx.add(1, seg.clone());
            }
            idx
        };

        (idx, DictionaryRegistry::new())
    }

    #[test]
    fn scan_metrics_full_range() {