OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/installer/gcapi/gcapi_omaha_experiment.h" | 5 #include "chrome/installer/gcapi/gcapi_omaha_experiment.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include "base/strings/utf_string_conversions.h" | 9 #include "base/strings/utf_string_conversions.h" |
| 10 #include "base/test/test_reg_util_win.h" |
10 #include "chrome/installer/gcapi/gcapi.h" | 11 #include "chrome/installer/gcapi/gcapi.h" |
11 #include "chrome/installer/gcapi/gcapi_test_registry_overrider.h" | |
12 #include "chrome/installer/util/google_update_constants.h" | 12 #include "chrome/installer/util/google_update_constants.h" |
13 #include "chrome/installer/util/google_update_settings.h" | 13 #include "chrome/installer/util/google_update_settings.h" |
14 #include "components/variations/variations_experiment_util.h" | 14 #include "components/variations/variations_experiment_util.h" |
15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
16 | 16 |
17 using base::ASCIIToUTF16; | 17 using base::ASCIIToUTF16; |
18 | 18 |
19 namespace { | 19 namespace { |
20 | 20 |
21 const wchar_t kBrand[] = L"ABCD"; | 21 const wchar_t kBrand[] = L"ABCD"; |
22 const uint16_t kUserLevel = GCAPI_INVOKED_STANDARD_SHELL; | 22 const uint16_t kUserLevel = GCAPI_INVOKED_STANDARD_SHELL; |
23 | 23 |
24 const wchar_t kSomeExperiments[] = L"myexp=1|Aug 2;yourexp=2|Sep 5"; | 24 const wchar_t kSomeExperiments[] = L"myexp=1|Aug 2;yourexp=2|Sep 5"; |
25 const wchar_t kSomeOtherExperiments[] = L"anotherexp=joe|Jun 7 2008"; | 25 const wchar_t kSomeOtherExperiments[] = L"anotherexp=joe|Jun 7 2008"; |
26 const wchar_t kSomeMoreExperiments[] = L"moreexp=foo|Jul 31 1999"; | 26 const wchar_t kSomeMoreExperiments[] = L"moreexp=foo|Jul 31 1999"; |
27 | 27 |
28 class GCAPIOmahaExperimentTest : public ::testing::Test { | 28 class GCAPIOmahaExperimentTest : public ::testing::Test { |
29 protected: | 29 protected: |
30 GCAPIOmahaExperimentTest() | 30 GCAPIOmahaExperimentTest() |
31 : brand_(kBrand), | 31 : brand_(kBrand), |
32 reactivation_label_(gcapi_internals::GetGCAPIExperimentLabel( | 32 reactivation_label_(gcapi_internals::GetGCAPIExperimentLabel( |
33 kBrand, gcapi_internals::kReactivationLabel)), | 33 kBrand, gcapi_internals::kReactivationLabel)), |
34 relaunch_label_(gcapi_internals::GetGCAPIExperimentLabel( | 34 relaunch_label_(gcapi_internals::GetGCAPIExperimentLabel( |
35 kBrand, gcapi_internals::kRelaunchLabel)) { | 35 kBrand, gcapi_internals::kRelaunchLabel)) { |
36 } | 36 } |
37 | 37 |
| 38 void SetUp() override { |
| 39 ASSERT_NO_FATAL_FAILURE( |
| 40 override_manager_.OverrideRegistry(HKEY_CURRENT_USER)); |
| 41 ASSERT_NO_FATAL_FAILURE( |
| 42 override_manager_.OverrideRegistry(HKEY_LOCAL_MACHINE)); |
| 43 } |
| 44 |
38 void VerifyExperimentLabels(const base::string16& expected_labels) { | 45 void VerifyExperimentLabels(const base::string16& expected_labels) { |
39 base::string16 actual_labels; | 46 base::string16 actual_labels; |
40 EXPECT_TRUE(GoogleUpdateSettings::ReadExperimentLabels(false, | 47 EXPECT_TRUE(GoogleUpdateSettings::ReadExperimentLabels(false, |
41 &actual_labels)); | 48 &actual_labels)); |
42 EXPECT_EQ(expected_labels, actual_labels); | 49 EXPECT_EQ(expected_labels, actual_labels); |
43 } | 50 } |
44 | 51 |
| 52 registry_util::RegistryOverrideManager override_manager_; |
45 base::string16 brand_; | 53 base::string16 brand_; |
46 base::string16 reactivation_label_; | 54 base::string16 reactivation_label_; |
47 base::string16 relaunch_label_; | 55 base::string16 relaunch_label_; |
48 | |
49 const GCAPITestRegistryOverrider gcapi_test_registry_overrider_; | |
50 }; | 56 }; |
51 | 57 |
52 TEST_F(GCAPIOmahaExperimentTest, SetReactivationLabelFromEmptyExperiments) { | 58 TEST_F(GCAPIOmahaExperimentTest, SetReactivationLabelFromEmptyExperiments) { |
53 ASSERT_TRUE(SetReactivationExperimentLabels(kBrand, kUserLevel)); | 59 ASSERT_TRUE(SetReactivationExperimentLabels(kBrand, kUserLevel)); |
54 VerifyExperimentLabels(reactivation_label_); | 60 VerifyExperimentLabels(reactivation_label_); |
55 } | 61 } |
56 | 62 |
57 // Test the relaunch label once; all other tests go more in depth, but since | 63 // Test the relaunch label once; all other tests go more in depth, but since |
58 // both labels use the same logic underneath there is no need to test both in | 64 // both labels use the same logic underneath there is no need to test both in |
59 // depth. | 65 // depth. |
(...skipping 117 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
177 base::string16 expected_labels(kSomeExperiments); | 183 base::string16 expected_labels(kSomeExperiments); |
178 expected_labels += variations::kExperimentLabelSeparator; | 184 expected_labels += variations::kExperimentLabelSeparator; |
179 expected_labels.append(gcapi_internals::kReactivationLabel); | 185 expected_labels.append(gcapi_internals::kReactivationLabel); |
180 expected_labels.append(kSomeOtherExperiments); | 186 expected_labels.append(kSomeOtherExperiments); |
181 expected_labels += variations::kExperimentLabelSeparator; | 187 expected_labels += variations::kExperimentLabelSeparator; |
182 expected_labels.append(reactivation_label_); | 188 expected_labels.append(reactivation_label_); |
183 VerifyExperimentLabels(expected_labels); | 189 VerifyExperimentLabels(expected_labels); |
184 } | 190 } |
185 | 191 |
186 } // namespace | 192 } // namespace |
OLD | NEW |