Adding more tests for the migration script.