MDL-36804: Add support for assignment upgrade tool