Skip to content

Conversation

@liviazhu
Copy link
Contributor

@liviazhu liviazhu commented Nov 4, 2025

What changes were proposed in this pull request?

The SPARK-40492: maintenance before unload test is occasionally flaky even after refactors in #52783. This PR rewrites the test by using a new state store provider that counts number of times maintenance is triggered to test the functionality more directly. It also removes the hooks that were only used by the old test.

Why are the changes needed?

Deflake the test

Does this PR introduce any user-facing change?

No

How was this patch tested?

Unit test, ran over 60 times

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

Generated-by: Claude 4

@volatile private var maintenancePaused: Boolean = false
private[spark] def setMaintenancePaused(maintPaused: Boolean): Unit = {
maintenancePaused = maintPaused
}
Copy link
Member

Choose a reason for hiding this comment

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

Thank you for making a PR, @liviazhu . However, the PR title looks wrong to me because this PR change the main source code under sql/core/src/main. Could you revise the PR title more properly?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Done!

@liviazhu liviazhu changed the title [SPARK-54078][SS] New test for StateStoreSuite SPARK-40492: maintenance before unload [SPARK-54078][SS] New test for StateStoreSuite SPARK-40492: maintenance before unload and remove infra from old test Nov 4, 2025
Copy link
Contributor

@anishshri-db anishshri-db left a comment

Choose a reason for hiding this comment

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

lgtm thanks

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