Add tests for snap-back to midroll

These tests require longer content so that the ads are spaced apart far enough
to trigger preloading while content is playing (rather than immediately as soon
as a position is reported).

PiperOrigin-RevId: 309733826

extensions/ima/src/androidTest/java/com/google/android/exoplayer2/ext/ima/ImaPlaybackTest.java

@@ -28,6 +28,7 @@ import androidx.test.ext.junit.runners.AndroidJUnit4;
 import androidx.test.rule.ActivityTestRule;
 import com.google.android.exoplayer2.C;
 import com.google.android.exoplayer2.MediaItem;
+import com.google.android.exoplayer2.Player;
 import com.google.android.exoplayer2.Player.DiscontinuityReason;
 import com.google.android.exoplayer2.Player.EventListener;
 import com.google.android.exoplayer2.Player.TimelineChangeReason;
@@ -40,6 +41,7 @@ import com.google.android.exoplayer2.source.DefaultMediaSourceFactory;
 import com.google.android.exoplayer2.source.MediaSource;
 import com.google.android.exoplayer2.source.ads.AdsLoader.AdViewProvider;
 import com.google.android.exoplayer2.source.ads.AdsMediaSource;
+import com.google.android.exoplayer2.testutil.ActionSchedule;
 import com.google.android.exoplayer2.testutil.ExoHostedTest;
 import com.google.android.exoplayer2.testutil.HostActivity;
 import com.google.android.exoplayer2.testutil.TestUtil;
@@ -52,6 +54,7 @@ import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import org.checkerframework.checker.nullness.qual.MonotonicNonNull;
+import org.junit.Ignore;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -60,11 +63,14 @@ import org.junit.runner.RunWith;
 @RunWith(AndroidJUnit4.class)
 public final class ImaPlaybackTest {
 
+  private static final String TAG = "ImaPlaybackTest";
+
   private static final long TIMEOUT_MS = 5 * 60 * C.MILLIS_PER_SECOND;
 
-  private static final String CONTENT_URI =
+  private static final String CONTENT_URI_SHORT =
       "https://storage.googleapis.com/exoplayer-test-media-1/mp4/android-screens-10s.mp4";
-
+  private static final String CONTENT_URI_LONG =
+      "https://storage.googleapis.com/exoplayer-test-media-1/mp4/android-screens-25s.mp4";
   private static final AdId CONTENT = new AdId(C.INDEX_UNSET, C.INDEX_UNSET);
 
   @Rule public ActivityTestRule<HostActivity> testRule = new ActivityTestRule<>(HostActivity.class);
@@ -75,7 +81,7 @@ public final class ImaPlaybackTest {
         TestUtil.getString(/* context= */ testRule.getActivity(), "ad-responses/preroll.xml");
     AdId[] expectedAdIds = new AdId[] {ad(0), CONTENT};
     ImaHostedTest hostedTest =
-        new ImaHostedTest(Uri.parse(CONTENT_URI), adsResponse, expectedAdIds);
+        new ImaHostedTest(Uri.parse(CONTENT_URI_SHORT), adsResponse, expectedAdIds);
 
     testRule.getActivity().runTest(hostedTest, TIMEOUT_MS);
   }
@@ -87,7 +93,7 @@ public final class ImaPlaybackTest {
             /* context= */ testRule.getActivity(), "ad-responses/preroll_midroll6s_postroll.xml");
     AdId[] expectedAdIds = new AdId[] {ad(0), CONTENT, ad(1), CONTENT, ad(2), CONTENT};
     ImaHostedTest hostedTest =
-        new ImaHostedTest(Uri.parse(CONTENT_URI), adsResponse, expectedAdIds);
+        new ImaHostedTest(Uri.parse(CONTENT_URI_SHORT), adsResponse, expectedAdIds);
 
     testRule.getActivity().runTest(hostedTest, TIMEOUT_MS);
   }
@@ -99,11 +105,44 @@ public final class ImaPlaybackTest {
             /* context= */ testRule.getActivity(), "ad-responses/midroll1s_midroll7s.xml");
     AdId[] expectedAdIds = new AdId[] {CONTENT, ad(0), CONTENT, ad(1), CONTENT};
     ImaHostedTest hostedTest =
-        new ImaHostedTest(Uri.parse(CONTENT_URI), adsResponse, expectedAdIds);
+        new ImaHostedTest(Uri.parse(CONTENT_URI_SHORT), adsResponse, expectedAdIds);
 
     testRule.getActivity().runTest(hostedTest, TIMEOUT_MS);
   }
 
