Skip to content

SDK / exoplayer

Go to a project
Commit 8a31e33d by Marc Baechinger
Options

Merge pull request #69 from ittiam-systems:rtp_amr_test 

PiperOrigin-RevId: 453905355
parents 20561528 718cea8d
Inline Side-by-side
Showing with 220 additions and 0 deletions
library/rtsp/src/test/java/com/google/android/exoplayer2/source/rtsp/reader/RtpAmrReaderTest.java 0 → 100644
/*
* Copyright 2022 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
*
* http://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 com.google.android.exoplayer2.source.rtsp.reader;
import static com.google.android.exoplayer2.util.Util.getBytesFromHexString;
import static com.google.common.truth.Truth.assertThat;
import static org.junit.Assert.assertThrows;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import com.google.android.exoplayer2.Format;
import com.google.android.exoplayer2.source.rtsp.RtpPayloadFormat;
import com.google.android.exoplayer2.testutil.FakeExtractorOutput;
import com.google.android.exoplayer2.testutil.FakeTrackOutput;
import com.google.android.exoplayer2.util.MimeTypes;
import com.google.android.exoplayer2.util.ParsableByteArray;
import com.google.common.collect.ImmutableMap;
import com.google.common.primitives.Bytes;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
/** Unit test for {@link RtpAmrReader}. */
@RunWith(AndroidJUnit4.class)
public final class RtpAmrReaderTest {
private FakeExtractorOutput extractorOutput;
@Before
public void setUp() {
extractorOutput = new FakeExtractorOutput();
}
@Test
public void consume_validAmrNbPackets() {
RtpAmrReader amrReader = createAmrReader(MimeTypes.AUDIO_AMR);
amrReader.createTracks(extractorOutput, /* trackId= */ 0);
amrReader.onReceivingFirstPacket(/* timestamp= */ 2599168056L, /* sequenceNumber= */ 40289);
amrReader.consume(
new ParsableByteArray(
Bytes.concat(
getBytesFromHexString("00"), // AMR-NB Codec Mode = 0.
getBytesFromHexString("40"), // AMR-NB ToC, F=0, FT=8.
getBytesFromHexString("0102030405"))),
/* timestamp= */ 2599168056L,
/* sequenceNumber= */ 40289,
/* rtpMarker= */ false);
amrReader.consume(
new ParsableByteArray(
Bytes.concat(
getBytesFromHexString("60"), // AMR-NB Codec Mode = 6.
getBytesFromHexString("C0"), // AMR-NB ToC, F=1, FT=8.
getBytesFromHexString("060708090A"))),
/* timestamp= */ 2599169592L,
/* sequenceNumber= */ 40290,
/* rtpMarker= */ false);
FakeTrackOutput trackOutput = extractorOutput.trackOutputs.get(0);
assertThat(trackOutput.getSampleCount()).isEqualTo(2);
assertThat(trackOutput.getSampleData(0)).isEqualTo(getBytesFromHexString("400102030405"));
assertThat(trackOutput.getSampleTimeUs(0)).isEqualTo(0);
assertThat(trackOutput.getSampleData(1)).isEqualTo(getBytesFromHexString("C0060708090A"));
assertThat(trackOutput.getSampleTimeUs(1)).isEqualTo(192000);
}
@Test
public void consume_amrNbPacketWithInvalidFrameType_throwsIllegalArgumentException() {
RtpAmrReader amrReader = createAmrReader(MimeTypes.AUDIO_AMR);
amrReader.createTracks(extractorOutput, /* trackId= */ 0);
amrReader.onReceivingFirstPacket(/* timestamp= */ 2599168056L, /* sequenceNumber= */ 40289);
assertThrows(
IllegalArgumentException.class,
() ->
amrReader.consume(
new ParsableByteArray(
Bytes.concat(
getBytesFromHexString("00"), // AMR-NB Codec Mode = 0.
getBytesFromHexString("68"), // AMR-NB ToC, F=0, FT=13.
getBytesFromHexString("0102030405"))),
/* timestamp= */ 2599168056L,
/* sequenceNumber= */ 40289,
/* rtpMarker= */ false));
}
@Test
public void consume_amrNbPacketWithInvalidFrameSize_throwsIllegalArgumentException() {
// The payload frame type is 8, expecting five bytes of data, getting four.
RtpAmrReader amrReader = createAmrReader(MimeTypes.AUDIO_AMR);
amrReader.createTracks(extractorOutput, /* trackId= */ 0);
amrReader.onReceivingFirstPacket(/* timestamp= */ 2599168056L, /* sequenceNumber= */ 40289);
assertThrows(
IllegalArgumentException.class,
() ->
amrReader.consume(
new ParsableByteArray(
Bytes.concat(
getBytesFromHexString("00"), // AMR-NB Codec Mode = 0.
getBytesFromHexString("40"), // AMR-NB ToC, F=0, FT=8.
getBytesFromHexString("01020304"))),
/* timestamp= */ 2599168056L,
/* sequenceNumber= */ 40289,
/* rtpMarker= */ false));
}
@Test
public void consume_validAmrWbPackets() {
RtpAmrReader amrReader = createAmrReader(MimeTypes.AUDIO_AMR_WB);
amrReader.createTracks(extractorOutput, /* trackId= */ 0);
amrReader.onReceivingFirstPacket(/* timestamp= */ 2599168056L, /* sequenceNumber= */ 40289);
amrReader.consume(
new ParsableByteArray(
Bytes.concat(
getBytesFromHexString("60"), // AMR-WB Codec Mode = 6.
getBytesFromHexString("87"), // AMR-WB ToC, F=0, FT=0.
getBytesFromHexString("0102030405060708090A0B0C0D0E0F1011"))),
/* timestamp= */ 2599168056L,
/* sequenceNumber= */ 40289,
/* rtpMarker= */ false);
amrReader.consume(
new ParsableByteArray(
Bytes.concat(
getBytesFromHexString("00"), // AMR-WB Codec Mode = 0.
getBytesFromHexString("7E"))), // AMR-WB ToC, F=0, FT=15.
/* timestamp= */ 2599169592L,
/* sequenceNumber= */ 40290,
/* rtpMarker= */ false);
FakeTrackOutput trackOutput = extractorOutput.trackOutputs.get(0);
assertThat(trackOutput.getSampleCount()).isEqualTo(2);
assertThat(trackOutput.getSampleData(0))
.isEqualTo(getBytesFromHexString("870102030405060708090A0B0C0D0E0F1011"));
assertThat(trackOutput.getSampleTimeUs(0)).isEqualTo(0);
assertThat(trackOutput.getSampleData(1)).isEqualTo(getBytesFromHexString("7E"));
assertThat(trackOutput.getSampleTimeUs(1)).isEqualTo(96000);
}
@Test
public void consume_amrWbPacketWithInvalidFrameType_throwsIllegalArgumentException() {
RtpAmrReader amrReader = createAmrReader(MimeTypes.AUDIO_AMR_WB);
amrReader.createTracks(extractorOutput, /* trackId= */ 0);
amrReader.onReceivingFirstPacket(/* timestamp= */ 2599168056L, /* sequenceNumber= */ 40289);
assertThrows(
IllegalArgumentException.class,
() ->
amrReader.consume(
new ParsableByteArray(
Bytes.concat(
getBytesFromHexString("00"), // AMR-WB Codec Mode = 0.
getBytesFromHexString("D5"), // AMR-WB ToC, F=1, FT=10.
getBytesFromHexString("01020304"))),
/* timestamp= */ 2599168056L,
/* sequenceNumber= */ 40289,
/* rtpMarker= */ false));
}
@Test
public void consume_amrWbPacketWithInvalidFrameSize_throwsIllegalArgumentException() {
// The payload frame type is 4, expecting 40 bytes of data, getting two.
RtpAmrReader amrReader = createAmrReader(MimeTypes.AUDIO_AMR_WB);
amrReader.createTracks(extractorOutput, /* trackId= */ 0);
amrReader.onReceivingFirstPacket(/* timestamp= */ 2599168056L, /* sequenceNumber= */ 40289);
assertThrows(
IllegalArgumentException.class,
() ->
amrReader.consume(
new ParsableByteArray(
Bytes.concat(
getBytesFromHexString("00"), // AMR-WB Codec Mode = 0.
getBytesFromHexString("A5"), // AMR-WB ToC, F=1, FT=4.
getBytesFromHexString("0102"))),
/* timestamp= */ 2599168056L,
/* sequenceNumber= */ 40289,
/* rtpMarker= */ false));
}
private static RtpAmrReader createAmrReader(String mimeType) {
switch (mimeType) {
case MimeTypes.AUDIO_AMR:
return new RtpAmrReader(createRtpPayloadFormat(mimeType, /* sampleRate= */ 8000));
case MimeTypes.AUDIO_AMR_WB:
return new RtpAmrReader(createRtpPayloadFormat(mimeType, /* sampleRate= */ 16000));
default:
throw new IllegalArgumentException("MimeType " + mimeType + " not supported.");
}
}
private static RtpPayloadFormat createRtpPayloadFormat(String mimeType, int sampleRate) {
return new RtpPayloadFormat(
new Format.Builder()
.setChannelCount(1)
.setSampleMimeType(mimeType)
.setSampleRate(sampleRate)
.build(),
/* rtpPayloadType= */ 97,
/* clockRate= */ sampleRate,
/* fmtpParameters= */ ImmutableMap.of());
}
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment