cmd/jujud: rework upgrade tests

Take advantage of extracted upgrade steps worker. The upgrade tests
are now significantly simpler and faster.
upgradeWorkerFactory refactored to upgradeWorkerContext

Somewhat cleaner.