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

chrome/browser/metrics/variations_service_unittest.cc

Index: chrome/browser/metrics/variations_service_unittest.cc
diff --git a/chrome/browser/metrics/variations_service_unittest.cc b/chrome/browser/metrics/variations_service_unittest.cc
index d51149f1a1b93396d8efd06b83e836874fc7388d..e8a5447900f197d64f0034d1a850f9a9b40d49e7 100644
--- a/chrome/browser/metrics/variations_service_unittest.cc
+++ b/chrome/browser/metrics/variations_service_unittest.cc
@@ -6,15 +6,47 @@
 #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 {
 
+// A test class used to validate expected functionality in VariationsService.
+class TestVariationsService : public VariationsService {
+ public:
+  TestVariationsService() {}
+  virtual ~TestVariationsService() {}
+
+  bool fetch_attempted() { return fetch_attempted_; }
+
+  void SimulateUpgradeAvailable() {
+    // Simulate that an auto-update is ready by sending the appropriate

[Alexei Svitkine (slow), 2012/07/24 14:44:49]

Nit: Comment should be above the method.

[SteveT, 2012/07/24 19:22:54]

Done.

+    // notification to the Observe method. Note that the source and details are
+    // not necessary for this notification type.
+    Observe(chrome::NOTIFICATION_UPGRADE_RECOMMENDED,
+        content::Source<UpgradeDetector>(NULL),
+        content::NotificationService::NoDetails());
+  }
+
+ protected:
+  virtual void FetchVariationsSeed() {
+    fetch_attempted_ = true;
+  }
+
+ private:
+  bool fetch_attempted_;

[Alexei Svitkine (slow), 2012/07/24 14:44:49]

Please initialize |fetch_attempted_| in the constructor.

[SteveT, 2012/07/24 19:22:54]

Yeah this caused a test failure earlier :(

Done.

+
+  DISALLOW_COPY_AND_ASSIGN(TestVariationsService);
+};
+
 // Converts |time| to Study proto format.
 int64 TimeToProtoTime(const base::Time& time) {
   return (time - base::Time::UnixEpoch()).InSeconds();
@@ -37,6 +69,18 @@ chrome_variations::TrialsSeed CreateTestSeed() {
 
 }  // 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,