Home | History | Annotate | Download | only in search_engines 
      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 "chrome/browser/chrome_notification_types.h"
      6 #include "chrome/browser/profiles/profile.h"
      7 #include "chrome/browser/search_engines/template_url_prepopulate_data.h"
      8 #include "chrome/browser/search_engines/template_url_service.h"
      9 #include "chrome/browser/search_engines/template_url_service_factory.h"
     10 #include "chrome/browser/ui/browser.h"
     11 #include "chrome/test/base/in_process_browser_test.h"
     12 #include "chrome/test/base/ui_test_utils.h"
     13 #include "content/public/browser/notification_registrar.h"
     14 #include "content/public/browser/notification_source.h"
     15 #include "net/base/net_util.h"
     16 #include "net/dns/mock_host_resolver.h"
     17 
     18 namespace {
     19 class TemplateURLScraperTest : public InProcessBrowserTest {
     20  public:
     21   TemplateURLScraperTest() {
     22   }
     23 
     24  private:
     25   DISALLOW_COPY_AND_ASSIGN(TemplateURLScraperTest);
     26 };
     27 
     28 class TemplateURLServiceLoader : public content::NotificationObserver {
     29  public:
     30   explicit TemplateURLServiceLoader(TemplateURLService* model) : model_(model) {
     31     registrar_.Add(this, chrome::NOTIFICATION_TEMPLATE_URL_SERVICE_LOADED,
     32                    content::Source<TemplateURLService>(model));
     33     model_->Load();
     34     content::RunMessageLoop();
     35   }
     36 
     37   virtual void Observe(int type,
     38                        const content::NotificationSource& source,
     39                        const content::NotificationDetails& details) {
     40     if (type == chrome::NOTIFICATION_TEMPLATE_URL_SERVICE_LOADED &&
     41         content::Source<TemplateURLService>(source).ptr() == model_) {
     42       base::MessageLoop::current()->Quit();
     43     }
     44   }
     45 
     46  private:
     47   content::NotificationRegistrar registrar_;
     48 
     49   TemplateURLService* model_;
     50 
     51   DISALLOW_COPY_AND_ASSIGN(TemplateURLServiceLoader);
     52 };
     53 
     54 }  // namespace
     55 
     56 /*
     57 IN_PROC_BROWSER_TEST_F(TemplateURLScraperTest, ScrapeWithOnSubmit) {
     58   host_resolver()->AddRule("*.foo.com", "localhost");
     59 
     60   TemplateURLService* template_urls =
     61       TemplateURLServiceFactory::GetInstance(browser()->profile());
     62   TemplateURLServiceLoader loader(template_urls);
     63 
     64   TemplateURLService::TemplateURLVector all_urls =
     65       template_urls->GetTemplateURLs();
     66 
     67   // We need to substract the default pre-populated engines that the profile is
     68   // set up with.
     69   size_t default_index = 0;
     70   std::vector<TemplateURL*> prepopulate_urls;
     71   TemplateURLPrepopulateData::GetPrepopulatedEngines(
     72       browser()->profile()->GetPrefs(),
     73       &prepopulate_urls,
     74       &default_index);
     75 
     76   EXPECT_EQ(prepopulate_urls.size(), all_urls.size());
     77 
     78   scoped_refptr<HTTPTestServer> server(
     79       HTTPTestServer::CreateServerWithFileRootURL(
     80           L"chrome/test/data/template_url_scraper/submit_handler", L"/",
     81           g_browser_process->io_thread()->message_loop()));
     82   ui_test_utils::NavigateToURLBlockUntilNavigationsComplete(
     83       browser(), GURL("http://www.foo.com:1337/"), 2);
     84 
     85   all_urls = template_urls->GetTemplateURLs();
     86   EXPECT_EQ(1, all_urls.size() - prepopulate_urls.size());
     87 }
     88 */
     89