| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 "base/stringprintf.h" | 5 #include "base/stringprintf.h" |
| 6 #include "base/system_monitor/system_monitor.h" | 6 #include "base/system_monitor/system_monitor.h" |
| 7 #include "base/utf_string_conversions.h" | 7 #include "base/utf_string_conversions.h" |
| 8 #include "chrome/browser/extensions/api/media_galleries_private/media_galleries_
private_api.h" | 8 #include "chrome/browser/extensions/api/media_galleries_private/media_galleries_
private_api.h" |
| 9 #include "chrome/browser/extensions/extension_apitest.h" | 9 #include "chrome/browser/extensions/extension_apitest.h" |
| 10 #include "chrome/browser/extensions/extension_service.h" | 10 #include "chrome/browser/extensions/extension_service.h" |
| (...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 73 } | 73 } |
| 74 | 74 |
| 75 void ChangeListener(content::RenderViewHost* host, | 75 void ChangeListener(content::RenderViewHost* host, |
| 76 const std::string& js_command, | 76 const std::string& js_command, |
| 77 const std::string& ok_message) { | 77 const std::string& ok_message) { |
| 78 ExtensionTestMessageListener listener(ok_message, false /* no reply */); | 78 ExtensionTestMessageListener listener(ok_message, false /* no reply */); |
| 79 host->ExecuteJavascriptInWebFrame(string16(), ASCIIToUTF16(js_command)); | 79 host->ExecuteJavascriptInWebFrame(string16(), ASCIIToUTF16(js_command)); |
| 80 EXPECT_TRUE(listener.WaitUntilSatisfied()); | 80 EXPECT_TRUE(listener.WaitUntilSatisfied()); |
| 81 } | 81 } |
| 82 | 82 |
| 83 void Attach(bool listening) { | 83 void AttachDetach() { |
| 84 const std::string expect_attach_msg = | 84 Attach(); |
| 85 base::StringPrintf("%s,%s", kAttachTestOk, kDeviceName); | 85 Detach(); |
| 86 ExtensionTestMessageListener attach_finished_listener( | 86 } |
| 87 expect_attach_msg, | 87 |
| 88 false /* no reply */); | 88 void Attach() { |
| 89 base::SystemMonitor::Get()->ProcessRemovableStorageAttached( | 89 base::SystemMonitor::Get()->ProcessRemovableStorageAttached( |
| 90 device_id_, ASCIIToUTF16(kDeviceName), kDevicePath); | 90 device_id_, ASCIIToUTF16(kDeviceName), kDevicePath); |
| 91 WaitForDeviceEvents(); | 91 WaitForDeviceEvents(); |
| 92 if (listening) | |
| 93 EXPECT_TRUE(attach_finished_listener.WaitUntilSatisfied()); | |
| 94 } | 92 } |
| 95 | 93 |
| 96 void Detach(bool listening) { | 94 void Detach() { |
| 97 ExtensionTestMessageListener detach_finished_listener( | |
| 98 kDetachTestOk, | |
| 99 false /* no reply */); | |
| 100 base::SystemMonitor::Get()->ProcessRemovableStorageDetached(device_id_); | 95 base::SystemMonitor::Get()->ProcessRemovableStorageDetached(device_id_); |
| 101 WaitForDeviceEvents(); | 96 WaitForDeviceEvents(); |
| 102 if (listening) | |
| 103 EXPECT_TRUE(detach_finished_listener.WaitUntilSatisfied()); | |
| 104 } | 97 } |
| 105 | 98 |
| 106 private: | 99 private: |
| 107 void WaitForDeviceEvents() { | 100 void WaitForDeviceEvents() { |
| 108 content::RunAllPendingInMessageLoop(); | 101 content::RunAllPendingInMessageLoop(); |
| 109 } | 102 } |
| 110 | 103 |
| 111 std::string device_id_; | 104 std::string device_id_; |
| 112 | 105 |
| 113 DISALLOW_COPY_AND_ASSIGN(MediaGalleriesPrivateApiTest); | 106 DISALLOW_COPY_AND_ASSIGN(MediaGalleriesPrivateApiTest); |
| 114 }; | 107 }; |
| 115 | 108 |
| 116 IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateApiTest, DeviceAttachDetachEvents) { | 109 IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateApiTest, DeviceAttachDetachEvents) { |
| 117 // Setup. | 110 // Setup. |
| 118 const extensions::Extension* extension = | 111 const extensions::Extension* extension = |
| 119 LoadExtension(test_data_dir_.AppendASCII(kTestExtensionPath)); | 112 LoadExtension(test_data_dir_.AppendASCII(kTestExtensionPath)); |
| 120 ASSERT_TRUE(extension); | 113 ASSERT_TRUE(extension); |
| 121 | 114 |
| 122 content::RenderViewHost* host = | 115 content::RenderViewHost* host = |
| 123 extensions::ExtensionSystem::Get(browser()->profile())-> | 116 extensions::ExtensionSystem::Get(browser()->profile())-> |
| 124 process_manager()->GetBackgroundHostForExtension(extension->id())-> | 117 process_manager()->GetBackgroundHostForExtension(extension->id())-> |
| 125 render_view_host(); | 118 render_view_host(); |
| 126 ASSERT_TRUE(host); | 119 ASSERT_TRUE(host); |
| 127 | 120 |
| 128 // No listeners, attach and detach a couple times. | 121 // No listeners, attach and detach a couple times. |
| 129 Attach(false); | 122 AttachDetach(); |
| 130 Detach(false); | 123 AttachDetach(); |
| 131 Attach(false); | |
| 132 Detach(false); | |
| 133 | 124 |
| 134 // Add attach listener. | 125 // Add attach listener. |
| 135 ChangeListener(host, kAddAttachListenerCmd, kAddAttachListenerOk); | 126 ChangeListener(host, kAddAttachListenerCmd, kAddAttachListenerOk); |
| 136 | 127 |
| 137 // Attach / detach | 128 // Attach / detach |
| 138 Attach(true); | 129 const std::string expect_attach_msg = |
| 139 Detach(false); | 130 base::StringPrintf("%s,%s", kAttachTestOk, kDeviceName); |
| 131 ExtensionTestMessageListener attach_finished_listener(expect_attach_msg, |
| 132 false /* no reply */); |
| 133 Attach(); |
| 134 EXPECT_TRUE(attach_finished_listener.WaitUntilSatisfied()); |
| 135 Detach(); |
| 140 | 136 |
| 141 // Attach / detach | 137 // Attach / detach |
| 142 Attach(true); | 138 Attach(); |
| 143 Detach(false); | 139 EXPECT_TRUE(attach_finished_listener.WaitUntilSatisfied()); |
| 140 // Detach |
| 141 Detach(); |
| 144 | 142 |
| 145 // Remove attach listener. | 143 // Remove attach listener. |
| 146 ChangeListener(host, kRemoveAttachListenerCmd, kRemoveAttachListenerOk); | 144 ChangeListener(host, kRemoveAttachListenerCmd, kRemoveAttachListenerOk); |
| 147 | 145 |
| 148 // No listeners, attach and detach a couple times. | 146 // No listeners, attach and detach a couple times. |
| 149 Attach(false); | 147 AttachDetach(); |
| 150 Detach(false); | 148 AttachDetach(); |
| 151 Attach(false); | |
| 152 Detach(false); | |
| 153 | 149 |
| 154 // Add detach listener. | 150 // Add detach listener. |
| 155 ChangeListener(host, kAddDummyDetachListenerCmd, kAddDummyDetachListenerOk); | 151 ChangeListener(host, kAddDummyDetachListenerCmd, kAddDummyDetachListenerOk); |
| 156 | 152 |
| 157 // Attach / detach | 153 // Attach / detach |
| 158 Attach(false); | 154 Attach(); |
| 159 Detach(true); | 155 |
| 156 ExtensionTestMessageListener detach_finished_listener(kDetachTestOk, |
| 157 false /* no reply */); |
| 158 Detach(); |
| 159 EXPECT_TRUE(detach_finished_listener.WaitUntilSatisfied()); |
| 160 | 160 |
| 161 // Attach / detach | 161 // Attach / detach |
| 162 Attach(false); | 162 Attach(); |
| 163 Detach(true); | 163 Detach(); |
| 164 EXPECT_TRUE(detach_finished_listener.WaitUntilSatisfied()); |
| 164 | 165 |
| 165 // Switch ok dummy detach listener for the regular one. | 166 // Switch ok dummy detach listener for the regular one. |
| 166 ChangeListener(host, kRemoveDummyDetachListenerCmd, | 167 ChangeListener(host, kRemoveDummyDetachListenerCmd, |
| 167 kRemoveDummyDetachListenerOk); | 168 kRemoveDummyDetachListenerOk); |
| 168 ChangeListener(host, kAddDetachListenerCmd, kAddDetachListenerOk); | 169 ChangeListener(host, kAddDetachListenerCmd, kAddDetachListenerOk); |
| 169 | 170 |
| 170 // Add attach listener. | 171 // Add attach listener. |
| 171 ChangeListener(host, kAddAttachListenerCmd, kAddAttachListenerOk); | 172 ChangeListener(host, kAddAttachListenerCmd, kAddAttachListenerOk); |
| 172 | 173 |
| 173 Attach(true); | 174 Attach(); |
| 174 Detach(true); | 175 EXPECT_TRUE(attach_finished_listener.WaitUntilSatisfied()); |
| 176 Detach(); |
| 177 EXPECT_TRUE(detach_finished_listener.WaitUntilSatisfied()); |
| 175 | 178 |
| 176 Attach(true); | 179 Attach(); |
| 177 Detach(true); | 180 EXPECT_TRUE(attach_finished_listener.WaitUntilSatisfied()); |
| 178 CloseShellWindowsAndWaitForAppToExit(); | 181 Detach(); |
| 182 EXPECT_TRUE(detach_finished_listener.WaitUntilSatisfied()); |
| 179 } | 183 } |
| OLD | NEW |