Have the VariationsService attempt to fetch the seed when an update is ready.

chrome/browser/metrics/variations_service_unittest.cc

 // Copyright (c) 2012 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 "base/base64.h"
 #include "base/string_split.h"
 #include "chrome/browser/metrics/proto/study.pb.h"
 #include "chrome/browser/metrics/variations_service.h"
+#include "chrome/browser/upgrade_detector.h"
 #include "chrome/common/chrome_version_info.h"
 #include "chrome/common/pref_names.h"
+#include "chrome/common/chrome_notification_types.h"
 #include "chrome/test/base/testing_pref_service.h"
+#include "content/public/browser/notification_service.h"
+#include "content/public/test/test_browser_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chrome_variations {
 
 namespace {
 
+class TestVariationsService : public VariationsService {

[Alexei Svitkine (slow), 2012/07/23 20:26:05]

Add a short comment.

[SteveT, 2012/07/24 14:42:58]

Done.

+ public:
+  TestVariationsService() {}
+  virtual ~TestVariationsService() {}

[Alexei Svitkine (slow), 2012/07/23 20:26:05]

Nit: Blank line after this.

[SteveT, 2012/07/24 14:42:58]

Done.

+  bool fetch_attempted() { return fetch_attempted_; }
+
+  void SimulateUpgradeAvailable() {

[Alexei Svitkine (slow), 2012/07/23 20:26:05]

Add a comment.

[SteveT, 2012/07/24 14:42:58]

Done.

+    Observe(chrome::NOTIFICATION_UPGRADE_RECOMMENDED,
+        content::Source<UpgradeDetector>(NULL),
+        content::NotificationService::NoDetails());
+  }
+
+ protected:
+  virtual void FetchVariationsSeed();
+
+ private:
+  bool fetch_attempted_;
+};

[Alexei Svitkine (slow), 2012/07/23 20:26:05]

DISALLOW_COPY_AND_ASSIGN?

[SteveT, 2012/07/24 14:42:58]

Done.

+
+void TestVariationsService::FetchVariationsSeed() {
+  fetch_attempted_ = true;

[Alexei Svitkine (slow), 2012/07/23 20:26:05]

Nit: Inline.

[SteveT, 2012/07/24 14:42:58]

Done.

+}
+
 // Converts |time| to Study proto format.
 int64 TimeToProtoTime(const base::Time& time) {
   return (time - base::Time::UnixEpoch()).InSeconds();
 }
 
 // Populates |seed| with simple test data. The resulting seed will contain one
 // study called "test", which contains one experiment called "abc" with
 // probability weight 100. |seed|'s study field will be cleared before adding
 // the new study.
 chrome_variations::TrialsSeed CreateTestSeed() {
   chrome_variations::TrialsSeed seed;
   chrome_variations::Study* study = seed.add_study();
   study->set_name("test");
   study->set_default_experiment_name("abc");
   chrome_variations::Study_Experiment* experiment = study->add_experiment();
   experiment->set_name("abc");
   experiment->set_probability_weight(100);
   return seed;
 }
 
 }  // namespace
 
+TEST(VariationsServiceTest, AttemptFetchOnAutoUpdate) {
+  // Simulate an auto-update and ensure that the VariationsService attempts
+  // to fetch the variations seed.
+  MessageLoopForUI message_loop;
+  content::TestBrowserThread ui_thread(content::BrowserThread::UI,
+                                       &message_loop);
+  TestVariationsService test_service;
+  ASSERT_FALSE(test_service.fetch_attempted());
+  test_service.SimulateUpgradeAvailable();
+  ASSERT_TRUE(test_service.fetch_attempted());
+}
+
 TEST(VariationsServiceTest, CheckStudyChannel) {
   const chrome::VersionInfo::Channel channels[] = {
     chrome::VersionInfo::CHANNEL_CANARY,
     chrome::VersionInfo::CHANNEL_DEV,
     chrome::VersionInfo::CHANNEL_BETA,
     chrome::VersionInfo::CHANNEL_STABLE,
   };
   const Study_Channel study_channels[] = {
     Study_Channel_CANARY,
     Study_Channel_DEV,
@@ skipping 399 matching lines @@
   ASSERT_TRUE(valid);
   Study_Experiment* repeated_group = study.add_experiment();
   repeated_group->set_name("abc");
   repeated_group->set_probability_weight(1);
   valid = VariationsService::ValidateStudyAndComputeTotalProbability(study,
       &total_probability);
   EXPECT_FALSE(valid);
 }
 
 }  // namespace chrome_variations