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