Refactor MediaResource to be multi-stream capable

chromecast/media/cma/test/frame_segmenter_for_test.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 "chromecast/media/cma/test/frame_segmenter_for_test.h"
 
 #include <stdint.h>
 
 #include "base/bind.h"
 #include "base/callback.h"
@@ skipping 305 matching lines @@
   ::media::FFmpegDemuxer demuxer(
       base::ThreadTaskRunnerHandle::Get(), &data_source,
       base::Bind(&OnEncryptedMediaInitData), base::Bind(&OnMediaTracksUpdated),
       new ::media::MediaLog());
   ::media::WaitableMessageLoopEvent init_event;
   demuxer.Initialize(&fake_demuxer_host,
                      init_event.GetPipelineStatusCB(),
                      false);
   init_event.RunAndWaitForStatus(::media::PIPELINE_OK);
 
-  ::media::DemuxerStream* stream = demuxer.GetStream(
-      audio ? ::media::DemuxerStream::AUDIO : ::media::DemuxerStream::VIDEO);
+  std::vector<::media::DemuxerStream*> streams = demuxer.GetStreams();
+  ::media::DemuxerStream* stream = nullptr;
+  ::media::DemuxerStream::Type target_type =
+      audio ? ::media::DemuxerStream::AUDIO : ::media::DemuxerStream::VIDEO;
+  for (const auto& demuxer_stream : streams) {
+    if (demuxer_stream->type() == target_type) {
+      stream = demuxer_stream;
+      break;
+    }
+  }
   CHECK(stream);
 
   DemuxResult demux_result;
   if (audio) {
     demux_result.audio_config = stream->audio_decoder_config();
   } else {
     demux_result.video_config = stream->video_decoder_config();
   }
 
   bool end_of_stream = false;
   while (!end_of_stream) {
     base::RunLoop run_loop;
     stream->Read(base::Bind(&OnNewBuffer,
                             base::Unretained(&demux_result.frames),
                             run_loop.QuitClosure()));
     run_loop.Run();
     CHECK(!demux_result.frames.empty());
     end_of_stream = demux_result.frames.back()->end_of_stream();
   }
 
   demuxer.Stop();
   return demux_result;
 }
 
 }  // namespace media
 }  // namespace chromecast