mozilla-central/toolkit/components/extensions/test/xpcshell/test_ext_native_messaging_perf.js (file symbol)

Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";
const MAX_ROUND_TRIP_TIME_MS =
AppConstants.DEBUG || AppConstants.ASAN ? 60 : 30;
const MAX_RETRIES = 5;
const ECHO_BODY = String.raw`
import struct
import sys
stdin = getattr(sys.stdin, 'buffer', sys.stdin)
stdout = getattr(sys.stdout, 'buffer', sys.stdout)
while True:
rawlen = stdin.read(4)
if len(rawlen) == 0:
sys.exit(0)
msglen = struct.unpack('@I', rawlen)[0]
msg = stdin.read(msglen)
stdout.write(struct.pack('@I', msglen))
stdout.write(msg)
`;
const SCRIPTS = [
{
name: "echo",
description: "A native app that echoes back messages it receives",
script: ECHO_BODY.replace(/^ {2}/gm, ""),
},
];
add_task(async function setup() {
await setupHosts(SCRIPTS);
});
add_task(async function test_round_trip_perf() {
let extension = ExtensionTestUtils.loadExtension({
background() {
browser.test.onMessage.addListener(msg => {
if (msg != "run-tests") {
return;
}
let port = browser.runtime.connectNative("echo");
function next() {
port.postMessage({
Lorem: {
ipsum: {
dolor: [
"sit amet",
"consectetur adipiscing elit",
"sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.",
],
"Ut enim": [
"ad minim veniam",
"quis nostrud exercitation ullamco",
"laboris nisi ut aliquip ex ea commodo consequat.",
],
Duis: [
"aute irure dolor in reprehenderit in",
"voluptate velit esse cillum dolore eu",
"fugiat nulla pariatur.",
],
Excepteur: [
"sint occaecat cupidatat non proident",
"sunt in culpa qui officia deserunt",
"mollit anim id est laborum.",
],
},
},
});
}
const COUNT = 1000;
let now;
function finish() {
let roundTripTime = (Date.now() - now) / COUNT;
port.disconnect();
browser.test.sendMessage("result", roundTripTime);
}
let count = 0;
port.onMessage.addListener(() => {
if (count == 0) {
// Skip the first round, since it includes the time it takes
// the app to start up.
now = Date.now();
}
if (count++ <= COUNT) {
next();
} else {
finish();
}
});
next();
});
},
manifest: {
browser_specific_settings: { gecko: { id: ID } },
permissions: ["nativeMessaging"],
},
});
await extension.startup();
let roundTripTime = Infinity;
for (
let i = 0;
i < MAX_RETRIES && roundTripTime > MAX_ROUND_TRIP_TIME_MS;
i++
) {
extension.sendMessage("run-tests");
roundTripTime = await extension.awaitMessage("result");
}
await extension.unload();
Assert.lessOrEqual(
roundTripTime,
MAX_ROUND_TRIP_TIME_MS,
`Expected round trip time (${roundTripTime}ms) to be less than ${MAX_ROUND_TRIP_TIME_MS}ms`
);
});