Home | History | Annotate | Download | only in media 
      1 // Copyright 2014 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef CONTENT_RENDERER_MEDIA_MOCK_MEDIA_STREAM_VIDEO_SOURCE_H_
      6 #define CONTENT_RENDERER_MEDIA_MOCK_MEDIA_STREAM_VIDEO_SOURCE_H_
      7 
      8 #include "content/renderer/media/media_stream_video_source.h"
      9 
     10 #include "testing/gmock/include/gmock/gmock.h"
     11 
     12 namespace content {
     13 
     14 class MockMediaStreamVideoSource : public MediaStreamVideoSource {
     15  public:
     16   explicit MockMediaStreamVideoSource(bool manual_get_supported_formats);
     17   virtual ~MockMediaStreamVideoSource();
     18 
     19   MOCK_METHOD1(DoSetMutedState, void(bool muted_state));
     20 
     21   // Simulate that the underlying source start successfully.
     22   void StartMockedSource();
     23 
     24   // Simulate that the underlying source fail to start.
     25   void FailToStartMockedSource();
     26 
     27   // Returns true if StartSource  has been called and StartMockedSource
     28   // or FailToStartMockedSource has not been called.
     29   bool SourceHasAttemptedToStart() { return attempted_to_start_; }
     30 
     31   void SetSupportedFormats(const media::VideoCaptureFormats& formats) {
     32     supported_formats_ = formats;
     33   }
     34 
     35   // Delivers |frame| to all registered tracks on the IO thread. Its up to the
     36   // call to make sure MockMediaStreamVideoSource is not destroyed before the
     37   // frame has been delivered.
     38   void DeliverVideoFrame(const scoped_refptr<media::VideoFrame>& frame);
     39 
     40   void CompleteGetSupportedFormats();
     41 
     42   const media::VideoCaptureFormat& start_format() const { return format_; }
     43   int max_requested_height() const { return max_requested_height_; }
     44   int max_requested_width() const { return max_requested_width_; }
     45   double max_requested_frame_rate() const { return max_requested_frame_rate_; }
     46 
     47   virtual void SetMutedState(bool muted_state) OVERRIDE {
     48     MediaStreamVideoSource::SetMutedState(muted_state);
     49     DoSetMutedState(muted_state);
     50   }
     51 
     52  protected:
     53   void DeliverVideoFrameOnIO(const scoped_refptr<media::VideoFrame>& frame,
     54                              media::VideoCaptureFormat format,
     55                              const base::TimeTicks& estimated_capture_time,
     56                              const VideoCaptureDeliverFrameCB& frame_callback);
     57 
     58   // Implements MediaStreamVideoSource.
     59   virtual void GetCurrentSupportedFormats(
     60       int max_requested_height,
     61       int max_requested_width,
     62       double max_requested_frame_rate,
     63       const VideoCaptureDeviceFormatsCB& callback) OVERRIDE;
     64   virtual void StartSourceImpl(
     65       const media::VideoCaptureFormat& format,
     66       const VideoCaptureDeliverFrameCB& frame_callback) OVERRIDE;
     67   virtual void StopSourceImpl() OVERRIDE;
     68 
     69  private:
     70   media::VideoCaptureFormat format_;
     71   media::VideoCaptureFormats supported_formats_;
     72   bool manual_get_supported_formats_;
     73   int max_requested_height_;
     74   int max_requested_width_;
     75   double max_requested_frame_rate_;
     76   bool attempted_to_start_;
     77   VideoCaptureDeviceFormatsCB formats_callback_;
     78   VideoCaptureDeliverFrameCB frame_callback_;
     79 
     80   DISALLOW_COPY_AND_ASSIGN(MockMediaStreamVideoSource);
     81 };
     82 
     83 }  // namespace content
     84 
     85 #endif  // CONTENT_RENDERER_MEDIA_MOCK_MEDIA_STREAM_VIDEO_SOURCE_H_
     86