Home | History | Annotate | Download | only in tests 
      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 "public/platform/Platform.h"
     39 #include "public/platform/WebString.h"
     40 #include "public/platform/WebURL.h"
     41 #include "public/platform/WebURLRequest.h"
     42 #include "public/platform/WebURLResponse.h"
     43 #include "public/platform/WebUnitTestSupport.h"
     44 
     45 #include <gtest/gtest.h>
     46 
     47 using namespace WebKit;
     48 using WebKit::URLTestHelpers::toKURL;
     49 
     50 namespace {
     51 
     52 class TestWebFrameClient : public WebFrameClient {
     53 };
     54 
     55 class WebPageSerializerTest : public testing::Test {
     56 public:
     57     WebPageSerializerTest() : m_webView(0), m_supportedSchemes(static_cast<size_t>(3))
     58     {
     59         m_supportedSchemes[0] = "http";
     60         m_supportedSchemes[1] = "https";
     61         m_supportedSchemes[2] = "file";
     62     }
     63 
     64 protected:
     65     virtual void SetUp()
     66     {
     67         // Create and initialize the WebView.
     68         m_webView = WebView::create(0);
     69         m_webView->initializeMainFrame(&m_webFrameClient);
     70     }
     71 
     72     virtual void TearDown()
     73     {
     74         Platform::current()->unitTestSupport()->unregisterAllMockedURLs();
     75         m_webView->close();
     76     }
     77 
     78     void registerMockedURLLoad(const std::string& url, const WebString& fileName)
     79     {
     80         URLTestHelpers::registerMockedURLLoad(toKURL(url), fileName, WebString::fromUTF8("pageserialization/"), WebString::fromUTF8("text/html"));
     81     }
     82 
     83     void loadURLInTopFrame(const WebURL& url)
     84     {
     85         WebURLRequest urlRequest;
     86         urlRequest.initialize();
     87         urlRequest.setURL(url);
     88         m_webView->mainFrame()->loadRequest(urlRequest);
     89         // Make sure any pending request get served.
     90         Platform::current()->unitTestSupport()->serveAsynchronousMockedRequests();
     91     }
     92 
     93     static bool webVectorContains(const WebVector<WebURL>& vector, const char* url)
     94     {
     95         return vector.contains(WebURL(toKURL(std::string(url))));
     96     }
     97 
     98     // Useful for debugging.
     99     static void printWebURLs(const WebVector<WebURL>& urls)
    100     {
    101         for (size_t i = 0; i < urls.size(); i++)
    102             printf("%s\n", urls[i].spec().data());
    103     }
    104 
    105     WebView* m_webView;
    106     WebVector<WebCString> m_supportedSchemes;
    107 
    108 private:
    109     TestWebFrameClient m_webFrameClient;
    110 };
    111 
    112 TEST_F(WebPageSerializerTest, HTMLNodes)
    113 {
    114     // Register the mocked frame and load it.
    115     WebURL topFrameURL = toKURL("http://www.test.com");
    116     registerMockedURLLoad("http://www.test.com", WebString::fromUTF8("simple_page.html"));
    117     registerMockedURLLoad("http://www.example.com/beautifull.css", WebString::fromUTF8("beautifull.css"));
    118     loadURLInTopFrame(topFrameURL);
    119 
    120     // Retrieve all resources.
    121     WebVector<WebURL> frames;
    122     WebVector<WebURL> resources;
    123     ASSERT_TRUE(WebPageSerializer::retrieveAllResources(
    124         m_webView, m_supportedSchemes, &resources, &frames));
    125 
    126     // Tests that all resources from the frame have been retrieved.
    127     EXPECT_EQ(1U, frames.size()); // There should be no duplicates.
    128     EXPECT_TRUE(webVectorContains(frames, "http://www.test.com"));
    129 
    130     EXPECT_EQ(14U, resources.size()); // There should be no duplicates.
    131     EXPECT_TRUE(webVectorContains(resources, "http://www.example.com/beautifull.css"));
    132     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/awesome.js"));
    133     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/bodyBackground.jpg"));
    134     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/awesome.png"));
    135     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/imageButton.png"));
    136     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/tableBackground.png"));
    137     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/trBackground.png"));
    138     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/tdBackground.png"));
    139     EXPECT_TRUE(webVectorContains(resources, "http://www.evene.fr/citations/auteur.php?ida=46"));
    140     EXPECT_TRUE(webVectorContains(resources, "http://www.brainyquote.com/quotes/authors/c/charles_darwin.html"));
    141     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/why_deleted.html"));
    142     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/why_inserted.html"));
    143     EXPECT_TRUE(webVectorContains(resources, "https://www.secure.com/https.gif"));
    144     EXPECT_TRUE(webVectorContains(resources, "file://c/my_folder/file.gif"));
    145 }
    146 
    147 TEST_F(WebPageSerializerTest, MultipleFrames)
    148 {
    149     // Register the mocked frames.
    150     WebURL topFrameURL = toKURL("http://www.test.com");
    151     registerMockedURLLoad("http://www.test.com", WebString::fromUTF8("top_frame.html"));
    152     registerMockedURLLoad("http://www.test.com/simple_iframe.html",
    153                           WebString::fromUTF8("simple_iframe.html"));
    154     registerMockedURLLoad("http://www.test.com/object_iframe.html",
    155                           WebString::fromUTF8("object_iframe.html"));
    156     registerMockedURLLoad("http://www.test.com/embed_iframe.html",
    157                           WebString::fromUTF8("embed_iframe.html"));
    158     // If we don't register a mocked resource for awesome.png, it causes the
    159     // document loader of the iframe that has it as its src to assert on close,
    160     // not sure why.
    161     registerMockedURLLoad("http://www.test.com/awesome.png",
    162                           WebString::fromUTF8("awesome.png"));
    163 
    164     loadURLInTopFrame(topFrameURL);
    165 
    166     // Retrieve all resources.
    167     WebVector<WebURL> frames;
    168     WebVector<WebURL> resources;
    169     ASSERT_TRUE(WebPageSerializer::retrieveAllResources(
    170         m_webView, m_supportedSchemes, &resources, &frames));
    171 
    172     // Tests that all resources from the frame have been retrieved.
    173     EXPECT_EQ(4U, frames.size()); // There should be no duplicates.
    174     EXPECT_TRUE(webVectorContains(frames, "http://www.test.com"));
    175     EXPECT_TRUE(webVectorContains(frames, "http://www.test.com/simple_iframe.html"));
    176     EXPECT_TRUE(webVectorContains(frames, "http://www.test.com/object_iframe.html"));
    177     EXPECT_TRUE(webVectorContains(frames, "http://www.test.com/embed_iframe.html"));
    178 
    179     EXPECT_EQ(5U, resources.size()); // There should be no duplicates.
    180     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/awesome.png"));
    181     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/innerFrame.png"));
    182     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/flash.swf"));
    183     // FIXME: for some reason the following resources is missing on one of the bot
    184     //        causing the test to fail. Probably a plugin issue.
    185     // EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/music.mid"));
    186     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/object.png"));
    187     EXPECT_TRUE(webVectorContains(resources, "http://www.test.com/embed.png"));
    188 }
    189 
    190 }
    191