fix integration tests

Signed-off-by: Bugen Zhao <[email protected]>
risingwavelabs · Sep 29, 2024 · 7a83d1a · 7a83d1a
1 parent b25d28e
commit 7a83d1a
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/src/tests/simulation/tests/integration_tests/scale/shared_source.rs b/src/tests/simulation/tests/integration_tests/scale/shared_source.rs
@@ -112,13 +112,13 @@ async fn test_shared_source() -> Result<()> {
         .collect_vec();
     validate_splits_aligned(&mut cluster).await?;
     expect_test::expect![[r#"
-        1 1 HASH {2} {} {SOURCE} 6
-        2 3 HASH {4,3} {3} {MVIEW} 6
-        3 3 HASH {5} {1} {SOURCE_SCAN} 6"#]]
+        1 1 HASH {2} {} {SOURCE} 6 256
+        2 3 HASH {4,3} {3} {MVIEW} 6 256
+        3 3 HASH {5} {1} {SOURCE_SCAN} 6 256"#]]
     .assert_eq(&cluster.run("select * from rw_fragments;").await?);
     expect_test::expect![[r#"
-        1 CREATED ADAPTIVE
-        3 CREATED ADAPTIVE"#]]
+        1 CREATED ADAPTIVE 256
+        3 CREATED ADAPTIVE 256"#]]
     .assert_eq(&cluster.run("select * from rw_table_fragments;").await?);
 
     // SourceBackfill cannot be scaled because of NoShuffle.
@@ -137,9 +137,9 @@ async fn test_shared_source() -> Result<()> {
         .await
         .unwrap();
     expect_test::expect![[r#"
-        1 1 HASH {2} {} {SOURCE} 6
-        2 3 HASH {4,3} {3} {MVIEW} 5
-        3 3 HASH {5} {1} {SOURCE_SCAN} 6"#]]
+        1 1 HASH {2} {} {SOURCE} 6 256
+        2 3 HASH {4,3} {3} {MVIEW} 5 256
+        3 3 HASH {5} {1} {SOURCE_SCAN} 6 256"#]]
     .assert_eq(&cluster.run("select * from rw_fragments;").await?);
 
     // source is the NoShuffle upstream. It can be scaled, and the downstream SourceBackfill will be scaled together.
@@ -156,13 +156,13 @@ async fn test_shared_source() -> Result<()> {
         .unwrap();
     validate_splits_aligned(&mut cluster).await?;
     expect_test::expect![[r#"
-        1 1 HASH {2} {} {SOURCE} 3
-        2 3 HASH {4,3} {3} {MVIEW} 5
-        3 3 HASH {5} {1} {SOURCE_SCAN} 3"#]]
+        1 1 HASH {2} {} {SOURCE} 3 256
+        2 3 HASH {4,3} {3} {MVIEW} 5 256
+        3 3 HASH {5} {1} {SOURCE_SCAN} 3 256"#]]
     .assert_eq(&cluster.run("select * from rw_fragments;").await?);
     expect_test::expect![[r#"
-        1 CREATED CUSTOM
-        3 CREATED CUSTOM"#]]
+        1 CREATED CUSTOM 256
+        3 CREATED CUSTOM 256"#]]
     .assert_eq(&cluster.run("select * from rw_table_fragments;").await?);
 
     // resolve_no_shuffle for backfill fragment is OK, which will scale the upstream together.
@@ -180,13 +180,13 @@ async fn test_shared_source() -> Result<()> {
         .unwrap();
     validate_splits_aligned(&mut cluster).await?;
     expect_test::expect![[r#"
-        1 1 HASH {2} {} {SOURCE} 7
-        2 3 HASH {4,3} {3} {MVIEW} 5
-        3 3 HASH {5} {1} {SOURCE_SCAN} 7"#]]
+        1 1 HASH {2} {} {SOURCE} 7 256
+        2 3 HASH {4,3} {3} {MVIEW} 5 256
+        3 3 HASH {5} {1} {SOURCE_SCAN} 7 256"#]]
     .assert_eq(&cluster.run("select * from rw_fragments;").await?);
     expect_test::expect![[r#"
-1 CREATED CUSTOM
-3 CREATED CUSTOM"#]]
+1 CREATED CUSTOM 256
+3 CREATED CUSTOM 256"#]]
     .assert_eq(&cluster.run("select * from rw_table_fragments;").await?);
     Ok(())
 }