You need to actually assert on the return value of wait_until. The wait_for timed out for me but the test still passes.