Add additional assertions to try to figure out why the tests are flaky when run in the test harness. Failure could not be reproduced locally even after 4000 run. PiperOrigin-RevId: 362282251
