1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
// 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/macros.h"
#include "base/strings/stringprintf.h"
#include "base/strings/utf_string_conversions.h"
#include "components/storage_monitor/storage_info.h"
#include "components/storage_monitor/storage_monitor.h"
#include "components/storage_monitor/test_storage_monitor.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/test/test_utils.h"
#include "extensions/browser/api/system_storage/storage_api_test_util.h"
#include "extensions/browser/api/system_storage/storage_info_provider.h"
#include "extensions/browser/extension_host.h"
#include "extensions/browser/process_manager.h"
#include "extensions/common/extension.h"
#include "extensions/shell/test/shell_apitest.h"
#include "extensions/test/extension_test_message_listener.h"
namespace {
using extensions::test::TestStorageUnitInfo;
using extensions::test::kRemovableStorageData;
using storage_monitor::StorageMonitor;
using storage_monitor::TestStorageMonitor;
} // namespace
class SystemStorageEjectApiTest : public extensions::ShellApiTest {
public:
SystemStorageEjectApiTest() : monitor_(NULL) {}
~SystemStorageEjectApiTest() override {}
protected:
void SetUpOnMainThread() override {
monitor_ = TestStorageMonitor::CreateForBrowserTests();
ShellApiTest::SetUpOnMainThread();
}
content::RenderViewHost* GetHost() {
ExtensionTestMessageListener listener("loaded", false);
const extensions::Extension* extension = LoadApp("system/storage_eject");
// Wait for the extension to load completely so we can execute
// the JavaScript function from test case.
EXPECT_TRUE(listener.WaitUntilSatisfied());
return extensions::ProcessManager::Get(browser_context())
->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->GetMainFrame()->ExecuteJavaScriptForTests(
base::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());
}
|
