Correct failing unit tests and add convenience method.
Adding unit tests for new migration script options.