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
|
// Copyright 2020 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_WEBSHARE_WIN_FAKE_DATA_TRANSFER_MANAGER_INTEROP_H_
#define CHROME_BROWSER_WEBSHARE_WIN_FAKE_DATA_TRANSFER_MANAGER_INTEROP_H_
#include <shlobj.h>
#include <wrl/implements.h>
#include <map>
#include "base/functional/callback_forward.h"
namespace webshare {
class FakeDataTransferManager;
// Provides an implementation of IDataTransferManagerInterop for use in GTests.
//
// Like the Windows implementation, this class cloaks its implementation of
// IDataTransferManagerInterop to closely match casting behaviors.
class FakeDataTransferManagerInterop final
: public Microsoft::WRL::RuntimeClass<
Microsoft::WRL::RuntimeClassFlags<
Microsoft::WRL::RuntimeClassType::WinRtClassicComMix>,
Microsoft::WRL::CloakedIid<IDataTransferManagerInterop>> {
public:
// Behavior options for the ShowShareUIForWindow API
enum ShowShareUIForWindowBehavior {
// Returns a failed value without invoking/scheduling the DataRequested
// event/handler.
FailImmediately,
// Invokes the DataRequested event/handler synchronously as part of the
// original invoking call. This matches the behavior exposed by Windows
// under various edge-case scenarios.
InvokeEventSynchronously,
// Invokes the DataRequested event/handler synchronously as part of the
// original invoking call, but then returns a failure result.
InvokeEventSynchronouslyAndReturnFailure,
// Schedules the invocation of the DataRequested event/handler to happen
// automatically, outside of the original invoking call. This matches the
// the most common behavior exposed by Windows.
ScheduleEvent,
// Returns a success value without invoking/scheduling the DataRequested
// event/handler. To later invoke the DataRequested event/handler, see
// |GetDataRequestedInvoker|.
SucceedWithoutAction
};
FakeDataTransferManagerInterop();
FakeDataTransferManagerInterop(const FakeDataTransferManagerInterop&) =
delete;
FakeDataTransferManagerInterop& operator=(
const FakeDataTransferManagerInterop&) = delete;
~FakeDataTransferManagerInterop() final;
// IDataTransferManagerInterop:
IFACEMETHODIMP GetForWindow(HWND app_window,
REFIID riid,
void** data_transfer_manager) final;
IFACEMETHODIMP ShowShareUIForWindow(HWND app_window) final;
// Returns a callback that captures a reference to the current DataRequested
// event handler and, when invoked, triggers that handler.
//
// If the registered handler changes after this method is called the callback
// will still trigger the previous event handler, not a newly registered one.
base::OnceClosure GetDataRequestedInvoker(HWND app_window);
// Checks if there are any listeners registered for the DataRequested event
// on the given |app_window|.
bool HasDataRequestedListener(HWND app_window);
void SetShowShareUIForWindowBehavior(ShowShareUIForWindowBehavior behavior);
private:
ShowShareUIForWindowBehavior show_share_ui_for_window_behavior_ =
ShowShareUIForWindowBehavior::ScheduleEvent;
std::map<HWND, Microsoft::WRL::ComPtr<FakeDataTransferManager>> managers_;
};
} // namespace webshare
#endif // CHROME_BROWSER_WEBSHARE_WIN_FAKE_DATA_TRANSFER_MANAGER_INTEROP_H_
|
