Add missing @Player.State in action schedule

PiperOrigin-RevId: 309735092

testutils/src/main/java/com/google/android/exoplayer2/testutil/Action.java

@@ -890,13 +890,13 @@ public abstract class Action {
    */
    */
   public static final class WaitForPlaybackState extends Action {
   public static final class WaitForPlaybackState extends Action {
 
 
-    private final int targetPlaybackState;
+    @Player.State private final int targetPlaybackState;
 
 
     /**
     /**
      * @param tag A tag to use for logging.
      * @param tag A tag to use for logging.
      * @param targetPlaybackState The playback state to wait for.
      * @param targetPlaybackState The playback state to wait for.
      */
      */
-    public WaitForPlaybackState(String tag, int targetPlaybackState) {
+    public WaitForPlaybackState(String tag, @Player.State int targetPlaybackState) {
       super(tag, "WaitForPlaybackState");
       super(tag, "WaitForPlaybackState");
       this.targetPlaybackState = targetPlaybackState;
       this.targetPlaybackState = targetPlaybackState;
     }
     }

testutils/src/main/java/com/google/android/exoplayer2/testutil/ActionSchedule.java

@@ -537,7 +537,7 @@ public final class ActionSchedule {
      * @param targetPlaybackState The target playback state.
      * @param targetPlaybackState The target playback state.
      * @return The builder, for convenience.
      * @return The builder, for convenience.
      */
      */
-    public Builder waitForPlaybackState(int targetPlaybackState) {
+    public Builder waitForPlaybackState(@Player.State int targetPlaybackState) {
       return apply(new WaitForPlaybackState(tag, targetPlaybackState));
       return apply(new WaitForPlaybackState(tag, targetPlaybackState));
     }
     }