| // Copyright 2014 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 "base/command_line.h" |
| #include "content/browser/transition_request_manager.h" |
| #include "content/public/common/content_switches.h" |
| #include "content/public/test/test_browser_thread_bundle.h" |
| #include "net/http/http_response_headers.h" |
| #include "net/http/http_util.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace content { |
| |
| class TransitionRequestManagerTest : public testing::Test { |
| public: |
| TransitionRequestManagerTest() |
| : thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP) { |
| base::CommandLine::ForCurrentProcess()->AppendSwitch( |
| switches::kEnableExperimentalWebPlatformFeatures); |
| } |
| |
| ~TransitionRequestManagerTest() override {} |
| TestBrowserThreadBundle thread_bundle_; |
| }; |
| |
| TEST_F(TransitionRequestManagerTest, |
| ParseTransitionStylesheetsFromNullHeaders) { |
| const GURL url("http://www.test.com/"); |
| std::vector<GURL> entering_stylesheets; |
| scoped_refptr<net::HttpResponseHeaders> headers; |
| |
| TransitionRequestManager::ParseTransitionStylesheetsFromHeaders( |
| headers, entering_stylesheets, url); |
| ASSERT_TRUE(entering_stylesheets.empty()); |
| } |
| |
| TEST_F(TransitionRequestManagerTest, |
| ParseTransitionStylesheetsFromEmptyHeaders) { |
| const GURL url("http://www.test.com/"); |
| std::vector<GURL> entering_stylesheets; |
| |
| char headers_string[] = ""; |
| scoped_refptr<net::HttpResponseHeaders> headers( |
| new net::HttpResponseHeaders(net::HttpUtil::AssembleRawHeaders( |
| headers_string, sizeof(headers_string)))); |
| |
| TransitionRequestManager::ParseTransitionStylesheetsFromHeaders( |
| headers, entering_stylesheets, url); |
| ASSERT_TRUE(entering_stylesheets.empty()); |
| } |
| |
| TEST_F(TransitionRequestManagerTest, |
| ParseTransitionStylesheetsFromHeadersForInvalidURL) { |
| const GURL url; |
| std::vector<GURL> entering_stylesheets; |
| |
| char headers_string[] = |
| "HTTP/1.0 200 OK\r\n" |
| "link: <transition.css>;rel=transition-entering-stylesheet;scope=*\r\n" |
| "\r\n"; |
| scoped_refptr<net::HttpResponseHeaders> headers( |
| new net::HttpResponseHeaders(net::HttpUtil::AssembleRawHeaders( |
| headers_string, sizeof(headers_string)))); |
| |
| TransitionRequestManager::ParseTransitionStylesheetsFromHeaders( |
| headers, entering_stylesheets, url); |
| ASSERT_TRUE(entering_stylesheets.empty()); |
| } |
| |
| TEST_F(TransitionRequestManagerTest, ParseTransitionStylesheetsFromHeaders) { |
| const GURL url("http://www.test.com/"); |
| std::vector<GURL> entering_stylesheets; |
| |
| char headers_string[] = |
| "HTTP/1.0 200 OK\r\n" |
| "link: <transition.css>;rel=transition-entering-stylesheet;scope=*\r\n" |
| "\r\n"; |
| scoped_refptr<net::HttpResponseHeaders> headers( |
| new net::HttpResponseHeaders(net::HttpUtil::AssembleRawHeaders( |
| headers_string, sizeof(headers_string)))); |
| |
| TransitionRequestManager::ParseTransitionStylesheetsFromHeaders( |
| headers, entering_stylesheets, url); |
| ASSERT_TRUE(entering_stylesheets.size() == 1); |
| ASSERT_STREQ((url.spec() + "transition.css").c_str(), |
| entering_stylesheets[0].spec().c_str()); |
| } |
| |
| TEST_F(TransitionRequestManagerTest, |
| ParseMultipleTransitionStylesheetsFromHeaders) { |
| const GURL url("http://www.test.com/"); |
| std::vector<GURL> entering_stylesheets; |
| |
| char headers_string[] = |
| "HTTP/1.0 200 OK\r\n" |
| "link: <transition0.css>;rel=transition-entering-stylesheet;scope=*\r\n" |
| "link: <transition1.css>;rel=transition-entering-stylesheet;scope=*\r\n" |
| "link: <transition2.css>;rel=transition-entering-stylesheet;scope=*\r\n" |
| "\r\n"; |
| scoped_refptr<net::HttpResponseHeaders> headers( |
| new net::HttpResponseHeaders(net::HttpUtil::AssembleRawHeaders( |
| headers_string, sizeof(headers_string)))); |
| |
| TransitionRequestManager::ParseTransitionStylesheetsFromHeaders( |
| headers, entering_stylesheets, url); |
| ASSERT_TRUE(entering_stylesheets.size() == 3); |
| ASSERT_STREQ((url.spec() + "transition0.css").c_str(), |
| entering_stylesheets[0].spec().c_str()); |
| ASSERT_STREQ((url.spec() + "transition1.css").c_str(), |
| entering_stylesheets[1].spec().c_str()); |
| ASSERT_STREQ((url.spec() + "transition2.css").c_str(), |
| entering_stylesheets[2].spec().c_str()); |
| } |
| |
| // Tests that the TransitionRequestManager correctly performs origin checks |
| // and fetches the correct transition data. |
| // |
| // We add data for http://www.foo.com and http://www.test.com URLs, then check |
| // that a navigation to test.com gives the latter data. A third URL should |
| // give no fallback data, before we add a fallback and check that it's now |
| // provided for the third URL. |
| TEST_F(TransitionRequestManagerTest, AllowedHostCheck) { |
| TransitionRequestManager* request_manager = |
| TransitionRequestManager::GetInstance(); |
| |
| const int render_process_id = 42; |
| const int render_frame_id = 4242; |
| |
| const std::string test_dot_com_css_selector("#correctTransitionElement"); |
| const std::string fallback_css_selector("#fallbackTransitionElement"); |
| const GURL test_dot_com_url("http://www.test.com"); |
| const std::string markup("<b>Hello World</b>"); |
| |
| // 1. Add transition data for http://www.foo.com URLs. |
| request_manager->AddPendingTransitionRequestData( |
| render_process_id, render_frame_id, "http://www.foo.com", |
| "#wrongTransition", markup, std::vector<TransitionElement>()); |
| // 2. Add transition data for http://www.test.com URLs |
| request_manager->AddPendingTransitionRequestData( |
| render_process_id, render_frame_id, test_dot_com_url.spec(), |
| test_dot_com_css_selector, markup, std::vector<TransitionElement>()); |
| |
| // 3. Check that a navigation to http://www.test.com finds the data from 2). |
| TransitionLayerData transition_layer_data; |
| bool found = request_manager->GetPendingTransitionRequest( |
| render_process_id, render_frame_id, test_dot_com_url, |
| &transition_layer_data); |
| ASSERT_TRUE(found); |
| EXPECT_TRUE(transition_layer_data.css_selector == test_dot_com_css_selector); |
| |
| // 4. Check that a navigation to http://www.unrelated.com finds no data. |
| const GURL unrelated_dot_com_url("http://www.unrelated.com"); |
| found = request_manager->GetPendingTransitionRequest( |
| render_process_id, render_frame_id, unrelated_dot_com_url, |
| &transition_layer_data); |
| EXPECT_FALSE(found); |
| |
| // 5. Add transition data for '*', i.e. matching any navigation. |
| request_manager->AddPendingTransitionRequestData( |
| render_process_id, render_frame_id, "*", fallback_css_selector, markup, |
| std::vector<TransitionElement>()); |
| |
| // 6. Check that a navigation to http://wwww.unrelated.com now finds an entry. |
| found = request_manager->GetPendingTransitionRequest( |
| render_process_id, render_frame_id, unrelated_dot_com_url, |
| &transition_layer_data); |
| ASSERT_TRUE(found); |
| EXPECT_TRUE(transition_layer_data.css_selector == fallback_css_selector); |
| } |
| |
| } // namespace content |