Added tests for MigrationController