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
|
// 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.
#ifndef CHROME_BROWSER_DEVICE_NOTIFICATIONS_DEVICE_CONNECTION_TRACKER_UNITTEST_H_
#define CHROME_BROWSER_DEVICE_NOTIFICATIONS_DEVICE_CONNECTION_TRACKER_UNITTEST_H_
#include <memory>
#include <string>
#include "build/build_config.h"
#include "build/chromeos_buildflags.h"
#include "chrome/browser/device_notifications/device_connection_tracker.h"
#include "chrome/browser/device_notifications/device_system_tray_icon.h"
#include "chrome/browser/device_notifications/device_test_utils.h"
#include "chrome/test/base/browser_with_test_window_test.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "url/origin.h"
#if BUILDFLAG(ENABLE_EXTENSIONS)
#include "base/command_line.h"
#include "base/values.h"
#include "chrome/browser/extensions/extension_service.h"
#include "chrome/browser/extensions/test_extension_system.h"
#include "extensions/common/extension.h"
#include "extensions/common/extension_builder.h"
#endif // BUILDFLAG(ENABLE_EXTENSIONS)
class DeviceConnectionTrackerTestBase : public BrowserWithTestWindowTest {
public:
DeviceConnectionTrackerTestBase();
DeviceConnectionTrackerTestBase(const DeviceConnectionTrackerTestBase&) =
delete;
DeviceConnectionTrackerTestBase& operator=(
const DeviceConnectionTrackerTestBase&) = delete;
~DeviceConnectionTrackerTestBase() override;
void SetUp() override;
virtual DeviceConnectionTracker* GetDeviceConnectionTracker(Profile* profile,
bool create) = 0;
virtual MockDeviceSystemTrayIcon* GetMockDeviceSystemTrayIcon() = 0;
Profile* CreateTestingProfile(const std::string& profile_name);
void TestDeviceConnection(
bool has_system_tray_icon,
const std::vector<std::pair<url::Origin, std::string>>&
origin_name_pairs);
// Test the scenario when the origin is whitelisted.
void TestWhitelistedOrigin(
const std::pair<url::Origin, std::string> whitelisted_origin,
const std::pair<url::Origin, std::string> origin2);
#if BUILDFLAG(ENABLE_EXTENSIONS)
scoped_refptr<const extensions::Extension> CreateExtensionWithName(
const std::string& extension_name);
// Create a extension with |extension_name| and |extension_id|.
scoped_refptr<const extensions::Extension> CreateExtensionWithNameAndId(
const std::string& extension_name,
const std::string& extension_id);
void AddExtensionToProfile(Profile* profile,
const extensions::Extension* extension);
void TestDeviceConnectionExtensionOrigins(bool has_system_tray_icon);
void TestProfileDestroyedExtensionOrigin();
// Run TestWhitelistedOrigin with a whitelisted origin and a non-whitelisted
// origin.
void TestSingleProfileWhitelistedExtension(
std::string whitelisted_extension_name,
std::string whitelisted_extension_id);
#endif // BUILDFLAG(ENABLE_EXTENSIONS)
};
#endif // CHROME_BROWSER_DEVICE_NOTIFICATIONS_DEVICE_CONNECTION_TRACKER_UNITTEST_H_
|
