Add test run_phase to set drain time