Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/chrome_notification_types.h" | |
| 6 #include "chrome/browser/profiles/profile.h" | 5 #include "chrome/browser/profiles/profile.h" |
| 7 #include "chrome/browser/search_engines/template_url_prepopulate_data.h" | 6 #include "chrome/browser/search_engines/template_url_prepopulate_data.h" |
| 8 #include "chrome/browser/search_engines/template_url_service.h" | 7 #include "chrome/browser/search_engines/template_url_service.h" |
| 9 #include "chrome/browser/search_engines/template_url_service_factory.h" | 8 #include "chrome/browser/search_engines/template_url_service_factory.h" |
| 10 #include "chrome/browser/ui/browser.h" | 9 #include "chrome/browser/ui/browser.h" |
| 11 #include "chrome/test/base/in_process_browser_test.h" | 10 #include "chrome/test/base/in_process_browser_test.h" |
| 12 #include "chrome/test/base/ui_test_utils.h" | 11 #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" | 12 #include "net/base/net_util.h" |
| 16 #include "net/dns/mock_host_resolver.h" | 13 #include "net/dns/mock_host_resolver.h" |
| 17 | 14 |
| 18 namespace { | 15 namespace { |
| 19 class TemplateURLScraperTest : public InProcessBrowserTest { | 16 class TemplateURLScraperTest : public InProcessBrowserTest { |
| 20 public: | 17 public: |
| 21 TemplateURLScraperTest() { | 18 TemplateURLScraperTest() { |
| 22 } | 19 } |
| 23 | 20 |
| 24 private: | 21 private: |
| 25 DISALLOW_COPY_AND_ASSIGN(TemplateURLScraperTest); | 22 DISALLOW_COPY_AND_ASSIGN(TemplateURLScraperTest); |
| 26 }; | 23 }; |
| 27 | 24 |
| 28 class TemplateURLServiceLoader : public content::NotificationObserver { | 25 class TemplateURLServiceLoader { |
| 29 public: | 26 public: |
| 30 explicit TemplateURLServiceLoader(TemplateURLService* model) : model_(model) { | 27 explicit TemplateURLServiceLoader(TemplateURLService* model) : model_(model) { |
| 31 registrar_.Add(this, chrome::NOTIFICATION_TEMPLATE_URL_SERVICE_LOADED, | 28 scoped_ptr<TemplateURLService::Subscription> subscription = |
| 32 content::Source<TemplateURLService>(model)); | 29 model_->RegisterOnLoadedCallback( |
| 30 base::MessageLoop::current()->QuitClosure()); | |
| 33 model_->Load(); | 31 model_->Load(); |
| 34 content::RunMessageLoop(); | 32 content::RunMessageLoop(); |
|
Avi (use Gerrit)
2013/10/02 22:27:52
Er... This smells. Do you want MessageLoopRunner?
Cait (Slow)
2013/10/03 15:36:45
Done.
| |
| 35 } | 33 } |
| 36 | 34 |
| 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: | 35 private: |
| 47 content::NotificationRegistrar registrar_; | |
| 48 | |
| 49 TemplateURLService* model_; | 36 TemplateURLService* model_; |
| 50 | 37 |
| 51 DISALLOW_COPY_AND_ASSIGN(TemplateURLServiceLoader); | 38 DISALLOW_COPY_AND_ASSIGN(TemplateURLServiceLoader); |
| 52 }; | 39 }; |
| 53 | 40 |
| 54 } // namespace | 41 } // namespace |
| 55 | 42 |
| 56 /* | 43 /* |
| 57 IN_PROC_BROWSER_TEST_F(TemplateURLScraperTest, ScrapeWithOnSubmit) { | 44 IN_PROC_BROWSER_TEST_F(TemplateURLScraperTest, ScrapeWithOnSubmit) { |
| 58 host_resolver()->AddRule("*.foo.com", "localhost"); | 45 host_resolver()->AddRule("*.foo.com", "localhost"); |
| (...skipping 20 matching lines...) Expand all Loading... | |
| 79 HTTPTestServer::CreateServerWithFileRootURL( | 66 HTTPTestServer::CreateServerWithFileRootURL( |
| 80 L"chrome/test/data/template_url_scraper/submit_handler", L"/", | 67 L"chrome/test/data/template_url_scraper/submit_handler", L"/", |
| 81 g_browser_process->io_thread()->message_loop())); | 68 g_browser_process->io_thread()->message_loop())); |
| 82 ui_test_utils::NavigateToURLBlockUntilNavigationsComplete( | 69 ui_test_utils::NavigateToURLBlockUntilNavigationsComplete( |
| 83 browser(), GURL("http://www.foo.com:1337/"), 2); | 70 browser(), GURL("http://www.foo.com:1337/"), 2); |
| 84 | 71 |
| 85 all_urls = template_urls->GetTemplateURLs(); | 72 all_urls = template_urls->GetTemplateURLs(); |
| 86 EXPECT_EQ(1, all_urls.size() - prepopulate_urls.size()); | 73 EXPECT_EQ(1, all_urls.size() - prepopulate_urls.size()); |
| 87 } | 74 } |
| 88 */ | 75 */ |
| OLD | NEW |