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
|
// 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.
#include "chrome/browser/webshare/win/fake_data_transfer_manager_interop.h"
#include "base/functional/callback.h"
#include "base/functional/callback_helpers.h"
#include "chrome/browser/webshare/win/fake_data_transfer_manager.h"
#include "content/public/browser/browser_task_traits.h"
#include "content/public/browser/browser_thread.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace webshare {
FakeDataTransferManagerInterop::FakeDataTransferManagerInterop() = default;
FakeDataTransferManagerInterop::~FakeDataTransferManagerInterop() {
// Though it is legal for consuming code to hold on to a DataTransferManager
// after releasing all references to the DataTransferManagerInterop, in a
// test environment the DataTransferManagerInterop is only expected to be
// torn down as part of the test cleanup, at which point the
// DataTransferManager references should also have been cleaned up.
for (auto& manager : managers_)
EXPECT_EQ(0u, manager.second.Reset());
}
IFACEMETHODIMP
FakeDataTransferManagerInterop::GetForWindow(HWND app_window,
REFIID riid,
void** data_transfer_manager) {
auto it = managers_.find(app_window);
if (it != managers_.end()) {
*data_transfer_manager = it->second.Get();
it->second->AddRef();
} else {
auto mock = Microsoft::WRL::Make<FakeDataTransferManager>();
managers_.insert({app_window, mock});
*data_transfer_manager = mock.Get();
mock->AddRef();
}
return S_OK;
}
IFACEMETHODIMP FakeDataTransferManagerInterop::ShowShareUIForWindow(
HWND app_window) {
auto it = managers_.find(app_window);
if (it == managers_.end()) {
ADD_FAILURE() << "ShowShareUIForWindow called for HWND with no "
"DataTransferManager (or DataRequested handler) defined.";
return E_FAIL;
}
switch (show_share_ui_for_window_behavior_) {
case ShowShareUIForWindowBehavior::FailImmediately:
return E_FAIL;
case ShowShareUIForWindowBehavior::InvokeEventSynchronously:
std::move(it->second->GetDataRequestedInvoker()).Run();
return S_OK;
case ShowShareUIForWindowBehavior::InvokeEventSynchronouslyAndReturnFailure:
std::move(it->second->GetDataRequestedInvoker()).Run();
return E_FAIL;
case ShowShareUIForWindowBehavior::ScheduleEvent:
EXPECT_TRUE(content::GetUIThreadTaskRunner({})->PostTask(
FROM_HERE, it->second->GetDataRequestedInvoker()));
return S_OK;
case ShowShareUIForWindowBehavior::SucceedWithoutAction:
return S_OK;
}
NOTREACHED();
}
base::OnceClosure FakeDataTransferManagerInterop::GetDataRequestedInvoker(
HWND app_window) {
auto it = managers_.find(app_window);
if (it == managers_.end()) {
ADD_FAILURE() << "GetDataRequestedInvoker called when no DataRequested "
"event handler has been registered";
return base::DoNothing();
}
return it->second->GetDataRequestedInvoker();
}
bool FakeDataTransferManagerInterop::HasDataRequestedListener(HWND app_window) {
auto it = managers_.find(app_window);
if (it == managers_.end())
return false;
return it->second->HasDataRequestedListener();
}
void FakeDataTransferManagerInterop::SetShowShareUIForWindowBehavior(
ShowShareUIForWindowBehavior behavior) {
show_share_ui_for_window_behavior_ = behavior;
}
} // namespace webshare
|
