Skip to content

Commit

Permalink
fix integration tests
Browse files Browse the repository at this point in the history
Signed-off-by: Bugen Zhao <[email protected]>
  • Loading branch information
BugenZhao committed Sep 30, 2024
1 parent 264b372 commit 07ec687
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions src/tests/simulation/tests/integration_tests/scale/shared_source.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand All @@ -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.
Expand All @@ -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.
Expand All @@ -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(())
}

0 comments on commit 07ec687

Please sign in to comment.