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