OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (C) 2011 Google Inc. All rights reserved. | |
3 * | |
4 * Redistribution and use in source and binary forms, with or without | |
5 * modification, are permitted provided that the following conditions are | |
6 * met: | |
7 * | |
8 * * Redistributions of source code must retain the above copyright | |
9 * notice, this list of conditions and the following disclaimer. | |
10 * * Redistributions in binary form must reproduce the above | |
11 * copyright notice, this list of conditions and the following disclaimer | |
12 * in the documentation and/or other materials provided with the | |
13 * distribution. | |
14 * * Neither the name of Google Inc. nor the names of its | |
15 * contributors may be used to endorse or promote products derived from | |
16 * this software without specific prior written permission. | |
17 * | |
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | |
21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | |
22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | |
24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | |
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | |
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
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. | |
29 */ | |
30 | |
31 #include "config.h" | |
32 #include "WebPageSerializer.h" | |
33 | |
34 #include "URLTestHelpers.h" | |
35 #include "WebFrame.h" | |
36 #include "WebFrameClient.h" | |
37 #include "WebView.h" | |
38 #include "core/dom/Document.h" | |
39 #include "public/platform/Platform.h" | |
40 #include "public/platform/WebString.h" | |
41 #include "public/platform/WebURL.h" | |
42 #include "public/platform/WebURLRequest.h" | |
43 #include "public/platform/WebURLResponse.h" | |
44 #include "public/platform/WebUnitTestSupport.h" | |
45 #include "public/web/WebDocument.h" | |
46 | |
47 #include <gtest/gtest.h> | |
48 | |
49 using namespace blink; | |
50 using WebCore::Document; | |
51 using blink::URLTestHelpers::toKURL; | |
52 | |
53 namespace { | |
54 | |
55 class TestWebFrameClient : public WebFrameClient { | |
56 }; | |
57 | |
58 class WebPageSerializerTest : public testing::Test { | |
59 public: | |
60 WebPageSerializerTest() : m_webView(0), m_supportedSchemes(static_cast<size_
t>(3)) | |
61 { | |
62 m_supportedSchemes[0] = "http"; | |
63 m_supportedSchemes[1] = "https"; | |
64 m_supportedSchemes[2] = "file"; | |
65 } | |
66 | |
67 protected: | |
68 virtual void SetUp() | |
69 { | |
70 // Create and initialize the WebView. | |
71 m_webView = WebView::create(0); | |
72 m_mainFrame = WebFrame::create(&m_webFrameClient); | |
73 m_webView->setMainFrame(m_mainFrame); | |
74 } | |
75 | |
76 virtual void TearDown() | |
77 { | |
78 Platform::current()->unitTestSupport()->unregisterAllMockedURLs(); | |
79 m_webView->close(); | |
80 m_mainFrame->close(); | |
81 } | |
82 | |
83 void registerMockedURLLoad(const std::string& url, const WebString& fileName
) | |
84 { | |
85 URLTestHelpers::registerMockedURLLoad(toKURL(url), fileName, WebString::
fromUTF8("pageserialization/"), WebString::fromUTF8("text/html")); | |
86 } | |
87 | |
88 void loadURLInTopFrame(const WebURL& url) | |
89 { | |
90 WebURLRequest urlRequest; | |
91 urlRequest.initialize(); | |
92 urlRequest.setURL(url); | |
93 m_webView->mainFrame()->loadRequest(urlRequest); | |
94 // Make sure any pending request get served. | |
95 Platform::current()->unitTestSupport()->serveAsynchronousMockedRequests(
); | |
96 } | |
97 | |
98 static bool webVectorContains(const WebVector<WebURL>& vector, const char* u
rl) | |
99 { | |
100 return vector.contains(WebURL(toKURL(std::string(url)))); | |
101 } | |
102 | |
103 // Useful for debugging. | |
104 static void printWebURLs(const WebVector<WebURL>& urls) | |
105 { | |
106 for (size_t i = 0; i < urls.size(); i++) | |
107 printf("%s\n", urls[i].spec().data()); | |
108 } | |
109 | |
110 WebView* m_webView; | |
111 WebVector<WebCString> m_supportedSchemes; | |
112 | |
113 private: | |
114 TestWebFrameClient m_webFrameClient; | |
115 WebFrame* m_mainFrame; | |
116 }; | |
117 | |
118 TEST_F(WebPageSerializerTest, HTMLNodes) | |
119 { | |
120 // Register the mocked frame and load it. | |
121 WebURL topFrameURL = toKURL("http://www.test.com"); | |
122 registerMockedURLLoad("http://www.test.com", WebString::fromUTF8("simple_pag
e.html")); | |
123 registerMockedURLLoad("http://www.example.com/beautifull.css", WebString::fr
omUTF8("beautifull.css")); | |
124 loadURLInTopFrame(topFrameURL); | |
125 | |
126 // Retrieve all resources. | |
127 WebVector<WebURL> frames; | |
128 WebVector<WebURL> resources; | |
129 ASSERT_TRUE(WebPageSerializer::retrieveAllResources( | |
130 m_webView, m_supportedSchemes, &resources, &frames)); | |
131 | |
132 // Tests that all resources from the frame have been retrieved. | |
133 EXPECT_EQ(1U, frames.size()); // There should be no duplicates. | |
134 EXPECT_TRUE(webVectorContains(frames, "http://www.test.com")); | |
135 | |
136 EXPECT_EQ(14U, resources.size()); // There should be no duplicates. | |
137 EXPECT_TRUE(webVectorContains(resources, "http://www.example.com/beautifull.
css")); | |
138 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/awesome.js")); | |
139 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/bodyBackground
.jpg")); | |
140 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/awesome.png"))
; | |
141 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/imageButton.pn
g")); | |
142 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/tableBackgroun
d.png")); | |
143 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/trBackground.p
ng")); | |
144 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/tdBackground.p
ng")); | |
145 EXPECT_TRUE(webVectorContains(resources, "http://www.evene.fr/citations/aute
ur.php?ida=46")); | |
146 EXPECT_TRUE(webVectorContains(resources, "http://www.brainyquote.com/quotes/
authors/c/charles_darwin.html")); | |
147 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/why_deleted.ht
ml")); | |
148 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/why_inserted.h
tml")); | |
149 EXPECT_TRUE(webVectorContains(resources, "https://www.secure.com/https.gif")
); | |
150 EXPECT_TRUE(webVectorContains(resources, "file://c/my_folder/file.gif")); | |
151 } | |
152 | |
153 TEST_F(WebPageSerializerTest, MultipleFrames) | |
154 { | |
155 // Register the mocked frames. | |
156 WebURL topFrameURL = toKURL("http://www.test.com"); | |
157 registerMockedURLLoad("http://www.test.com", WebString::fromUTF8("top_frame.
html")); | |
158 registerMockedURLLoad("http://www.test.com/simple_iframe.html", | |
159 WebString::fromUTF8("simple_iframe.html")); | |
160 registerMockedURLLoad("http://www.test.com/object_iframe.html", | |
161 WebString::fromUTF8("object_iframe.html")); | |
162 registerMockedURLLoad("http://www.test.com/embed_iframe.html", | |
163 WebString::fromUTF8("embed_iframe.html")); | |
164 // If we don't register a mocked resource for awesome.png, it causes the | |
165 // document loader of the iframe that has it as its src to assert on close, | |
166 // not sure why. | |
167 registerMockedURLLoad("http://www.test.com/awesome.png", | |
168 WebString::fromUTF8("awesome.png")); | |
169 | |
170 loadURLInTopFrame(topFrameURL); | |
171 // OBJECT/EMBED have some delay to start to load their content. The first | |
172 // serveAsynchronousMockedRequests call in loadURLInTopFrame() finishes | |
173 // before the start. | |
174 RefPtr<Document> document = static_cast<PassRefPtr<Document> >(m_webView->ma
inFrame()->document()); | |
175 document->updateLayoutIgnorePendingStylesheets(Document::RunPostLayoutTasksS
ynchronously); | |
176 Platform::current()->unitTestSupport()->serveAsynchronousMockedRequests(); | |
177 | |
178 // Retrieve all resources. | |
179 WebVector<WebURL> frames; | |
180 WebVector<WebURL> resources; | |
181 ASSERT_TRUE(WebPageSerializer::retrieveAllResources( | |
182 m_webView, m_supportedSchemes, &resources, &frames)); | |
183 | |
184 // Tests that all resources from the frame have been retrieved. | |
185 EXPECT_EQ(4U, frames.size()); // There should be no duplicates. | |
186 EXPECT_TRUE(webVectorContains(frames, "http://www.test.com")); | |
187 EXPECT_TRUE(webVectorContains(frames, "http://www.test.com/simple_iframe.htm
l")); | |
188 EXPECT_TRUE(webVectorContains(frames, "http://www.test.com/object_iframe.htm
l")); | |
189 EXPECT_TRUE(webVectorContains(frames, "http://www.test.com/embed_iframe.html
")); | |
190 | |
191 EXPECT_EQ(5U, resources.size()); // There should be no duplicates. | |
192 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/awesome.png"))
; | |
193 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/innerFrame.png
")); | |
194 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/flash.swf")); | |
195 // FIXME: for some reason the following resources is missing on one of the b
ot | |
196 // causing the test to fail. Probably a plugin issue. | |
197 // EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/music.mid")
); | |
198 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/object.png")); | |
199 EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/embed.png")); | |
200 } | |
201 | |
202 } | |
OLD | NEW |