Added tests for migration.