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   void SetMutedState(bool muted_state) override {
  48     MediaStreamVideoSource::SetMutedState(muted_state);
  49     DoSetMutedState(muted_state);
  50   }
  51
  52  protected:
  53   // Implements MediaStreamVideoSource.
  54   void GetCurrentSupportedFormats(
  55       int max_requested_height,
  56       int max_requested_width,
  57       double max_requested_frame_rate,
  58       const VideoCaptureDeviceFormatsCB& callback) override;
  59   void StartSourceImpl(
  60       const media::VideoCaptureFormat& format,
  61       const blink::WebMediaConstraints& constraints,
  62       const VideoCaptureDeliverFrameCB& frame_callback) override;
  63   void StopSourceImpl() override;
  64
  65  private:
  66   media::VideoCaptureFormat format_;
  67   media::VideoCaptureFormats supported_formats_;
  68   bool manual_get_supported_formats_;
  69   int max_requested_height_;
  70   int max_requested_width_;
  71   double max_requested_frame_rate_;
  72   bool attempted_to_start_;
  73   VideoCaptureDeviceFormatsCB formats_callback_;
  74   VideoCaptureDeliverFrameCB frame_callback_;
  75
  76   DISALLOW_COPY_AND_ASSIGN(MockMediaStreamVideoSource);
  77 };
  78
  79 }  // namespace content
  80
  81 #endif  // CONTENT_RENDERER_MEDIA_MOCK_MEDIA_STREAM_VIDEO_SOURCE_H_