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/custom_handlers/protocol_handler_registry.h" | 5 #include "chrome/browser/custom_handlers/protocol_handler_registry.h" |
6 | 6 |
7 #include <set> | 7 #include <set> |
8 | 8 |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
10 #include "base/message_loop.h" | 10 #include "base/message_loop.h" |
11 #include "base/utf_string_conversions.h" | 11 #include "base/utf_string_conversions.h" |
12 #include "chrome/common/chrome_notification_types.h" | 12 #include "chrome/common/chrome_notification_types.h" |
13 #include "chrome/common/custom_handlers/protocol_handler.h" | 13 #include "chrome/common/custom_handlers/protocol_handler.h" |
14 #include "chrome/test/base/testing_browser_process.h" | 14 #include "chrome/test/base/testing_browser_process.h" |
15 #include "chrome/test/base/testing_pref_service.h" | 15 #include "chrome/test/base/testing_pref_service.h" |
16 #include "chrome/test/base/testing_profile.h" | 16 #include "chrome/test/base/testing_profile.h" |
17 #include "content/public/browser/notification_observer.h" | 17 #include "content/public/browser/notification_observer.h" |
18 #include "content/public/browser/notification_registrar.h" | 18 #include "content/public/browser/notification_registrar.h" |
19 #include "content/public/browser/notification_source.h" | 19 #include "content/public/browser/notification_source.h" |
20 #include "content/public/test/test_browser_thread.h" | 20 #include "content/public/test/test_browser_thread.h" |
21 #include "content/public/test/test_renderer_host.h" | 21 #include "content/public/test/test_renderer_host.h" |
22 #include "net/url_request/url_request.h" | 22 #include "net/url_request/url_request.h" |
| 23 #include "net/url_request/url_request_context.h" |
23 | 24 |
24 using content::BrowserThread; | 25 using content::BrowserThread; |
25 | 26 |
26 namespace { | 27 namespace { |
27 | 28 |
28 class FakeDelegate : public ProtocolHandlerRegistry::Delegate { | 29 class FakeDelegate : public ProtocolHandlerRegistry::Delegate { |
29 public: | 30 public: |
30 FakeDelegate() : force_os_failure_(false) {} | 31 FakeDelegate() : force_os_failure_(false) {} |
31 virtual ~FakeDelegate() { } | 32 virtual ~FakeDelegate() { } |
32 virtual void RegisterExternalHandler(const std::string& protocol) { | 33 virtual void RegisterExternalHandler(const std::string& protocol) { |
(...skipping 608 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
641 delegate()->set_force_os_failure(true); | 642 delegate()->set_force_os_failure(true); |
642 registry()->OnAcceptRegisterProtocolHandler(ph_dont); | 643 registry()->OnAcceptRegisterProtocolHandler(ph_dont); |
643 MessageLoop::current()->Run(); | 644 MessageLoop::current()->Run(); |
644 ASSERT_TRUE(registry()->IsHandledProtocol("do")); | 645 ASSERT_TRUE(registry()->IsHandledProtocol("do")); |
645 ASSERT_EQ(static_cast<size_t>(1), registry()->GetHandlersFor("do").size()); | 646 ASSERT_EQ(static_cast<size_t>(1), registry()->GetHandlersFor("do").size()); |
646 ASSERT_FALSE(registry()->IsHandledProtocol("dont")); | 647 ASSERT_FALSE(registry()->IsHandledProtocol("dont")); |
647 ASSERT_EQ(static_cast<size_t>(1), registry()->GetHandlersFor("dont").size()); | 648 ASSERT_EQ(static_cast<size_t>(1), registry()->GetHandlersFor("dont").size()); |
648 } | 649 } |
649 | 650 |
650 static void MakeRequest(const GURL& url, ProtocolHandlerRegistry* registry) { | 651 static void MakeRequest(const GURL& url, ProtocolHandlerRegistry* registry) { |
651 net::URLRequest request(url, NULL); | 652 net::URLRequestContext context; |
| 653 net::URLRequest request(url, NULL, &context); |
652 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, | 654 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
653 MessageLoop::QuitClosure()); | 655 MessageLoop::QuitClosure()); |
654 scoped_refptr<net::URLRequestJob> job(registry->MaybeCreateJob(&request)); | 656 scoped_refptr<net::URLRequestJob> job(registry->MaybeCreateJob(&request)); |
655 ASSERT_TRUE(job.get() != NULL); | 657 ASSERT_TRUE(job.get() != NULL); |
656 } | 658 } |
657 | 659 |
658 TEST_F(ProtocolHandlerRegistryTest, TestMaybeCreateTaskWorksFromIOThread) { | 660 TEST_F(ProtocolHandlerRegistryTest, TestMaybeCreateTaskWorksFromIOThread) { |
659 ProtocolHandler ph1 = CreateProtocolHandler("mailto", "test1"); | 661 ProtocolHandler ph1 = CreateProtocolHandler("mailto", "test1"); |
660 registry()->OnAcceptRegisterProtocolHandler(ph1); | 662 registry()->OnAcceptRegisterProtocolHandler(ph1); |
661 GURL url("mailto:someone@something.com"); | 663 GURL url("mailto:someone@something.com"); |
(...skipping 143 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
805 ASSERT_EQ(ph3.url().GetOrigin() == ph2.url().GetOrigin(), | 807 ASSERT_EQ(ph3.url().GetOrigin() == ph2.url().GetOrigin(), |
806 ph3.IsSameOrigin(ph2)); | 808 ph3.IsSameOrigin(ph2)); |
807 } | 809 } |
808 | 810 |
809 TEST_F(ProtocolHandlerRegistryTest, TestInstallDefaultHandler) { | 811 TEST_F(ProtocolHandlerRegistryTest, TestInstallDefaultHandler) { |
810 ReloadProtocolHandlerRegistryAndInstallDefaultHandler(); | 812 ReloadProtocolHandlerRegistryAndInstallDefaultHandler(); |
811 std::vector<std::string> protocols; | 813 std::vector<std::string> protocols; |
812 registry()->GetRegisteredProtocols(&protocols); | 814 registry()->GetRegisteredProtocols(&protocols); |
813 ASSERT_EQ(static_cast<size_t>(1), protocols.size()); | 815 ASSERT_EQ(static_cast<size_t>(1), protocols.size()); |
814 } | 816 } |
OLD | NEW |