content/renderer/media/mock_media_stream_video_source.h

   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_