Add a test for try_wait()