OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef CHROME_BROWSER_EXTENSIONS_TEST_MANAGEMENT_POLICY_H_ | |
6 #define CHROME_BROWSER_EXTENSIONS_TEST_MANAGEMENT_POLICY_H_ | |
7 | |
8 #include <string> | |
9 | |
10 #include "base/strings/string16.h" | |
11 #include "chrome/browser/extensions/extension_service.h" | |
12 #include "chrome/browser/extensions/management_policy.h" | |
13 | |
14 namespace extensions { | |
15 | |
16 // This class provides a simple way to create providers with specific | |
17 // restrictions and a known error message, for use in testing. | |
18 class TestManagementPolicyProvider : public ManagementPolicy::Provider { | |
19 public: | |
20 enum AllowedActionFlag { | |
21 ALLOW_ALL = 0, | |
22 PROHIBIT_LOAD = 1 << 0, | |
23 PROHIBIT_MODIFY_STATUS = 1 << 1, | |
24 MUST_REMAIN_ENABLED = 1 << 2, | |
25 MUST_REMAIN_DISABLED = 1 << 3 | |
26 }; | |
27 | |
28 static std::string expected_error() { | |
29 return "Action prohibited by test provider."; | |
30 } | |
31 | |
32 TestManagementPolicyProvider(); | |
33 explicit TestManagementPolicyProvider(int prohibited_actions); | |
34 | |
35 void SetProhibitedActions(int prohibited_actions); | |
36 void SetDisableReason(Extension::DisableReason reason); | |
37 | |
38 virtual std::string GetDebugPolicyProviderName() const OVERRIDE; | |
39 | |
40 virtual bool UserMayLoad(const Extension* extension, | |
41 string16* error) const OVERRIDE; | |
42 | |
43 virtual bool UserMayModifySettings(const Extension* extension, | |
44 string16* error) const OVERRIDE; | |
45 | |
46 virtual bool MustRemainEnabled(const Extension* extension, | |
47 string16* error) const OVERRIDE; | |
48 | |
49 virtual bool MustRemainDisabled(const Extension* extension, | |
50 Extension::DisableReason* reason, | |
51 string16* error) const OVERRIDE; | |
52 | |
53 private: | |
54 bool may_load_; | |
55 bool may_modify_status_; | |
56 bool must_remain_enabled_; | |
57 bool must_remain_disabled_; | |
58 Extension::DisableReason disable_reason_; | |
59 | |
60 string16 error_message_; | |
61 }; | |
62 | |
63 } // namespace extensions | |
64 | |
65 #endif // CHROME_BROWSER_EXTENSIONS_TEST_MANAGEMENT_POLICY_H_ | |
OLD | NEW |