Fix check write for spark 3.5 unit tests

[wForget]

:mag: Description

Issue References 🔗

KyuubiSparkSQLExtensionTest#withListener in Spark 3.5 may not work, make the following adjustments:

• skip AdaptiveSparkPlanExec when collecting writes
• execute callback function in main thread to avoid assert failed not working
• add collectRebalancePartitions methoad
• fix unit tests

Describe Your Solution 🔧

Types of changes :bookmark:

• [X] Bugfix (non-breaking change which fixes an issue)
• [ ] New feature (non-breaking change which adds functionality)
• [ ] Breaking change (fix or feature that would cause existing functionality to change)

Test Plan 🧪

Behavior Without This Pull Request :coffin:

Behavior With This Pull Request :tada:

Related Unit Tests

---

Checklist 📝

• [X] This patch was not authored or co-authored using Generative Tooling

Be nice. Be informative.

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 0.00%. Comparing base (b4838b4) to head (a989de5).

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #6793 +/- ## ====================================== Coverage 0.00% 0.00% ====================================== Files 687 687 Lines 42442 42442 Branches 5793 5793 ====================================== Misses 42442 42442 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.