Chromium Code Reviews| Index: chrome/install_static/user_data_dir_win_unittest.cc |
| diff --git a/chrome/install_static/user_data_dir_win_unittest.cc b/chrome/install_static/user_data_dir_win_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..30afd0287ca91ed234c9f71158ffbc6a1e68d56a |
| --- /dev/null |
| +++ b/chrome/install_static/user_data_dir_win_unittest.cc |
| @@ -0,0 +1,160 @@ |
| +// Copyright 2016 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include <algorithm> |
| + |
| +#include "base/test/test_reg_util_win.h" |
| +#include "chrome/install_static/install_details.h" |
| +#include "chrome/install_static/user_data_dir.h" |
| +#include "chrome_elf/nt_registry/nt_registry.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +namespace install_static { |
| +namespace { |
| + |
| +inline bool EndsWith(const std::wstring& value, const std::wstring& ending) { |
| + if (ending.size() > value.size()) |
| + return false; |
| + return std::equal(ending.rbegin(), ending.rend(), value.rbegin()); |
| +} |
| + |
| +#if defined(GOOGLE_CHROME_BUILD) |
| +const wchar_t kPolicyRegistryKey[] = L"SOFTWARE\\Policies\\Google\\Chrome"; |
| +const wchar_t kUserDataDirNameSuffix[] = L"\\Google\\Chrome\\User Data"; |
| +#else |
| +const wchar_t kPolicyRegistryKey[] = L"SOFTWARE\\Policies\\Chromium"; |
| +const wchar_t kUserDataDirNameSuffix[] = L"\\Chromium\\User Data"; |
| +#endif |
| + |
| +const wchar_t kUserDataDirRegistryKey[] = L"UserDataDir"; |
| + |
| +const InstallConstants kFakeInstallConstants = {sizeof(InstallConstants), 0, |
| + L""}; |
| + |
| +class ScopedNTRegistryTestingOverride { |
| + public: |
| + ScopedNTRegistryTestingOverride(nt::ROOT_KEY root, const std::wstring& path) |
| + : root_(root) { |
| + EXPECT_TRUE(nt::SetTestingOverride(root_, path)); |
| + } |
| + ~ScopedNTRegistryTestingOverride() { |
| + nt::SetTestingOverride(root_, base::string16()); |
| + } |
| + |
| + private: |
| + nt::ROOT_KEY root_; |
| +}; |
| + |
| +TEST(UserDataDir, EmptyResultsInDefault) { |
| + std::wstring result, invalid; |
| + |
| + install_static::GetUserDataDirectoryImpl(L"", kFakeInstallConstants, &result, |
| + &invalid); |
| + EXPECT_TRUE(EndsWith(result, kUserDataDirNameSuffix)); |
| + EXPECT_EQ(std::wstring(), invalid); |
| +} |
| + |
| +TEST(UserDataDir, InvalidResultsInDefault) { |
| + std::wstring result, invalid; |
| + |
| + install_static::GetUserDataDirectoryImpl(L"<>|:", kFakeInstallConstants, |
| + &result, &invalid); |
| + EXPECT_TRUE(EndsWith(result, kUserDataDirNameSuffix)); |
| + EXPECT_EQ(L"<>|:", invalid); |
| +} |
| + |
| +TEST(UserDataDir, RegistrySettingsInHKLMOverrides) { |
| + std::wstring result, invalid; |
| + |
| + // Override the registry to say one value in HKLM, and confirm it takes |
| + // precedence over the command line in both implementations. |
| + registry_util::RegistryOverrideManager override_manager; |
| + base::string16 temp; |
| + override_manager.OverrideRegistry(HKEY_LOCAL_MACHINE, &temp); |
| + ScopedNTRegistryTestingOverride nt_override(nt::HKLM, temp); |
| + |
| + base::win::RegKey key(HKEY_LOCAL_MACHINE, kPolicyRegistryKey, KEY_WRITE); |
| + LONG rv = key.WriteValue(kUserDataDirRegistryKey, L"yyy"); |
| + ASSERT_EQ(rv, ERROR_SUCCESS); |
| + |
| + install_static::GetUserDataDirectoryImpl(L"xxx", kFakeInstallConstants, |
| + &result, &invalid); |
| + |
| + EXPECT_TRUE(EndsWith(result, L"\\yyy")); |
| + EXPECT_EQ(std::wstring(), invalid); |
| +} |
| + |
| +TEST(UserDataDir, RegistrySettingsInHKCUOverrides) { |
| + std::wstring result, invalid; |
| + |
| + // Override the registry to say one value in HKCU, and confirm it takes |
| + // precedence over the command line in both implementations. |
| + registry_util::RegistryOverrideManager override_manager; |
| + base::string16 temp; |
| + override_manager.OverrideRegistry(HKEY_CURRENT_USER, &temp); |
| + ScopedNTRegistryTestingOverride nt_override(nt::HKCU, temp); |
| + |
| + base::win::RegKey key(HKEY_CURRENT_USER, kPolicyRegistryKey, KEY_WRITE); |
| + LONG rv = key.WriteValue(kUserDataDirRegistryKey, L"yyy"); |
| + ASSERT_EQ(rv, ERROR_SUCCESS); |
| + |
| + install_static::GetUserDataDirectoryImpl(L"xxx", kFakeInstallConstants, |
| + &result, &invalid); |
| + |
| + EXPECT_TRUE(EndsWith(result, L"\\yyy")); |
| + EXPECT_EQ(std::wstring(), invalid); |
| +} |
| + |
| +TEST(UserDataDir, RegistrySettingsInHKLMTakesPrecedenceOverHKCU) { |
| + std::wstring result, invalid; |
| + |
| + // Override the registry in both HKLM and HKCU, and confirm HKLM takes |
| + // precedence. |
| + registry_util::RegistryOverrideManager override_manager; |
| + base::string16 temp; |
| + override_manager.OverrideRegistry(HKEY_LOCAL_MACHINE, &temp); |
| + ScopedNTRegistryTestingOverride nt_override(nt::HKLM, temp); |
| + LONG rv; |
| + base::win::RegKey key1(HKEY_LOCAL_MACHINE, kPolicyRegistryKey, KEY_WRITE); |
| + rv = key1.WriteValue(kUserDataDirRegistryKey, L"111"); |
|
tapted
2017/02/05 23:59:46
Some tests have started failing with output like
|
| + ASSERT_EQ(rv, ERROR_SUCCESS); |
| + |
| + override_manager.OverrideRegistry(HKEY_CURRENT_USER, &temp); |
| + ScopedNTRegistryTestingOverride nt_override2(nt::HKCU, temp); |
| + base::win::RegKey key2(HKEY_CURRENT_USER, kPolicyRegistryKey, KEY_WRITE); |
| + rv = key2.WriteValue(kUserDataDirRegistryKey, L"222"); |
| + ASSERT_EQ(rv, ERROR_SUCCESS); |
| + |
| + install_static::GetUserDataDirectoryImpl(L"xxx", kFakeInstallConstants, |
| + &result, &invalid); |
| + |
| + EXPECT_TRUE(EndsWith(result, L"\\111")); |
| + EXPECT_EQ(std::wstring(), invalid); |
| +} |
| + |
| +TEST(UserDataDir, RegistrySettingWithPathExpansionHKCU) { |
| + std::wstring result, invalid; |
| + |
| + registry_util::RegistryOverrideManager override_manager; |
| + base::string16 temp; |
| + override_manager.OverrideRegistry(HKEY_CURRENT_USER, &temp); |
| + ScopedNTRegistryTestingOverride nt_override(nt::HKCU, temp); |
| + base::win::RegKey key(HKEY_CURRENT_USER, kPolicyRegistryKey, KEY_WRITE); |
| + LONG rv = key.WriteValue(kUserDataDirRegistryKey, L"${windows}"); |
| + ASSERT_EQ(rv, ERROR_SUCCESS); |
| + |
| + install_static::GetUserDataDirectoryImpl(L"xxx", kFakeInstallConstants, |
| + &result, &invalid); |
| + |
| + EXPECT_EQ(strlen("X:\\WINDOWS"), result.size()); |
| + EXPECT_EQ(std::wstring::npos, result.find(L"${windows}")); |
| + std::wstring upper; |
| + std::transform(result.begin(), result.end(), std::back_inserter(upper), |
| + toupper); |
| + EXPECT_TRUE(EndsWith(upper, L"\\WINDOWS")); |
| + EXPECT_EQ(std::wstring(), invalid); |
| +} |
| + |
| +} // namespace |
| +} // namespace install_static |