check correct state

appgate · May 21, 2024 · ee1eeb4 · ee1eeb4
1 parent f153f9d
commit ee1eeb4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmd/appliance/switch_partition.go b/cmd/appliance/switch_partition.go
@@ -190,6 +190,10 @@ func switchPartitionRunE(opts *options) error {
 		return fmt.Errorf("failed to get appliance stats: %w", err)
 	}
 
+	if err := api.ApplianceStats.WaitForApplianceStatus(ctx, *appliance, appliancepkg.StatusNotBusy, t); err != nil {
+		return err
+	}
+
 	if p != nil {
 		p.Wait()
 	}