Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[SPARK-51096][SQL][TESTS] Splitting TransformWithStateSuite into UnsafeRow and Avro encoding suites #49815

Closed
wants to merge 3 commits into from

Conversation

ericm-db
Copy link
Contributor

@ericm-db ericm-db commented Feb 5, 2025

What changes were proposed in this pull request?

Splitting the TransformWithStateSuite into TransformWithStateUnsafeRowEncodingSuite and TransformWithStateAvroEncodingSuite, in order to remove it from the slow test category

Why are the changes needed?

To remove TransformWithState-related suites from the Slow test category

Does this PR introduce any user-facing change?

No

How was this patch tested?

Just moving tests around, no need to test

Was this patch authored or co-authored using generative AI tooling?

No

@ericm-db ericm-db changed the title Splitting TransformWithStateSuite into two different test suites with encoding Splitting TransformWithStateSuite into UnsafeRow and Avro encoding suites Feb 5, 2025
@ericm-db ericm-db changed the title Splitting TransformWithStateSuite into UnsafeRow and Avro encoding suites [SPARK-51096] Splitting TransformWithStateSuite into UnsafeRow and Avro encoding suites Feb 5, 2025
}
}

test("test that introducing TTL after restart fails query") {
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This applies to both modes right ?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yeah it does, but the expected error is different based on the encoding type. Couldn't find a good way to expect a different error in the test

@HyukjinKwon HyukjinKwon changed the title [SPARK-51096] Splitting TransformWithStateSuite into UnsafeRow and Avro encoding suites [SPARK-51096][SQL] Splitting TransformWithStateSuite into UnsafeRow and Avro encoding suites Feb 6, 2025
Copy link
Contributor

@HeartSaVioR HeartSaVioR left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for making this change. Only a nit.

@@ -1815,7 +1352,6 @@ class TransformWithStateSuite extends StateStoreMetricsTest
TransformWithStateSuiteUtils.NUM_SHUFFLE_PARTITIONS.toString) {
withTempDir { checkpointDir =>
// When Avro is used, we want to set the StructFields to nullable
val shouldBeNullable = usingAvroEncoding()
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks like this is coupled with #49751. Shall we just do pure refactor here so that we can merge this first? Looks like #49751 needs some time to review.

@HeartSaVioR
Copy link
Contributor

@ericm-db
https://github.com/ericm-db/spark/actions/runs/13167171290/job/36750002304

Looks like failure happens from refactored suite?

Copy link
Contributor

@HeartSaVioR HeartSaVioR left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

+1 pending CI

@HeartSaVioR
Copy link
Contributor

The CI failure is not relevant - it only failed from pyspark-pandas and this PR only touched "Scala tests".

@HeartSaVioR HeartSaVioR changed the title [SPARK-51096][SQL] Splitting TransformWithStateSuite into UnsafeRow and Avro encoding suites [SPARK-51096][SQL][TESTS] Splitting TransformWithStateSuite into UnsafeRow and Avro encoding suites Feb 6, 2025
@HeartSaVioR
Copy link
Contributor

Thanks! Merging to master/4.0.

HeartSaVioR pushed a commit that referenced this pull request Feb 6, 2025
…feRow and Avro encoding suites

### What changes were proposed in this pull request?

Splitting the TransformWithStateSuite into TransformWithStateUnsafeRowEncodingSuite and TransformWithStateAvroEncodingSuite, in order to remove it from the slow test category

### Why are the changes needed?

To remove TransformWithState-related suites from the Slow test category

### Does this PR introduce _any_ user-facing change?

No

### How was this patch tested?

Just moving tests around, no need to test

### Was this patch authored or co-authored using generative AI tooling?

No

Closes #49815 from ericm-db/split-tws-suite.

Authored-by: Eric Marnadi <[email protected]>
Signed-off-by: Jungtaek Lim <[email protected]>
(cherry picked from commit 9dbbb5b)
Signed-off-by: Jungtaek Lim <[email protected]>
zecookiez pushed a commit to zecookiez/spark that referenced this pull request Feb 6, 2025
…feRow and Avro encoding suites

### What changes were proposed in this pull request?

Splitting the TransformWithStateSuite into TransformWithStateUnsafeRowEncodingSuite and TransformWithStateAvroEncodingSuite, in order to remove it from the slow test category

### Why are the changes needed?

To remove TransformWithState-related suites from the Slow test category

### Does this PR introduce _any_ user-facing change?

No

### How was this patch tested?

Just moving tests around, no need to test

### Was this patch authored or co-authored using generative AI tooling?

No

Closes apache#49815 from ericm-db/split-tws-suite.

Authored-by: Eric Marnadi <[email protected]>
Signed-off-by: Jungtaek Lim <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants