Ability for demuxer clients to get to FFmpeg's AVStream object exposed as an ...

media/filters/ffmpeg_demuxer_unittest.cc

 // Copyright (c) 2009 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 <deque>
 
 #include "media/base/filter_host.h"
 #include "media/base/filters.h"
 #include "media/base/mock_filter_host.h"
 #include "media/base/mock_media_filters.h"
@@ skipping 223 matching lines @@
   EXPECT_EQ(PIPELINE_OK, pipeline.GetError());
 
   // Since we ignore data streams, the duration should be equal to the video
   // stream's duration.
   EXPECT_EQ(g_streams[1].duration, pipeline.GetDuration().InMicroseconds());
 
   // Verify that 2 out of 3 streams were created.
   EXPECT_EQ(2, demuxer->GetNumberOfStreams());
 
   // First stream should be video.
-  DemuxerStream* stream = demuxer->GetStream(0);
+  scoped_refptr<DemuxerStream> stream = demuxer->GetStream(0);
   ASSERT_TRUE(stream);
   const MediaFormat* stream_format = stream->GetMediaFormat();
   std::string mime_type;
   int result;
   EXPECT_TRUE(stream_format->GetAsString(MediaFormat::kMimeType, &mime_type));
   EXPECT_STREQ(mime_type::kFFmpegVideo, mime_type.c_str());
   EXPECT_TRUE(stream_format->GetAsInteger(kFFmpegCodecID, &result));
   EXPECT_EQ(CODEC_ID_THEORA, static_cast<CodecID>(result));
   EXPECT_TRUE(stream_format->GetAsInteger(MediaFormat::kHeight, &result));
   EXPECT_EQ(g_video_codec.height, result);
@@ skipping 49 matching lines @@
   EXPECT_TRUE(demuxer);
   MockFilterHost<Demuxer> filter_host_b(&pipeline, demuxer);
   EXPECT_TRUE(demuxer->Initialize(data_source));
   EXPECT_TRUE(filter_host_b.IsInitialized());
   EXPECT_EQ(PIPELINE_OK, pipeline.GetError());
 
   // Verify both streams were created.
   EXPECT_EQ(2, demuxer->GetNumberOfStreams());
 
   // Get our streams.
-  DemuxerStream* audio_stream = demuxer->GetStream(kAudio);
-  DemuxerStream* video_stream = demuxer->GetStream(kVideo);
+  scoped_refptr<DemuxerStream> audio_stream = demuxer->GetStream(kAudio);
+  scoped_refptr<DemuxerStream> video_stream = demuxer->GetStream(kVideo);
   ASSERT_TRUE(audio_stream);
   ASSERT_TRUE(video_stream);
 
+  // Both streams should support FFmpegDemuxerStream interface.
+  scoped_refptr<FFmpegDemuxerStream> ffmpeg_demuxer_stream;
+  EXPECT_TRUE(audio_stream->QueryInterface(&ffmpeg_demuxer_stream));
+  EXPECT_TRUE(ffmpeg_demuxer_stream);
+  ffmpeg_demuxer_stream = NULL;
+  EXPECT_TRUE(video_stream->QueryInterface(&ffmpeg_demuxer_stream));
+  EXPECT_TRUE(ffmpeg_demuxer_stream);
+
   // Prepare our test audio packet.
   g_packet.stream_index = kAudio;
   g_packet.data = audio_data;
   g_packet.size = kDataSize;
 
   // Attempt a read from the audio stream and run the message loop until done.
   scoped_refptr<TestBuffer> buffer(new TestBuffer());
   audio_stream->Read(buffer);
   pipeline.RunAllTasks();
   EXPECT_TRUE(buffer->assigned());
@@ skipping 22 matching lines @@
   buffer = new TestBuffer();
   audio_stream->Read(buffer);
   pipeline.RunAllTasks();
   EXPECT_FALSE(buffer->assigned());
   EXPECT_FALSE(buffer->buffer());
 
   // Manually release buffer, which should release any remaining AVPackets.
   buffer = NULL;
   EXPECT_EQ(0, g_oustanding_packets);
 }