blob: 8ddbc4a2e0997c1ef0e805cdcd364c77e57bcdd7 [file] [log] [blame]
// Copyright 2019 The Chromium Authors. All rights reserved.
// 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 "base/macros.h"
#include "chrome/browser/ash/crostini/crostini_features.h"
#include "third_party/abseil-cpp/absl/types/optional.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;
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;
}
private:
// Original global static when this instance is created. It is captured when
// FakeCrostiniFeatures is created and replaced at destruction.
CrostiniFeatures* original_features_;
absl::optional<bool> could_be_allowed_;
absl::optional<bool> allowed_now_;
absl::optional<bool> enabled_;
absl::optional<bool> export_import_ui_allowed_;
absl::optional<bool> root_access_allowed_;
absl::optional<bool> container_upgrade_ui_allowed_;
absl::optional<bool> can_change_adb_sideloading_;
absl::optional<bool> port_forwarding_allowed_;
};
} // namespace crostini
#endif // CHROME_BROWSER_ASH_CROSTINI_FAKE_CROSTINI_FEATURES_H_