Make the embedder responsible for creating the WebFrame

Source/web/tests/FrameTestHelpers.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 30 matching lines @@
 #include "public/platform/Platform.h"
 #include "public/platform/WebString.h"
 #include "public/platform/WebThread.h"
 #include "public/platform/WebURLRequest.h"
 #include "public/platform/WebURLResponse.h"
 #include "public/platform/WebUnitTestSupport.h"
 
 namespace WebKit {
 namespace FrameTestHelpers {
 
+namespace {
+
+class QuitTask : public WebThread::Task {
+public:
+    virtual void run()
+    {
+        Platform::current()->currentThread()->exitRunLoop();
+    }
+};
+
+class TestWebFrameClient : public WebFrameClient {
+};
+
+WebFrameClient* defaultWebFrameClient()
+{
+    DEFINE_STATIC_LOCAL(TestWebFrameClient, client, ());
+    return &client;
+}
+
+class WebViewHelperClient : public WebViewClient {
+};
+
+WebViewClient* defaultWebViewClient()
+{
+    DEFINE_STATIC_LOCAL(WebViewHelperClient,  client, ());
+    return &client;
+}
+
+}  // namespace
+
 void loadFrame(WebFrame* frame, const std::string& url)
 {
     WebURLRequest urlRequest;
     urlRequest.initialize();
     urlRequest.setURL(URLTestHelpers::toKURL(url));
     frame->loadRequest(urlRequest);
 }
 
-class TestWebFrameClient : public WebFrameClient {
-};
-
-static WebFrameClient* defaultWebFrameClient()
-{
-    DEFINE_STATIC_LOCAL(TestWebFrameClient, client, ());
-    return &client;
-}
-
-class TestWebViewClient : public WebViewClient {
-};
-
-static WebViewClient* defaultWebViewClient()
-{
-    DEFINE_STATIC_LOCAL(TestWebViewClient,  client, ());
-    return &client;
-}
-
-WebView* createWebView(bool enableJavascript, WebFrameClient* webFrameClient, WebViewClient* webViewClient)
-{
-    if (!webFrameClient)
-        webFrameClient = defaultWebFrameClient();
-    if (!webViewClient)
-        webViewClient = defaultWebViewClient();
-    WebView* webView = WebView::create(webViewClient);
-    webView->settings()->setJavaScriptEnabled(enableJavascript);
-    webView->settings()->setDeviceSupportsMouse(false);
-    webView->settings()->setForceCompositingMode(true);
-    webView->initializeMainFrame(webFrameClient);
-
-    return webView;
-}
-
-WebView* createWebViewAndLoad(const std::string& url, bool enableJavascript, WebFrameClient* webFrameClient, WebViewClient* webViewClient)
-{
-    WebView* webView = createWebView(enableJavascript, webFrameClient, webViewClient);
-
-    loadFrame(webView->mainFrame(), url);
-    Platform::current()->unitTestSupport()->serveAsynchronousMockedRequests();
-
-    return webView;
-}
-
-class QuitTask : public WebThread::Task {
-public:
-    virtual void run()
-    {
-        Platform::current()->currentThread()->exitRunLoop();
-    }
-};
-
 void runPendingTasks()
 {
     Platform::current()->currentThread()->postTask(new QuitTask);
     Platform::current()->currentThread()->enterRunLoop();
 }
 
+WebViewHelper::WebViewHelper()
+    : m_mainFrame(0)
+    , m_webView(0)
+{
+}
+
+WebViewHelper::~WebViewHelper()
+{
+    reset();
+}
+
+WebView* WebViewHelper::initialize(bool enableJavascript, WebFrameClient* webFrameClient, WebViewClient* webViewClient)
+{
+    ASSERT(!m_webView);
+    ASSERT(!m_mainFrame);
+
+    if (!webFrameClient)
+        webFrameClient = defaultWebFrameClient();
+    if (!webViewClient)
+        webViewClient = defaultWebViewClient();
+    m_mainFrame = WebFrame::create(webFrameClient);

[darin (slow to review), 2013/09/13 04:02:47]

nit: allocate m_mainFrame closer to where it is first used?

[awong, 2013/09/18 22:04:47]

Done.

+    m_webView = WebView::create(webViewClient);
+    m_webView->settings()->setJavaScriptEnabled(enableJavascript);
+    m_webView->settings()->setDeviceSupportsMouse(false);
+    m_webView->settings()->setForceCompositingMode(true);
+    m_webView->initializeMainFrame(m_mainFrame);
+
+    return m_webView;
+}
+
+WebView* WebViewHelper::initializeAndLoad(const std::string& url, bool enableJavascript, WebFrameClient* webFrameClient, WebViewClient* webViewClient)
+{
+    ASSERT(!m_webView);
+    ASSERT(!m_mainFrame);
+
+    initialize(enableJavascript, webFrameClient, webViewClient);
+
+    loadFrame(webView()->mainFrame(), url);
+    Platform::current()->unitTestSupport()->serveAsynchronousMockedRequests();
+
+    return webView();
+}
+
+void WebViewHelper::reset() {
+    if (m_webView)
+        m_webView->close();
+    if (m_mainFrame)
+        m_mainFrame->close();
+}
+
 } // namespace FrameTestHelpers
 } // namespace WebKit