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
12namespace content {
13
14class 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