OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 "base/file_path.h" | 5 #include "base/file_path.h" |
6 #include "base/file_version_info.h" | 6 #include "base/file_version_info.h" |
7 #include "base/file_version_info_win.h" | 7 #include "base/file_version_info_win.h" |
8 #include "base/string_util.h" | 8 #include "base/string_util.h" |
9 #include "base/utf_string_conversions.h" | 9 #include "base/utf_string_conversions.h" |
10 #include "base/win/registry.h" | 10 #include "base/win/registry.h" |
11 #include "chrome_frame/navigation_constraints.h" | 11 #include "chrome_frame/navigation_constraints.h" |
12 #include "chrome_frame/test/chrome_frame_test_utils.h" | 12 #include "chrome_frame/test/chrome_frame_test_utils.h" |
13 #include "chrome_frame/utils.h" | 13 #include "chrome_frame/utils.h" |
14 | 14 |
15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
16 #include "testing/gmock/include/gmock/gmock.h" | 16 #include "testing/gmock/include/gmock/gmock.h" |
17 | 17 |
18 using base::win::RegKey; | 18 using base::win::RegKey; |
19 using chrome_frame_test::TempRegKeyOverride; | 19 using registry_util::ScopedRegistryKeyOverride; |
20 | 20 |
21 const wchar_t kChannelName[] = L"-dev"; | 21 const wchar_t kChannelName[] = L"-dev"; |
22 const wchar_t kSuffix[] = L"-fix"; | 22 const wchar_t kSuffix[] = L"-fix"; |
23 | 23 |
24 // Registry override in the UtilsTest will cause shell APIs to fail | 24 // Registry override in the UtilsTest will cause shell APIs to fail |
25 // So separate this test from the rest | 25 // So separate this test from the rest |
26 TEST(SimpleUtilTests, GetTempInternetFiles) { | 26 TEST(SimpleUtilTests, GetTempInternetFiles) { |
27 FilePath path = GetIETemporaryFilesFolder(); | 27 FilePath path = GetIETemporaryFilesFolder(); |
28 EXPECT_FALSE(path.empty()); | 28 EXPECT_FALSE(path.empty()); |
29 } | 29 } |
30 | 30 |
31 class UtilTests : public testing::Test { | 31 class UtilTests : public testing::Test { |
32 protected: | 32 protected: |
33 void SetUp() { | 33 void SetUp() { |
34 TempRegKeyOverride::DeleteAllTempKeys(); | 34 ScopedRegistryKeyOverride::DeleteAllTempKeys(); |
35 DeleteAllSingletons(); | 35 DeleteAllSingletons(); |
36 | 36 |
37 hklm_pol_.reset(new TempRegKeyOverride(HKEY_LOCAL_MACHINE, L"hklm_fake")); | 37 hklm_pol_.reset(new ScopedRegistryKeyOverride(HKEY_LOCAL_MACHINE, |
38 hkcu_pol_.reset(new TempRegKeyOverride(HKEY_CURRENT_USER, L"hkcu_fake")); | 38 L"hklm_fake")); |
| 39 hkcu_pol_.reset(new ScopedRegistryKeyOverride(HKEY_CURRENT_USER, |
| 40 L"hkcu_fake")); |
39 } | 41 } |
40 | 42 |
41 void TearDown() { | 43 void TearDown() { |
42 hkcu_pol_.reset(NULL); | 44 hkcu_pol_.reset(NULL); |
43 hklm_pol_.reset(NULL); | 45 hklm_pol_.reset(NULL); |
44 TempRegKeyOverride::DeleteAllTempKeys(); | 46 ScopedRegistryKeyOverride::DeleteAllTempKeys(); |
45 } | 47 } |
46 | 48 |
47 // This is used to manage life cycle of PolicySettings singleton. | 49 // This is used to manage life cycle of PolicySettings singleton. |
48 // base::ShadowingAtExitManager at_exit_manager_; | 50 // base::ShadowingAtExitManager at_exit_manager_; |
49 scoped_ptr<TempRegKeyOverride> hklm_pol_; | 51 scoped_ptr<ScopedRegistryKeyOverride> hklm_pol_; |
50 scoped_ptr<TempRegKeyOverride> hkcu_pol_; | 52 scoped_ptr<ScopedRegistryKeyOverride> hkcu_pol_; |
51 }; | 53 }; |
52 | 54 |
53 TEST_F(UtilTests, GetModuleVersionTest) { | 55 TEST_F(UtilTests, GetModuleVersionTest) { |
54 HMODULE mod = GetModuleHandle(L"kernel32.dll"); | 56 HMODULE mod = GetModuleHandle(L"kernel32.dll"); |
55 EXPECT_NE(mod, static_cast<HMODULE>(NULL)); | 57 EXPECT_NE(mod, static_cast<HMODULE>(NULL)); |
56 wchar_t path[MAX_PATH] = {0}; | 58 wchar_t path[MAX_PATH] = {0}; |
57 GetModuleFileName(mod, path, arraysize(path)); | 59 GetModuleFileName(mod, path, arraysize(path)); |
58 | 60 |
59 // Use the method that goes to disk | 61 // Use the method that goes to disk |
60 scoped_ptr<FileVersionInfo> base_info( | 62 scoped_ptr<FileVersionInfo> base_info( |
(...skipping 482 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
543 | 545 |
544 ASSERT_EQ(expect_match, | 546 ASSERT_EQ(expect_match, |
545 CheckXUaCompatibleDirective(test.header_value, | 547 CheckXUaCompatibleDirective(test.header_value, |
546 all_versions[version_index])) | 548 all_versions[version_index])) |
547 << "Expect '" << test.header_value << "' to " | 549 << "Expect '" << test.header_value << "' to " |
548 << (expect_match ? "match" : "not match") << " IE major version " | 550 << (expect_match ? "match" : "not match") << " IE major version " |
549 << all_versions[version_index]; | 551 << all_versions[version_index]; |
550 } | 552 } |
551 } | 553 } |
552 } | 554 } |
OLD | NEW |