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

Side by Side Diff: chrome/browser/rlz/rlz_unittest.cc

Issue 6090006: Regkey functions return error code instead of bool (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src/
Patch Set: '' Created 9 years, 11 months 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 unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2006-2008 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/rlz/rlz.h" 5 #include "chrome/browser/rlz/rlz.h"
6 6
7 #include "base/path_service.h" 7 #include "base/path_service.h"
8 #include "base/win/registry.h" 8 #include "base/win/registry.h"
9 #include "testing/gtest/include/gtest/gtest.h" 9 #include "testing/gtest/include/gtest/gtest.h"
10 10
11 using base::win::RegKey; 11 using base::win::RegKey;
12 12
13 namespace { 13 namespace {
14 14
15 // Gets rid of registry leftovers from testing. Returns false if there 15 // Gets rid of registry leftovers from testing. Returns false if there
16 // is nothing to clean. 16 // is nothing to clean.
17 bool CleanValue(const wchar_t* key_name, const wchar_t* value) { 17 bool CleanValue(const wchar_t* key_name, const wchar_t* value) {
18 RegKey key; 18 RegKey key;
19 if (!key.Open(HKEY_CURRENT_USER, key_name, KEY_READ | KEY_WRITE)) 19 if (key.Open(HKEY_CURRENT_USER, key_name, KEY_READ | KEY_WRITE) !=
20 ERROR_SUCCESS)
20 return false; 21 return false;
21 EXPECT_TRUE(key.DeleteValue(value)); 22 EXPECT_EQ(ERROR_SUCCESS, key.DeleteValue(value));
22 return true; 23 return true;
23 } 24 }
24 25
25 // The chrome events RLZ key lives here. 26 // The chrome events RLZ key lives here.
26 const wchar_t kKeyName[] = L"Software\\Google\\Common\\Rlz\\Events\\C"; 27 const wchar_t kKeyName[] = L"Software\\Google\\Common\\Rlz\\Events\\C";
27 28
28 } // namespace 29 } // namespace
29 30
30 TEST(RlzLibTest, RecordProductEvent) { 31 TEST(RlzLibTest, RecordProductEvent) {
31 DWORD recorded_value = 0; 32 DWORD recorded_value = 0;
32 EXPECT_TRUE(RLZTracker::RecordProductEvent(rlz_lib::CHROME, 33 EXPECT_TRUE(RLZTracker::RecordProductEvent(rlz_lib::CHROME,
33 rlz_lib::CHROME_OMNIBOX, rlz_lib::FIRST_SEARCH)); 34 rlz_lib::CHROME_OMNIBOX, rlz_lib::FIRST_SEARCH));
34 const wchar_t kEvent1[] = L"C1F"; 35 const wchar_t kEvent1[] = L"C1F";
35 RegKey key1; 36 RegKey key1;
36 EXPECT_TRUE(key1.Open(HKEY_CURRENT_USER, kKeyName, KEY_READ)); 37 EXPECT_EQ(ERROR_SUCCESS, key1.Open(HKEY_CURRENT_USER, kKeyName, KEY_READ));
37 EXPECT_TRUE(key1.ReadValueDW(kEvent1, &recorded_value)); 38 EXPECT_EQ(ERROR_SUCCESS, key1.ReadValueDW(kEvent1, &recorded_value));
38 EXPECT_EQ(1, recorded_value); 39 EXPECT_EQ(1, recorded_value);
39 EXPECT_TRUE(CleanValue(kKeyName, kEvent1)); 40 EXPECT_TRUE(CleanValue(kKeyName, kEvent1));
40 41
41 EXPECT_TRUE(RLZTracker::RecordProductEvent(rlz_lib::CHROME, 42 EXPECT_TRUE(RLZTracker::RecordProductEvent(rlz_lib::CHROME,
42 rlz_lib::CHROME_HOME_PAGE, rlz_lib::SET_TO_GOOGLE)); 43 rlz_lib::CHROME_HOME_PAGE, rlz_lib::SET_TO_GOOGLE));
43 const wchar_t kEvent2[] = L"C2S"; 44 const wchar_t kEvent2[] = L"C2S";
44 RegKey key2; 45 RegKey key2;
45 EXPECT_TRUE(key2.Open(HKEY_CURRENT_USER, kKeyName, KEY_READ)); 46 EXPECT_EQ(ERROR_SUCCESS, key2.Open(HKEY_CURRENT_USER, kKeyName, KEY_READ));
46 DWORD value = 0; 47 DWORD value = 0;
47 EXPECT_TRUE(key2.ReadValueDW(kEvent2, &recorded_value)); 48 EXPECT_EQ(ERROR_SUCCESS, key2.ReadValueDW(kEvent2, &recorded_value));
48 EXPECT_EQ(1, recorded_value); 49 EXPECT_EQ(1, recorded_value);
49 EXPECT_TRUE(CleanValue(kKeyName, kEvent2)); 50 EXPECT_TRUE(CleanValue(kKeyName, kEvent2));
50 } 51 }
51 52
52 TEST(RlzLibTest, CleanProductEvents) { 53 TEST(RlzLibTest, CleanProductEvents) {
53 DWORD recorded_value = 0; 54 DWORD recorded_value = 0;
54 EXPECT_TRUE(RLZTracker::RecordProductEvent(rlz_lib::CHROME, 55 EXPECT_TRUE(RLZTracker::RecordProductEvent(rlz_lib::CHROME,
55 rlz_lib::CHROME_OMNIBOX, rlz_lib::FIRST_SEARCH)); 56 rlz_lib::CHROME_OMNIBOX, rlz_lib::FIRST_SEARCH));
56 const wchar_t kEvent1[] = L"C1F"; 57 const wchar_t kEvent1[] = L"C1F";
57 RegKey key1; 58 RegKey key1;
58 EXPECT_TRUE(key1.Open(HKEY_CURRENT_USER, kKeyName, KEY_READ)); 59 EXPECT_EQ(ERROR_SUCCESS, key1.Open(HKEY_CURRENT_USER, kKeyName, KEY_READ));
59 EXPECT_TRUE(key1.ReadValueDW(kEvent1, &recorded_value)); 60 EXPECT_EQ(ERROR_SUCCESS, key1.ReadValueDW(kEvent1, &recorded_value));
60 EXPECT_EQ(1, recorded_value); 61 EXPECT_EQ(1, recorded_value);
61 62
62 EXPECT_TRUE(RLZTracker::ClearAllProductEvents(rlz_lib::CHROME)); 63 EXPECT_TRUE(RLZTracker::ClearAllProductEvents(rlz_lib::CHROME));
63 EXPECT_FALSE(CleanValue(kKeyName, kEvent1)); 64 EXPECT_FALSE(CleanValue(kKeyName, kEvent1));
64 } 65 }
OLDNEW
« no previous file with comments | « chrome/browser/policy/configuration_policy_provider_delegate_win.cc ('k') | chrome/browser/shell_integration_win.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698