Index: chromecast/media/cma/test/frame_segmenter_for_test.cc |
diff --git a/chromecast/media/cma/test/frame_segmenter_for_test.cc b/chromecast/media/cma/test/frame_segmenter_for_test.cc |
index 5d9406ec01950def579fbb77597fd7403ae85c24..f7d497636f129ca86d77553d5e431dd7b33ab92f 100644 |
--- a/chromecast/media/cma/test/frame_segmenter_for_test.cc |
+++ b/chromecast/media/cma/test/frame_segmenter_for_test.cc |
@@ -323,8 +323,16 @@ DemuxResult FFmpegDemuxForTest(const base::FilePath& filepath, |
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& s : streams) { |
+ if (s->type() == target_type) { |
+ stream = s; |
+ break; |
+ } |
+ } |
CHECK(stream); |
DemuxResult demux_result; |