Cast string to details.State

elastic · Dec 14, 2023 · 05cb012 · 05cb012
1 parent 77488a7
commit 05cb012
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/testing/integration/upgrade_rollback_test.go b/testing/integration/upgrade_rollback_test.go
@@ -136,7 +136,7 @@ inputs:
 		require.NoError(t, err)
 
 		require.NotNil(t, state.UpgradeDetails)
-		require.Equal(t, details.StateRollback, state.UpgradeDetails.State)
+		require.Equal(t, details.StateRollback, details.State(state.UpgradeDetails.State))
 	}
 
 	// rollback should stop the watcher
@@ -268,7 +268,7 @@ func TestStandaloneUpgradeRollbackOnRestarts(t *testing.T) {
 		require.NoError(t, err)
 
 		require.NotNil(t, state.UpgradeDetails)
-		require.Equal(t, details.StateRollback, state.UpgradeDetails.State)
+		require.Equal(t, details.StateRollback, details.State(state.UpgradeDetails.State))
 	}
 
 	// rollback should stop the watcher