How to use

LocalDrmSessionManager

in

com.novoda.noplayer.internal.exoplayer.drm

  @Override
  public DrmSessionManager<FrameworkMediaCrypto> create(DefaultDrmSessionEventListener eventListener) {
    return new LocalDrmSessionManager(
        downloadedModularDrm.getKeySetId(),
        mediaDrmCreator.create(WIDEVINE_MODULAR_UUID),
        WIDEVINE_MODULAR_UUID,
        handler,
        eventListener
    );
  }
}

@RequiresApi(api = Build.VERSION_CODES.KITKAT)
@Test
public void givenOpeningSessionError_whenAcquiringSession_thenNotifiesErrorEventListenerOnHandler() throws MediaDrmException {
  given(mediaDrm.openSession()).willThrow(new ResourceBusyException("resource is busy"));
  localDrmSessionManager.acquireSession(IGNORED_LOOPER, IGNORED_DRM_DATA);
  ArgumentCaptor<Runnable> argumentCaptor = ArgumentCaptor.forClass(Runnable.class);
  verify(handler).post(argumentCaptor.capture());
  argumentCaptor.getValue().run();
  verify(eventListener).onDrmSessionManagerError(any(DrmSession.DrmSessionException.class));
}

@Test
public void givenDrmDataDoesNotContainDrmScheme_whenCheckingCanAcquireSession_thenReturnsFalse() {
  DrmInitData.SchemeData unrecognisedSchemeData = new DrmInitData.SchemeData(
      UUID.randomUUID(), "ANY_MIME_TYPE", new byte[]{}
  );
  DrmInitData drmInitData = new DrmInitData(Collections.singletonList(unrecognisedSchemeData));
  boolean canAcquireSession = localDrmSessionManager.canAcquireSession(drmInitData);
  assertThat(canAcquireSession).isFalse();
}

  @Test
  public void givenAcquiredSession_whenReleasingSession_thenClosesCurrentSession() {
    DrmSession<FrameworkMediaCrypto> drmSession = new LocalDrmSession(frameworkMediaCrypto, KEY_SET_ID_TO_RESTORE, SESSION_ID);

    localDrmSessionManager.releaseSession(drmSession);

    verify(mediaDrm).closeSession(SESSION_ID.asBytes());
  }
}

@SuppressWarnings("PMD.AvoidCatchingGenericException") // We are forced to catch Exception as ResourceBusyException is minSdk 19
@TargetApi(Build.VERSION_CODES.JELLY_BEAN_MR2)
@Override
public DrmSession<FrameworkMediaCrypto> acquireSession(Looper playbackLooper, DrmInitData drmInitData) {
  DrmSession<FrameworkMediaCrypto> drmSession;
  try {
    SessionId sessionId = SessionId.of(mediaDrm.openSession());
    FrameworkMediaCrypto mediaCrypto = mediaDrm.createMediaCrypto(sessionId.asBytes());
    mediaDrm.restoreKeys(sessionId.asBytes(), keySetIdToRestore.asBytes());
    drmSession = new LocalDrmSession(mediaCrypto, keySetIdToRestore, sessionId);
  } catch (Exception exception) {
    drmSession = new InvalidDrmSession(new DrmSession.DrmSessionException(exception));
    notifyErrorListener(drmSession);
  }
  return drmSession;
}

@Before
public void setUp() throws MediaDrmException, MediaCryptoException {
  frameworkMediaCrypto = FrameworkMediaCryptoFixture.aFrameworkMediaCrypto().build();
  given(mediaDrm.openSession()).willReturn(SESSION_ID.asBytes());
  localDrmSessionManager = new LocalDrmSessionManager(
      KEY_SET_ID_TO_RESTORE,
      mediaDrm,
      DRM_SCHEME,
      handler,
      eventListener
  );
}

@Test
public void givenValidMediaDrm_whenAcquiringSession_thenRestoresKeys() throws MediaCryptoException {
  given(mediaDrm.createMediaCrypto(SESSION_ID.asBytes())).willReturn(frameworkMediaCrypto);
  localDrmSessionManager.acquireSession(IGNORED_LOOPER, IGNORED_DRM_DATA);
  verify(mediaDrm).restoreKeys(SESSION_ID.asBytes(), KEY_SET_ID_TO_RESTORE.asBytes());
}

@Test
public void givenDrmDataContainsDrmScheme_whenCheckingCanAcquireSession_thenReturnsTrue() {
  DrmInitData.SchemeData recognisedSchemeData = new DrmInitData.SchemeData(
      DRM_SCHEME, "ANY_MIME_TYPE", new byte[]{}
  );
  DrmInitData drmInitData = new DrmInitData(Collections.singletonList(recognisedSchemeData));
  boolean canAcquireSession = localDrmSessionManager.canAcquireSession(drmInitData);
  assertThat(canAcquireSession).isTrue();
}

@RequiresApi(api = Build.VERSION_CODES.KITKAT)
@Test
public void givenOpeningSessionError_whenAcquiringSession_thenReturnsInvalidDrmSession() throws MediaDrmException {
  ResourceBusyException resourceBusyException = new ResourceBusyException("resource is busy");
  given(mediaDrm.openSession()).willThrow(resourceBusyException);
  DrmSession<FrameworkMediaCrypto> drmSession = localDrmSessionManager.acquireSession(IGNORED_LOOPER, IGNORED_DRM_DATA);
  assertThat(drmSession).isInstanceOf(InvalidDrmSession.class);
  assertThat(drmSession.getError().getCause()).isEqualTo(resourceBusyException);
}

@Test
public void givenValidMediaDrm_whenAcquiringSession_thenReturnsLocalDrmSession() throws MediaCryptoException {
  given(mediaDrm.createMediaCrypto(SESSION_ID.asBytes())).willReturn(frameworkMediaCrypto);
  DrmSession<FrameworkMediaCrypto> drmSession = localDrmSessionManager.acquireSession(IGNORED_LOOPER, IGNORED_DRM_DATA);
  LocalDrmSession localDrmSession = new LocalDrmSession(frameworkMediaCrypto, KEY_SET_ID_TO_RESTORE, SESSION_ID);
  assertThat(drmSession).isEqualTo(localDrmSession);
}