Remove flacky test and replace with Mocking test (#80159)
This PR removes the flaky test that was introduce in the previous commit, and replace it with a mocking test. The actual test of queue depth is added in the next commit.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80159
Approved by: https://github.com/robieta