Migrate ParameterizedAudioExportTest to use trackTypes

PiperOrigin-RevId: 819247538

Author: shahdDaghash

libraries/transformer/src/test/java/androidx/media3/transformer/ParameterizedAudioExportTest.java

@@ -16,6 +16,8 @@
 
 package androidx.media3.transformer;
 
+import static androidx.media3.common.C.TRACK_TYPE_AUDIO;
+import static androidx.media3.common.C.TRACK_TYPE_VIDEO;
 import static androidx.media3.test.utils.robolectric.ShadowMediaCodecConfig.CODEC_INFO_RAW;
 import static androidx.media3.transformer.TestUtil.ASSET_URI_PREFIX;
 import static androidx.media3.transformer.TestUtil.FILE_AUDIO_RAW;
@@ -27,6 +29,7 @@
 import static java.util.stream.Collectors.toList;
 
 import android.content.Context;
+import androidx.media3.common.C.TrackType;
 import androidx.media3.common.MediaItem;
 import androidx.media3.common.util.Util;
 import androidx.media3.test.utils.DumpFileAsserts;
@@ -60,17 +63,20 @@ public final class ParameterizedAudioExportTest {
   @Parameters(name = "{0}")
   public static List<SequenceConfig> params() {
     return new ImmutableList.Builder<SequenceConfig>()
-        .addAll(getAllPermutationsOfAllCombinations(AUDIO_ITEMS))
-        .addAll(getAllPermutationsOfAllCombinations(AUDIO_VIDEO_ITEMS))
+        .addAll(getAllPermutationsOfAllCombinations(AUDIO_ITEMS, ImmutableSet.of(TRACK_TYPE_AUDIO)))
+        .addAll(
+            getAllPermutationsOfAllCombinations(
+                AUDIO_VIDEO_ITEMS, ImmutableSet.of(TRACK_TYPE_AUDIO, TRACK_TYPE_VIDEO)))
         .build();
   }
 
-  private static List<SequenceConfig> getAllPermutationsOfAllCombinations(Set<ItemConfig> items) {
+  private static List<SequenceConfig> getAllPermutationsOfAllCombinations(
+      Set<ItemConfig> items, Set<@TrackType Integer> trackTypes) {
     return Sets.powerSet(items).stream()
         .filter(s -> !s.isEmpty())
         .flatMap(s -> Collections2.permutations(s).stream())
         .filter(permutation -> permutation.size() < 4)
-        .map(SequenceConfig::new)
+        .map(permutation -> new SequenceConfig(trackTypes, permutation))
         .collect(toList());
   }
 
@@ -151,9 +157,11 @@ public void export() throws Exception {
   }
 
   private static class SequenceConfig {
+    private final ImmutableSet<@TrackType Integer> trackTypes;
     private final List<ItemConfig> itemConfigs;
 
-    public SequenceConfig(List<ItemConfig> itemConfigs) {
+    private SequenceConfig(Set<@TrackType Integer> trackTypes, List<ItemConfig> itemConfigs) {
+      this.trackTypes = ImmutableSet.copyOf(trackTypes);
       this.itemConfigs = itemConfigs;
     }
 
@@ -164,9 +172,7 @@ public Composition asComposition() {
       }
 
       return new Composition.Builder(
-              new EditedMediaItemSequence.Builder(items.build())
-                  .experimentalSetForceAudioTrack(true)
-                  .build())
+              new EditedMediaItemSequence.Builder(trackTypes).addItems(items.build()).build())
           .setTransmuxVideo(true)
           .build();
     }