worker/diskformatter: fix tests
apiserver/diskmanager: update to state changes
worker/diskmanager: fix import grouping
cmd/jujud: fix DiskManager test