Fix flaky InvalidListPruneTest

In InvalidListPruneTest, prune state information is written in an async thread. This has lead to sleep statements in the test that lead to flakiness. We should use a lock or other deterministic mechanisms to wait for the state to be written to fix the flakiness.