Fix race condition that causes the test flaky.#36235

shunping merged 2 commits into

apache:masterapache/beam:masterfrom

shunping:fix-test-server-run-then-cancelshunping/beam:fix-test-server-run-then-cancelCopy head branch name to clipboard