Fix flaky ExactlyOnce and PartialUpsertRebalance integration tests#17834

xiangfu0 wants to merge 2 commits intoapache:masterapache/pinot:masterfrom

xiangfu0:flaky-kafka-exactlyOnce-testxiangfu0/pinot:flaky-kafka-exactlyOnce-testCopy head branch name to clipboard