Assign correct track ids to buffers emitted from StreamParsers

media/formats/common/stream_parser_test_base.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "media/formats/common/stream_parser_test_base.h"
 
 #include <memory>
 #include <utility>
 
 #include "base/bind.h"
@@ skipping 75 matching lines @@
            << params.auto_update_timestamp_offset << ")";
 }
 
 bool StreamParserTestBase::OnNewConfig(
     std::unique_ptr<MediaTracks> tracks,
     const StreamParser::TextTrackConfigMap& text_config) {
   DVLOG(1) << __FUNCTION__ << ": got " << tracks->tracks().size() << " tracks";
   EXPECT_EQ(tracks->tracks().size(), 1u);
   const auto& track = tracks->tracks()[0];
   EXPECT_EQ(track->type(), MediaTrack::Audio);
+  audio_track_id_ = track->bytestream_track_id();
   last_audio_config_ = tracks->getAudioConfig(track->bytestream_track_id());
   EXPECT_TRUE(last_audio_config_.IsValidConfig());
   return true;
 }
 
 bool StreamParserTestBase::OnNewBuffers(
     const StreamParser::BufferQueue& audio_buffers,
     const StreamParser::BufferQueue& video_buffers,
     const StreamParser::TextBufferQueueMap& text_map) {
   EXPECT_FALSE(audio_buffers.empty());
   EXPECT_TRUE(video_buffers.empty());
 
+  // Ensure that track ids are properly assigned on all emitted buffers.
+  for (const auto& buf : audio_buffers) {
+    EXPECT_EQ(audio_track_id_, buf->track_id());
+  }
+
   // TODO(wolenetz/acolwell): Add text track support to more MSE parsers. See
   // http://crbug.com/336926.
   EXPECT_TRUE(text_map.empty());
 
   const std::string buffers_str = BufferQueueToString(audio_buffers);
   DVLOG(1) << __FUNCTION__ << " : " << buffers_str;
   results_stream_ << buffers_str;
   return true;
 }
 
 void StreamParserTestBase::OnKeyNeeded(EmeInitDataType type,
                                        const std::vector<uint8_t>& init_data) {
   DVLOG(1) << __FUNCTION__ << "(" << static_cast<int>(type) << ", "
            << init_data.size() << ")";
 }
 
 void StreamParserTestBase::OnNewSegment() {
   DVLOG(1) << __FUNCTION__;
   results_stream_ << "NewSegment";
 }
 
 void StreamParserTestBase::OnEndOfSegment() {
   DVLOG(1) << __FUNCTION__;
   results_stream_ << "EndOfSegment";
 }
 
 }  // namespace media