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