init

Bin

2025-12-16 9e0b2ba2c317b1a86212f24cbae3195ad1f3dbfa

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
const { I } = inject();
const Helpers = require("../tests/helpers");
 
module.exports = {
  init({ events = {}, ...params }) {
    I.executeScript(Helpers.createLabelStudioInitFunction(params));
    for (const [eventName, callback] of Object.entries(events)) {
      this.on(eventName, callback);
    }
 
    const defaultCpuThrottling = process.env.DEFAULT_CPU_THROTTLING;
 
    if (defaultCpuThrottling) {
      I.wait(1); // gives some time to page to load
      I.throttleCPU(+defaultCpuThrottling);
    }
  },
  on(eventName, callback) {
    I.executeScript(Helpers.createAddEventListenerScript(eventName, callback));
  },
  async serialize() {
    const result = await I.executeScript(Helpers.serialize);
 
    return result;
  },
 
  hasFF(fflag) {
    return I.executeScript(Helpers.hasFF, fflag);
  },
 
  setFeatureFlagsDefaultValue(value) {
    I.executeScript(Helpers.setFeatureFlagsDefaultValue, value);
  },
 
  setFeatureFlags(featureFlags) {
    I.executeScript(Helpers.setFeatureFlags, featureFlags);
  },
 
  clearModalIfPresent() {
    I.executeScript(Helpers.clearModalIfPresent);
  },
 
  async waitForObjectsReady() {
    await I.executeScript(Helpers.waitForObjectsReady);
  },
 
  async resultsNotChanged(result, fractionDigits = 2) {
    const serialized = await this.serialize();
 
    I.assertDeepEqualWithTolerance(result, serialized, fractionDigits, "Results must be equal");
  },
 
  async resultsChanged(result, fractionDigits = 2) {
    const serialized = await this.serialize();
 
    I.assertNotDeepEqualWithTolerance(result, serialized, fractionDigits, "Results must be different");
  },
 
  async grabUserLabels() {
    const userLabels = await I.executeScript(() => {
      return Object.fromEntries(
        Object.entries(window.Htx.userLabels?.controls).map(([control, labels]) => {
          return [control, labels.map((label) => label.path)];
        }),
      );
    });
 
    return userLabels;
  },
 
  initUserLabels(userLabels) {
    return I.executeScript((userLabels) => {
      window.Htx.userLabels?.init(userLabels);
    }, userLabels);
  },
 
  enableSetting(settingName) {
    I.say("Attempting to open settings menu");
    I.click("[aria-label=Settings]");
    I.see("Settings");
    I.say("Attempt to enable setting");
    I.click(settingName);
    I.seeCheckboxIsChecked(settingName);
    I.click("[aria-label=Close]");
    I.dontSee("Settings");
  },
 
  disableSetting(settingName) {
    I.say("Attempting to open settings menu");
    I.click("[aria-label=Settings]");
    I.see("Settings");
    I.say("Attempt to disable setting");
    I.click(settingName);
    I.seeCheckboxIsChecked(settingName);
    I.click("[aria-label=Close]");
    I.dontSee("Settings");
  },
};