[SearchView][Predictive Back] Fix issue where predictive back is not …

…enabled in SearchView after rotating the screen or restarting the activity when the SearchView is already expanded

Resolves #3594
Resolves #3604

PiperOrigin-RevId: 568869369

lib/java/com/google/android/material/search/SearchView.java

@@ -564,6 +564,7 @@ public void setupWithSearchBar(@Nullable SearchBar searchBar) {
     }
     updateNavigationIconIfNeeded();
     setUpBackgroundViewElevationOverlay();
+    updateListeningForBackCallbacks(getCurrentTransitionState());
   }
 
   /**
@@ -814,6 +815,10 @@ private void setTransitionState(
       listener.onStateChanged(this, previousState, state);
     }
 
+    updateListeningForBackCallbacks(state);
+  }
+
+  private void updateListeningForBackCallbacks(@NonNull TransitionState state) {
     // Only automatically handle back if we have a search bar to collapse to, and if back handling
     // is enabled for the SearchView.
     if (searchBar != null && backHandlingEnabled) {