Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(4190)

Unified Diff: chrome/install_static/user_data_dir_win_unittest.cc

Issue 2487783002: Make Crashpad use the user data dir, rather than always default location (Closed)
Patch Set: install_static for swarming Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « chrome/install_static/user_data_dir.cc ('k') | chrome_elf/chrome_elf.def » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
« no previous file with comments | « chrome/install_static/user_data_dir.cc ('k') | chrome_elf/chrome_elf.def » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698