Index: chrome/common/variations/child_process_field_trial_syncer_unittest.cc |
diff --git a/chrome/common/variations/child_process_field_trial_syncer_unittest.cc b/chrome/common/variations/child_process_field_trial_syncer_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..7d1992a05597add9f1a4539fdaec93874823e7fc |
--- /dev/null |
+++ b/chrome/common/variations/child_process_field_trial_syncer_unittest.cc |
@@ -0,0 +1,94 @@ |
+// Copyright 2016 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 "chrome/common/variations/child_process_field_trial_syncer.h" |
+ |
+#include <string> |
+#include <utility> |
+#include <vector> |
+ |
+#include "base/base_switches.h" |
+#include "base/command_line.h" |
+#include "base/message_loop/message_loop.h" |
+#include "base/metrics/field_trial.h" |
+#include "base/run_loop.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace chrome_variations { |
+ |
+namespace { |
+ |
+// TestFieldTrialObserver to listen to be notified by the child process syncer. |
+class TestFieldTrialObserver : public base::FieldTrialList::Observer { |
+ public: |
+ TestFieldTrialObserver() {} |
+ ~TestFieldTrialObserver() override {} |
+ |
+ // base::FieldTrial::Observer: |
+ void OnFieldTrialGroupFinalized(const std::string& trial_name, |
+ const std::string& group_name) override { |
+ observed_entries_.push_back(std::make_pair(trial_name, group_name)); |
+ } |
+ |
+ size_t observed_entries_count() const { return observed_entries_.size(); } |
+ |
+ std::pair<std::string, std::string> get_observed_entry(int i) const { |
+ return observed_entries_[i]; |
+ } |
+ |
+ private: |
+ std::vector<std::pair<std::string, std::string>> observed_entries_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(TestFieldTrialObserver); |
+}; |
+ |
+// Needed because make_pair("a", "b") doesn't convert to std::string pair. |
+std::pair<std::string, std::string> MakeStringPair(const std::string& a, |
+ const std::string& b) { |
+ return std::make_pair(a, b); |
+} |
+ |
+} // namespace |
+ |
+TEST(ChildProcessFieldTrialSyncerTest, FieldTrialState) { |
+ base::MessageLoop message_loop; |
+ base::FieldTrialList field_trial_list(nullptr); |
+ |
+ base::FieldTrial* trial1 = base::FieldTrialList::CreateFieldTrial("A", "G1"); |
+ base::FieldTrial* trial2 = base::FieldTrialList::CreateFieldTrial("B", "G2"); |
+ base::FieldTrial* trial3 = base::FieldTrialList::CreateFieldTrial("C", "G3"); |
+ // Activate trial3 before command line is produced. |
+ trial1->group(); |
+ |
+ std::string states_string; |
+ base::FieldTrialList::AllStatesToString(&states_string); |
+ |
+ base::CommandLine::ForCurrentProcess()->AppendSwitchASCII( |
+ switches::kForceFieldTrials, states_string); |
+ EXPECT_EQ("*A/G1/B/G2/C/G3/", states_string); |
+ |
+ // Active trial 2 before creating the syncer. |
+ trial2->group(); |
+ |
+ TestFieldTrialObserver observer; |
+ ChildProcessFieldTrialSyncer syncer(&observer); |
+ syncer.InitFieldTrialObserving(*base::CommandLine::ForCurrentProcess()); |
+ |
+ // The observer should be notified of activated entries that were not activate |
+ // on the command line. In this case, trial 2. (Trial 1 was already active via |
+ // command line and so its state shouldn't be notified.) |
+ ASSERT_EQ(1U, observer.observed_entries_count()); |
+ EXPECT_EQ(MakeStringPair("B", "G2"), observer.get_observed_entry(0)); |
+ |
+ // Now, activate trial 3, which should also get reflected. |
+ trial3->group(); |
+ // Notifications from field trial activation actually happen via posted tasks, |
+ // so invoke the run loop. |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ ASSERT_EQ(2U, observer.observed_entries_count()); |
+ EXPECT_EQ(MakeStringPair("C", "G3"), observer.get_observed_entry(1)); |
+} |
+ |
+} // namespace chrome_variations |