// Copyright (c) 2011 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 "chrome/browser/profiles/profile.h" #include "chrome/browser/search_engines/template_url_model.h" #include "chrome/browser/search_engines/template_url_prepopulate_data.h" #include "chrome/browser/ui/browser.h" #include "chrome/test/in_process_browser_test.h" #include "chrome/test/ui_test_utils.h" #include "content/common/notification_registrar.h" #include "content/common/notification_source.h" #include "content/common/notification_type.h" #include "net/base/mock_host_resolver.h" #include "net/base/net_util.h" namespace { class TemplateURLScraperTest : public InProcessBrowserTest { public: TemplateURLScraperTest() { } private: DISALLOW_COPY_AND_ASSIGN(TemplateURLScraperTest); }; class TemplateURLModelLoader : public NotificationObserver { public: explicit TemplateURLModelLoader(TemplateURLModel* model) : model_(model) { registrar_.Add(this, NotificationType::TEMPLATE_URL_MODEL_LOADED, Source<TemplateURLModel>(model)); model_->Load(); ui_test_utils::RunMessageLoop(); } virtual void Observe(NotificationType type, const NotificationSource& source, const NotificationDetails& details) { if (type == NotificationType::TEMPLATE_URL_MODEL_LOADED && Source<TemplateURLModel>(source).ptr() == model_) { MessageLoop::current()->Quit(); } } private: NotificationRegistrar registrar_; TemplateURLModel* model_; DISALLOW_COPY_AND_ASSIGN(TemplateURLModelLoader); }; } // namespace /* IN_PROC_BROWSER_TEST_F(TemplateURLScraperTest, ScrapeWithOnSubmit) { host_resolver()->AddRule("*.foo.com", "localhost"); TemplateURLModel* template_urls = browser()->profile()->GetTemplateURLModel(); TemplateURLModelLoader loader(template_urls); std::vector<const TemplateURL*> all_urls = template_urls->GetTemplateURLs(); // We need to substract the default pre-populated engines that the profile is // set up with. size_t default_index = 0; std::vector<TemplateURL*> prepopulate_urls; TemplateURLPrepopulateData::GetPrepopulatedEngines( browser()->profile()->GetPrefs(), &prepopulate_urls, &default_index); EXPECT_EQ(prepopulate_urls.size(), all_urls.size()); scoped_refptr<HTTPTestServer> server( HTTPTestServer::CreateServerWithFileRootURL( L"chrome/test/data/template_url_scraper/submit_handler", L"/", g_browser_process->io_thread()->message_loop())); ui_test_utils::NavigateToURLBlockUntilNavigationsComplete( browser(), GURL("http://www.foo.com:1337/"), 2); all_urls = template_urls->GetTemplateURLs(); EXPECT_EQ(1, all_urls.size() - prepopulate_urls.size()); } */