Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 <string> | 5 #include <string> |
| 6 | 6 |
| 7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
| 8 #include "base/string_number_conversions.h" | 8 #include "base/string_number_conversions.h" |
| 9 #include "base/stringprintf.h" | 9 #include "base/stringprintf.h" |
| 10 #include "base/test/test_reg_util_win.h" | 10 #include "base/test/test_reg_util_win.h" |
| (...skipping 107 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 118 } | 118 } |
| 119 | 119 |
| 120 private: | 120 private: |
| 121 registry_util::RegistryOverrideManager override_manager_; | 121 registry_util::RegistryOverrideManager override_manager_; |
| 122 }; | 122 }; |
| 123 | 123 |
| 124 TEST_F(GCAPIReactivationTest, CheckSetReactivationBrandCode) { | 124 TEST_F(GCAPIReactivationTest, CheckSetReactivationBrandCode) { |
| 125 EXPECT_TRUE(SetReactivationBrandCode(L"GAGA", GCAPI_INVOKED_STANDARD_SHELL)); | 125 EXPECT_TRUE(SetReactivationBrandCode(L"GAGA", GCAPI_INVOKED_STANDARD_SHELL)); |
| 126 EXPECT_EQ(L"GAGA", GetReactivationString(HKEY_CURRENT_USER)); | 126 EXPECT_EQ(L"GAGA", GetReactivationString(HKEY_CURRENT_USER)); |
| 127 | 127 |
| 128 std::vector<std::wstring> check_codes; | 128 std::vector<std::wstring> check_codes; |
|
robertshield
2012/03/23 17:52:16
you can delete check_codes and just have a single
mgraboski
2012/03/23 21:40:35
Done.
| |
| 129 check_codes.push_back(L"GAGA"); | 129 check_codes.push_back(L"GAGA"); |
| 130 EXPECT_TRUE(HasBeenReactivatedByBrandCodes(check_codes)); | 130 EXPECT_TRUE(HasBeenReactivated()); |
| 131 | 131 |
| 132 check_codes.push_back(L"GOOGOO"); | 132 check_codes.push_back(L"GOOGOO"); |
| 133 EXPECT_TRUE(HasBeenReactivatedByBrandCodes(check_codes)); | 133 EXPECT_TRUE(HasBeenReactivated()); |
| 134 | 134 |
| 135 check_codes.erase(check_codes.begin()); | 135 check_codes.erase(check_codes.begin()); |
| 136 EXPECT_FALSE(HasBeenReactivatedByBrandCodes(check_codes)); | 136 EXPECT_TRUE(HasBeenReactivated()); |
| 137 } | 137 } |
| 138 | 138 |
| 139 TEST_F(GCAPIReactivationTest, CanOfferReactivation_Basic) { | 139 TEST_F(GCAPIReactivationTest, CanOfferReactivation_Basic) { |
| 140 const wchar_t* previous_brands[] = {L"GOOGOO", L"MAMA", L"DADA"}; | 140 const wchar_t* previous_brands[] = {L"GOOGOO", L"MAMA", L"DADA"}; |
| 141 DWORD error; | 141 DWORD error; |
| 142 | 142 |
| 143 // We're not installed yet. Make sure CanOfferReactivation fails. | 143 // We're not installed yet. Make sure CanOfferReactivation fails. |
| 144 EXPECT_FALSE(CanOfferReactivation(L"GAGA", arraysize(previous_brands), | 144 EXPECT_FALSE(CanOfferReactivation(L"GAGA", arraysize(previous_brands), |
| 145 previous_brands, | 145 previous_brands, |
| 146 GCAPI_INVOKED_STANDARD_SHELL, | 146 GCAPI_INVOKED_STANDARD_SHELL, |
| (...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 216 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, | 216 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, |
| 217 &error)); | 217 &error)); |
| 218 EXPECT_EQ(L"GAGA", GetReactivationString(HKEY_CURRENT_USER)); | 218 EXPECT_EQ(L"GAGA", GetReactivationString(HKEY_CURRENT_USER)); |
| 219 | 219 |
| 220 // Make sure we can't reactivate again: | 220 // Make sure we can't reactivate again: |
| 221 EXPECT_FALSE(ReactivateChrome(L"GAGA", arraysize(previous_brands), | 221 EXPECT_FALSE(ReactivateChrome(L"GAGA", arraysize(previous_brands), |
| 222 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, | 222 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, |
| 223 &error)); | 223 &error)); |
| 224 EXPECT_EQ(REACTIVATE_ERROR_ALREADY_REACTIVATED, error); | 224 EXPECT_EQ(REACTIVATE_ERROR_ALREADY_REACTIVATED, error); |
| 225 | 225 |
| 226 // Should still be able to reactivate under other brands: | 226 // Should not be able to reactivate under other brands: |
| 227 EXPECT_TRUE(ReactivateChrome(L"MAMA", arraysize(previous_brands), | 227 EXPECT_FALSE(ReactivateChrome(L"MAMA", arraysize(previous_brands), |
| 228 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, | 228 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, |
| 229 &error)); | 229 &error)); |
| 230 EXPECT_EQ(L"MAMA", GetReactivationString(HKEY_CURRENT_USER)); | 230 EXPECT_EQ(L"GAGA", GetReactivationString(HKEY_CURRENT_USER)); |
| 231 | 231 |
| 232 // Validate that previous_brands are rejected: | 232 // Validate that previous_brands are rejected: |
| 233 EXPECT_FALSE(ReactivateChrome(L"PFFT", arraysize(previous_brands), | 233 EXPECT_FALSE(ReactivateChrome(L"PFFT", arraysize(previous_brands), |
| 234 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, | 234 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, |
| 235 &error)); | 235 &error)); |
| 236 EXPECT_EQ(REACTIVATE_ERROR_ALREADY_REACTIVATED, error); | 236 EXPECT_EQ(REACTIVATE_ERROR_ALREADY_REACTIVATED, error); |
| 237 EXPECT_EQ(L"MAMA", GetReactivationString(HKEY_CURRENT_USER)); | 237 EXPECT_EQ(L"GAGA", GetReactivationString(HKEY_CURRENT_USER)); |
| 238 } | 238 } |
| 239 | 239 |
| 240 TEST_F(GCAPIReactivationTest, ExperimentLabelCheck) { | 240 TEST_F(GCAPIReactivationTest, ExperimentLabelCheck) { |
| 241 const wchar_t* previous_brands[] = {L"GOOGOO", L"MAMA", L"DADA"}; | 241 const wchar_t* previous_brands[] = {L"GOOGOO", L"MAMA", L"DADA"}; |
| 242 DWORD error; | 242 DWORD error; |
| 243 | 243 |
| 244 // Set us up as a candidate for reactivation. | 244 // Set us up as a candidate for reactivation. |
| 245 EXPECT_TRUE(SetChromeInstallMarker(HKEY_CURRENT_USER)); | 245 EXPECT_TRUE(SetChromeInstallMarker(HKEY_CURRENT_USER)); |
| 246 | 246 |
| 247 Time hkcu_last_run = Time::NowFromSystemTime() - | 247 Time hkcu_last_run = Time::NowFromSystemTime() - |
| 248 TimeDelta::FromDays(kReactivationMinDaysDormant); | 248 TimeDelta::FromDays(kReactivationMinDaysDormant); |
| 249 EXPECT_TRUE(SetLastRunTime(HKEY_CURRENT_USER, | 249 EXPECT_TRUE(SetLastRunTime(HKEY_CURRENT_USER, |
| 250 hkcu_last_run.ToInternalValue())); | 250 hkcu_last_run.ToInternalValue())); |
| 251 | 251 |
| 252 EXPECT_TRUE(ReactivateChrome(L"GAGA", arraysize(previous_brands), | 252 EXPECT_TRUE(ReactivateChrome(L"GAGA", arraysize(previous_brands), |
| 253 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, | 253 previous_brands, GCAPI_INVOKED_STANDARD_SHELL, |
| 254 &error)); | 254 &error)); |
| 255 EXPECT_EQ(L"GAGA", GetReactivationString(HKEY_CURRENT_USER)); | 255 EXPECT_EQ(L"GAGA", GetReactivationString(HKEY_CURRENT_USER)); |
| 256 | 256 |
| 257 EXPECT_TRUE(HasExperimentLabels(HKEY_CURRENT_USER)); | 257 EXPECT_TRUE(HasExperimentLabels(HKEY_CURRENT_USER)); |
| 258 } | 258 } |
| OLD | NEW |