mirror of
https://github.com/androidx/media.git
synced 2025-04-30 06:46:50 +08:00
Add EditingMetricsCollector test for export success
PiperOrigin-RevId: 727826326 (cherry picked from commit 3ce6a2e6b8b0b7147a06c5ec38aaf2ee36282720)
This commit is contained in:
shahddaghash
oceanjules
parent
e58d9120bc
commit
16add9922d
@ -0,0 +1,227 @@
|
||||
/*
|
||||
* Copyright 2025 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package androidx.media3.transformer;
|
||||
|
||||
import static androidx.media3.common.util.Util.usToMs;
|
||||
import static androidx.media3.transformer.AndroidTestUtil.JPG_ASSET;
|
||||
import static androidx.media3.transformer.AndroidTestUtil.MP4_ASSET;
|
||||
import static androidx.media3.transformer.AndroidTestUtil.assumeFormatsSupported;
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
import static org.junit.Assume.assumeTrue;
|
||||
|
||||
import android.content.Context;
|
||||
import android.media.metrics.EditingEndedEvent;
|
||||
import android.media.metrics.MediaItemInfo;
|
||||
import android.util.Size;
|
||||
import androidx.media3.common.MediaItem;
|
||||
import androidx.media3.common.MediaLibraryInfo;
|
||||
import androidx.media3.common.util.Util;
|
||||
import androidx.test.ext.junit.runners.AndroidJUnit4;
|
||||
import androidx.test.platform.app.InstrumentationRegistry;
|
||||
import java.util.concurrent.atomic.AtomicReference;
|
||||
import org.junit.Before;
|
||||
import org.junit.Rule;
|
||||
import org.junit.Test;
|
||||
import org.junit.rules.TestName;
|
||||
import org.junit.runner.RunWith;
|
||||
|
||||
/** Instrumentation tests for metrics reporting using {@link EditingMetricsCollector} */
|
||||
@RunWith(AndroidJUnit4.class)
|
||||
public class EditingMetricsCollectorTest {
|
||||
private static final String EXPORTER_NAME =
|
||||
"androidx.media3.media3-transformer:" + MediaLibraryInfo.VERSION;
|
||||
|
||||
@Rule public final TestName testName = new TestName();
|
||||
private final Context context = InstrumentationRegistry.getInstrumentation().getContext();
|
||||
private String testId;
|
||||
|
||||
@Before
|
||||
public void setUpTestId() {
|
||||
testId = testName.getMethodName();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void exportSuccess_populatesEditingEndedEvent() throws Exception {
|
||||
assumeTrue("Reporting metrics requires API 35", Util.SDK_INT >= 35);
|
||||
assumeFormatsSupported(
|
||||
context,
|
||||
testId,
|
||||
/* inputFormat= */ MP4_ASSET.videoFormat,
|
||||
/* outputFormat= */ MP4_ASSET.videoFormat);
|
||||
AtomicReference<EditingEndedEvent> editingEndedEventAtomicReference = new AtomicReference<>();
|
||||
Transformer transformer =
|
||||
new Transformer.Builder(context)
|
||||
.setUsePlatformDiagnostics(true)
|
||||
.setMetricsReporterFactory(
|
||||
new TestMetricsReporterFactory(context, editingEndedEventAtomicReference::set))
|
||||
.build();
|
||||
EditedMediaItem audioVideoItem =
|
||||
new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri)).build();
|
||||
EditedMediaItem imageItem =
|
||||
new EditedMediaItem.Builder(
|
||||
new MediaItem.Builder().setUri(JPG_ASSET.uri).setImageDurationMs(1500).build())
|
||||
.setFrameRate(30)
|
||||
.build();
|
||||
EditedMediaItemSequence videoImageSequence =
|
||||
new EditedMediaItemSequence.Builder(audioVideoItem, imageItem).build();
|
||||
EditedMediaItemSequence audioSequence =
|
||||
new EditedMediaItemSequence.Builder(
|
||||
new EditedMediaItem.Builder(MediaItem.fromUri(MP4_ASSET.uri))
|
||||
.setRemoveVideo(true)
|
||||
.build())
|
||||
.build();
|
||||
Composition composition = new Composition.Builder(videoImageSequence, audioSequence).build();
|
||||
|
||||
ExportTestResult exportTestResult =
|
||||
new TransformerAndroidTestRunner.Builder(context, transformer)
|
||||
.build()
|
||||
.run(testId, composition);
|
||||
|
||||
EditingEndedEvent editingEndedEvent = editingEndedEventAtomicReference.get();
|
||||
assertThat(editingEndedEvent.getFinalState())
|
||||
.isEqualTo(EditingEndedEvent.FINAL_STATE_SUCCEEDED);
|
||||
assertThat(editingEndedEvent.getTimeSinceCreatedMillis()).isAtLeast(0);
|
||||
assertThat(editingEndedEvent.getExporterName()).isEqualTo(EXPORTER_NAME);
|
||||
assertThat(editingEndedEvent.getMuxerName()).isEqualTo(DefaultMuxer.MUXER_NAME);
|
||||
assertThat(editingEndedEvent.getFinalProgressPercent()).isEqualTo(100);
|
||||
// Assert video input media item information
|
||||
MediaItemInfo firstMediaItemInfo = editingEndedEvent.getInputMediaItemInfos().get(0);
|
||||
ExportResult.ProcessedInput firstProcessedInput =
|
||||
exportTestResult.exportResult.processedInputs.get(0);
|
||||
assertThat(firstMediaItemInfo.getClipDurationMillis())
|
||||
.isEqualTo(usToMs(firstProcessedInput.durationUs));
|
||||
assertThat(firstMediaItemInfo.getDataTypes())
|
||||
.isEqualTo(MediaItemInfo.DATA_TYPE_VIDEO | MediaItemInfo.DATA_TYPE_AUDIO);
|
||||
assertThat(firstMediaItemInfo.getContainerMimeType())
|
||||
.isEqualTo(firstProcessedInput.videoFormat.containerMimeType);
|
||||
assertThat(firstMediaItemInfo.getSampleMimeTypes().get(0))
|
||||
.isEqualTo(firstProcessedInput.videoFormat.sampleMimeType);
|
||||
assertThat(firstMediaItemInfo.getVideoFrameRate())
|
||||
.isEqualTo(firstProcessedInput.videoFormat.frameRate);
|
||||
assertThat(firstMediaItemInfo.getVideoSize())
|
||||
.isEqualTo(
|
||||
new Size(
|
||||
firstProcessedInput.videoFormat.width, firstProcessedInput.videoFormat.height));
|
||||
assertThat(firstMediaItemInfo.getSampleMimeTypes().get(1))
|
||||
.isEqualTo(firstProcessedInput.audioFormat.sampleMimeType);
|
||||
assertThat(firstMediaItemInfo.getAudioChannelCount())
|
||||
.isEqualTo(firstProcessedInput.audioFormat.channelCount);
|
||||
assertThat(firstMediaItemInfo.getAudioSampleRateHz())
|
||||
.isEqualTo(firstProcessedInput.audioFormat.sampleRate);
|
||||
// Assert image input media item and its silent audio
|
||||
MediaItemInfo secondMediaItemInfo = editingEndedEvent.getInputMediaItemInfos().get(1);
|
||||
ExportResult.ProcessedInput secondProcessedInput =
|
||||
exportTestResult.exportResult.processedInputs.get(1);
|
||||
assertThat(secondMediaItemInfo.getClipDurationMillis())
|
||||
.isEqualTo(usToMs(secondProcessedInput.durationUs));
|
||||
assertThat(secondMediaItemInfo.getDataTypes())
|
||||
.isEqualTo(MediaItemInfo.DATA_TYPE_IMAGE | MediaItemInfo.DATA_TYPE_AUDIO);
|
||||
assertThat(secondMediaItemInfo.getContainerMimeType())
|
||||
.isEqualTo(secondProcessedInput.videoFormat.containerMimeType);
|
||||
assertThat(secondMediaItemInfo.getSampleMimeTypes().get(0))
|
||||
.isEqualTo(secondProcessedInput.videoFormat.sampleMimeType);
|
||||
assertThat(secondMediaItemInfo.getVideoFrameRate())
|
||||
.isEqualTo(secondProcessedInput.videoFormat.frameRate);
|
||||
assertThat(secondMediaItemInfo.getVideoSize())
|
||||
.isEqualTo(
|
||||
new Size(
|
||||
secondProcessedInput.videoFormat.width, secondProcessedInput.videoFormat.height));
|
||||
assertThat(secondMediaItemInfo.getSampleMimeTypes().get(1))
|
||||
.isEqualTo(secondProcessedInput.audioFormat.sampleMimeType);
|
||||
assertThat(secondMediaItemInfo.getAudioChannelCount())
|
||||
.isEqualTo(secondProcessedInput.audioFormat.channelCount);
|
||||
assertThat(secondMediaItemInfo.getAudioSampleRateHz())
|
||||
.isEqualTo(secondProcessedInput.audioFormat.sampleRate);
|
||||
// Assert audio input media item information
|
||||
MediaItemInfo thirdMediaItemInfo = editingEndedEvent.getInputMediaItemInfos().get(2);
|
||||
ExportResult.ProcessedInput thirdProcessedInput =
|
||||
exportTestResult.exportResult.processedInputs.get(2);
|
||||
assertThat(thirdMediaItemInfo.getClipDurationMillis())
|
||||
.isEqualTo(usToMs(thirdProcessedInput.durationUs));
|
||||
assertThat(thirdMediaItemInfo.getDataTypes()).isEqualTo(MediaItemInfo.DATA_TYPE_AUDIO);
|
||||
assertThat(thirdMediaItemInfo.getSampleMimeTypes().get(0))
|
||||
.isEqualTo(thirdProcessedInput.audioFormat.sampleMimeType);
|
||||
assertThat(thirdMediaItemInfo.getAudioChannelCount())
|
||||
.isEqualTo(thirdProcessedInput.audioFormat.channelCount);
|
||||
assertThat(thirdMediaItemInfo.getAudioSampleRateHz())
|
||||
.isEqualTo(thirdProcessedInput.audioFormat.sampleRate);
|
||||
// Assert output media item information
|
||||
MediaItemInfo outputMediaItemInfo = editingEndedEvent.getOutputMediaItemInfo();
|
||||
assertThat(outputMediaItemInfo).isNotNull();
|
||||
assertThat(outputMediaItemInfo.getDurationMillis())
|
||||
.isEqualTo(exportTestResult.exportResult.durationMs);
|
||||
assertThat(outputMediaItemInfo.getSampleMimeTypes()).isNotEmpty();
|
||||
assertThat(outputMediaItemInfo.getAudioChannelCount())
|
||||
.isEqualTo(exportTestResult.exportResult.channelCount);
|
||||
assertThat(outputMediaItemInfo.getAudioSampleRateHz())
|
||||
.isEqualTo(exportTestResult.exportResult.sampleRate);
|
||||
assertThat(outputMediaItemInfo.getCodecNames().get(0))
|
||||
.isEqualTo(exportTestResult.exportResult.audioEncoderName);
|
||||
assertThat(outputMediaItemInfo.getCodecNames().get(1))
|
||||
.isEqualTo(exportTestResult.exportResult.videoEncoderName);
|
||||
assertThat(outputMediaItemInfo.getVideoSampleCount())
|
||||
.isEqualTo(exportTestResult.exportResult.videoFrameCount);
|
||||
assertThat(outputMediaItemInfo.getVideoSize())
|
||||
.isEqualTo(
|
||||
new Size(exportTestResult.exportResult.width, exportTestResult.exportResult.height));
|
||||
}
|
||||
|
||||
private static final class TestMetricsReporterFactory
|
||||
implements EditingMetricsCollector.MetricsReporter.Factory {
|
||||
|
||||
private final EditingMetricsCollector.MetricsReporter.Factory wrappedMetricsReporterFactory;
|
||||
private final TestMetricsReporter.Listener listener;
|
||||
|
||||
public TestMetricsReporterFactory(Context context, TestMetricsReporter.Listener listener) {
|
||||
this.wrappedMetricsReporterFactory =
|
||||
new EditingMetricsCollector.DefaultMetricsReporter.Factory(context);
|
||||
this.listener = listener;
|
||||
}
|
||||
|
||||
@Override
|
||||
public EditingMetricsCollector.MetricsReporter create() {
|
||||
return new TestMetricsReporter(wrappedMetricsReporterFactory.create(), listener);
|
||||
}
|
||||
}
|
||||
|
||||
private static final class TestMetricsReporter
|
||||
implements EditingMetricsCollector.MetricsReporter {
|
||||
public interface Listener {
|
||||
void onMetricsReported(EditingEndedEvent editingEndedEvent);
|
||||
}
|
||||
|
||||
private final EditingMetricsCollector.MetricsReporter wrappedMetricsReporter;
|
||||
private final TestMetricsReporter.Listener listener;
|
||||
|
||||
private TestMetricsReporter(
|
||||
EditingMetricsCollector.MetricsReporter metricsReporter,
|
||||
TestMetricsReporter.Listener listener) {
|
||||
this.wrappedMetricsReporter = metricsReporter;
|
||||
this.listener = listener;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void reportMetrics(EditingEndedEvent editingEndedEvent) {
|
||||
listener.onMetricsReported(editingEndedEvent);
|
||||
wrappedMetricsReporter.reportMetrics(editingEndedEvent);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() throws Exception {
|
||||
wrappedMetricsReporter.close();
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user