Call  on timeout argument for migrate command
Refactor tests for migrate command