Add pyauto hook for getting and manipulating the data underneath the NTP.

chrome/browser/history/top_sites_unittest.cc

 // Copyright (c) 2010 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 "app/l10n_util.h"
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/scoped_temp_dir.h"
 #include "base/string_util.h"
@@ skipping 1325 matching lines @@
   profile()->CreateTopSites();
   WaitForTopSites();
   EXPECT_FALSE(IsTopSitesLoaded());
 
   // Load history, which should make TopSites finish loading too.
   profile()->CreateHistoryService(false, false);
   profile()->BlockUntilTopSitesLoaded();
   EXPECT_TRUE(IsTopSitesLoaded());
 }
 
+// Used for running the refresh method of TopSites synchronously.
+class RefreshRunner {
+ public:
+  RefreshRunner() : number_of_callbacks_(0), waiting_(false) {}
+
+  void Refresh(TopSites* top_sites) {
+    int start_number_of_callbacks = number_of_callbacks_;
+    top_sites->RefreshAndCallback(NewCallback(this, &RefreshRunner::OnRefresh));
+    if (start_number_of_callbacks == number_of_callbacks_) {
+      waiting_ = true;
+      MessageLoop::current()->Run();
+    }
+  }
+
+  int number_of_callbacks() const { return number_of_callbacks_; }
+
+ private:
+  // Callback for TopSites::RefreshAndCallback.
+  void OnRefresh() {
+    number_of_callbacks_++;
+    if (waiting_) {
+      MessageLoop::current()->Quit();
+      waiting_ = false;
+    }
+  }
+
+  int number_of_callbacks_;
+  bool waiting_;
+
+  DISALLOW_COPY_AND_ASSIGN(RefreshRunner);
+};
+
+// This tests RefreshAndCallback, a method only used in higher-level testing.
+TEST_F(TopSitesTest, RefreshCallbackInvoked) {
+  // Test that the callback is invoked.
+  RefreshRunner refresh_runner;
+  refresh_runner.Refresh(top_sites());
+  ASSERT_EQ(1, refresh_runner.number_of_callbacks());
+
+  // Tests that the callback can be invoked again.
+  refresh_runner.Refresh(top_sites());
+  ASSERT_EQ(2, refresh_runner.number_of_callbacks());
+
+  // Test that the callback is not invoked accidentally.
+  StartQueryForMostVisited();
+  WaitForHistory();
+  WaitForTopSites();
+
+  ASSERT_EQ(2, refresh_runner.number_of_callbacks());
+}
+
+// This tests RefreshAndCallback, a method only used in higher-level testing.
+TEST_F(TopSitesTest, RefreshWorks) {
+  GURL google("http://google.com/");
+  AddPageToHistory(google);
+  RefreshRunner refresh_runner;
+  refresh_runner.Refresh(top_sites());
+
+  TopSitesQuerier querier;
+  querier.QueryTopSites(top_sites(), true);
+
+  ASSERT_EQ(1u + GetPrepopulatePages().size(), querier.urls().size());
+}
+
 }  // namespace history