| /* |
| * Copyright (C) 2011, 2012 Google Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions are |
| * met: |
| * |
| * * Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * * Redistributions in binary form must reproduce the above |
| * copyright notice, this list of conditions and the following disclaimer |
| * in the documentation and/or other materials provided with the |
| * distribution. |
| * * Neither the name of Google Inc. nor the names of its |
| * contributors may be used to endorse or promote products derived from |
| * this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include "config.h" |
| |
| #include "web/FrameLoaderClientImpl.h" |
| |
| #include "core/loader/FrameLoader.h" |
| #include "platform/weborigin/KURL.h" |
| #include "public/web/WebFrameClient.h" |
| #include "public/web/WebPluginParams.h" |
| #include "public/web/WebPluginPlaceholder.h" |
| #include "public/web/WebSettings.h" |
| #include "public/web/WebView.h" |
| #include "web/PluginPlaceholderImpl.h" |
| #include "web/WebLocalFrameImpl.h" |
| #include "web/tests/FrameTestHelpers.h" |
| #include "wtf/text/CString.h" |
| #include "wtf/text/WTFString.h" |
| #include <gmock/gmock.h> |
| #include <gtest/gtest.h> |
| |
| using testing::_; |
| using testing::Mock; |
| using testing::Return; |
| |
| namespace blink { |
| namespace { |
| |
| class MockWebFrameClient : public WebFrameClient { |
| public: |
| virtual ~MockWebFrameClient() { } |
| |
| MOCK_METHOD2(userAgentOverride, WebString(WebLocalFrame*, const WebURL&)); |
| MOCK_METHOD2(createPluginPlaceholder, WebPluginPlaceholder*(WebLocalFrame*, const WebPluginParams&)); |
| }; |
| |
| class FrameLoaderClientImplTest : public ::testing::Test { |
| protected: |
| virtual void SetUp() override |
| { |
| ON_CALL(m_webFrameClient, userAgentOverride(_, _)).WillByDefault(Return(WebString())); |
| |
| FrameTestHelpers::TestWebViewClient webViewClient; |
| m_webView = WebView::create(&webViewClient); |
| // FIXME: http://crbug.com/363843. This needs to find a better way to |
| // not create graphics layers. |
| m_webView->settings()->setAcceleratedCompositingEnabled(false); |
| m_mainFrame = WebLocalFrame::create(&m_webFrameClient); |
| m_webView->setMainFrame(m_mainFrame); |
| m_frameLoaderClientImpl = toFrameLoaderClientImpl(toWebLocalFrameImpl(m_webView->mainFrame())->frame()->loader().client()); |
| } |
| |
| virtual void TearDown() override |
| { |
| m_webView->close(); |
| m_mainFrame->close(); |
| } |
| |
| WebString userAgent() |
| { |
| // The test always returns the same user agent, regardless of the URL passed in. |
| KURL dummyURL(ParsedURLString, "about:blank"); |
| WTF::CString userAgent = m_frameLoaderClientImpl->userAgent(dummyURL).utf8(); |
| return WebString::fromUTF8(userAgent.data(), userAgent.length()); |
| } |
| |
| WebLocalFrameImpl* mainFrame() { return toWebLocalFrameImpl(m_webView->mainFrame()); } |
| Document& document() { return *toWebLocalFrameImpl(m_mainFrame)->frame()->document(); } |
| MockWebFrameClient& webFrameClient() { return m_webFrameClient; } |
| FrameLoaderClient& frameLoaderClient() { return *m_frameLoaderClientImpl; } |
| |
| private: |
| MockWebFrameClient m_webFrameClient; |
| FrameLoaderClientImpl* m_frameLoaderClientImpl; |
| WebView* m_webView; |
| WebFrame* m_mainFrame; |
| }; |
| |
| TEST_F(FrameLoaderClientImplTest, UserAgentOverride) |
| { |
| const WebString defaultUserAgent = userAgent(); |
| const WebString overrideUserAgent = WebString::fromUTF8("dummy override"); |
| |
| // Override the user agent and make sure we get it back. |
| EXPECT_CALL(webFrameClient(), userAgentOverride(_, _)).WillOnce(Return(overrideUserAgent)); |
| EXPECT_TRUE(overrideUserAgent.equals(userAgent())); |
| Mock::VerifyAndClearExpectations(&webFrameClient()); |
| |
| // Remove the override and make sure we get the original back. |
| EXPECT_CALL(webFrameClient(), userAgentOverride(_, _)).WillOnce(Return(WebString())); |
| EXPECT_TRUE(defaultUserAgent.equals(userAgent())); |
| } |
| |
| TEST_F(FrameLoaderClientImplTest, CreatePluginPlaceholderForwardsToWebFrameClient) |
| { |
| KURL url(ParsedURLString, "http://www.example.com/plugin.swf"); |
| Vector<String> paramNames(1, "param"); |
| Vector<String> paramValues(1, "value"); |
| String mimeType = "application/x-shockwave-flash"; |
| bool loadManually = false; |
| |
| // Test with a valid WebPluginPlaceholder. |
| { |
| WebPluginPlaceholder* webPluginPlaceholder = new WebPluginPlaceholder; |
| EXPECT_CALL(webFrameClient(), createPluginPlaceholder(mainFrame(), _)) |
| .WillOnce(Return(webPluginPlaceholder)); |
| OwnPtrWillBeRawPtr<PluginPlaceholder> pluginPlaceholder = frameLoaderClient().createPluginPlaceholder( |
| document(), url, paramNames, paramValues, mimeType, loadManually); |
| ASSERT_TRUE(pluginPlaceholder); |
| EXPECT_EQ(webPluginPlaceholder, static_cast<PluginPlaceholderImpl*>(pluginPlaceholder.get())->webPluginPlaceholder()); |
| } |
| |
| // Test with no WebPluginPlaceholder. |
| { |
| EXPECT_CALL(webFrameClient(), createPluginPlaceholder(mainFrame(), _)) |
| .WillOnce(Return(nullptr)); |
| OwnPtrWillBeRawPtr<PluginPlaceholder> pluginPlaceholder = frameLoaderClient().createPluginPlaceholder( |
| document(), url, paramNames, paramValues, mimeType, loadManually); |
| ASSERT_FALSE(pluginPlaceholder); |
| } |
| } |
| |
| } // namespace |
| } // namespace blink |