|
| 1 | +import { TestRunner } from "./test-runner.mjs"; |
| 2 | +import { Params } from "./params.mjs"; |
| 3 | + |
| 4 | +/** |
| 5 | + * BenchmarkStep |
| 6 | + * |
| 7 | + * A single test step, with a common interface to interact with. |
| 8 | + */ |
| 9 | +export class BenchmarkStep { |
| 10 | + constructor(name, run) { |
| 11 | + this.name = name; |
| 12 | + this.run = run; |
| 13 | + } |
| 14 | + |
| 15 | + async runAndRecord(params, suite, test, callback) { |
| 16 | + const testRunner = new TestRunner(null, null, params, suite, test, callback); |
| 17 | + const result = await testRunner.runTest(); |
| 18 | + return result; |
| 19 | + } |
| 20 | +} |
| 21 | + |
| 22 | +/** |
| 23 | + * BenchmarkSuite |
| 24 | + * |
| 25 | + * A single test suite that contains one or more test steps. |
| 26 | + */ |
| 27 | +export class BenchmarkSuite { |
| 28 | + constructor(name, tests) { |
| 29 | + this.name = name; |
| 30 | + this.tests = tests; |
| 31 | + } |
| 32 | + |
| 33 | + record(_test, syncTime, asyncTime) { |
| 34 | + const total = syncTime + asyncTime; |
| 35 | + const results = { |
| 36 | + tests: { Sync: syncTime, Async: asyncTime }, |
| 37 | + total: total, |
| 38 | + }; |
| 39 | + |
| 40 | + return results; |
| 41 | + } |
| 42 | + |
| 43 | + async runAndRecord(params, onProgress) { |
| 44 | + const measuredValues = { |
| 45 | + tests: {}, |
| 46 | + total: 0, |
| 47 | + }; |
| 48 | + const suiteStartLabel = `suite-${this.name}-start`; |
| 49 | + const suiteEndLabel = `suite-${this.name}-end`; |
| 50 | + |
| 51 | + performance.mark(suiteStartLabel); |
| 52 | + |
| 53 | + for (const test of this.tests) { |
| 54 | + const result = await test.runAndRecord(params, this, test, this.record); |
| 55 | + measuredValues.tests[test.name] = result; |
| 56 | + measuredValues.total += result.total; |
| 57 | + onProgress?.(test.name); |
| 58 | + } |
| 59 | + |
| 60 | + performance.mark(suiteEndLabel); |
| 61 | + performance.measure(`suite-${this.name}`, suiteStartLabel, suiteEndLabel); |
| 62 | + |
| 63 | + return { |
| 64 | + type: "suite-tests-complete", |
| 65 | + status: "success", |
| 66 | + result: measuredValues, |
| 67 | + suitename: this.name, |
| 68 | + }; |
| 69 | + } |
| 70 | +} |
| 71 | + |
| 72 | +/** |
| 73 | + * BenchmarkSuitesManager |
| 74 | + * |
| 75 | + * A collection of test suites for a single workload. |
| 76 | + */ |
| 77 | +export class BenchmarkSuitesManager { |
| 78 | + constructor(name, suites) { |
| 79 | + this.name = name; |
| 80 | + this.suites = suites; |
| 81 | + } |
| 82 | + |
| 83 | + getSuiteByName(name) { |
| 84 | + return this.suites.find((suite) => suite.name === name); |
| 85 | + } |
| 86 | +} |
| 87 | + |
| 88 | +/** ********************************************************************** |
| 89 | + * BenchmarkConnector |
| 90 | + * |
| 91 | + * postMessage is used to communicate between app and benchmark. |
| 92 | + * When the app is ready, an 'app-ready' message is sent to signal that the app can receive instructions. |
| 93 | + * |
| 94 | + * A prepare script within the apps appends window.name and window.version from the package.json file. |
| 95 | + * The appId is build by appending name-version |
| 96 | + * It's used as an additional safe-guard to ensure the correct app responds to a message. |
| 97 | + *************************************************************************/ |
| 98 | +export class BenchmarkConnector { |
| 99 | + constructor(benchmarkSuitesManager, name, version) { |
| 100 | + this.benchmarkSuitesManager = benchmarkSuitesManager; |
| 101 | + this.name = name; |
| 102 | + this.version = version; |
| 103 | + |
| 104 | + if (!name || !version) |
| 105 | + console.warn("No name or version supplied, to create a unique appId"); |
| 106 | + |
| 107 | + this.appId = name && version ? `${name}-${version}` : -1; |
| 108 | + this.onMessage = this.onMessage.bind(this); |
| 109 | + } |
| 110 | + |
| 111 | + async onMessage(event) { |
| 112 | + if (event.data.id !== this.appId || event.data.key !== "benchmark-connector") |
| 113 | + return; |
| 114 | + |
| 115 | + switch (event.data.type) { |
| 116 | + case "benchmark-suite": |
| 117 | + // eslint-disable-next-line no-case-declarations |
| 118 | + const params = new Params(new URLSearchParams(window.location.search)); |
| 119 | + // eslint-disable-next-line no-case-declarations |
| 120 | + const { result } = await this.benchmarkSuitesManager.getSuiteByName(event.data.name).runAndRecord(params, (test) => this.sendMessage({ type: "step-complete", status: "success", appId: this.appId, name: this.name, test })); |
| 121 | + this.sendMessage({ type: "suite-complete", status: "success", appId: this.appId, result }); |
| 122 | + this.disconnect(); |
| 123 | + break; |
| 124 | + default: |
| 125 | + console.error(`Message data type not supported: ${event.data.type}`); |
| 126 | + } |
| 127 | + } |
| 128 | + |
| 129 | + sendMessage(message) { |
| 130 | + window.top.postMessage(message, "*"); |
| 131 | + } |
| 132 | + |
| 133 | + connect() { |
| 134 | + window.addEventListener("message", this.onMessage); |
| 135 | + this.sendMessage({ type: "app-ready", status: "success", appId: this.appId }); |
| 136 | + } |
| 137 | + |
| 138 | + disconnect() { |
| 139 | + window.removeEventListener("message", this.onMessage); |
| 140 | + } |
| 141 | +} |
0 commit comments