diff options
Diffstat (limited to 'policy/mock_device_policy.h')
-rw-r--r-- | policy/mock_device_policy.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/policy/mock_device_policy.h b/policy/mock_device_policy.h index f26433f..90470e2 100644 --- a/policy/mock_device_policy.h +++ b/policy/mock_device_policy.h @@ -7,6 +7,7 @@ #include <set> #include <string> +#include <utility> #include <vector> #include <gmock/gmock.h> @@ -103,8 +104,8 @@ class MockDevicePolicy : public DevicePolicy { MOCK_CONST_METHOD1(GetSecondFactorAuthenticationMode, bool(int*)); MOCK_CONST_METHOD1(GetDisallowedTimeIntervals, bool(std::vector<WeeklyTimeInterval>*)); - MOCK_CONST_METHOD1(GetDeviceUpdateStagingSchedule, bool(std::vector<int> *)); - + MOCK_CONST_METHOD1(GetDeviceUpdateStagingSchedule, + bool(std::vector<DayPercentagePair>*)); MOCK_METHOD0(VerifyPolicyFiles, bool(void)); MOCK_METHOD0(VerifyPolicySignature, bool(void)); }; |