risingwave_common/catalog/
test_utils.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 risingwave_pb::data::DataType;
16use risingwave_pb::plan_common::{AdditionalColumn, ColumnDesc, ColumnDescVersion};
17
18pub trait ColumnDescTestExt {
19    /// Create a [`ColumnDesc`] with the given name and type.
20    ///
21    /// **Note: Only used for tests.**
22    fn new(data_type: DataType, name: &str, column_id: i32) -> Self;
23}
24
25impl ColumnDescTestExt for ColumnDesc {
26    fn new(data_type: DataType, name: &str, column_id: i32) -> Self {
27        Self {
28            column_type: Some(data_type),
29            column_id,
30            name: name.to_owned(),
31            additional_column: Some(AdditionalColumn { column_type: None }),
32            version: ColumnDescVersion::LATEST as _,
33            nullable: true,
34            ..Default::default()
35        }
36    }
37}
risingwave_common/catalog/test_utils.rs

risingwave_common/catalog/
test_utils.rs
