summaryrefslogtreecommitdiffstats
path: root/testing/raptor/browsertime/utils/NetworkThrottlingUtils.js
blob: 8a0f78150ecdc88d5e4a4248655c61259e41be60 (plain)
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
/* 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 lazy = {};

ChromeUtils.defineESModuleGetters(lazy, {
  NetworkObserver:
    "resource://devtools/shared/network-observer/NetworkObserver.sys.mjs",
});

/**
 * The NetworkThrottler uses the dev tools NetworkObserver to provide api to throttle all network activity.
 * This can be used to fix network conditions in browsertime pageload tests.
 *
 */

// A minimal struct for onNetworkEvent handling
class NetworkEventRecord {
  addRequestPostData() {}
  addResponseStart() {}
  addSecurityInfo() {}
  addEventTimings() {}
  addResponseCache() {}
  addResponseContent() {}
  addServerTimings() {}
  addServiceWorkerTimings() {}
}

class NetworkThrottler {
  #devtoolsNetworkObserver;
  #throttling;

  constructor() {
    this.#throttling = false;
  }

  destroy() {
    this.stop();
  }

  start(throttleData) {
    if (this.#throttling) {
      console.error("NetworkThrottler already started");
      return;
    }

    this.#devtoolsNetworkObserver = new lazy.NetworkObserver({
      ignoreChannelFunction: this.#ignoreChannelFunction,
      onNetworkEvent: this.#onNetworkEvent,
    });

    this.#devtoolsNetworkObserver.setThrottleData(throttleData);

    this.#throttling = true;
  }

  stop() {
    if (!this.#throttling) {
      return;
    }

    this.#devtoolsNetworkObserver.destroy();
    this.#devtoolsNetworkObserver = null;

    this.#throttling = false;
  }

  #ignoreChannelFunction = channel => {
    // Ignore chrome-privileged or DevTools-initiated requests
    if (
      channel.loadInfo?.loadingDocument === null &&
      (channel.loadInfo.loadingPrincipal ===
        Services.scriptSecurityManager.getSystemPrincipal() ||
        channel.loadInfo.isInDevToolsContext)
    ) {
      return true;
    }
    return false;
  };

  #onNetworkEvent = (networkEvent, channel) => {
    return new NetworkEventRecord(networkEvent, channel, this);
  };
}