Tests for the migration script