risingwave_planner_test/
resolve_id.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::collections::HashMap;
16
17use anyhow::{Result, anyhow};
18use itertools::Itertools;
19
20use crate::{TestCase, TestInput};
21
22pub fn resolve_testcase_id(testcases: Vec<TestCase>) -> Result<Vec<TestCase>> {
23    let mut testcases_with_ids = HashMap::new();
24    for testcase in &testcases {
25        if let Some(id) = &testcase.input.id {
26            testcases_with_ids.insert(id.clone(), testcase.clone());
27        }
28    }
29
30    testcases
31        .into_iter()
32        .map(|testcase| {
33            let before_statements = if let Some(before) = &testcase.input.before {
34                Some(
35                    before
36                        .iter()
37                        .map(|id| {
38                            testcases_with_ids
39                                .get(id)
40                                .map(|case| case.sql().clone())
41                                .ok_or_else(|| anyhow!("failed to resolve {}: not found", id))
42                        })
43                        .try_collect()?,
44                )
45            } else {
46                None
47            };
48
49            Ok(TestCase {
50                input: TestInput {
51                    before_statements,
52                    ..testcase.input
53                },
54                ..testcase
55            })
56        })
57        .try_collect()
58}
risingwave_planner_test/resolve_id.rs

risingwave_planner_test/
resolve_id.rs
