OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/profile_resetter/triggered_profile_resetter.h" | |
6 | |
7 #include "base/memory/scoped_ptr.h" | |
8 #include "base/test/test_reg_util_win.h" | |
9 #include "base/win/registry.h" | |
10 #include "chrome/browser/prefs/browser_prefs.h" | |
engedy
2015/09/21 12:56:10
nit: If the default TestingProfile ctor can be use
robertshield
2015/09/21 22:04:30
Done.
| |
11 #include "chrome/browser/profile_resetter/triggered_profile_resetter_factory.h" | |
12 #include "chrome/common/pref_names.h" | |
13 #include "chrome/test/base/testing_pref_service_syncable.h" | |
engedy
2015/09/21 12:56:10
nit: If the default TestingProfile ctor can be use
robertshield
2015/09/21 22:04:30
Done.
| |
14 #include "chrome/test/base/testing_profile.h" | |
15 #include "content/public/test/test_browser_thread_bundle.h" | |
16 #include "testing/gtest/include/gtest/gtest.h" | |
17 | |
18 using base::win::RegKey; | |
19 | |
20 class TriggeredProfileResetterTest : public testing::Test { | |
21 protected: | |
22 void SetUp() { | |
23 override_manager_.OverrideRegistry(HKEY_CURRENT_USER); | |
24 | |
25 // Create the profile. | |
26 TestingProfile::Builder builder; | |
engedy
2015/09/21 12:56:10
nit: I think the default TestingProfile constructo
robertshield
2015/09/21 22:04:30
Done.
| |
27 TestingPrefServiceSyncable* pref_service = new TestingPrefServiceSyncable(); | |
28 chrome::RegisterUserProfilePrefs(pref_service->registry()); | |
29 builder.SetPrefService(make_scoped_ptr<PrefServiceSyncable>(pref_service)); | |
30 profile_ = builder.Build(); | |
31 } | |
32 | |
33 content::TestBrowserThreadBundle thread_bundle_; | |
34 scoped_ptr<TestingProfile> profile_; | |
35 | |
36 private: | |
37 registry_util::RegistryOverrideManager override_manager_; | |
38 }; | |
39 | |
40 TEST_F(TriggeredProfileResetterTest, HasResetTriggerAndClear) { | |
41 RegKey trigger_key(HKEY_CURRENT_USER, kTriggeredResetRegistryPath, | |
engedy
2015/09/21 12:56:10
optional nit: I'd personally introduce a helper me
robertshield
2015/09/21 22:04:30
Done.
| |
42 KEY_ALL_ACCESS); | |
43 ASSERT_TRUE(trigger_key.Valid()); | |
44 | |
45 FILETIME ft; | |
46 ::GetSystemTimeAsFileTime(&ft); | |
47 ASSERT_TRUE(trigger_key.WriteValue(kTriggeredResetTimestamp, &ft, sizeof(ft), | |
48 REG_QWORD) == ERROR_SUCCESS); | |
49 | |
50 TriggeredProfileResetter triggered_profile_resetter(profile_.get()); | |
51 triggered_profile_resetter.Activate(); | |
52 EXPECT_TRUE(triggered_profile_resetter.HasResetTrigger()); | |
53 triggered_profile_resetter.ClearResetTrigger(); | |
54 EXPECT_FALSE(triggered_profile_resetter.HasResetTrigger()); | |
55 } | |
56 | |
57 TEST_F(TriggeredProfileResetterTest, HasDuplicateResetTrigger) { | |
58 RegKey trigger_key(HKEY_CURRENT_USER, kTriggeredResetRegistryPath, | |
59 KEY_ALL_ACCESS); | |
60 ASSERT_TRUE(trigger_key.Valid()); | |
61 | |
62 FILETIME ft; | |
63 ::GetSystemTimeAsFileTime(&ft); | |
64 ASSERT_TRUE(trigger_key.WriteValue(kTriggeredResetTimestamp, &ft, sizeof(ft), | |
65 REG_QWORD) == ERROR_SUCCESS); | |
66 profile_->GetPrefs()->SetInt64(prefs::kLastProfileResetTimestamp, | |
67 bit_cast<int64, FILETIME>(ft)); | |
engedy
2015/09/21 12:56:10
Please use int64_t and include <stdint.h>. (int64
robertshield
2015/09/21 22:04:30
Done.
| |
68 | |
69 TriggeredProfileResetter triggered_profile_resetter(profile_.get()); | |
70 triggered_profile_resetter.Activate(); | |
71 EXPECT_FALSE(triggered_profile_resetter.HasResetTrigger()); | |
72 } | |
73 | |
74 TEST_F(TriggeredProfileResetterTest, HasToolName) { | |
75 const wchar_t kToolName[] = L"ToolyMcTool"; | |
76 RegKey trigger_key(HKEY_CURRENT_USER, kTriggeredResetRegistryPath, | |
77 KEY_ALL_ACCESS); | |
78 ASSERT_TRUE(trigger_key.Valid()); | |
79 FILETIME ft; | |
80 ::GetSystemTimeAsFileTime(&ft); | |
81 ASSERT_TRUE(trigger_key.WriteValue(kTriggeredResetTimestamp, &ft, sizeof(ft), | |
82 REG_QWORD) == ERROR_SUCCESS); | |
83 ASSERT_TRUE(trigger_key.WriteValue(kTriggeredResetToolName, kToolName) == | |
84 ERROR_SUCCESS); | |
85 | |
86 TriggeredProfileResetter triggered_profile_resetter(profile_.get()); | |
87 triggered_profile_resetter.Activate(); | |
88 EXPECT_TRUE(triggered_profile_resetter.HasResetTrigger()); | |
89 EXPECT_STREQ(kToolName, | |
90 triggered_profile_resetter.GetResetToolName().c_str()); | |
91 } | |
92 | |
93 TEST_F(TriggeredProfileResetterTest, HasLongToolName) { | |
94 const wchar_t kLongToolName[] = | |
95 L"ToolMcToolToolMcToolToolMcToolToolMcToolToolMcToolToolMcToolToolMcTool" | |
96 L"ToolMcToolToolMcToolToolMcToolThisIsTheToolThatNeverEndsYesItGoesOnAnd" | |
97 L"OnMyFriend"; | |
98 const wchar_t kExpectedToolName[] = | |
99 L"ToolMcToolToolMcToolToolMcToolToolMcToolToolMcToolToolMcToolToolMcTool" | |
100 L"ToolMcToolToolMcToolToolMcTool"; | |
101 RegKey trigger_key(HKEY_CURRENT_USER, kTriggeredResetRegistryPath, | |
102 KEY_ALL_ACCESS); | |
103 ASSERT_TRUE(trigger_key.Valid()); | |
104 FILETIME ft; | |
105 ::GetSystemTimeAsFileTime(&ft); | |
106 ASSERT_TRUE(trigger_key.WriteValue(kTriggeredResetTimestamp, &ft, sizeof(ft), | |
107 REG_QWORD) == ERROR_SUCCESS); | |
108 ASSERT_TRUE(trigger_key.WriteValue(kTriggeredResetToolName, kLongToolName) == | |
109 ERROR_SUCCESS); | |
110 | |
111 TriggeredProfileResetter triggered_profile_resetter(profile_.get()); | |
112 triggered_profile_resetter.Activate(); | |
113 EXPECT_TRUE(triggered_profile_resetter.HasResetTrigger()); | |
114 EXPECT_STREQ(kExpectedToolName, | |
115 triggered_profile_resetter.GetResetToolName().c_str()); | |
116 } | |
OLD | NEW |