OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/chromeos/settings/device_settings_test_helper.h" | 5 #include "chrome/browser/chromeos/settings/device_settings_test_helper.h" |
6 | 6 |
7 #include "base/message_loop/message_loop.h" | 7 #include "base/message_loop/message_loop.h" |
8 #include "base/run_loop.h" | 8 #include "base/run_loop.h" |
9 #include "base/threading/sequenced_worker_pool.h" | 9 #include "base/threading/sequenced_worker_pool.h" |
10 #include "chrome/browser/chromeos/ownership/owner_settings_service_chromeos.h" | 10 #include "chrome/browser/chromeos/ownership/owner_settings_service_chromeos.h" |
11 #include "chrome/browser/chromeos/ownership/owner_settings_service_chromeos_fact
ory.h" | 11 #include "chrome/browser/chromeos/ownership/owner_settings_service_chromeos_fact
ory.h" |
12 #include "chrome/browser/chromeos/policy/proto/chrome_device_policy.pb.h" | 12 #include "chrome/browser/chromeos/policy/proto/chrome_device_policy.pb.h" |
13 #include "chrome/browser/chromeos/profiles/profile_helper.h" | 13 #include "chrome/browser/chromeos/profiles/profile_helper.h" |
14 #include "chrome/browser/chromeos/settings/device_settings_service.h" | 14 #include "chrome/browser/chromeos/settings/device_settings_service.h" |
15 #include "chrome/test/base/testing_browser_process.h" | 15 #include "chrome/test/base/testing_browser_process.h" |
16 #include "chrome/test/base/testing_profile.h" | 16 #include "chrome/test/base/testing_profile.h" |
17 #include "chromeos/cryptohome/cryptohome_parameters.h" | 17 #include "chromeos/cryptohome/cryptohome_parameters.h" |
18 #include "chromeos/dbus/dbus_thread_manager.h" | 18 #include "chromeos/dbus/dbus_thread_manager.h" |
19 #include "components/ownership/mock_owner_key_util.h" | 19 #include "components/ownership/mock_owner_key_util.h" |
20 #include "content/public/browser/browser_thread.h" | 20 #include "content/public/browser/browser_thread.h" |
21 #include "content/public/test/test_utils.h" | 21 #include "content/public/test/test_utils.h" |
22 | 22 |
23 using RetrievePolicyResponseType = | |
24 chromeos::DeviceSettingsTestHelper::RetrievePolicyResponseType; | |
25 | |
26 namespace chromeos { | 23 namespace chromeos { |
27 | 24 |
28 DeviceSettingsTestHelper::DeviceSettingsTestHelper() {} | 25 DeviceSettingsTestHelper::DeviceSettingsTestHelper() {} |
29 | 26 |
30 DeviceSettingsTestHelper::~DeviceSettingsTestHelper() {} | 27 DeviceSettingsTestHelper::~DeviceSettingsTestHelper() {} |
31 | 28 |
32 void DeviceSettingsTestHelper::FlushStore() { | 29 void DeviceSettingsTestHelper::FlushStore() { |
33 std::vector<StorePolicyCallback> callbacks; | 30 std::vector<StorePolicyCallback> callbacks; |
34 callbacks.swap(device_policy_.store_callbacks_); | 31 callbacks.swap(device_policy_.store_callbacks_); |
35 for (std::vector<StorePolicyCallback>::iterator cb(callbacks.begin()); | 32 for (std::vector<StorePolicyCallback>::iterator cb(callbacks.begin()); |
(...skipping 11 matching lines...) Expand all Loading... |
47 cb->Run(device_local_account_state->second.store_result_); | 44 cb->Run(device_local_account_state->second.store_result_); |
48 } | 45 } |
49 } | 46 } |
50 } | 47 } |
51 | 48 |
52 void DeviceSettingsTestHelper::FlushRetrieve() { | 49 void DeviceSettingsTestHelper::FlushRetrieve() { |
53 std::vector<RetrievePolicyCallback> callbacks; | 50 std::vector<RetrievePolicyCallback> callbacks; |
54 callbacks.swap(device_policy_.retrieve_callbacks_); | 51 callbacks.swap(device_policy_.retrieve_callbacks_); |
55 for (std::vector<RetrievePolicyCallback>::iterator cb(callbacks.begin()); | 52 for (std::vector<RetrievePolicyCallback>::iterator cb(callbacks.begin()); |
56 cb != callbacks.end(); ++cb) { | 53 cb != callbacks.end(); ++cb) { |
57 cb->Run(device_policy_.policy_blob_, RetrievePolicyResponseType::SUCCESS); | 54 cb->Run(device_policy_.policy_blob_); |
58 } | 55 } |
59 | 56 |
60 std::map<std::string, PolicyState>::iterator device_local_account_state; | 57 std::map<std::string, PolicyState>::iterator device_local_account_state; |
61 for (device_local_account_state = device_local_account_policy_.begin(); | 58 for (device_local_account_state = device_local_account_policy_.begin(); |
62 device_local_account_state != device_local_account_policy_.end(); | 59 device_local_account_state != device_local_account_policy_.end(); |
63 ++device_local_account_state) { | 60 ++device_local_account_state) { |
64 std::vector<RetrievePolicyCallback> callbacks; | 61 std::vector<RetrievePolicyCallback> callbacks; |
65 callbacks.swap(device_local_account_state->second.retrieve_callbacks_); | 62 callbacks.swap(device_local_account_state->second.retrieve_callbacks_); |
66 for (std::vector<RetrievePolicyCallback>::iterator cb(callbacks.begin()); | 63 for (std::vector<RetrievePolicyCallback>::iterator cb(callbacks.begin()); |
67 cb != callbacks.end(); ++cb) { | 64 cb != callbacks.end(); ++cb) { |
68 cb->Run(device_local_account_state->second.policy_blob_, | 65 cb->Run(device_local_account_state->second.policy_blob_); |
69 RetrievePolicyResponseType::SUCCESS); | |
70 } | 66 } |
71 } | 67 } |
72 } | 68 } |
73 | 69 |
74 void DeviceSettingsTestHelper::Flush() { | 70 void DeviceSettingsTestHelper::Flush() { |
75 do { | 71 do { |
76 content::RunAllBlockingPoolTasksUntilIdle(); | 72 content::RunAllBlockingPoolTasksUntilIdle(); |
77 FlushStore(); | 73 FlushStore(); |
78 content::RunAllBlockingPoolTasksUntilIdle(); | 74 content::RunAllBlockingPoolTasksUntilIdle(); |
79 FlushRetrieve(); | 75 FlushRetrieve(); |
(...skipping 14 matching lines...) Expand all Loading... |
94 } | 90 } |
95 | 91 |
96 return false; | 92 return false; |
97 } | 93 } |
98 | 94 |
99 void DeviceSettingsTestHelper::RetrieveDevicePolicy( | 95 void DeviceSettingsTestHelper::RetrieveDevicePolicy( |
100 const RetrievePolicyCallback& callback) { | 96 const RetrievePolicyCallback& callback) { |
101 device_policy_.retrieve_callbacks_.push_back(callback); | 97 device_policy_.retrieve_callbacks_.push_back(callback); |
102 } | 98 } |
103 | 99 |
104 RetrievePolicyResponseType | 100 std::string DeviceSettingsTestHelper::BlockingRetrieveDevicePolicy() { |
105 DeviceSettingsTestHelper::BlockingRetrieveDevicePolicy( | 101 return device_policy_.policy_blob_; |
106 std::string* policy_out) { | |
107 *policy_out = device_policy_.policy_blob_; | |
108 return RetrievePolicyResponseType::SUCCESS; | |
109 } | 102 } |
110 | 103 |
111 void DeviceSettingsTestHelper::RetrieveDeviceLocalAccountPolicy( | 104 void DeviceSettingsTestHelper::RetrieveDeviceLocalAccountPolicy( |
112 const std::string& account_id, | 105 const std::string& account_id, |
113 const RetrievePolicyCallback& callback) { | 106 const RetrievePolicyCallback& callback) { |
114 device_local_account_policy_[account_id].retrieve_callbacks_.push_back( | 107 device_local_account_policy_[account_id].retrieve_callbacks_.push_back( |
115 callback); | 108 callback); |
116 } | 109 } |
117 | 110 |
118 RetrievePolicyResponseType | 111 std::string DeviceSettingsTestHelper::BlockingRetrieveDeviceLocalAccountPolicy( |
119 DeviceSettingsTestHelper::BlockingRetrieveDeviceLocalAccountPolicy( | 112 const std::string& account_id) { |
120 const std::string& account_id, | 113 return ""; |
121 std::string* policy_out) { | |
122 *policy_out = ""; | |
123 return RetrievePolicyResponseType::SUCCESS; | |
124 } | 114 } |
125 | 115 |
126 void DeviceSettingsTestHelper::StoreDevicePolicy( | 116 void DeviceSettingsTestHelper::StoreDevicePolicy( |
127 const std::string& policy_blob, | 117 const std::string& policy_blob, |
128 const StorePolicyCallback& callback) { | 118 const StorePolicyCallback& callback) { |
129 device_policy_.policy_blob_ = policy_blob; | 119 device_policy_.policy_blob_ = policy_blob; |
130 device_policy_.store_callbacks_.push_back(callback); | 120 device_policy_.store_callbacks_.push_back(callback); |
131 } | 121 } |
132 | 122 |
133 void DeviceSettingsTestHelper::StoreDeviceLocalAccountPolicy( | 123 void DeviceSettingsTestHelper::StoreDeviceLocalAccountPolicy( |
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
220 const_cast<user_manager::User*>(user)); | 210 const_cast<user_manager::User*>(user)); |
221 } | 211 } |
222 OwnerSettingsServiceChromeOS* service = | 212 OwnerSettingsServiceChromeOS* service = |
223 OwnerSettingsServiceChromeOSFactory::GetForBrowserContext(profile_.get()); | 213 OwnerSettingsServiceChromeOSFactory::GetForBrowserContext(profile_.get()); |
224 CHECK(service); | 214 CHECK(service); |
225 if (tpm_is_ready) | 215 if (tpm_is_ready) |
226 service->OnTPMTokenReady(true /* token is enabled */); | 216 service->OnTPMTokenReady(true /* token is enabled */); |
227 } | 217 } |
228 | 218 |
229 } // namespace chromeos | 219 } // namespace chromeos |
OLD | NEW |