Back to Repositories

Validating DASH DRM Initialization in SmartTube

This test suite validates the DashUtil class functionality in ExoPlayer’s DASH module, focusing on DRM initialization data handling and manifest parsing. The tests ensure proper extraction and validation of DRM-related information from DASH media presentations.

Test Coverage Overview

The test suite provides comprehensive coverage of DRM initialization data loading scenarios in DASH manifests:

Successful DRM init data loading from manifest
Handling missing DRM initialization data
Edge cases with empty representations
Null handling for missing adaptation sets

Implementation Analysis

The testing approach utilizes JUnit with AndroidJUnit4 runner for Android-specific testing context. The implementation follows a structured pattern using helper methods to construct test data hierarchies, simulating DASH manifest components like Period, AdaptationSet, and Representation objects.

Technical Details

Key technical components include:

AndroidJUnit4 test runner
DummyDataSource for simulated data access
Custom helper methods for test data construction
Truth assertion library for test validations

Best Practices Demonstrated

The test suite exemplifies several testing best practices:

Isolated test cases for specific scenarios
Clear test method naming conventions
Proper test data setup and helper methods
Comprehensive edge case coverage

yuliskov/smarttube

exoplayer-amzn-2.10.6/library/dash/src/test/java/com/google/android/exoplayer2/source/dash/DashUtilTest.java

            
/*
 * Copyright (C) 2017 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.dash;

import static com.google.common.truth.Truth.assertThat;

import androidx.test.ext.junit.runners.AndroidJUnit4;
import com.google.android.exoplayer2.C;
import com.google.android.exoplayer2.Format;
import com.google.android.exoplayer2.drm.DrmInitData;
import com.google.android.exoplayer2.drm.DrmInitData.SchemeData;
import com.google.android.exoplayer2.source.dash.manifest.AdaptationSet;
import com.google.android.exoplayer2.source.dash.manifest.Period;
import com.google.android.exoplayer2.source.dash.manifest.Representation;
import com.google.android.exoplayer2.source.dash.manifest.SegmentBase.SingleSegmentBase;
import com.google.android.exoplayer2.upstream.DummyDataSource;
import com.google.android.exoplayer2.util.MimeTypes;
import java.util.Arrays;
import org.junit.Test;
import org.junit.runner.RunWith;

/** Unit tests for {@link DashUtil}. */
@RunWith(AndroidJUnit4.class)
public final class DashUtilTest {

  @Test
  public void testLoadDrmInitDataFromManifest() throws Exception {
    Period period = newPeriod(newAdaptationSets(newRepresentations(newDrmInitData())));
    DrmInitData drmInitData = DashUtil.loadDrmInitData(DummyDataSource.INSTANCE, period);
    assertThat(drmInitData).isEqualTo(newDrmInitData());
  }

  @Test
  public void testLoadDrmInitDataMissing() throws Exception {
    Period period = newPeriod(newAdaptationSets(newRepresentations(null /* no init data */)));
    DrmInitData drmInitData = DashUtil.loadDrmInitData(DummyDataSource.INSTANCE, period);
    assertThat(drmInitData).isNull();
  }

  @Test
  public void testLoadDrmInitDataNoRepresentations() throws Exception {
    Period period = newPeriod(newAdaptationSets(/* no representation */ ));
    DrmInitData drmInitData = DashUtil.loadDrmInitData(DummyDataSource.INSTANCE, period);
    assertThat(drmInitData).isNull();
  }

  @Test
  public void testLoadDrmInitDataNoAdaptationSets() throws Exception {
    Period period = newPeriod(/* no adaptation set */ );
    DrmInitData drmInitData = DashUtil.loadDrmInitData(DummyDataSource.INSTANCE, period);
    assertThat(drmInitData).isNull();
  }

  private static Period newPeriod(AdaptationSet... adaptationSets) {
    return new Period("", 0, Arrays.asList(adaptationSets));
  }

  private static AdaptationSet newAdaptationSets(Representation... representations) {
    return new AdaptationSet(0, C.TRACK_TYPE_VIDEO, Arrays.asList(representations), null, null);
  }

  private static Representation newRepresentations(DrmInitData drmInitData) {
    Format format =
        Format.createVideoContainerFormat(
            "id",
            "label",
            MimeTypes.VIDEO_MP4,
            MimeTypes.VIDEO_H264,
            /* codecs= */ "",
            /* metadata= */ null,
            Format.NO_VALUE,
            /* width= */ 1024,
            /* height= */ 768,
            Format.NO_VALUE,
            /* initializationData= */ null,
            /* selectionFlags= */ 0,
            /* roleFlags= */ 0);
    if (drmInitData != null) {
      format = format.copyWithDrmInitData(drmInitData);
    }
    return Representation.newInstance(0, format, "", new SingleSegmentBase());
  }

  private static DrmInitData newDrmInitData() {
    return new DrmInitData(
        new SchemeData(C.WIDEVINE_UUID, "mimeType", new byte[] {1, 4, 7, 0, 3, 6}));
  }
}