Remove MediaTracks::getFirstAudio/VideoConfig

media/formats/mp2t/mp2t_stream_parser_unittest.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/mp2t/mp2t_stream_parser.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <algorithm>
@@ skipping 100 matching lines @@
     return true;
   }
 
   void OnInit(const StreamParser::InitParameters& params) {
     DVLOG(1) << "OnInit: dur=" << params.duration.InMilliseconds()
              << ", autoTimestampOffset=" << params.auto_update_timestamp_offset;
   }
 
   bool OnNewConfig(std::unique_ptr<MediaTracks> tracks,
                    const StreamParser::TextTrackConfigMap& tc) {
-    const AudioDecoderConfig& ac = tracks->getFirstAudioConfig();
-    const VideoDecoderConfig& vc = tracks->getFirstVideoConfig();
-    DVLOG(1) << "OnNewConfig: media tracks count=" << tracks->tracks().size()
-             << ", audio=" << ac.IsValidConfig()
-             << ", video=" << vc.IsValidConfig();
+    DVLOG(1) << "OnNewConfig: got " << tracks->tracks().size() << " tracks";
+    bool found_audio_track = false;
+    bool found_video_track = false;
+    for (const auto& track : tracks->tracks()) {
+      const auto& track_id = track->bytestream_track_id();
+      if (track->type() == MediaTrack::Audio) {
+        found_audio_track = true;
+        EXPECT_TRUE(tracks->getAudioConfig(track_id).IsValidConfig());
+      } else if (track->type() == MediaTrack::Video) {
+        found_video_track = true;
+        EXPECT_TRUE(tracks->getVideoConfig(track_id).IsValidConfig());
+      } else {
+        // Unexpected track type.
+        LOG(ERROR) << "Unexpected track type " << track->type();
+        EXPECT_TRUE(false);
+      }
+    }
+    EXPECT_TRUE(found_audio_track);
+    EXPECT_EQ(has_video_, found_video_track);
     config_count_++;
-    EXPECT_TRUE(ac.IsValidConfig());
-    EXPECT_EQ(vc.IsValidConfig(), has_video_);
     return true;
   }
 
 
   void DumpBuffers(const std::string& label,
                    const StreamParser::BufferQueue& buffers) {
     DVLOG(2) << "DumpBuffers: " << label << " size " << buffers.size();
     for (StreamParser::BufferQueue::const_iterator buf = buffers.begin();
          buf != buffers.end(); buf++) {
       DVLOG(3) << "  n=" << buf - buffers.begin()
@@ skipping 39 matching lines @@
       audio_max_dts_ = last_dts;
     }
 
     audio_frame_count_ += audio_buffers.size();
     video_frame_count_ += video_buffers.size();
     return true;
   }
 
   void OnKeyNeeded(EmeInitDataType type,
                    const std::vector<uint8_t>& init_data) {
-    NOTREACHED() << "OnKeyNeeded not expected in the Mpeg2 TS parser";
+    LOG(ERROR) << "OnKeyNeeded not expected in the Mpeg2 TS parser";
+    EXPECT_TRUE(false);
   }
 
   void OnNewSegment() {
     DVLOG(1) << "OnNewSegment";
     segment_count_++;
   }
 
   void OnEndOfSegment() {
-    NOTREACHED() << "OnEndOfSegment not expected in the Mpeg2 TS parser";
+    LOG(ERROR) << "OnEndOfSegment not expected in the Mpeg2 TS parser";
+    EXPECT_TRUE(false);
   }
 
   void InitializeParser() {
     parser_->Init(
         base::Bind(&Mp2tStreamParserTest::OnInit, base::Unretained(this)),
         base::Bind(&Mp2tStreamParserTest::OnNewConfig, base::Unretained(this)),
         base::Bind(&Mp2tStreamParserTest::OnNewBuffers, base::Unretained(this)),
         true,
         base::Bind(&Mp2tStreamParserTest::OnKeyNeeded, base::Unretained(this)),
         base::Bind(&Mp2tStreamParserTest::OnNewSegment, base::Unretained(this)),
@@ skipping 89 matching lines @@
   parser_->Flush();
   EXPECT_EQ(audio_frame_count_, 40);
   EXPECT_EQ(video_frame_count_, 0);
   // This stream has no mid-stream configuration change.
   EXPECT_EQ(config_count_, 1);
   EXPECT_EQ(segment_count_, 1);
 }
 
 }  // namespace mp2t
 }  // namespace media