content/browser/plugin_data_remover_impl_browsertest.cc

   1 // Copyright (c) 2012 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 "base/base_paths.h"
   6 #include "base/callback.h"
   7 #include "base/command_line.h"
   8 #include "base/path_service.h"
   9 #include "base/synchronization/waitable_event_watcher.h"
  10 #include "content/browser/plugin_data_remover_impl.h"
  11 #include "content/public/browser/web_contents.h"
  12 #include "content/public/common/content_switches.h"
  13 #include "content/public/test/content_browser_test.h"
  14 #include "content/public/test/test_utils.h"
  15 #include "content/shell/browser/shell.h"
  16
  17 namespace content {
  18
  19 namespace {
  20 const char* kNPAPITestPluginMimeType = "application/vnd.npapi-test";
  21 }
  22
  23 class PluginDataRemoverTest : public ContentBrowserTest {
  24  public:
  25   PluginDataRemoverTest() {}
  26
  27   void OnWaitableEventSignaled(base::WaitableEvent* waitable_event) {
  28     base::MessageLoop::current()->Quit();
  29   }
  30
  31   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
  32 #if defined(OS_MACOSX)
  33     base::FilePath browser_directory;
  34     PathService::Get(base::DIR_MODULE, &browser_directory);
  35     command_line->AppendSwitchPath(switches::kExtraPluginDir,
  36                                    browser_directory.AppendASCII("plugins"));
  37 #endif
  38     // TODO(jam): since these plugin tests are running under Chrome, we need to
  39     // tell it to disable its security features for old plugins. Once this is
  40     // running under content_browsertests, these flags won't be needed.
  41     // http://crbug.com/90448
  42     // switches::kAlwaysAuthorizePlugins
  43     command_line->AppendSwitch("always-authorize-plugins");
  44   }
  45 };
  46
  47 IN_PROC_BROWSER_TEST_F(PluginDataRemoverTest, RemoveData) {
  48   PluginDataRemoverImpl plugin_data_remover(
  49       shell()->web_contents()->GetBrowserContext());
  50   plugin_data_remover.set_mime_type(kNPAPITestPluginMimeType);
  51   base::WaitableEventWatcher watcher;
  52   base::WaitableEvent* event =
  53       plugin_data_remover.StartRemoving(base::Time());
  54   watcher.StartWatching(
  55       event,
  56       base::Bind(&PluginDataRemoverTest::OnWaitableEventSignaled, this));
  57   RunMessageLoop();
  58 }
  59
  60 }  // namespace content