1 // Copyright 2020 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 #include "chrome/browser/media/webrtc/media_capture_devices_dispatcher.h"
6
7 #include <memory>
8
9 #include "build/build_config.h"
10 #include "chrome/browser/media/media_access_handler.h"
11 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
12 #include "testing/gmock/include/gmock/gmock.h"
13 #include "testing/gtest/include/gtest/gtest.h"
14 #include "third_party/blink/public/common/mediastream/media_stream_request.h"
15
16 using testing::_;
17
18 namespace {
19
20 class MockMediaAccessHandler : public MediaAccessHandler {
21 public:
MockMediaAccessHandler(const blink::mojom::MediaStreamType supported_type)22 explicit MockMediaAccessHandler(
23 const blink::mojom::MediaStreamType supported_type)
24 : supported_type_(supported_type) {}
25
SupportsStreamType(content::WebContents * web_contents,const blink::mojom::MediaStreamType type,const extensions::Extension * extension)26 bool SupportsStreamType(content::WebContents* web_contents,
27 const blink::mojom::MediaStreamType type,
28 const extensions::Extension* extension) override {
29 return supported_type_ == type;
30 }
31
32 MOCK_METHOD4(CheckMediaAccessPermission,
33 bool(content::RenderFrameHost* render_frame_host,
34 const GURL& security_origin,
35 blink::mojom::MediaStreamType type,
36 const extensions::Extension* extension));
37 MOCK_METHOD4(HandleRequest,
38 void(content::WebContents* web_contents,
39 const content::MediaStreamRequest& request,
40 content::MediaResponseCallback callback,
41 const extensions::Extension* extension));
42 MOCK_METHOD4(UpdateVideoScreenCaptureStatus,
43 void(int render_process_id,
44 int render_frame_id,
45 int page_request_id,
46 bool is_secure));
47
48 const blink::mojom::MediaStreamType supported_type_;
49 };
50
51 } // namespace
52
53 class MediaCaptureDevicesDispatcherTest
54 : public ChromeRenderViewHostTestHarness {
55 public:
56 MediaCaptureDevicesDispatcherTest() = default;
57 ~MediaCaptureDevicesDispatcherTest() override = default;
58
SetUp()59 void SetUp() override {
60 ChromeRenderViewHostTestHarness::SetUp();
61 dispatcher_ = MediaCaptureDevicesDispatcher::GetInstance();
62 }
63
media_access_handlers()64 std::vector<std::unique_ptr<MediaAccessHandler>>& media_access_handlers() {
65 return dispatcher_->media_access_handlers_;
66 }
67
UpdateVideoScreenCaptureStatus(const blink::mojom::MediaStreamType type)68 void UpdateVideoScreenCaptureStatus(
69 const blink::mojom::MediaStreamType type) {
70 dispatcher_->UpdateVideoScreenCaptureStatus(0, 0, 0, type, false);
71 }
72
73 protected:
74 MediaCaptureDevicesDispatcher* dispatcher_;
75 };
76
TEST_F(MediaCaptureDevicesDispatcherTest,DISABLED_LoopsAllMediaAccessHandlers)77 TEST_F(MediaCaptureDevicesDispatcherTest,
78 DISABLED_LoopsAllMediaAccessHandlers) {
79 media_access_handlers().clear();
80
81 // Add two handlers.
82 auto stream_type1 = blink::mojom::MediaStreamType::DISPLAY_VIDEO_CAPTURE;
83 auto mock1 = std::make_unique<MockMediaAccessHandler>(stream_type1);
84 MockMediaAccessHandler* handler1 = mock1.get();
85 media_access_handlers().push_back(std::move(mock1));
86 auto stream_type2 = blink::mojom::MediaStreamType::GUM_DESKTOP_VIDEO_CAPTURE;
87 auto mock2 = std::make_unique<MockMediaAccessHandler>(stream_type2);
88 MockMediaAccessHandler* handler2 = mock2.get();
89 media_access_handlers().push_back(std::move(mock2));
90
91 // Expect both to be called.
92 EXPECT_CALL(*handler1, UpdateVideoScreenCaptureStatus(_, _, _, _));
93 UpdateVideoScreenCaptureStatus(stream_type1);
94 EXPECT_CALL(*handler2, UpdateVideoScreenCaptureStatus(_, _, _, _));
95 UpdateVideoScreenCaptureStatus(stream_type2);
96 }
97