OLD | NEW |
1 /* | 1 /* |
2 * Copyright (C) 2011, 2012 Google Inc. All rights reserved. | 2 * Copyright (C) 2011, 2012 Google Inc. All rights reserved. |
3 * | 3 * |
4 * Redistribution and use in source and binary forms, with or without | 4 * Redistribution and use in source and binary forms, with or without |
5 * modification, are permitted provided that the following conditions are | 5 * modification, are permitted provided that the following conditions are |
6 * met: | 6 * met: |
7 * | 7 * |
8 * * Redistributions of source code must retain the above copyright | 8 * * Redistributions of source code must retain the above copyright |
9 * notice, this list of conditions and the following disclaimer. | 9 * notice, this list of conditions and the following disclaimer. |
10 * * Redistributions in binary form must reproduce the above | 10 * * Redistributions in binary form must reproduce the above |
(...skipping 12 matching lines...) Expand all Loading... |
23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | 23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | 24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | 25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
29 */ | 29 */ |
30 | 30 |
31 #include "web/LocalFrameClientImpl.h" | 31 #include "web/LocalFrameClientImpl.h" |
32 | 32 |
| 33 #include "core/frame/WebLocalFrameBase.h" |
33 #include "core/loader/FrameLoader.h" | 34 #include "core/loader/FrameLoader.h" |
34 #include "platform/weborigin/KURL.h" | 35 #include "platform/weborigin/KURL.h" |
35 #include "platform/wtf/text/CString.h" | 36 #include "platform/wtf/text/CString.h" |
36 #include "platform/wtf/text/WTFString.h" | 37 #include "platform/wtf/text/WTFString.h" |
37 #include "public/web/WebFrameClient.h" | 38 #include "public/web/WebFrameClient.h" |
38 #include "public/web/WebSettings.h" | 39 #include "public/web/WebSettings.h" |
39 #include "public/web/WebView.h" | 40 #include "public/web/WebView.h" |
40 #include "testing/gmock/include/gmock/gmock.h" | 41 #include "testing/gmock/include/gmock/gmock.h" |
41 #include "testing/gtest/include/gtest/gtest.h" | 42 #include "testing/gtest/include/gtest/gtest.h" |
42 #include "web/WebLocalFrameImpl.h" | |
43 #include "web/tests/FrameTestHelpers.h" | 43 #include "web/tests/FrameTestHelpers.h" |
44 | 44 |
45 using testing::_; | 45 using testing::_; |
46 using testing::Mock; | 46 using testing::Mock; |
47 using testing::Return; | 47 using testing::Return; |
48 | 48 |
49 namespace blink { | 49 namespace blink { |
50 namespace { | 50 namespace { |
51 | 51 |
52 class MockWebFrameClient : public FrameTestHelpers::TestWebFrameClient { | 52 class MockWebFrameClient : public FrameTestHelpers::TestWebFrameClient { |
(...skipping 21 matching lines...) Expand all Loading... |
74 } | 74 } |
75 | 75 |
76 void TearDown() override { web_view_->Close(); } | 76 void TearDown() override { web_view_->Close(); } |
77 | 77 |
78 WebString UserAgent() { | 78 WebString UserAgent() { |
79 // The test always returns the same user agent . | 79 // The test always returns the same user agent . |
80 WTF::CString user_agent = GetLocalFrameClient().UserAgent().Utf8(); | 80 WTF::CString user_agent = GetLocalFrameClient().UserAgent().Utf8(); |
81 return WebString::FromUTF8(user_agent.data(), user_agent.length()); | 81 return WebString::FromUTF8(user_agent.data(), user_agent.length()); |
82 } | 82 } |
83 | 83 |
84 WebLocalFrameImpl* MainFrame() { | 84 WebLocalFrameBase* MainFrame() { |
85 return ToWebLocalFrameImpl(web_view_->MainFrame()); | 85 return ToWebLocalFrameBase(web_view_->MainFrame()); |
86 } | 86 } |
87 Document& GetDocument() { | 87 Document& GetDocument() { |
88 return *ToWebLocalFrameImpl(main_frame_)->GetFrame()->GetDocument(); | 88 return *ToWebLocalFrameBase(main_frame_)->GetFrame()->GetDocument(); |
89 } | 89 } |
90 MockWebFrameClient& WebFrameClient() { return web_frame_client_; } | 90 MockWebFrameClient& WebFrameClient() { return web_frame_client_; } |
91 LocalFrameClient& GetLocalFrameClient() { | 91 LocalFrameClient& GetLocalFrameClient() { |
92 return *ToLocalFrameClientImpl(ToWebLocalFrameImpl(web_view_->MainFrame()) | 92 return *ToLocalFrameClientImpl(ToWebLocalFrameBase(web_view_->MainFrame()) |
93 ->GetFrame() | 93 ->GetFrame() |
94 ->Loader() | 94 ->Loader() |
95 .Client()); | 95 .Client()); |
96 } | 96 } |
97 | 97 |
98 private: | 98 private: |
99 MockWebFrameClient web_frame_client_; | 99 MockWebFrameClient web_frame_client_; |
100 WebView* web_view_; | 100 WebView* web_view_; |
101 WebLocalFrame* main_frame_; | 101 WebLocalFrame* main_frame_; |
102 }; | 102 }; |
103 | 103 |
104 TEST_F(LocalFrameClientImplTest, UserAgentOverride) { | 104 TEST_F(LocalFrameClientImplTest, UserAgentOverride) { |
105 const WebString default_user_agent = UserAgent(); | 105 const WebString default_user_agent = UserAgent(); |
106 const WebString override_user_agent = WebString::FromUTF8("dummy override"); | 106 const WebString override_user_agent = WebString::FromUTF8("dummy override"); |
107 | 107 |
108 // Override the user agent and make sure we get it back. | 108 // Override the user agent and make sure we get it back. |
109 EXPECT_CALL(WebFrameClient(), UserAgentOverride()) | 109 EXPECT_CALL(WebFrameClient(), UserAgentOverride()) |
110 .WillOnce(Return(override_user_agent)); | 110 .WillOnce(Return(override_user_agent)); |
111 EXPECT_TRUE(override_user_agent.Equals(UserAgent())); | 111 EXPECT_TRUE(override_user_agent.Equals(UserAgent())); |
112 Mock::VerifyAndClearExpectations(&WebFrameClient()); | 112 Mock::VerifyAndClearExpectations(&WebFrameClient()); |
113 | 113 |
114 // Remove the override and make sure we get the original back. | 114 // Remove the override and make sure we get the original back. |
115 EXPECT_CALL(WebFrameClient(), UserAgentOverride()) | 115 EXPECT_CALL(WebFrameClient(), UserAgentOverride()) |
116 .WillOnce(Return(WebString())); | 116 .WillOnce(Return(WebString())); |
117 EXPECT_TRUE(default_user_agent.Equals(UserAgent())); | 117 EXPECT_TRUE(default_user_agent.Equals(UserAgent())); |
118 } | 118 } |
119 | 119 |
120 } // namespace | 120 } // namespace |
121 } // namespace blink | 121 } // namespace blink |
OLD | NEW |