Add tests for MIGRATE command