Home | History | Annotate | Download | only in top_sites 
      1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "base/values.h"
      6 #include "build/build_config.h"
      7 #include "chrome/browser/extensions/api/top_sites/top_sites_api.h"
      8 #include "chrome/browser/extensions/extension_function_test_utils.h"
      9 #include "chrome/browser/history/top_sites.h"
     10 #include "chrome/browser/profiles/profile.h"
     11 #include "chrome/browser/ui/browser.h"
     12 #include "chrome/test/base/in_process_browser_test.h"
     13 
     14 namespace utils = extension_function_test_utils;
     15 
     16 namespace extensions {
     17 
     18 namespace {
     19 
     20 class TopSitesExtensionTest : public InProcessBrowserTest {
     21  public:
     22   TopSitesExtensionTest() : top_sites_inited_(false), waiting_(false) {
     23   }
     24 
     25   virtual void SetUpOnMainThread() OVERRIDE {
     26     history::TopSites* top_sites = browser()->profile()->GetTopSites();
     27 
     28     // This may return async or sync. If sync, top_sites_inited_ will be true
     29     // before we get to the conditional below. Otherwise, we'll run a nested
     30     // message loop until the async callback.
     31     top_sites->GetMostVisitedURLs(
     32         base::Bind(&TopSitesExtensionTest::OnTopSitesAvailable, this));
     33 
     34     if (!top_sites_inited_) {
     35       waiting_ = true;
     36       base::MessageLoop::current()->Run();
     37     }
     38 
     39     // By this point, we know topsites has loaded. We can run the tests now.
     40   }
     41 
     42  private:
     43   void OnTopSitesAvailable(const history::MostVisitedURLList& data) {
     44     if (waiting_) {
     45       base::MessageLoop::current()->Quit();
     46       waiting_ = false;
     47     }
     48     top_sites_inited_ = true;
     49   }
     50 
     51   bool top_sites_inited_;
     52   bool waiting_;
     53 };
     54 
     55 }  // namespace
     56 
     57 IN_PROC_BROWSER_TEST_F(TopSitesExtensionTest, GetTopSites) {
     58   scoped_refptr<TopSitesGetFunction> get_top_sites_function(
     59       new TopSitesGetFunction());
     60   // Without a callback the function will not generate a result.
     61   get_top_sites_function->set_has_callback(true);
     62 
     63   scoped_ptr<base::Value> result(utils::RunFunctionAndReturnSingleResult(
     64       get_top_sites_function.get(), "[]", browser()));
     65   base::ListValue* list;
     66   ASSERT_TRUE(result->GetAsList(&list));
     67   EXPECT_GE(list->GetSize(), arraysize(history::kPrepopulatedPages));
     68 }
     69 
     70 }  // namespace extensions
     71