Add tests for db:migrate task