1 // Copyright (c) 2011 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 "chrome/browser/plugin_data_remover.h"
6
7 #include "base/command_line.h"
8 #include "base/path_service.h"
9 #include "base/synchronization/waitable_event_watcher.h"
10 #include "chrome/browser/ui/browser.h"
11 #include "chrome/browser/browser_process.h"
12 #include "chrome/browser/prefs/pref_service.h"
13 #include "chrome/browser/profiles/profile.h"
14 #include "chrome/common/chrome_paths.h"
15 #include "chrome/common/chrome_switches.h"
16 #include "chrome/common/pref_names.h"
17 #include "chrome/test/in_process_browser_test.h"
18 #include "chrome/test/ui_test_utils.h"
19
20 namespace {
21 const char* kNPAPITestPluginMimeType = "application/vnd.npapi-test";
22 }
23
24 class PluginDataRemoverTest : public InProcessBrowserTest,
25 public base::WaitableEventWatcher::Delegate {
26 public:
PluginDataRemoverTest()27 PluginDataRemoverTest() : InProcessBrowserTest() { }
28
SetUpOnMainThread()29 virtual void SetUpOnMainThread() {
30 old_plugin_data_remover_mime_type_ =
31 g_browser_process->plugin_data_remover_mime_type();
32 g_browser_process->set_plugin_data_remover_mime_type(
33 kNPAPITestPluginMimeType);
34 }
35
TearDownOnMainThread()36 virtual void TearDownOnMainThread() {
37 g_browser_process->set_plugin_data_remover_mime_type(
38 old_plugin_data_remover_mime_type_);
39 }
40
OnWaitableEventSignaled(base::WaitableEvent * waitable_event)41 virtual void OnWaitableEventSignaled(base::WaitableEvent* waitable_event) {
42 MessageLoop::current()->Quit();
43 }
44
SetUpCommandLine(CommandLine * command_line)45 virtual void SetUpCommandLine(CommandLine* command_line) {
46 #ifdef OS_MACOSX
47 FilePath browser_directory;
48 PathService::Get(chrome::DIR_APP, &browser_directory);
49 command_line->AppendSwitchPath(switches::kExtraPluginDir,
50 browser_directory.AppendASCII("plugins"));
51 #endif
52 }
53
54 private:
55 std::string old_plugin_data_remover_mime_type_;
56 };
57
IN_PROC_BROWSER_TEST_F(PluginDataRemoverTest,RemoveData)58 IN_PROC_BROWSER_TEST_F(PluginDataRemoverTest, RemoveData) {
59 scoped_refptr<PluginDataRemover> plugin_data_remover(new PluginDataRemover());
60 plugin_data_remover->set_mime_type(kNPAPITestPluginMimeType);
61 base::WaitableEventWatcher watcher;
62 base::WaitableEvent* event =
63 plugin_data_remover->StartRemoving(base::Time());
64 watcher.StartWatching(event, this);
65 ui_test_utils::RunMessageLoop();
66 }
67
IN_PROC_BROWSER_TEST_F(PluginDataRemoverTest,AtShutdown)68 IN_PROC_BROWSER_TEST_F(PluginDataRemoverTest, AtShutdown) {
69 browser()->profile()->GetPrefs()->SetBoolean(
70 prefs::kClearSiteDataOnExit, true);
71 g_browser_process->local_state()->SetBoolean(
72 prefs::kClearPluginLSODataEnabled, true);
73 }
74