third_party/WebKit/Source/web/tests/WebHelperPluginTest.cpp

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "config.h"
   6 #include "public/web/WebHelperPlugin.h"
   7
   8 #include "platform/testing/UnitTestHelpers.h"
   9 #include "public/web/WebFrameClient.h"
  10 #include "public/web/WebLocalFrame.h"
  11 #include "web/tests/FakeWebPlugin.h"
  12 #include "web/tests/FrameTestHelpers.h"
  13 #include <gtest/gtest.h>
  14
  15 namespace blink {
  16
  17 namespace {
  18
  19 class FakePlaceholderWebPlugin : public FakeWebPlugin {
  20 public:
  21     FakePlaceholderWebPlugin(WebFrame* frame, const WebPluginParams& params)
  22         : FakeWebPlugin(frame, params)
  23     {
  24     }
  25     ~FakePlaceholderWebPlugin() override {}
  26
  27     bool isPlaceholder() override { return true; }
  28 };
  29
  30 class WebHelperPluginFrameClient : public FrameTestHelpers::TestWebFrameClient {
  31 public:
  32     WebHelperPluginFrameClient() : m_createPlaceholder(false) {}
  33     ~WebHelperPluginFrameClient() override {}
  34
  35     WebPlugin* createPlugin(WebLocalFrame* frame, const WebPluginParams& params) override
  36     {
  37         return m_createPlaceholder ? new FakePlaceholderWebPlugin(frame, params) : new FakeWebPlugin(frame, params);
  38     }
  39
  40     void setCreatePlaceholder(bool createPlaceholder) { m_createPlaceholder = createPlaceholder; }
  41
  42 private:
  43     bool m_createPlaceholder;
  44 };
  45
  46 class WebHelperPluginTest : public ::testing::Test {
  47 protected:
  48     void SetUp() override
  49     {
  50         m_helper.initializeAndLoad("about:blank", false, &m_frameClient);
  51     }
  52
  53
  54     void destroyHelperPlugin()
  55     {
  56         m_plugin.clear();
  57         // WebHelperPlugin is destroyed by a task posted to the message loop.
  58         testing::runPendingTasks();
  59     }
  60
  61     FrameTestHelpers::WebViewHelper m_helper;
  62     WebHelperPluginFrameClient m_frameClient;
  63     OwnPtr<WebHelperPlugin> m_plugin;
  64 };
  65
  66 TEST_F(WebHelperPluginTest, CreateAndDestroyAfterWebViewDestruction)
  67 {
  68     m_plugin = adoptPtr(WebHelperPlugin::create("hello", m_helper.webView()->mainFrame()->toWebLocalFrame()));
  69     EXPECT_TRUE(m_plugin);
  70     EXPECT_TRUE(m_plugin->getPlugin());
  71
  72     m_helper.reset();
  73     destroyHelperPlugin();
  74 }
  75
  76 TEST_F(WebHelperPluginTest, CreateAndDestroyBeforeWebViewDestruction)
  77 {
  78     m_plugin = adoptPtr(WebHelperPlugin::create("hello", m_helper.webView()->mainFrame()->toWebLocalFrame()));
  79     EXPECT_TRUE(m_plugin);
  80     EXPECT_TRUE(m_plugin->getPlugin());
  81
  82     destroyHelperPlugin();
  83     m_helper.reset();
  84 }
  85
  86 TEST_F(WebHelperPluginTest, CreateFailsWithPlaceholder)
  87 {
  88     m_frameClient.setCreatePlaceholder(true);
  89
  90     m_plugin = adoptPtr(WebHelperPlugin::create("hello", m_helper.webView()->mainFrame()->toWebLocalFrame()));
  91     EXPECT_EQ(0, m_plugin.get());
  92 }
  93
  94 } // namespace
  95
  96 } // namespace