| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/plugin_service.h" | 5 #include "content/browser/plugin_service.h" |
| 6 | 6 |
| 7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
| 8 #include "base/path_service.h" | 8 #include "base/path_service.h" |
| 9 #include "chrome/test/base/in_process_browser_test.h" | 9 #include "chrome/test/base/in_process_browser_test.h" |
| 10 #include "chrome/test/base/ui_test_utils.h" | 10 #include "chrome/test/base/ui_test_utils.h" |
| 11 #include "content/browser/browser_thread.h" | 11 #include "content/browser/browser_thread.h" |
| 12 #include "content/browser/resource_context.h" |
| 12 #include "content/common/content_switches.h" | 13 #include "content/common/content_switches.h" |
| 13 #include "testing/gmock/include/gmock/gmock.h" | 14 #include "testing/gmock/include/gmock/gmock.h" |
| 14 #include "webkit/plugins/npapi/plugin_list.h" | 15 #include "webkit/plugins/npapi/plugin_list.h" |
| 15 | 16 |
| 16 namespace { | 17 namespace { |
| 17 | 18 |
| 18 const char* kNPAPITestPluginMimeType = "application/vnd.npapi-test"; | 19 const char* kNPAPITestPluginMimeType = "application/vnd.npapi-test"; |
| 19 | 20 |
| 21 class MockResourceContext : public content::ResourceContext { |
| 22 public: |
| 23 MockResourceContext() {} |
| 24 virtual ~MockResourceContext() {} |
| 25 |
| 26 private: |
| 27 virtual void EnsureInitialized() const OVERRIDE {} |
| 28 }; |
| 29 |
| 20 // Mock up of the Client and the Listener classes that would supply the | 30 // Mock up of the Client and the Listener classes that would supply the |
| 21 // communication channel with the plugin. | 31 // communication channel with the plugin. |
| 22 class MockPluginProcessHostClient : public PluginProcessHost::Client, | 32 class MockPluginProcessHostClient : public PluginProcessHost::Client, |
| 23 public IPC::Channel::Listener { | 33 public IPC::Channel::Listener { |
| 24 public: | 34 public: |
| 25 MockPluginProcessHostClient() | 35 MockPluginProcessHostClient() |
| 26 : channel_(NULL), | 36 : channel_(NULL), |
| 27 set_plugin_info_called_(false) { | 37 set_plugin_info_called_(false) { |
| 28 } | 38 } |
| 29 | 39 |
| 30 virtual ~MockPluginProcessHostClient() { | 40 virtual ~MockPluginProcessHostClient() { |
| 31 if (channel_) | 41 if (channel_) |
| 32 BrowserThread::DeleteSoon(BrowserThread::IO, FROM_HERE, channel_); | 42 BrowserThread::DeleteSoon(BrowserThread::IO, FROM_HERE, channel_); |
| 33 } | 43 } |
| 34 | 44 |
| 35 // Client implementation. | 45 // Client implementation. |
| 36 int ID() { return 42; } | 46 virtual int ID() OVERRIDE { return 42; } |
| 37 bool OffTheRecord() { return false; } | 47 virtual bool OffTheRecord() OVERRIDE { return false; } |
| 48 virtual const content::ResourceContext* GetResourceContext() OVERRIDE { |
| 49 return &context_; |
| 50 } |
| 38 | 51 |
| 39 void OnChannelOpened(const IPC::ChannelHandle& handle) { | 52 void OnChannelOpened(const IPC::ChannelHandle& handle) { |
| 40 ASSERT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 53 ASSERT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| 41 ASSERT_TRUE(set_plugin_info_called_); | 54 ASSERT_TRUE(set_plugin_info_called_); |
| 42 ASSERT_TRUE(!channel_); | 55 ASSERT_TRUE(!channel_); |
| 43 channel_ = new IPC::Channel(handle, IPC::Channel::MODE_CLIENT, this); | 56 channel_ = new IPC::Channel(handle, IPC::Channel::MODE_CLIENT, this); |
| 44 ASSERT_TRUE(channel_->Connect()); | 57 ASSERT_TRUE(channel_->Connect()); |
| 45 } | 58 } |
| 46 | 59 |
| 47 void SetPluginInfo(const webkit::npapi::WebPluginInfo& info) { | 60 void SetPluginInfo(const webkit::npapi::WebPluginInfo& info) { |
| 48 ASSERT_TRUE(info.mime_types.size()); | 61 ASSERT_TRUE(info.mime_types.size()); |
| 49 ASSERT_EQ(kNPAPITestPluginMimeType, info.mime_types[0].mime_type); | 62 ASSERT_EQ(kNPAPITestPluginMimeType, info.mime_types[0].mime_type); |
| 50 set_plugin_info_called_ = true; | 63 set_plugin_info_called_ = true; |
| 51 } | 64 } |
| 52 | 65 |
| 53 MOCK_METHOD0(OnError, void()); | 66 MOCK_METHOD0(OnError, void()); |
| 54 | 67 |
| 55 // Listener implementation. | 68 // Listener implementation. |
| 56 MOCK_METHOD1(OnMessageReceived, bool(const IPC::Message& message)); | 69 MOCK_METHOD1(OnMessageReceived, bool(const IPC::Message& message)); |
| 57 void OnChannelConnected(int32 peer_pid) { | 70 void OnChannelConnected(int32 peer_pid) { |
| 58 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, | 71 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
| 59 new MessageLoop::QuitTask()); | 72 new MessageLoop::QuitTask()); |
| 60 } | 73 } |
| 61 MOCK_METHOD0(OnChannelError, void()); | 74 MOCK_METHOD0(OnChannelError, void()); |
| 62 MOCK_METHOD0(OnChannelDenied, void()); | 75 MOCK_METHOD0(OnChannelDenied, void()); |
| 63 MOCK_METHOD0(OnChannelListenError, void()); | 76 MOCK_METHOD0(OnChannelListenError, void()); |
| 64 | 77 |
| 65 private: | 78 private: |
| 79 MockResourceContext context_; |
| 66 IPC::Channel* channel_; | 80 IPC::Channel* channel_; |
| 67 bool set_plugin_info_called_; | 81 bool set_plugin_info_called_; |
| 68 DISALLOW_COPY_AND_ASSIGN(MockPluginProcessHostClient); | 82 DISALLOW_COPY_AND_ASSIGN(MockPluginProcessHostClient); |
| 69 }; | 83 }; |
| 70 | 84 |
| 71 } // namespace | 85 } // namespace |
| 72 | 86 |
| 73 class PluginServiceTest : public InProcessBrowserTest { | 87 class PluginServiceTest : public InProcessBrowserTest { |
| 74 public: | 88 public: |
| 75 PluginServiceTest() : InProcessBrowserTest() { } | 89 PluginServiceTest() : InProcessBrowserTest() { } |
| 76 | 90 |
| 77 virtual void SetUpCommandLine(CommandLine* command_line) { | 91 virtual void SetUpCommandLine(CommandLine* command_line) { |
| 78 #ifdef OS_MACOSX | 92 #ifdef OS_MACOSX |
| 79 FilePath browser_directory; | 93 FilePath browser_directory; |
| 80 PathService::Get(base::DIR_MODULE, &browser_directory); | 94 PathService::Get(base::DIR_MODULE, &browser_directory); |
| 81 command_line->AppendSwitchPath(switches::kExtraPluginDir, | 95 command_line->AppendSwitchPath(switches::kExtraPluginDir, |
| 82 browser_directory.AppendASCII("plugins")); | 96 browser_directory.AppendASCII("plugins")); |
| 83 #endif | 97 #endif |
| 84 } | 98 } |
| 99 |
| 100 static void OpenChannelOnIOThread(PluginProcessHost::Client* client) { |
| 101 PluginService::GetInstance()->OpenChannelToNpapiPlugin( |
| 102 0, 0, GURL(), GURL(), kNPAPITestPluginMimeType, client); |
| 103 } |
| 85 }; | 104 }; |
| 86 | 105 |
| 87 // Try to open a channel to the test plugin. Minimal plugin process spawning | 106 // Try to open a channel to the test plugin. Minimal plugin process spawning |
| 88 // test for the PluginService interface. | 107 // test for the PluginService interface. |
| 89 IN_PROC_BROWSER_TEST_F(PluginServiceTest, OpenChannelToPlugin) { | 108 IN_PROC_BROWSER_TEST_F(PluginServiceTest, OpenChannelToPlugin) { |
| 90 MockPluginProcessHostClient mock_client; | 109 ::testing::StrictMock<MockPluginProcessHostClient> mock_client; |
| 91 PluginService::GetInstance()->OpenChannelToNpapiPlugin( | 110 BrowserThread::PostTask( |
| 92 0, 0, GURL(), kNPAPITestPluginMimeType, &mock_client); | 111 BrowserThread::IO, FROM_HERE, |
| 112 NewRunnableFunction(&PluginServiceTest::OpenChannelOnIOThread, |
| 113 &mock_client)); |
| 93 ui_test_utils::RunMessageLoop(); | 114 ui_test_utils::RunMessageLoop(); |
| 94 } | 115 } |
| OLD | NEW |