| // Copyright 2013 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| // |
| // SystemStorage eject API browser tests. |
| |
| #include "base/files/file_path.h" |
| #include "base/strings/stringprintf.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "chrome/browser/extensions/api/system_storage/storage_api_test_util.h" |
| #include "chrome/browser/extensions/api/system_storage/storage_info_provider.h" |
| #include "chrome/browser/extensions/extension_apitest.h" |
| #include "chrome/browser/extensions/extension_system.h" |
| #include "chrome/browser/extensions/extension_test_message_listener.h" |
| #include "chrome/browser/storage_monitor/storage_info.h" |
| #include "chrome/browser/storage_monitor/storage_monitor.h" |
| #include "chrome/browser/storage_monitor/test_storage_monitor.h" |
| #include "content/public/browser/render_view_host.h" |
| #include "content/public/test/test_utils.h" |
| #include "extensions/browser/process_manager.h" |
| #include "extensions/common/extension.h" |
| |
| namespace { |
| |
| using extensions::test::TestStorageUnitInfo; |
| using extensions::test::kRemovableStorageData; |
| |
| } // namespace |
| |
| class SystemStorageEjectApiTest : public ExtensionApiTest { |
| public: |
| SystemStorageEjectApiTest() : monitor_(NULL) {} |
| virtual ~SystemStorageEjectApiTest() {} |
| |
| protected: |
| virtual void SetUpOnMainThread() OVERRIDE { |
| monitor_ = TestStorageMonitor::CreateForBrowserTests(); |
| ExtensionApiTest::SetUpOnMainThread(); |
| } |
| |
| content::RenderViewHost* GetHost() { |
| const extensions::Extension* extension = |
| LoadExtension(test_data_dir_.AppendASCII("system/storage_eject")); |
| return extensions::ExtensionSystem::Get(browser()->profile())-> |
| process_manager()->GetBackgroundHostForExtension(extension->id())-> |
| render_view_host(); |
| } |
| |
| void ExecuteCmdAndCheckReply(content::RenderViewHost* host, |
| const std::string& js_command, |
| const std::string& ok_message) { |
| ExtensionTestMessageListener listener(ok_message, false); |
| host->ExecuteJavascriptInWebFrame(base::string16(), |
| ASCIIToUTF16(js_command)); |
| EXPECT_TRUE(listener.WaitUntilSatisfied()); |
| } |
| |
| void Attach() { |
| DCHECK(StorageMonitor::GetInstance()->IsInitialized()); |
| StorageMonitor::GetInstance()->receiver()->ProcessAttach( |
| extensions::test::BuildStorageInfoFromTestStorageUnitInfo( |
| kRemovableStorageData)); |
| content::RunAllPendingInMessageLoop(); |
| } |
| |
| void Detach() { |
| DCHECK(StorageMonitor::GetInstance()->IsInitialized()); |
| StorageMonitor::GetInstance()->receiver()->ProcessDetach( |
| kRemovableStorageData.device_id); |
| content::RunAllPendingInMessageLoop(); |
| } |
| |
| protected: |
| TestStorageMonitor* monitor_; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(SystemStorageEjectApiTest); |
| }; |
| |
| |
| IN_PROC_BROWSER_TEST_F(SystemStorageEjectApiTest, EjectTest) { |
| content::RenderViewHost* host = GetHost(); |
| ExecuteCmdAndCheckReply(host, "addAttachListener()", "add_attach_ok"); |
| |
| // Attach / detach |
| const std::string expect_attach_msg = |
| base::StringPrintf("%s,%s", "attach_test_ok", |
| kRemovableStorageData.name); |
| ExtensionTestMessageListener attach_finished_listener(expect_attach_msg, |
| false /* no reply */); |
| Attach(); |
| EXPECT_TRUE(attach_finished_listener.WaitUntilSatisfied()); |
| |
| ExecuteCmdAndCheckReply(host, "ejectTest()", "eject_ok"); |
| EXPECT_EQ(kRemovableStorageData.device_id, monitor_->ejected_device()); |
| |
| Detach(); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(SystemStorageEjectApiTest, EjectBadDeviceTest) { |
| ExecuteCmdAndCheckReply(GetHost(), "ejectFailTest()", "eject_no_such_device"); |
| |
| EXPECT_EQ("", monitor_->ejected_device()); |
| } |