1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_MEDIA_WEBRTC_FAKE_DESKTOP_MEDIA_PICKER_FACTORY_H_
#define CHROME_BROWSER_MEDIA_WEBRTC_FAKE_DESKTOP_MEDIA_PICKER_FACTORY_H_
#include <memory>
#include <vector>
#include "base/memory/raw_ptr.h"
#include "base/types/expected.h"
#include "chrome/browser/media/webrtc/desktop_media_list.h"
#include "chrome/browser/media/webrtc/desktop_media_picker.h"
#include "chrome/browser/media/webrtc/desktop_media_picker_factory.h"
#include "content/public/browser/desktop_media_id.h"
#include "third_party/blink/public/mojom/mediastream/media_stream.mojom.h"
class FakeDesktopMediaPicker;
// Used in tests to supply fake picker.
class FakeDesktopMediaPickerFactory : public DesktopMediaPickerFactory {
public:
// TODO(crbug.com/40169647): Make this less error prone - use WithX() methods.
struct TestFlags {
bool expect_screens = false;
bool expect_windows = false;
bool expect_tabs = false;
bool expect_current_tab = false;
bool expect_audio = false;
std::optional<base::expected<content::DesktopMediaID,
blink::mojom::MediaStreamRequestResult>>
picker_result;
bool cancelled = false;
// Following flags are set by FakeDesktopMediaPicker when it's created and
// deleted.
bool picker_created = false;
bool picker_deleted = false;
};
FakeDesktopMediaPickerFactory();
FakeDesktopMediaPickerFactory(const FakeDesktopMediaPickerFactory&) = delete;
FakeDesktopMediaPickerFactory& operator=(
const FakeDesktopMediaPickerFactory&) = delete;
~FakeDesktopMediaPickerFactory() override;
// |test_flags| are expected to outlive the factory.
void SetTestFlags(TestFlags* test_flags, int tests_count);
FakeDesktopMediaPicker* picker() const { return picker_; }
bool IsWebContentsExcluded() const { return is_web_contents_excluded_; }
// DesktopMediaPickerFactory implementation
std::unique_ptr<DesktopMediaPicker> CreatePicker(
const content::MediaStreamRequest* request) override;
std::vector<std::unique_ptr<DesktopMediaList>> CreateMediaList(
const std::vector<DesktopMediaList::Type>& types,
content::WebContents* web_contents,
DesktopMediaList::WebContentsFilter includable_web_contents_filter)
override;
private:
raw_ptr<FakeDesktopMediaPicker, AcrossTasksDanglingUntriaged> picker_;
raw_ptr<TestFlags, AcrossTasksDanglingUntriaged | AllowPtrArithmetic>
test_flags_;
int tests_count_;
int current_test_;
bool is_web_contents_excluded_ = false;
};
class FakeDesktopMediaPicker : public DesktopMediaPicker {
public:
explicit FakeDesktopMediaPicker(
FakeDesktopMediaPickerFactory::TestFlags* expectation);
FakeDesktopMediaPicker(const FakeDesktopMediaPicker&) = delete;
FakeDesktopMediaPicker& operator=(const FakeDesktopMediaPicker&) = delete;
~FakeDesktopMediaPicker() override;
// DesktopMediaPicker interface.
void Show(const DesktopMediaPicker::Params& params,
std::vector<std::unique_ptr<DesktopMediaList>> source_lists,
DoneCallback done_callback) override;
DesktopMediaPicker::Params GetParams();
private:
void CallCallback(DoneCallback done_callback);
raw_ptr<FakeDesktopMediaPickerFactory::TestFlags> expectation_;
DoneCallback done_callback_;
DesktopMediaPicker::Params picker_params_;
base::WeakPtrFactory<FakeDesktopMediaPicker> weak_factory_{this};
};
#endif // CHROME_BROWSER_MEDIA_WEBRTC_FAKE_DESKTOP_MEDIA_PICKER_FACTORY_H_
|
