Add other variations (with ids) for NTPSnippets to chrome://flags

components/variations/variations_http_header_provider_unittest.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/variations/variations_http_header_provider.h"
 
 #include <string>
 
 #include "base/base64.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 47 matching lines @@
   ~VariationsHttpHeaderProviderTest() override {}
 
   void TearDown() override { testing::ClearAllVariationIDs(); }
 };
 
 TEST_F(VariationsHttpHeaderProviderTest, SetDefaultVariationIds_Valid) {
   base::MessageLoop loop;
   VariationsHttpHeaderProvider provider;
 
   // Valid experiment ids.
-  EXPECT_TRUE(provider.SetDefaultVariationIds("12,456,t789"));
+  EXPECT_TRUE(provider.SetDefaultVariationIds({"12", "456", "t789"}));
   provider.InitVariationIDsCacheIfNeeded();
   std::string variations = provider.GetClientDataHeader();
   EXPECT_FALSE(variations.empty());
   std::set<VariationID> variation_ids;
   std::set<VariationID> trigger_ids;
   ASSERT_TRUE(ExtractVariationIds(variations, &variation_ids, &trigger_ids));
   EXPECT_TRUE(variation_ids.find(12) != variation_ids.end());
   EXPECT_TRUE(variation_ids.find(456) != variation_ids.end());
   EXPECT_TRUE(trigger_ids.find(789) != trigger_ids.end());
   EXPECT_FALSE(variation_ids.find(789) != variation_ids.end());
 }
 
 TEST_F(VariationsHttpHeaderProviderTest, SetDefaultVariationIds_Invalid) {
   base::MessageLoop loop;
   VariationsHttpHeaderProvider provider;
 
   // Invalid experiment ids.
-  EXPECT_FALSE(provider.SetDefaultVariationIds("abcd12,456"));
+  EXPECT_FALSE(provider.SetDefaultVariationIds(
+      std::vector<std::string>{"abcd12", "456"}));
   provider.InitVariationIDsCacheIfNeeded();
   EXPECT_TRUE(provider.GetClientDataHeader().empty());
 
   // Invalid trigger experiment id
-  EXPECT_FALSE(provider.SetDefaultVariationIds("12,tabc456"));
+  EXPECT_FALSE(provider.SetDefaultVariationIds(
+      std::vector<std::string>{"12", "tabc456"}));
   provider.InitVariationIDsCacheIfNeeded();
   EXPECT_TRUE(provider.GetClientDataHeader().empty());
 }
 
 TEST_F(VariationsHttpHeaderProviderTest, OnFieldTrialGroupFinalized) {
   base::MessageLoop loop;
   base::FieldTrialList field_trial_list(nullptr);
   VariationsHttpHeaderProvider provider;
   provider.InitVariationIDsCacheIfNeeded();
 
@@ skipping 22 matching lines @@
 }
 
 TEST_F(VariationsHttpHeaderProviderTest, GetVariationsString) {
   base::MessageLoop loop;
   base::FieldTrialList field_trial_list(nullptr);
 
   CreateTrialAndAssociateId("t1", "g1", GOOGLE_WEB_PROPERTIES, 123);
   CreateTrialAndAssociateId("t2", "g2", GOOGLE_WEB_PROPERTIES, 124);
 
   VariationsHttpHeaderProvider provider;
-  provider.SetDefaultVariationIds("100,200");
+  std::vector<std::string> ids;
+  ids.push_back("100");
+  ids.push_back("200");
+  provider.SetDefaultVariationIds(ids);
   EXPECT_EQ(" 100 123 124 200 ", provider.GetVariationsString());
 }
 
 }  // namespace variations