@Test
public void testAdjustUpstreamFormat() {

  String label = "label";
  sampleQueue =
      new SampleQueue(allocator, mockDrmSessionManager) {
        @Override
        public Format getAdjustedUpstreamFormat(Format format) {
          return super.getAdjustedUpstreamFormat(format.copyWithLabel(label));
        }
      };
  writeFormat(FORMAT_1);
  assertReadFormat(/* formatRequired= */ false, FORMAT_1.copyWithLabel(label));
  assertReadEndOfStream(/* formatRequired= */ false);
}