summaryrefslogtreecommitdiff
path: root/tree/library/extractor/src/test/java/com/google/android/exoplayer2/extractor/ogg/OggPageHeaderTest.java
blob: 6b5ffe8f915bed06d22a24c375486ee283eb697e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
/*
 * Copyright (C) 2016 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.extractor.ogg;

import static com.google.android.exoplayer2.testutil.TestUtil.getByteArray;
import static com.google.common.truth.Truth.assertThat;

import androidx.test.core.app.ApplicationProvider;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import com.google.android.exoplayer2.testutil.FakeExtractorInput;
import com.google.android.exoplayer2.testutil.FakeExtractorInput.SimulatedIOException;
import com.google.android.exoplayer2.testutil.TestUtil;
import org.junit.Test;
import org.junit.runner.RunWith;

/** Unit test for {@link OggPageHeader}. */
@RunWith(AndroidJUnit4.class)
public final class OggPageHeaderTest {

  @Test
  public void populatePageHeader_success() throws Exception {
    byte[] data = getByteArray(ApplicationProvider.getApplicationContext(), "ogg/page_header");

    FakeExtractorInput input = createInput(data, /* simulateUnknownLength= */ true);
    OggPageHeader header = new OggPageHeader();
    populatePageHeader(input, header, /* quiet= */ false);

    assertThat(header.type).isEqualTo(0x01);
    assertThat(header.headerSize).isEqualTo(27 + 2);
    assertThat(header.bodySize).isEqualTo(4);
    assertThat(header.pageSegmentCount).isEqualTo(2);
    assertThat(header.granulePosition).isEqualTo(123456);
    assertThat(header.pageSequenceNumber).isEqualTo(4);
    assertThat(header.streamSerialNumber).isEqualTo(0x1000);
    assertThat(header.pageChecksum).isEqualTo(0x100000);
    assertThat(header.revision).isEqualTo(0);
  }

  @Test
  public void populatePageHeader_withLessThan27Bytes_returnFalseWithoutException()
      throws Exception {
    FakeExtractorInput input =
        createInput(TestUtil.createByteArray(2, 2), /* simulateUnknownLength= */ false);
    OggPageHeader header = new OggPageHeader();
    assertThat(populatePageHeader(input, header, /* quiet= */ true)).isFalse();
  }

  @Test
  public void populatePageHeader_withNotOgg_returnFalseWithoutException() throws Exception {
    byte[] data = getByteArray(ApplicationProvider.getApplicationContext(), "ogg/page_header");
    // change from 'O' to 'o'
    data[0] = 'o';
    FakeExtractorInput input = createInput(data, /* simulateUnknownLength= */ false);
    OggPageHeader header = new OggPageHeader();
    assertThat(populatePageHeader(input, header, /* quiet= */ true)).isFalse();
  }

  @Test
  public void populatePageHeader_withWrongRevision_returnFalseWithoutException() throws Exception {
    byte[] data = getByteArray(ApplicationProvider.getApplicationContext(), "ogg/page_header");
    // change revision from 0 to 1
    data[4] = 0x01;
    FakeExtractorInput input = createInput(data, /* simulateUnknownLength= */ false);
    OggPageHeader header = new OggPageHeader();
    assertThat(populatePageHeader(input, header, /* quiet= */ true)).isFalse();
  }

  private static boolean populatePageHeader(
      FakeExtractorInput input, OggPageHeader header, boolean quiet) throws Exception {
    while (true) {
      try {
        return header.populate(input, quiet);
      } catch (SimulatedIOException e) {
        // ignored
      }
    }
  }

  private static FakeExtractorInput createInput(byte[] data, boolean simulateUnknownLength) {
    return new FakeExtractorInput.Builder()
        .setData(data)
        .setSimulateIOErrors(true)
        .setSimulateUnknownLength(simulateUnknownLength)
        .setSimulatePartialReads(true)
        .build();
  }
}