Add UT for TimedWait.