Adds unit tests for wait_time() hook.