Add unit test for migration script.