From 1374eb859f578bdf7d3aacee7f64b8fce4b7e7ab Mon Sep 17 00:00:00 2001
From: xx01cyx <caoyuanxin0531@outlook.com>
Date: Fri, 17 Jun 2022 15:54:02 +0800
Subject: [PATCH] fix ut

---
 src/frontend/src/handler/create_table.rs | 2 +-
 src/frontend/src/test_utils.rs           | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/handler/create_table.rs b/src/frontend/src/handler/create_table.rs
index 3adc289d389a8..85a5fb6c717d2 100644
--- a/src/frontend/src/handler/create_table.rs
+++ b/src/frontend/src/handler/create_table.rs
@@ -96,7 +96,7 @@ pub(crate) fn gen_create_table_plan(
     Ok((plan, source, table))
 }
 
-/// Generate a stream plan with `StreamSource` + `StreamMaterialize`, it ressembles a
+/// Generate a stream plan with `StreamSource` + `StreamMaterialize`, it resembles a
 /// `CREATE MATERIALIZED VIEW AS SELECT * FROM <source>`.
 pub(crate) fn gen_materialized_source_plan(
     context: OptimizerContextRef,
diff --git a/src/frontend/src/test_utils.rs b/src/frontend/src/test_utils.rs
index 281e32cf2f5bb..caf0dbad9acd7 100644
--- a/src/frontend/src/test_utils.rs
+++ b/src/frontend/src/test_utils.rs
@@ -28,6 +28,7 @@ use risingwave_pb::catalog::table::OptionalAssociatedSourceId;
 use risingwave_pb::catalog::{
     Database as ProstDatabase, Schema as ProstSchema, Source as ProstSource, Table as ProstTable,
 };
+use risingwave_pb::common::ParallelUnitMapping;
 use risingwave_pb::stream_plan::StreamFragmentGraph;
 use risingwave_pb::user::{GrantPrivilege, UserInfo};
 use risingwave_rpc_client::error::Result as RpcResult;
@@ -163,6 +164,11 @@ impl CatalogWriter for MockCatalogWriter {
         _graph: StreamFragmentGraph,
     ) -> Result<()> {
         table.id = self.gen_id();
+        table.mapping = Some(ParallelUnitMapping {
+            table_id: table.id,
+            original_indices: [0, 10, 20].to_vec(),
+            data: [1, 2, 3].to_vec(),
+        });
         self.catalog.write().create_table(&table);
         self.add_table_or_source_id(table.id, table.schema_id, table.database_id);
         Ok(())