OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "components/variations/variations_http_header_provider.h" | 5 #include "components/variations/variations_http_header_provider.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/base64.h" | 9 #include "base/base64.h" |
10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
(...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
60 void TearDown() override { testing::ClearAllVariationIDs(); } | 60 void TearDown() override { testing::ClearAllVariationIDs(); } |
61 }; | 61 }; |
62 | 62 |
63 TEST_F(VariationsHttpHeaderProviderTest, SetDefaultVariationIds_Valid) { | 63 TEST_F(VariationsHttpHeaderProviderTest, SetDefaultVariationIds_Valid) { |
64 base::MessageLoop loop; | 64 base::MessageLoop loop; |
65 VariationsHttpHeaderProvider provider; | 65 VariationsHttpHeaderProvider provider; |
66 | 66 |
67 // Valid experiment ids. | 67 // Valid experiment ids. |
68 EXPECT_TRUE(provider.SetDefaultVariationIds({"12", "456", "t789"})); | 68 EXPECT_TRUE(provider.SetDefaultVariationIds({"12", "456", "t789"})); |
69 provider.InitVariationIDsCacheIfNeeded(); | 69 provider.InitVariationIDsCacheIfNeeded(); |
70 std::string variations = provider.GetClientDataHeader(false); | 70 std::string variations = provider.GetClientDataHeader(); |
71 EXPECT_FALSE(variations.empty()); | 71 EXPECT_FALSE(variations.empty()); |
72 std::set<VariationID> variation_ids; | 72 std::set<VariationID> variation_ids; |
73 std::set<VariationID> trigger_ids; | 73 std::set<VariationID> trigger_ids; |
74 ASSERT_TRUE(ExtractVariationIds(variations, &variation_ids, &trigger_ids)); | 74 ASSERT_TRUE(ExtractVariationIds(variations, &variation_ids, &trigger_ids)); |
75 EXPECT_TRUE(variation_ids.find(12) != variation_ids.end()); | 75 EXPECT_TRUE(variation_ids.find(12) != variation_ids.end()); |
76 EXPECT_TRUE(variation_ids.find(456) != variation_ids.end()); | 76 EXPECT_TRUE(variation_ids.find(456) != variation_ids.end()); |
77 EXPECT_TRUE(trigger_ids.find(789) != trigger_ids.end()); | 77 EXPECT_TRUE(trigger_ids.find(789) != trigger_ids.end()); |
78 EXPECT_FALSE(variation_ids.find(789) != variation_ids.end()); | 78 EXPECT_FALSE(variation_ids.find(789) != variation_ids.end()); |
79 } | 79 } |
80 | 80 |
81 TEST_F(VariationsHttpHeaderProviderTest, SetDefaultVariationIds_Invalid) { | 81 TEST_F(VariationsHttpHeaderProviderTest, SetDefaultVariationIds_Invalid) { |
82 base::MessageLoop loop; | 82 base::MessageLoop loop; |
83 VariationsHttpHeaderProvider provider; | 83 VariationsHttpHeaderProvider provider; |
84 | 84 |
85 // Invalid experiment ids. | 85 // Invalid experiment ids. |
86 EXPECT_FALSE(provider.SetDefaultVariationIds( | 86 EXPECT_FALSE(provider.SetDefaultVariationIds( |
87 std::vector<std::string>{"abcd12", "456"})); | 87 std::vector<std::string>{"abcd12", "456"})); |
88 provider.InitVariationIDsCacheIfNeeded(); | 88 provider.InitVariationIDsCacheIfNeeded(); |
89 EXPECT_TRUE(provider.GetClientDataHeader(false).empty()); | 89 EXPECT_TRUE(provider.GetClientDataHeader().empty()); |
90 | 90 |
91 // Invalid trigger experiment id | 91 // Invalid trigger experiment id |
92 EXPECT_FALSE(provider.SetDefaultVariationIds( | 92 EXPECT_FALSE(provider.SetDefaultVariationIds( |
93 std::vector<std::string>{"12", "tabc456"})); | 93 std::vector<std::string>{"12", "tabc456"})); |
94 provider.InitVariationIDsCacheIfNeeded(); | 94 provider.InitVariationIDsCacheIfNeeded(); |
95 EXPECT_TRUE(provider.GetClientDataHeader(false).empty()); | 95 EXPECT_TRUE(provider.GetClientDataHeader().empty()); |
96 } | 96 } |
97 | 97 |
98 TEST_F(VariationsHttpHeaderProviderTest, OnFieldTrialGroupFinalized) { | 98 TEST_F(VariationsHttpHeaderProviderTest, OnFieldTrialGroupFinalized) { |
99 base::MessageLoop loop; | 99 base::MessageLoop loop; |
100 base::FieldTrialList field_trial_list(nullptr); | 100 base::FieldTrialList field_trial_list(nullptr); |
101 VariationsHttpHeaderProvider provider; | 101 VariationsHttpHeaderProvider provider; |
102 provider.InitVariationIDsCacheIfNeeded(); | 102 provider.InitVariationIDsCacheIfNeeded(); |
103 | 103 |
104 const std::string default_name = "default"; | 104 const std::string default_name = "default"; |
105 scoped_refptr<base::FieldTrial> trial_1(CreateTrialAndAssociateId( | 105 scoped_refptr<base::FieldTrial> trial_1(CreateTrialAndAssociateId( |
106 "t1", default_name, GOOGLE_WEB_PROPERTIES, 123)); | 106 "t1", default_name, GOOGLE_WEB_PROPERTIES, 123)); |
| 107 |
107 ASSERT_EQ(default_name, trial_1->group_name()); | 108 ASSERT_EQ(default_name, trial_1->group_name()); |
108 | 109 |
109 scoped_refptr<base::FieldTrial> trial_2(CreateTrialAndAssociateId( | 110 scoped_refptr<base::FieldTrial> trial_2(CreateTrialAndAssociateId( |
110 "t2", default_name, GOOGLE_WEB_PROPERTIES_TRIGGER, 456)); | 111 "t2", default_name, GOOGLE_WEB_PROPERTIES_TRIGGER, 456)); |
| 112 |
111 ASSERT_EQ(default_name, trial_2->group_name()); | 113 ASSERT_EQ(default_name, trial_2->group_name()); |
112 | 114 |
113 scoped_refptr<base::FieldTrial> trial_3(CreateTrialAndAssociateId( | |
114 "t3", default_name, GOOGLE_WEB_PROPERTIES_SIGNED_IN, 789)); | |
115 ASSERT_EQ(default_name, trial_3->group_name()); | |
116 | |
117 // Run the message loop to make sure OnFieldTrialGroupFinalized is called for | 115 // Run the message loop to make sure OnFieldTrialGroupFinalized is called for |
118 // the two field trials. | 116 // the two field trials. |
119 base::RunLoop().RunUntilIdle(); | 117 base::RunLoop().RunUntilIdle(); |
120 | 118 |
121 // Get non-signed in ids. | 119 std::string variations = provider.GetClientDataHeader(); |
122 { | |
123 std::string variations = provider.GetClientDataHeader(false); | |
124 std::set<VariationID> variation_ids; | |
125 std::set<VariationID> trigger_ids; | |
126 ASSERT_TRUE(ExtractVariationIds(variations, &variation_ids, &trigger_ids)); | |
127 EXPECT_EQ(1U, variation_ids.size()); | |
128 EXPECT_TRUE(variation_ids.find(123) != variation_ids.end()); | |
129 EXPECT_EQ(1U, trigger_ids.size()); | |
130 EXPECT_TRUE(trigger_ids.find(456) != trigger_ids.end()); | |
131 } | |
132 | 120 |
133 // Now, get signed-in ids. | 121 std::set<VariationID> variation_ids; |
134 { | 122 std::set<VariationID> trigger_ids; |
135 std::string variations = provider.GetClientDataHeader(true); | 123 ASSERT_TRUE(ExtractVariationIds(variations, &variation_ids, &trigger_ids)); |
136 std::set<VariationID> variation_ids; | 124 EXPECT_TRUE(variation_ids.find(123) != variation_ids.end()); |
137 std::set<VariationID> trigger_ids; | 125 EXPECT_TRUE(trigger_ids.find(456) != trigger_ids.end()); |
138 ASSERT_TRUE(ExtractVariationIds(variations, &variation_ids, &trigger_ids)); | |
139 EXPECT_EQ(2U, variation_ids.size()); | |
140 EXPECT_TRUE(variation_ids.find(123) != variation_ids.end()); | |
141 EXPECT_TRUE(variation_ids.find(789) != variation_ids.end()); | |
142 EXPECT_EQ(1U, trigger_ids.size()); | |
143 EXPECT_TRUE(trigger_ids.find(456) != trigger_ids.end()); | |
144 } | |
145 } | 126 } |
146 | 127 |
147 TEST_F(VariationsHttpHeaderProviderTest, GetVariationsString) { | 128 TEST_F(VariationsHttpHeaderProviderTest, GetVariationsString) { |
148 base::MessageLoop loop; | 129 base::MessageLoop loop; |
149 base::FieldTrialList field_trial_list(nullptr); | 130 base::FieldTrialList field_trial_list(nullptr); |
150 | 131 |
151 CreateTrialAndAssociateId("t1", "g1", GOOGLE_WEB_PROPERTIES, 123); | 132 CreateTrialAndAssociateId("t1", "g1", GOOGLE_WEB_PROPERTIES, 123); |
152 CreateTrialAndAssociateId("t2", "g2", GOOGLE_WEB_PROPERTIES, 124); | 133 CreateTrialAndAssociateId("t2", "g2", GOOGLE_WEB_PROPERTIES, 124); |
153 // SIGNED_IN ids shouldn't be included. | |
154 CreateTrialAndAssociateId("t3", "g3", GOOGLE_WEB_PROPERTIES_SIGNED_IN, 125); | |
155 | 134 |
156 VariationsHttpHeaderProvider provider; | 135 VariationsHttpHeaderProvider provider; |
157 std::vector<std::string> ids; | 136 std::vector<std::string> ids; |
158 ids.push_back("100"); | 137 ids.push_back("100"); |
159 ids.push_back("200"); | 138 ids.push_back("200"); |
160 provider.SetDefaultVariationIds(ids); | 139 provider.SetDefaultVariationIds(ids); |
161 EXPECT_EQ(" 100 123 124 200 ", provider.GetVariationsString()); | 140 EXPECT_EQ(" 100 123 124 200 ", provider.GetVariationsString()); |
162 } | 141 } |
163 | 142 |
164 } // namespace variations | 143 } // namespace variations |
OLD | NEW |