Fix flaky unit test to pass post-submit checks#39562

shunping merged 1 commit into

apache:masterapache/beam:masterfrom

ian-Liaozy:masterian-Liaozy/beam:masterCopy head branch name to clipboard