blob: 0c5f79ae992b95c679879ad923cd3e1de19803dd [file] [log] [blame]
// Copyright (c) 2012 The Chromium OS 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 LIBBRILLO_POLICY_MOCK_DEVICE_POLICY_H_
#define LIBBRILLO_POLICY_MOCK_DEVICE_POLICY_H_
#include <set>
#include <string>
#include <vector>
#include <gmock/gmock.h>
#include "policy/device_policy.h"
#pragma GCC visibility push(default)
namespace policy {
// This is a generic mock class for the DevicePolicy that can be used by other
// subsystems for tests. It allows to mock out the reading of a real policy
// file and to simulate different policy values.
// The test that needs policies would then do something like this :
// // Prepare the action that would return a predefined policy value:
// ACTION_P(SetMetricsPolicy, enabled) {
// *arg0 = enabled;
// return true;
// }
// ...
// // Initialize the Mock class
// policy::MockDevicePolicy* device_policy = new policy::MockDevicePolicy();
// // We should expect calls to LoadPolicy almost always and return true.
// EXPECT_CALL(*device_policy_, LoadPolicy())
// .Times(AnyNumber())
// .WillRepeatedly(Return(true));
// // This example needs to simulate the Metrics Enabled policy being set.
// EXPECT_CALL(*device_policy_, GetMetricsEnabled(_))
// .Times(AnyNumber())
// .WillRepeatedly(SetMetricsPolicy(true));
// policy::PolicyProvider provider(device_policy);
// ...
// // In a test that needs other value of that policy we can do that:
// EXPECT_CALL(*device_policy_, GetMetricsEnabled(_))
// .WillOnce(SetMetricsPolicy(false));
//
// See metrics_library_test.cc for example.
class MockDevicePolicy : public DevicePolicy {
public:
MockDevicePolicy() {
ON_CALL(*this, LoadPolicy()).WillByDefault(testing::Return(true));
}
~MockDevicePolicy() override = default;
MOCK_METHOD0(LoadPolicy, bool(void));
MOCK_CONST_METHOD1(GetPolicyRefreshRate,
bool(int*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetUserWhitelist, bool(std::vector<std::string>*));
MOCK_CONST_METHOD1(GetGuestModeEnabled,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetCameraEnabled,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetShowUserNames,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetDataRoamingEnabled,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetAllowNewUsers,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetMetricsEnabled,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetReportVersionInfo,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetReportActivityTimes,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetReportBootMode,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetEphemeralUsersEnabled,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetReleaseChannel, bool(std::string*));
MOCK_CONST_METHOD1(GetReleaseChannelDelegated,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetUpdateDisabled,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetTargetVersionPrefix, bool(std::string*));
MOCK_CONST_METHOD1(GetScatterFactorInSeconds,
bool(int64_t*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetAllowedConnectionTypesForUpdate,
bool(std::set<std::string>*));
MOCK_CONST_METHOD1(GetOpenNetworkConfiguration, bool(std::string*));
MOCK_CONST_METHOD1(GetOwner, bool(std::string*));
MOCK_CONST_METHOD1(GetHttpDownloadsEnabled,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetAuP2PEnabled,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetAllowKioskAppControlChromeVersion,
bool(bool*)); // NOLINT(readability/function)
MOCK_CONST_METHOD1(GetUsbDetachableWhitelist,
bool(std::vector<DevicePolicy::UsbDeviceId>*));
MOCK_CONST_METHOD1(GetAutoLaunchedKioskAppId, bool(std::string*));
MOCK_CONST_METHOD0(IsEnterpriseManaged, bool());
MOCK_METHOD0(VerifyPolicyFiles, bool(void));
MOCK_METHOD0(VerifyPolicySignature, bool(void));
};
} // namespace policy
#pragma GCC visibility pop
#endif // LIBBRILLO_POLICY_MOCK_DEVICE_POLICY_H_