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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
const { FileUtils } = ChromeUtils.import(
"resource://gre/modules/FileUtils.jsm"
);
/**
* Test that the IOInterposer is working correctly to capture main thread IO.
*
* This test should not run on release or beta, as the IOInterposer is wrapped in
* an ifdef.
*/
add_task(async () => {
if (!AppConstants.MOZ_GECKO_PROFILER) {
return;
}
{
const filename = "profiler-mainthreadio-test-firstrun";
const { markers, schema } = await runProfilerWithFileIO(
["mainthreadio"],
filename
);
info("Check the FileIO markers when using the mainthreadio feature");
checkInflatedFileIOMarkers(markers, filename);
checkSchema(schema, {
name: "FileIO",
display: ["marker-chart", "marker-table", "timeline-fileio"],
data: [
{
key: "operation",
label: "Operation",
format: "string",
searchable: true,
},
{ key: "source", label: "Source", format: "string", searchable: true },
{
key: "filename",
label: "Filename",
format: "file-path",
searchable: true,
},
],
});
}
{
const filename = "profiler-mainthreadio-test-no-instrumentation";
const { markers } = await runProfilerWithFileIO([], filename);
equal(
markers.length,
0,
"No FileIO markers are found when the mainthreadio feature is not turned on " +
"in the profiler."
);
}
{
const filename = "profiler-mainthreadio-test-secondrun";
const { markers } = await runProfilerWithFileIO(["mainthreadio"], filename);
info("Check the FileIO markers when re-starting the mainthreadio feature");
checkInflatedFileIOMarkers(markers, filename);
}
});
/**
* Start the profiler and get FileIO markers and schema.
*
* @param {Array} features The list of profiler features
* @param {string} filename A filename to trigger a write operation
* @returns {{
* markers: InflatedMarkers[];
* schema: MarkerSchema;
* }}
*/
async function runProfilerWithFileIO(features, filename) {
const entries = 10000;
const interval = 10;
const threads = [];
Services.profiler.StartProfiler(entries, interval, features, threads);
info("Get the file");
const file = FileUtils.getFile("TmpD", [filename]);
if (file.exists()) {
console.warn(
"This test is triggering FileIO by writing to a file. However, the test found an " +
"existing file at the location it was trying to write to. This could happen " +
"because a previous run of the test failed to clean up after itself. This test " +
" will now clean up that file before running the test again."
);
file.remove(false);
}
info(
"Generate file IO on the main thread using FileUtils.openSafeFileOutputStream."
);
const outputStream = FileUtils.openSafeFileOutputStream(file);
const data = "Test data.";
info("Write to the file");
outputStream.write(data, data.length);
info("Close the file");
FileUtils.closeSafeFileOutputStream(outputStream);
info("Remove the file");
file.remove(false);
// Pause the profiler as we don't need to collect more samples as we retrieve
// and serialize the profile.
Services.profiler.Pause();
const profile = await Services.profiler.getProfileDataAsync();
Services.profiler.StopProfiler();
const mainThread = profile.threads.find(({ name }) => name === "GeckoMain");
const schema = getSchema(profile, "FileIO");
const markers = getInflatedFileIOMarkers(mainThread, filename);
return { schema, markers };
}
|