| OLD | NEW |
| 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 LONG result = key.Open(HKEY_CURRENT_USER, key_name, KEY_READ | KEY_WRITE); |
| 20 if (result != 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 } |
| OLD | NEW |