Merge "Add unit tests for database migration"