fix mocking of various files in webkit_unit_tests

Source/WebKit/chromium/tests/AssociatedURLLoaderTest.cpp

 /*
  * Copyright (C) 2011 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
@@ skipping 61 matching lines @@
         :  m_willSendRequest(false)
         ,  m_didSendData(false)
         ,  m_didReceiveResponse(false)
         ,  m_didReceiveData(false)
         ,  m_didReceiveCachedMetadata(false)
         ,  m_didFinishLoading(false)
         ,  m_didFail(false)
         ,  m_runningMessageLoop(false)
     {
         // Reuse one of the test files from WebFrameTest.
-        std::string filePath = std::string(Platform::current()->unitTestSupport()->webKitRootDir().utf8().data());
-        filePath += "/Source/WebKit/chromium/tests/data/iframes_test.html";
-        m_frameFilePath = WebString::fromUTF8(filePath.c_str());
+        m_baseFilePath = Platform::current()->unitTestSupport()->webKitRootDir();
+        m_baseFilePath.append("/Source/WebKit/chromium/tests/data/");
+        m_frameFilePath = m_baseFilePath;
+        m_frameFilePath.append("iframes_test.html");
+    }
+
+    WebCore::KURL RegisterMockedUrl(const std::string& urlRoot, const WTF::String& filename)
+    {
+        WebURLResponse response;
+        response.initialize();
+        response.setMIMEType("text/html");
+        WTF::String localPath = m_baseFilePath;
+        localPath.append(filename);
+        WebCore::KURL url = toKURL(urlRoot + filename.utf8().data());
+        Platform::current()->unitTestSupport()->registerMockedURL(url, response, localPath);
+        return url;
     }
 
     void SetUp()
     {
         m_webView = WebView::create(0);
         m_webView->initializeMainFrame(&m_webFrameClient);
 
-        // Load the frame before trying to load resources.
-        WebCore::KURL url = toKURL("http://www.test.com/iframes_test.html");
-        WebURLResponse response;
-        response.initialize();
-        response.setMIMEType("text/html");
-        Platform::current()->unitTestSupport()->registerMockedURL(url, response, m_frameFilePath);
+        std::string urlRoot = "http://www.test.com/";
+        WebCore::KURL url = RegisterMockedUrl(urlRoot, "iframes_test.html");
+        const char* iframeSupportFiles[] = {
+            "invisible_iframe.html",
+            "visible_iframe.html",
+            "zero_sized_iframe.html",
+        };
+        for (size_t i = 0; i < arraysize(iframeSupportFiles); ++i) {
+            RegisterMockedUrl(urlRoot, iframeSupportFiles[i]);
+        }
 
         WebURLRequest request;
         request.initialize();
         request.setURL(url);
         m_webView->mainFrame()->loadRequest(request);
         serveRequests();
 
         Platform::current()->unitTestSupport()->unregisterMockedURL(url);
     }
 
@@ skipping 147 matching lines @@
         m_expectedLoader->loadAsynchronously(request, this);
         serveRequests();
         EXPECT_TRUE(m_didReceiveResponse);
         EXPECT_TRUE(m_didReceiveData);
         EXPECT_TRUE(m_didFinishLoading);
 
         return !m_actualResponse.httpHeaderField(headerNameString).isEmpty();
     }
 
 protected:
-    WebString m_frameFilePath;
+    WTF::String m_baseFilePath;
+    WTF::String m_frameFilePath;
     TestWebFrameClient m_webFrameClient;
     WebView* m_webView;
 
     WebURLLoader* m_expectedLoader;
     WebURLResponse m_actualResponse;
     WebURLResponse m_expectedResponse;
     WebURLRequest m_expectedNewRequest;
     WebURLResponse m_expectedRedirectResponse;
     bool m_willSendRequest;
     bool m_didSendData;
@@ skipping 337 matching lines @@
     m_expectedLoader->loadAsynchronously(request, this);
     serveRequests();
     EXPECT_TRUE(m_didReceiveResponse);
     EXPECT_TRUE(m_didReceiveData);
     EXPECT_TRUE(m_didFinishLoading);
 
     EXPECT_FALSE(m_actualResponse.httpHeaderField(headerNameString).isEmpty());
 }
 
 }