| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 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/download/download_dir_policy_handler.h" | 5 #include "chrome/browser/download/download_dir_policy_handler.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/compiler_specific.h" | 9 #include "base/compiler_specific.h" |
| 10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| (...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 65 } | 65 } |
| 66 | 66 |
| 67 protected: | 67 protected: |
| 68 scoped_refptr<policy::ConfigurationPolicyPrefStore> recommended_store_; | 68 scoped_refptr<policy::ConfigurationPolicyPrefStore> recommended_store_; |
| 69 }; | 69 }; |
| 70 | 70 |
| 71 #if !defined(OS_CHROMEOS) | 71 #if !defined(OS_CHROMEOS) |
| 72 TEST_F(DownloadDirPolicyHandlerTest, SetDownloadDirectory) { | 72 TEST_F(DownloadDirPolicyHandlerTest, SetDownloadDirectory) { |
| 73 policy::PolicyMap policy; | 73 policy::PolicyMap policy; |
| 74 EXPECT_FALSE(store_->GetValue(prefs::kPromptForDownload, NULL)); | 74 EXPECT_FALSE(store_->GetValue(prefs::kPromptForDownload, NULL)); |
| 75 policy.Set(policy::key::kDownloadDirectory, | 75 policy.Set(policy::key::kDownloadDirectory, policy::POLICY_LEVEL_MANDATORY, |
| 76 policy::POLICY_LEVEL_MANDATORY, | 76 policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD, |
| 77 policy::POLICY_SCOPE_USER, | 77 base::WrapUnique(new base::StringValue(std::string())), nullptr); |
| 78 policy::POLICY_SOURCE_CLOUD, | |
| 79 new base::StringValue(std::string()), | |
| 80 NULL); | |
| 81 UpdateProviderPolicy(policy); | 78 UpdateProviderPolicy(policy); |
| 82 | 79 |
| 83 // Setting a DownloadDirectory should disable the PromptForDownload pref. | 80 // Setting a DownloadDirectory should disable the PromptForDownload pref. |
| 84 const base::Value* value = NULL; | 81 const base::Value* value = NULL; |
| 85 EXPECT_TRUE(store_->GetValue(prefs::kPromptForDownload, &value)); | 82 EXPECT_TRUE(store_->GetValue(prefs::kPromptForDownload, &value)); |
| 86 ASSERT_TRUE(value); | 83 ASSERT_TRUE(value); |
| 87 bool prompt_for_download = true; | 84 bool prompt_for_download = true; |
| 88 bool result = value->GetAsBoolean(&prompt_for_download); | 85 bool result = value->GetAsBoolean(&prompt_for_download); |
| 89 ASSERT_TRUE(result); | 86 ASSERT_TRUE(result); |
| 90 EXPECT_FALSE(prompt_for_download); | 87 EXPECT_FALSE(prompt_for_download); |
| 91 } | 88 } |
| 92 #endif | 89 #endif |
| 93 | 90 |
| 94 #if defined(OS_CHROMEOS) | 91 #if defined(OS_CHROMEOS) |
| 95 TEST_F(DownloadDirPolicyHandlerTest, SetDownloadToDrive) { | 92 TEST_F(DownloadDirPolicyHandlerTest, SetDownloadToDrive) { |
| 96 EXPECT_FALSE(store_->GetValue(prefs::kPromptForDownload, NULL)); | 93 EXPECT_FALSE(store_->GetValue(prefs::kPromptForDownload, NULL)); |
| 97 | 94 |
| 98 policy::PolicyMap policy; | 95 policy::PolicyMap policy; |
| 99 policy.Set(policy::key::kDownloadDirectory, | 96 policy.Set( |
| 100 policy::POLICY_LEVEL_MANDATORY, | 97 policy::key::kDownloadDirectory, policy::POLICY_LEVEL_MANDATORY, |
| 101 policy::POLICY_SCOPE_USER, | 98 policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD, |
| 102 policy::POLICY_SOURCE_CLOUD, | 99 base::WrapUnique(new base::StringValue(kDriveNamePolicyVariableName)), |
| 103 new base::StringValue(kDriveNamePolicyVariableName), | 100 nullptr); |
| 104 NULL); | |
| 105 UpdateProviderPolicy(policy); | 101 UpdateProviderPolicy(policy); |
| 106 | 102 |
| 107 const base::Value* value = NULL; | 103 const base::Value* value = NULL; |
| 108 bool prompt_for_download; | 104 bool prompt_for_download; |
| 109 EXPECT_TRUE(store_->GetValue(prefs::kPromptForDownload, &value)); | 105 EXPECT_TRUE(store_->GetValue(prefs::kPromptForDownload, &value)); |
| 110 EXPECT_TRUE(value); | 106 EXPECT_TRUE(value); |
| 111 EXPECT_TRUE(value->GetAsBoolean(&prompt_for_download)); | 107 EXPECT_TRUE(value->GetAsBoolean(&prompt_for_download)); |
| 112 EXPECT_FALSE(prompt_for_download); | 108 EXPECT_FALSE(prompt_for_download); |
| 113 | 109 |
| 114 bool disable_drive; | 110 bool disable_drive; |
| 115 EXPECT_TRUE(store_->GetValue(drive::prefs::kDisableDrive, &value)); | 111 EXPECT_TRUE(store_->GetValue(drive::prefs::kDisableDrive, &value)); |
| 116 EXPECT_TRUE(value); | 112 EXPECT_TRUE(value); |
| 117 EXPECT_TRUE(value->GetAsBoolean(&disable_drive)); | 113 EXPECT_TRUE(value->GetAsBoolean(&disable_drive)); |
| 118 EXPECT_FALSE(disable_drive); | 114 EXPECT_FALSE(disable_drive); |
| 119 | 115 |
| 120 std::string download_directory; | 116 std::string download_directory; |
| 121 EXPECT_TRUE(store_->GetValue(prefs::kDownloadDefaultDirectory, &value)); | 117 EXPECT_TRUE(store_->GetValue(prefs::kDownloadDefaultDirectory, &value)); |
| 122 EXPECT_TRUE(value); | 118 EXPECT_TRUE(value); |
| 123 EXPECT_TRUE(value->GetAsString(&download_directory)); | 119 EXPECT_TRUE(value->GetAsString(&download_directory)); |
| 124 EXPECT_EQ(GetExpectedDownloadDirectory(), download_directory); | 120 EXPECT_EQ(GetExpectedDownloadDirectory(), download_directory); |
| 125 | 121 |
| 126 policy.Set(policy::key::kDownloadDirectory, | 122 policy.Set(policy::key::kDownloadDirectory, policy::POLICY_LEVEL_MANDATORY, |
| 127 policy::POLICY_LEVEL_MANDATORY, | 123 policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD, |
| 128 policy::POLICY_SCOPE_USER, | 124 base::WrapUnique(new base::StringValue(kUserIDHash)), nullptr); |
| 129 policy::POLICY_SOURCE_CLOUD, | |
| 130 new base::StringValue(kUserIDHash), | |
| 131 NULL); | |
| 132 UpdateProviderPolicy(policy); | 125 UpdateProviderPolicy(policy); |
| 133 EXPECT_FALSE(recommended_store_->GetValue(drive::prefs::kDisableDrive, NULL)); | 126 EXPECT_FALSE(recommended_store_->GetValue(drive::prefs::kDisableDrive, NULL)); |
| 134 | 127 |
| 135 policy.Set(policy::key::kDownloadDirectory, | 128 policy.Set( |
| 136 policy::POLICY_LEVEL_RECOMMENDED, | 129 policy::key::kDownloadDirectory, policy::POLICY_LEVEL_RECOMMENDED, |
| 137 policy::POLICY_SCOPE_USER, | 130 policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD, |
| 138 policy::POLICY_SOURCE_CLOUD, | 131 base::WrapUnique(new base::StringValue( |
| 139 new base::StringValue(std::string(kDriveNamePolicyVariableName) + | 132 std::string(kDriveNamePolicyVariableName) + kRelativeToDriveRoot)), |
| 140 kRelativeToDriveRoot), | 133 nullptr); |
| 141 NULL); | |
| 142 UpdateProviderPolicy(policy); | 134 UpdateProviderPolicy(policy); |
| 143 | 135 |
| 144 EXPECT_FALSE(recommended_store_->GetValue(prefs::kPromptForDownload, NULL)); | 136 EXPECT_FALSE(recommended_store_->GetValue(prefs::kPromptForDownload, NULL)); |
| 145 EXPECT_FALSE(recommended_store_->GetValue(drive::prefs::kDisableDrive, NULL)); | 137 EXPECT_FALSE(recommended_store_->GetValue(drive::prefs::kDisableDrive, NULL)); |
| 146 | 138 |
| 147 EXPECT_TRUE( | 139 EXPECT_TRUE( |
| 148 recommended_store_->GetValue(prefs::kDownloadDefaultDirectory, &value)); | 140 recommended_store_->GetValue(prefs::kDownloadDefaultDirectory, &value)); |
| 149 EXPECT_TRUE(value); | 141 EXPECT_TRUE(value); |
| 150 EXPECT_TRUE(value->GetAsString(&download_directory)); | 142 EXPECT_TRUE(value->GetAsString(&download_directory)); |
| 151 EXPECT_EQ(GetExpectedDownloadDirectory() + kRelativeToDriveRoot, | 143 EXPECT_EQ(GetExpectedDownloadDirectory() + kRelativeToDriveRoot, |
| 152 download_directory); | 144 download_directory); |
| 153 | 145 |
| 154 policy.Set(policy::key::kDownloadDirectory, | 146 policy.Set(policy::key::kDownloadDirectory, policy::POLICY_LEVEL_RECOMMENDED, |
| 155 policy::POLICY_LEVEL_RECOMMENDED, | 147 policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD, |
| 156 policy::POLICY_SCOPE_USER, | 148 base::WrapUnique(new base::StringValue(kUserIDHash)), nullptr); |
| 157 policy::POLICY_SOURCE_CLOUD, | |
| 158 new base::StringValue(kUserIDHash), | |
| 159 NULL); | |
| 160 UpdateProviderPolicy(policy); | 149 UpdateProviderPolicy(policy); |
| 161 | 150 |
| 162 EXPECT_FALSE(recommended_store_->GetValue(prefs::kPromptForDownload, NULL)); | 151 EXPECT_FALSE(recommended_store_->GetValue(prefs::kPromptForDownload, NULL)); |
| 163 EXPECT_FALSE(recommended_store_->GetValue(drive::prefs::kDisableDrive, NULL)); | 152 EXPECT_FALSE(recommended_store_->GetValue(drive::prefs::kDisableDrive, NULL)); |
| 164 | 153 |
| 165 EXPECT_TRUE( | 154 EXPECT_TRUE( |
| 166 recommended_store_->GetValue(prefs::kDownloadDefaultDirectory, &value)); | 155 recommended_store_->GetValue(prefs::kDownloadDefaultDirectory, &value)); |
| 167 EXPECT_TRUE(value); | 156 EXPECT_TRUE(value); |
| 168 EXPECT_TRUE(value->GetAsString(&download_directory)); | 157 EXPECT_TRUE(value->GetAsString(&download_directory)); |
| 169 EXPECT_EQ(kUserIDHash, download_directory); | 158 EXPECT_EQ(kUserIDHash, download_directory); |
| 170 } | 159 } |
| 171 #endif | 160 #endif |
| OLD | NEW |