+  @Test
+  public void playbackWithMidrolls10And20WithSeekTo12_playsAdsAndContent() throws Exception {
+    String adsResponse =
+        TestUtil.getString(
+            /* context= */ testRule.getActivity(), "ad-responses/midroll10s_midroll20s.xml");
+    AdId[] expectedAdIds = new AdId[] {CONTENT, ad(0), CONTENT, ad(1), CONTENT};
+    ImaHostedTest hostedTest =
+        new ImaHostedTest(Uri.parse(CONTENT_URI_LONG), adsResponse, expectedAdIds);
+    hostedTest.setSchedule(
+        new ActionSchedule.Builder(TAG)
+            .waitForPlaybackState(Player.STATE_READY)
+            .seek(12 * C.MILLIS_PER_SECOND)
+            .build());
+    testRule.getActivity().runTest(hostedTest, TIMEOUT_MS);
+  }
+
+  @Ignore("The second ad doesn't preload so playback gets stuck. See [internal: b/155615925].")
+  @Test
+  public void playbackWithMidrolls10And20WithSeekTo18_playsAdsAndContent() throws Exception {
+    String adsResponse =
+        TestUtil.getString(
+            /* context= */ testRule.getActivity(), "ad-responses/midroll10s_midroll20s.xml");
+    AdId[] expectedAdIds = new AdId[] {CONTENT, ad(0), CONTENT, ad(1), CONTENT};
+    ImaHostedTest hostedTest =
+        new ImaHostedTest(Uri.parse(CONTENT_URI_LONG), adsResponse, expectedAdIds);
+    hostedTest.setSchedule(
+        new ActionSchedule.Builder(TAG)
+            .waitForPlaybackState(Player.STATE_READY)
+            .seek(18 * C.MILLIS_PER_SECOND)
+            .build());
+    testRule.getActivity().runTest(hostedTest, TIMEOUT_MS);
+  }
+
   private static AdId ad(int groupIndex) {
     return new AdId(groupIndex, /* indexInGroup= */ 0);
   }
@@ -181,8 +220,10 @@ public final class ImaPlaybackTest {
             @Override
             public void onPositionDiscontinuity(
                 EventTime eventTime, @DiscontinuityReason int reason) {
+              if (reason != Player.DISCONTINUITY_REASON_SEEK) {
                 maybeUpdateSeenAdIdentifiers();
               }
+            }
           });
       Context context = host.getApplicationContext();
       imaAdsLoader = new ImaAdsLoader.Builder(context).buildForAdsResponse(adsResponse);

testdata/src/test/assets/ad-responses/midroll10s_midroll20s.xml

@@ -0,0 +1,64 @@
+<vmap:VMAP xmlns:vmap="http://www.iab.net/videosuite/vmap" version="1.0">
+  <vmap:AdBreak timeOffset="00:00:10.000" breakType="linear" breakId="midroll-1">
+    <vmap:AdSource id="midroll-1-ad-1" allowMultipleAds="false" followRedirects="true">
+      <vmap:VASTAdData>
+        <VAST xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="vast.xsd" version="3.0">
+          <Ad id="710742616">
+            <InLine>
+              <AdSystem>GDFP</AdSystem>
+              <AdTitle>Midroll</AdTitle>
+              <Description>
+                <![CDATA[ Midroll ad ]]>
+              </Description>
+              <Creatives>
+                <Creative id="57861202456" sequence="1">
+                  <Linear>
+                    <Duration>00:00:05</Duration>
+                    <MediaFiles>
+                      <MediaFile id="GDFP" delivery="progressive" width="640" height="360" type="video/mp4" bitrate="450" scalable="true" maintainAspectRatio="true">
+                        <![CDATA[
+file:///android_asset/mp4/midroll-5s.mp4
+]]>
+                      </MediaFile>
+                    </MediaFiles>
+                  </Linear>
+                </Creative>
+              </Creatives>
+            </InLine>
+          </Ad>
+        </VAST>
+      </vmap:VASTAdData>
+    </vmap:AdSource>
+  </vmap:AdBreak>
+  <vmap:AdBreak timeOffset="00:00:20.000" breakType="linear" breakId="midroll-2">
+    <vmap:AdSource id="midroll-2-ad-1" allowMultipleAds="false" followRedirects="true">
+      <vmap:VASTAdData>
+        <VAST xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="vast.xsd" version="3.0">
+          <Ad id="710742616">
+            <InLine>
+              <AdSystem>GDFP</AdSystem>
+              <AdTitle>Midroll</AdTitle>
+              <Description>
+                <![CDATA[ Midroll ad ]]>
+              </Description>
+              <Creatives>
+                <Creative id="57861202456" sequence="1">
+                  <Linear>
+                    <Duration>00:00:05</Duration>
+                    <MediaFiles>
+                      <MediaFile id="GDFP" delivery="progressive" width="640" height="360" type="video/mp4" bitrate="450" scalable="true" maintainAspectRatio="true">
+                        <![CDATA[
+file:///android_asset/mp4/midroll-5s.mp4
+]]>
+                      </MediaFile>
+                    </MediaFiles>
+                  </Linear>
+                </Creative>
+              </Creatives>
+            </InLine>
+          </Ad>
+        </VAST>
+      </vmap:VASTAdData>
+    </vmap:AdSource>
+  </vmap:AdBreak>
+</vmap:VMAP>