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 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
62 | 62 |
63 private: | 63 private: |
64 WebString m_userAgentOverride; | 64 WebString m_userAgentOverride; |
65 }; | 65 }; |
66 | 66 |
67 class FrameLoaderClientImplTest : public testing::Test { | 67 class FrameLoaderClientImplTest : public testing::Test { |
68 public: | 68 public: |
69 void SetUp() | 69 void SetUp() |
70 { | 70 { |
71 m_webView = WebView::create(0); | 71 m_webView = WebView::create(0); |
72 m_webView->initializeMainFrame(&m_webFrameClient); | 72 m_mainFrame = WebFrame::create(&m_webFrameClient); |
| 73 m_webView->setMainFrame(m_mainFrame); |
73 m_frameLoaderClientImpl = static_cast<FrameLoaderClientImpl*>(toWebFrame
Impl(m_webView->mainFrame())->frame()->loader()->client()); | 74 m_frameLoaderClientImpl = static_cast<FrameLoaderClientImpl*>(toWebFrame
Impl(m_webView->mainFrame())->frame()->loader()->client()); |
74 } | 75 } |
75 | 76 |
76 void TearDown() | 77 void TearDown() |
77 { | 78 { |
78 m_webView->close(); | 79 m_webView->close(); |
| 80 m_mainFrame->close(); |
79 } | 81 } |
80 | 82 |
81 void setUserAgentOverride(const WebString& userAgent) | 83 void setUserAgentOverride(const WebString& userAgent) |
82 { | 84 { |
83 return m_webFrameClient.setUserAgentOverride(userAgent); | 85 return m_webFrameClient.setUserAgentOverride(userAgent); |
84 } | 86 } |
85 | 87 |
86 const WebString userAgent() | 88 const WebString userAgent() |
87 { | 89 { |
88 // The test always returns the same user agent, regardless of the URL pa
ssed in. | 90 // The test always returns the same user agent, regardless of the URL pa
ssed in. |
89 WebCore::KURL dummyURL(WebCore::ParsedURLString, "about:blank"); | 91 WebCore::KURL dummyURL(WebCore::ParsedURLString, "about:blank"); |
90 WTF::CString userAgent = m_frameLoaderClientImpl->userAgent(dummyURL).ut
f8(); | 92 WTF::CString userAgent = m_frameLoaderClientImpl->userAgent(dummyURL).ut
f8(); |
91 return WebString::fromUTF8(userAgent.data(), userAgent.length()); | 93 return WebString::fromUTF8(userAgent.data(), userAgent.length()); |
92 } | 94 } |
93 | 95 |
94 protected: | 96 protected: |
95 TestWebFrameClient m_webFrameClient; | 97 TestWebFrameClient m_webFrameClient; |
96 FrameLoaderClientImpl* m_frameLoaderClientImpl; | 98 FrameLoaderClientImpl* m_frameLoaderClientImpl; |
97 WebView* m_webView; | 99 WebView* m_webView; |
| 100 WebFrame* m_mainFrame; |
98 }; | 101 }; |
99 | 102 |
100 TEST_F(FrameLoaderClientImplTest, UserAgentOverride) | 103 TEST_F(FrameLoaderClientImplTest, UserAgentOverride) |
101 { | 104 { |
102 const WebString defaultUserAgent = userAgent(); | 105 const WebString defaultUserAgent = userAgent(); |
103 const WebString override = WebString::fromUTF8("dummy override"); | 106 const WebString override = WebString::fromUTF8("dummy override"); |
104 | 107 |
105 // Override the user agent and make sure we get it back. | 108 // Override the user agent and make sure we get it back. |
106 setUserAgentOverride(override); | 109 setUserAgentOverride(override); |
107 EXPECT_TRUE(override.equals(userAgent())); | 110 EXPECT_TRUE(override.equals(userAgent())); |
108 | 111 |
109 // Remove the override and make sure we get the original back. | 112 // Remove the override and make sure we get the original back. |
110 setUserAgentOverride(WebString()); | 113 setUserAgentOverride(WebString()); |
111 EXPECT_TRUE(defaultUserAgent.equals(userAgent())); | 114 EXPECT_TRUE(defaultUserAgent.equals(userAgent())); |
112 } | 115 } |
113 | 116 |
114 } // namespace | 117 } // namespace |
OLD | NEW |