Add tests for the upgrade step.