adding unit tests for the migration