OLD | NEW |
(Empty) | |
| 1 // Copyright 2014 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/bind.h" |
| 6 #include "content/browser/loader/cross_site_resource_handler.h" |
| 7 #include "content/browser/loader/resource_dispatcher_host_impl.h" |
| 8 #include "content/browser/loader/resource_request_info_impl.h" |
| 9 #include "content/browser/transition_request_manager.h" |
| 10 #include "content/browser/web_contents/web_contents_impl.h" |
| 11 #include "content/public/browser/web_contents_observer.h" |
| 12 #include "content/public/test/content_browser_test.h" |
| 13 #include "content/public/test/content_browser_test_utils.h" |
| 14 #include "content/public/test/test_utils.h" |
| 15 #include "content/shell/browser/shell.h" |
| 16 #include "content/shell/browser/shell_resource_dispatcher_host_delegate.h" |
| 17 #include "net/test/embedded_test_server/embedded_test_server.h" |
| 18 #include "net/url_request/url_request.h" |
| 19 |
| 20 namespace content { |
| 21 |
| 22 class TransitionBrowserTest : public ContentBrowserTest { |
| 23 public: |
| 24 TransitionBrowserTest() {} |
| 25 |
| 26 private: |
| 27 DISALLOW_COPY_AND_ASSIGN(TransitionBrowserTest); |
| 28 }; |
| 29 |
| 30 class TransitionBrowserTestObserver |
| 31 : public WebContentsObserver, |
| 32 public ShellResourceDispatcherHostDelegate { |
| 33 public: |
| 34 TransitionBrowserTestObserver(WebContents* web_contents) |
| 35 : WebContentsObserver(web_contents), |
| 36 request_(NULL), |
| 37 did_defer_response_(false), |
| 38 is_transition_request_(false) { |
| 39 } |
| 40 |
| 41 virtual void RequestBeginning( |
| 42 net::URLRequest* request, |
| 43 ResourceContext* resource_context, |
| 44 appcache::AppCacheService* appcache_service, |
| 45 ResourceType::Type resource_type, |
| 46 int child_id, |
| 47 int route_id, |
| 48 ScopedVector<ResourceThrottle>* throttles) OVERRIDE { |
| 49 CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| 50 ShellResourceDispatcherHostDelegate::RequestBeginning(request, |
| 51 resource_context, |
| 52 appcache_service, |
| 53 resource_type, |
| 54 child_id, |
| 55 route_id, |
| 56 throttles); |
| 57 request_ = request; |
| 58 |
| 59 ResourceRequestInfoImpl* info = |
| 60 ResourceRequestInfoImpl::ForRequest(request_); |
| 61 |
| 62 TransitionRequestManager::GetInstance()->SetHasPendingTransitionRequest( |
| 63 child_id, info->GetRenderFrameID(), is_transition_request_); |
| 64 } |
| 65 |
| 66 virtual void OnResponseStarted( |
| 67 net::URLRequest* request, |
| 68 ResourceContext* resource_context, |
| 69 ResourceResponse* response, |
| 70 IPC::Sender* sender) OVERRIDE { |
| 71 ResourceRequestInfoImpl* info = |
| 72 ResourceRequestInfoImpl::ForRequest(request_); |
| 73 |
| 74 did_defer_response_ = info->cross_site_handler()->did_defer_for_testing(); |
| 75 } |
| 76 |
| 77 void set_pending_transition_request(bool is_transition_request) { |
| 78 is_transition_request_ = is_transition_request; |
| 79 } |
| 80 |
| 81 bool did_defer_response() const { return did_defer_response_; } |
| 82 |
| 83 private: |
| 84 net::URLRequest* request_; |
| 85 bool did_defer_response_; |
| 86 bool is_transition_request_; |
| 87 }; |
| 88 |
| 89 // This tests that normal navigations don't defer at first response. |
| 90 IN_PROC_BROWSER_TEST_F(TransitionBrowserTest, |
| 91 NormalNavigationNotDeferred) { |
| 92 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); |
| 93 scoped_ptr<TransitionBrowserTestObserver> observer( |
| 94 new TransitionBrowserTestObserver(shell()->web_contents())); |
| 95 |
| 96 ResourceDispatcherHost::Get()->SetDelegate(observer.get()); |
| 97 |
| 98 NavigateToURL(shell(), embedded_test_server()->GetURL("/title1.html")); |
| 99 |
| 100 ASSERT_FALSE(observer->did_defer_response()); |
| 101 } |
| 102 |
| 103 // This tests that when a navigation transition is detected, the response is |
| 104 // deferred. |
| 105 IN_PROC_BROWSER_TEST_F(TransitionBrowserTest, |
| 106 TransitionNavigationIsDeferred) { |
| 107 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); |
| 108 scoped_ptr<TransitionBrowserTestObserver> observer( |
| 109 new TransitionBrowserTestObserver(shell()->web_contents())); |
| 110 |
| 111 ResourceDispatcherHost::Get()->SetDelegate(observer.get()); |
| 112 observer->set_pending_transition_request(true); |
| 113 |
| 114 NavigateToURL(shell(), embedded_test_server()->GetURL("/title1.html")); |
| 115 |
| 116 ASSERT_TRUE(observer->did_defer_response()); |
| 117 } |
| 118 |
| 119 } // namespace content |
OLD | NEW |