Adding tests for upgrade.