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
|
// Copyright 2019 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_ASH_CROSTINI_FAKE_CROSTINI_FEATURES_H_
#define CHROME_BROWSER_ASH_CROSTINI_FAKE_CROSTINI_FEATURES_H_
#include <optional>
#include "base/memory/raw_ptr.h"
#include "chrome/browser/ash/crostini/crostini_features.h"
class Profile;
namespace crostini {
// FakeCrostiniFeatures implements a fake version of CrostiniFeatures which can
// be used for testing. It captures the current global CrostiniFeatures object
// and replaces it for the scope of this object. It overrides only the
// features that you set and uses the previous object for other features.
class FakeCrostiniFeatures : public CrostiniFeatures {
public:
FakeCrostiniFeatures();
~FakeCrostiniFeatures() override;
// CrostiniFeatures:
bool CouldBeAllowed(Profile* profile, std::string* reason) override;
bool IsAllowedNow(Profile* profile, std::string* reason) override;
bool IsEnabled(Profile* profile) override;
bool IsExportImportUIAllowed(Profile* profile) override;
bool IsRootAccessAllowed(Profile* profile) override;
bool IsContainerUpgradeUIAllowed(Profile* profile) override;
void CanChangeAdbSideloading(
Profile* profile,
CanChangeAdbSideloadingCallback callback) override;
bool IsPortForwardingAllowed(Profile* profile) override;
bool IsMultiContainerAllowed(Profile* profile) override;
void SetAll(bool flag);
void ClearAll();
void set_could_be_allowed(bool allowed) { could_be_allowed_ = allowed; }
void set_is_allowed_now(bool allowed) { allowed_now_ = allowed; }
void set_enabled(bool enabled) { enabled_ = enabled; }
void set_export_import_ui_allowed(bool allowed) {
export_import_ui_allowed_ = allowed;
}
void set_root_access_allowed(bool allowed) { root_access_allowed_ = allowed; }
void set_container_upgrade_ui_allowed(bool allowed) {
container_upgrade_ui_allowed_ = allowed;
}
void set_can_change_adb_sideloading(bool can_change) {
can_change_adb_sideloading_ = can_change;
}
void set_port_forwarding_allowed(bool allowed) {
port_forwarding_allowed_ = allowed;
}
void set_multi_container_allowed(bool allowed) {
multi_container_allowed_ = allowed;
}
private:
// Original global static when this instance is created. It is captured when
// FakeCrostiniFeatures is created and replaced at destruction.
raw_ptr<CrostiniFeatures> original_features_;
std::optional<bool> could_be_allowed_;
std::optional<bool> allowed_now_;
std::optional<bool> enabled_;
std::optional<bool> export_import_ui_allowed_;
std::optional<bool> root_access_allowed_;
std::optional<bool> container_upgrade_ui_allowed_;
std::optional<bool> can_change_adb_sideloading_;
std::optional<bool> port_forwarding_allowed_;
std::optional<bool> multi_container_allowed_;
};
} // namespace crostini
#endif // CHROME_BROWSER_ASH_CROSTINI_FAKE_CROSTINI_FEATURES_H_
|
