| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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/policy/device_policy_cros_browser_test.h" | 5 #include "chrome/browser/chromeos/policy/device_policy_cros_browser_test.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| 11 #include "base/files/file_path.h" | 11 #include "base/files/file_path.h" |
| 12 #include "base/path_service.h" | 12 #include "base/path_service.h" |
| 13 #include "base/stl_util.h" | 13 #include "base/stl_util.h" |
| 14 #include "chrome/browser/chromeos/policy/device_policy_builder.h" | 14 #include "chrome/browser/chromeos/policy/device_policy_builder.h" |
| 15 #include "chrome/browser/chromeos/policy/enterprise_install_attributes.h" | 15 #include "chrome/browser/chromeos/policy/enterprise_install_attributes.h" |
| 16 #include "chrome/browser/chromeos/policy/proto/install_attributes.pb.h" | 16 #include "chrome/browser/chromeos/policy/proto/install_attributes.pb.h" |
| 17 #include "chrome/common/chrome_paths.h" | |
| 18 #include "chromeos/chromeos_paths.h" | 17 #include "chromeos/chromeos_paths.h" |
| 19 #include "chromeos/dbus/fake_dbus_thread_manager.h" | 18 #include "chromeos/dbus/fake_dbus_thread_manager.h" |
| 20 #include "chromeos/dbus/fake_session_manager_client.h" | 19 #include "chromeos/dbus/fake_session_manager_client.h" |
| 21 #include "crypto/rsa_private_key.h" | 20 #include "crypto/rsa_private_key.h" |
| 22 #include "testing/gmock/include/gmock/gmock.h" | 21 #include "testing/gmock/include/gmock/gmock.h" |
| 23 #include "testing/gtest/include/gtest/gtest.h" | 22 #include "testing/gtest/include/gtest/gtest.h" |
| 24 | 23 |
| 25 using ::testing::_; | 24 using ::testing::_; |
| 26 using ::testing::AnyNumber; | 25 using ::testing::AnyNumber; |
| 27 using ::testing::Return; | 26 using ::testing::Return; |
| 28 | 27 |
| 29 namespace policy { | 28 namespace policy { |
| 30 | 29 |
| 31 DevicePolicyCrosTestHelper::DevicePolicyCrosTestHelper() {} | 30 DevicePolicyCrosTestHelper::DevicePolicyCrosTestHelper() { |
| 31 CHECK(temp_dir_.CreateUniqueTempDir()); |
| 32 } |
| 32 | 33 |
| 33 DevicePolicyCrosTestHelper::~DevicePolicyCrosTestHelper() {} | 34 DevicePolicyCrosTestHelper::~DevicePolicyCrosTestHelper() {} |
| 34 | 35 |
| 35 void DevicePolicyCrosTestHelper::MarkAsEnterpriseOwned() { | 36 void DevicePolicyCrosTestHelper::MarkAsEnterpriseOwned() { |
| 36 OverridePaths(); | |
| 37 | |
| 38 cryptohome::SerializedInstallAttributes install_attrs_proto; | 37 cryptohome::SerializedInstallAttributes install_attrs_proto; |
| 39 cryptohome::SerializedInstallAttributes::Attribute* attribute = NULL; | 38 cryptohome::SerializedInstallAttributes::Attribute* attribute = NULL; |
| 40 | 39 |
| 41 attribute = install_attrs_proto.add_attributes(); | 40 attribute = install_attrs_proto.add_attributes(); |
| 42 attribute->set_name(EnterpriseInstallAttributes::kAttrEnterpriseOwned); | 41 attribute->set_name(EnterpriseInstallAttributes::kAttrEnterpriseOwned); |
| 43 attribute->set_value("true"); | 42 attribute->set_value("true"); |
| 44 | 43 |
| 45 attribute = install_attrs_proto.add_attributes(); | 44 attribute = install_attrs_proto.add_attributes(); |
| 46 attribute->set_name(EnterpriseInstallAttributes::kAttrEnterpriseUser); | 45 attribute->set_name(EnterpriseInstallAttributes::kAttrEnterpriseUser); |
| 47 attribute->set_value(device_policy_.policy_data().username()); | 46 attribute->set_value(device_policy_.policy_data().username()); |
| 48 | 47 |
| 49 base::FilePath install_attrs_file; | 48 base::FilePath install_attrs_file = |
| 50 ASSERT_TRUE( | 49 temp_dir_.path().AppendASCII("install_attributes.pb"); |
| 51 PathService::Get(chromeos::FILE_INSTALL_ATTRIBUTES, &install_attrs_file)); | |
| 52 const std::string install_attrs_blob( | 50 const std::string install_attrs_blob( |
| 53 install_attrs_proto.SerializeAsString()); | 51 install_attrs_proto.SerializeAsString()); |
| 54 ASSERT_EQ(static_cast<int>(install_attrs_blob.size()), | 52 ASSERT_EQ(static_cast<int>(install_attrs_blob.size()), |
| 55 base::WriteFile(install_attrs_file, | 53 base::WriteFile(install_attrs_file, |
| 56 install_attrs_blob.c_str(), | 54 install_attrs_blob.c_str(), |
| 57 install_attrs_blob.size())); | 55 install_attrs_blob.size())); |
| 56 ASSERT_TRUE(PathService::Override(chromeos::FILE_INSTALL_ATTRIBUTES, |
| 57 install_attrs_file)); |
| 58 } | 58 } |
| 59 | 59 |
| 60 void DevicePolicyCrosTestHelper::InstallOwnerKey() { | 60 void DevicePolicyCrosTestHelper::InstallOwnerKey() { |
| 61 OverridePaths(); | 61 base::FilePath owner_key_file = temp_dir_.path().AppendASCII("owner.key"); |
| 62 | |
| 63 base::FilePath owner_key_file; | |
| 64 ASSERT_TRUE(PathService::Get(chromeos::FILE_OWNER_KEY, &owner_key_file)); | |
| 65 std::vector<uint8> owner_key_bits; | 62 std::vector<uint8> owner_key_bits; |
| 66 ASSERT_TRUE( | 63 ASSERT_TRUE( |
| 67 device_policy()->GetSigningKey()->ExportPublicKey(&owner_key_bits)); | 64 device_policy()->GetSigningKey()->ExportPublicKey(&owner_key_bits)); |
| 68 ASSERT_EQ(base::WriteFile( | 65 ASSERT_EQ(base::WriteFile( |
| 69 owner_key_file, | 66 owner_key_file, |
| 70 reinterpret_cast<const char*>(vector_as_array(&owner_key_bits)), | 67 reinterpret_cast<const char*>(vector_as_array(&owner_key_bits)), |
| 71 owner_key_bits.size()), | 68 owner_key_bits.size()), |
| 72 static_cast<int>(owner_key_bits.size())); | 69 static_cast<int>(owner_key_bits.size())); |
| 73 } | 70 ASSERT_TRUE(PathService::Override(chromeos::FILE_OWNER_KEY, owner_key_file)); |
| 74 | |
| 75 void DevicePolicyCrosTestHelper::OverridePaths() { | |
| 76 // This is usually done by ChromeBrowserMainChromeOS, but some tests | |
| 77 // use the overridden paths before ChromeBrowserMain starts. Make sure that | |
| 78 // the paths are overridden before using them. | |
| 79 base::FilePath user_data_dir; | |
| 80 ASSERT_TRUE(PathService::Get(chrome::DIR_USER_DATA, &user_data_dir)); | |
| 81 chromeos::RegisterStubPathOverrides(user_data_dir); | |
| 82 } | 71 } |
| 83 | 72 |
| 84 DevicePolicyCrosBrowserTest::DevicePolicyCrosBrowserTest() | 73 DevicePolicyCrosBrowserTest::DevicePolicyCrosBrowserTest() |
| 85 : fake_dbus_thread_manager_(new chromeos::FakeDBusThreadManager), | 74 : fake_dbus_thread_manager_(new chromeos::FakeDBusThreadManager), |
| 86 fake_session_manager_client_(new chromeos::FakeSessionManagerClient) { | 75 fake_session_manager_client_(new chromeos::FakeSessionManagerClient) { |
| 87 fake_dbus_thread_manager_->SetFakeClients(); | 76 fake_dbus_thread_manager_->SetFakeClients(); |
| 88 fake_dbus_thread_manager_->SetSessionManagerClient( | 77 fake_dbus_thread_manager_->SetSessionManagerClient( |
| 89 scoped_ptr<chromeos::SessionManagerClient>(fake_session_manager_client_)); | 78 scoped_ptr<chromeos::SessionManagerClient>(fake_session_manager_client_)); |
| 90 } | 79 } |
| 91 | 80 |
| (...skipping 19 matching lines...) Expand all Loading... |
| 111 | 100 |
| 112 void DevicePolicyCrosBrowserTest::RefreshDevicePolicy() { | 101 void DevicePolicyCrosBrowserTest::RefreshDevicePolicy() { |
| 113 // Reset the key to its original state. | 102 // Reset the key to its original state. |
| 114 device_policy()->SetDefaultSigningKey(); | 103 device_policy()->SetDefaultSigningKey(); |
| 115 device_policy()->Build(); | 104 device_policy()->Build(); |
| 116 session_manager_client()->set_device_policy(device_policy()->GetBlob()); | 105 session_manager_client()->set_device_policy(device_policy()->GetBlob()); |
| 117 session_manager_client()->OnPropertyChangeComplete(true); | 106 session_manager_client()->OnPropertyChangeComplete(true); |
| 118 } | 107 } |
| 119 | 108 |
| 120 } // namespace policy | 109 } // namespace policy |
| OLD | NEW |