From b183bfb0cb16172c763de7cddaeb784e138a1636 Mon Sep 17 00:00:00 2001
From: kimvde <kimvde@google.com>
Date: Sun, 12 Jan 2025 23:59:04 -0800
Subject: [PATCH] CompositionPlayerTest: use correct video duration

PiperOrigin-RevId: 714847208
---
 .../transformer/CompositionPlayerTest.java    | 20 +++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/libraries/transformer/src/androidTest/java/androidx/media3/transformer/CompositionPlayerTest.java b/libraries/transformer/src/androidTest/java/androidx/media3/transformer/CompositionPlayerTest.java
index 9b4699e0fe..7ce91285f2 100644
--- a/libraries/transformer/src/androidTest/java/androidx/media3/transformer/CompositionPlayerTest.java
+++ b/libraries/transformer/src/androidTest/java/androidx/media3/transformer/CompositionPlayerTest.java
@@ -130,7 +130,7 @@ public class CompositionPlayerTest {
               new Composition.Builder(
                       new EditedMediaItemSequence.Builder(
                               new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-                                  .setDurationUs(1_000_000)
+                                  .setDurationUs(MP4_ASSET.videoDurationUs)
                                   .build())
                           .build())
                   .build());
@@ -151,7 +151,7 @@ public class CompositionPlayerTest {
               new Composition.Builder(
                       new EditedMediaItemSequence.Builder(
                               new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-                                  .setDurationUs(1_000_000)
+                                  .setDurationUs(MP4_ASSET.videoDurationUs)
                                   .build())
                           .build())
                   .build());
@@ -176,7 +176,7 @@ public class CompositionPlayerTest {
               new Composition.Builder(
                       new EditedMediaItemSequence.Builder(
                               new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-                                  .setDurationUs(1_000_000)
+                                  .setDurationUs(MP4_ASSET.videoDurationUs)
                                   .build())
                           .build())
                   .build());
@@ -199,7 +199,7 @@ public class CompositionPlayerTest {
               new Composition.Builder(
                       new EditedMediaItemSequence.Builder(
                               new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-                                  .setDurationUs(1_000_000)
+                                  .setDurationUs(MP4_ASSET.videoDurationUs)
                                   .build())
                           .build())
                   .build());
@@ -345,7 +345,7 @@ public class CompositionPlayerTest {
 
     EditedMediaItem video =
         new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-            .setDurationUs(1_000_000)
+            .setDurationUs(MP4_ASSET.videoDurationUs)
             .build();
 
     instrumentation.runOnMainSync(
@@ -370,7 +370,7 @@ public class CompositionPlayerTest {
     PlayerTestListener listener = new PlayerTestListener(TEST_TIMEOUT_MS);
     EditedMediaItem video =
         new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-            .setDurationUs(1_000_000)
+            .setDurationUs(MP4_ASSET.videoDurationUs)
             .build();
     EditedMediaItem image =
         new EditedMediaItem.Builder(
@@ -405,7 +405,7 @@ public class CompositionPlayerTest {
             TestSpeedProvider.createWithStartTimes(new long[] {0}, new float[] {2f}));
     EditedMediaItem video =
         new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-            .setDurationUs(1_000_000)
+            .setDurationUs(MP4_ASSET.videoDurationUs)
             .setEffects(
                 new Effects(ImmutableList.of(effects.first), ImmutableList.of(effects.second)))
             .build();
@@ -434,7 +434,7 @@ public class CompositionPlayerTest {
             TestSpeedProvider.createWithStartTimes(new long[] {0}, new float[] {0.5f}));
     EditedMediaItem video =
         new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-            .setDurationUs(1_000_000)
+            .setDurationUs(MP4_ASSET.videoDurationUs)
             .setEffects(
                 new Effects(ImmutableList.of(effects.first), ImmutableList.of(effects.second)))
             .build();
@@ -460,7 +460,7 @@ public class CompositionPlayerTest {
     PlayerTestListener listener = new PlayerTestListener(TEST_TIMEOUT_MS);
     EditedMediaItem video =
         new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-            .setDurationUs(1_000_000)
+            .setDurationUs(MP4_ASSET.videoDurationUs)
             .build();
 
     instrumentation.runOnMainSync(
@@ -497,7 +497,7 @@ public class CompositionPlayerTest {
     PlayerTestListener playerTestListener = new PlayerTestListener(TEST_TIMEOUT_MS);
     EditedMediaItem video =
         new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
-            .setDurationUs(1_000_000)
+            .setDurationUs(MP4_ASSET.videoDurationUs)
             .build();
     instrumentation.runOnMainSync(
         () -> {