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
|
// Copyright 2023 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/device_notifications/device_connection_tracker_unittest.h"
#include "chrome/browser/usb/usb_connection_tracker_factory.h"
#include "chrome/browser/usb/usb_test_utils.h"
#include "chrome/test/base/testing_browser_process.h"
#include "testing/gtest/include/gtest/gtest.h"
class UsbConnectionTrackerTest : public DeviceConnectionTrackerTestBase {
public:
UsbConnectionTrackerTest() = default;
UsbConnectionTrackerTest(const UsbConnectionTrackerTest&) = delete;
UsbConnectionTrackerTest& operator=(const UsbConnectionTrackerTest&) = delete;
~UsbConnectionTrackerTest() override = default;
void SetUp() override {
DeviceConnectionTrackerTestBase::SetUp();
auto usb_system_tray_icon = std::make_unique<TestUsbSystemTrayIcon>();
TestingBrowserProcess::GetGlobal()->SetUsbSystemTrayIcon(
std::move(usb_system_tray_icon));
}
void TearDown() override {
// Set the system tray icon to null to avoid uninteresting call to it during
// profile destruction.
TestingBrowserProcess::GetGlobal()->SetUsbSystemTrayIcon(nullptr);
DeviceConnectionTrackerTestBase::TearDown();
}
DeviceConnectionTracker* GetDeviceConnectionTracker(Profile* profile,
bool create) override {
return UsbConnectionTrackerFactory::GetForProfile(profile, create);
}
MockDeviceSystemTrayIcon* GetMockDeviceSystemTrayIcon() override {
TestUsbSystemTrayIcon* test_usb_system_tray_icon =
static_cast<TestUsbSystemTrayIcon*>(
TestingBrowserProcess::GetGlobal()->usb_system_tray_icon());
if (!test_usb_system_tray_icon) {
return nullptr;
}
return test_usb_system_tray_icon->mock_device_system_tray_icon();
}
};
#if BUILDFLAG(ENABLE_EXTENSIONS)
TEST_F(UsbConnectionTrackerTest, DeviceConnectionExtensionOrigins) {
TestDeviceConnectionExtensionOrigins(/*has_system_tray_icon=*/true);
}
// Test the scenario with null USB system tray icon and it doesn't cause crash.
TEST_F(UsbConnectionTrackerTest,
DeviceConnectionExtensionOriginsWithNullSystemTrayIcon) {
TestingBrowserProcess::GetGlobal()->SetUsbSystemTrayIcon(nullptr);
TestDeviceConnectionExtensionOrigins(/*has_system_tray_icon=*/false);
}
TEST_F(UsbConnectionTrackerTest, ProfileDestroyedExtensionOrigin) {
TestProfileDestroyedExtensionOrigin();
}
#endif // BUILDFLAG(ENABLE_EXTENSIONS)
|
