@testomatio/reporter 2.0.1-beta.6 → 2.0.1-beta.7

package/lib/adapter/codecept.js

@@ -126,6 +126,7 @@ function CodeceptReporter(config) {
         const logs = getTestLogs(test);
         const manuallyAttachedArtifacts = index_js_1.services.artifacts.get(test.fullTitle());
         const keyValues = index_js_1.services.keyValues.get(test.fullTitle());
+        const labels = index_js_1.services.labels.get(test.fullTitle());
         index_js_1.services.setContext(null);
         client.addTestRun(constants_js_1.STATUS.PASSED, {
             ...stripExampleFromTitle(title),
@@ -138,6 +139,7 @@ function CodeceptReporter(config) {
             logs,
             manuallyAttachedArtifacts,
             meta: keyValues,
+            labels: labels,
         });
         // output.stop();
     });
@@ -180,6 +182,7 @@ function CodeceptReporter(config) {
         const logs = getTestLogs(test);
         const manuallyAttachedArtifacts = index_js_1.services.artifacts.get(test.fullTitle());
         const keyValues = index_js_1.services.keyValues.get(test.fullTitle());
+        const labels = index_js_1.services.labels.get(test.fullTitle());
         index_js_1.services.setContext(null);
         client.addTestRun(constants_js_1.STATUS.FAILED, {
             ...stripExampleFromTitle(title),
@@ -194,6 +197,7 @@ function CodeceptReporter(config) {
             logs,
             manuallyAttachedArtifacts,
             meta: keyValues,
+            labels: labels,
         });
         debug('artifacts', artifacts);
         for (const aid in artifacts) {

package/lib/adapter/playwright.js

@@ -123,9 +123,18 @@ class PlaywrightReporter {
         }
         if (artifact.body) {
             let filePath = generateTmpFilepath(artifact.name);
-            const extension = artifact.contentType?.split('/')[1]?.replace('jpeg', 'jpg');
-            if (extension)
-                filePath += `.${extension}`;
+            // Check if file already has an extension
+            const hasExtension = artifact.name && path_1.default.extname(artifact.name);
+            if (!hasExtension && artifact.contentType) {
+                const mimeType = artifact.contentType.split('/')[1];
+                const extensionMap = {
+                    jpeg: 'jpg',
+                    plain: 'txt',
+                };
+                const extension = extensionMap[mimeType] || mimeType;
+                if (extension)
+                    filePath += `.${extension}`;
+            }
             fs_1.default.writeFileSync(filePath, artifact.body);
             return filePath;
         }

package/lib/client.js

@@ -179,7 +179,7 @@ class Client {
         /**
          * @type {TestData}
          */
-        const { rid, error = null, time = 0, example = null, files = [], filesBuffers = [], steps, code = null, title, file, suite_title, suite_id, test_id, timestamp, manuallyAttachedArtifacts, } = testData;
+        const { rid, error = null, time = 0, example = null, files = [], filesBuffers = [], steps, code = null, title, file, suite_title, suite_id, test_id, timestamp, manuallyAttachedArtifacts, labels, } = testData;
         let { message = '', meta = {} } = testData;
         // stringify meta values and limit keys and values length to 255
         meta = Object.entries(meta)
@@ -218,6 +218,7 @@ class Client {
                 acc[key] = value;
             return acc;
         }, {});
+        // Labels are simple array of strings, no processing needed
         let errorFormatted = '';
         if (error) {
             errorFormatted += this.formatError(error) || '';
@@ -265,6 +266,7 @@ class Client {
             timestamp,
             artifacts,
             meta,
+            labels,
             ...(rootSuiteId && { root_suite_id: rootSuiteId }),
         };
         // debug('Adding test run...', data);

package/lib/data-storage.d.ts

@@ -12,22 +12,22 @@ declare class DataStorage {
     /**
      * Puts any data to storage (file or global variable).
      * If file: stores data as text, if global variable – stores as array of data.
-     * @param {'log' | 'artifact' | 'keyvalue'} dataType
+     * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
      * @param {*} data anything you want to store (string, object, array, etc)
      * @param {*} context could be testId or any context (test name, suite name, including their IDs etc)
      * suite name + test name is used by default
      * @returns
      */
-    putData(dataType: "log" | "artifact" | "keyvalue", data: any, context?: any): void;
+    putData(dataType: "log" | "artifact" | "keyvalue" | "labels", data: any, context?: any): void;
     /**
      * Returns data, stored for specific test/context (or data which was stored without test id specified).
      * This method will get data from global variable and/or from from file (previosly saved with put method).
      *
-     * @param {'log' | 'artifact' | 'keyvalue'} dataType
+     * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
      * @param {string} context
      * @returns {any []} array of data (any type), null (if no data found for context) or string (if data type is log)
      */
-    getData(dataType: "log" | "artifact" | "keyvalue", context: string): any[];
+    getData(dataType: "log" | "artifact" | "keyvalue" | "labels", context: string): any[];
     #private;
 }
 export function stringToMD5Hash(str: any): string;

package/lib/data-storage.js

@@ -75,7 +75,7 @@ class DataStorage {
     /**
      * Puts any data to storage (file or global variable).
      * If file: stores data as text, if global variable – stores as array of data.
-     * @param {'log' | 'artifact' | 'keyvalue'} dataType
+     * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
      * @param {*} data anything you want to store (string, object, array, etc)
      * @param {*} context could be testId or any context (test name, suite name, including their IDs etc)
      * suite name + test name is used by default
@@ -103,7 +103,7 @@ class DataStorage {
      * Returns data, stored for specific test/context (or data which was stored without test id specified).
      * This method will get data from global variable and/or from from file (previosly saved with put method).
      *
-     * @param {'log' | 'artifact' | 'keyvalue'} dataType
+     * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
      * @param {string} context
      * @returns {any []} array of data (any type), null (if no data found for context) or string (if data type is log)
      */
@@ -134,7 +134,7 @@ class DataStorage {
         return null;
     }
     /**
-     * @param {'log' | 'artifact' | 'keyvalue'} dataType
+     * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
      * @param {string} context
      * @returns aray of data (any type)
      */
@@ -154,7 +154,7 @@ class DataStorage {
         }
     }
     /**
-     * @param {'log' | 'artifact' | 'keyvalue'} dataType
+     * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
      * @param {*} context
      * @returns array of data (any type)
      */
@@ -179,7 +179,7 @@ class DataStorage {
     }
     /**
      * Puts data to global variable. Unlike the file storage, stores data in array (file storage just append as string).
-     * @param {'log' | 'artifact' | 'keyvalue'} dataType
+     * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
      * @param {*} data
      * @param {*} context
      */
@@ -195,7 +195,7 @@ class DataStorage {
     }
     /**
      * Puts data to file. Unlike the global variable storage, stores data as string
-     * @param {'log' | 'artifact' | 'keyvalue'} dataType
+     * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
      * @param {*} data
      * @param {string} context
      * @returns

package/lib/reporter-functions.d.ts

@@ -3,6 +3,7 @@ declare namespace _default {
     export { logMessage as log };
     export { addStep as step };
     export { setKeyValue as keyValue };
+    export { setLabel as label };
 }
 export default _default;
 /**
@@ -32,3 +33,9 @@ declare function addStep(message: string): void;
 declare function setKeyValue(keyValue: {
     [key: string]: string;
 } | string, value?: string | null): void;
+/**
+ * Add a single label to the test report
+ * @param {string} key - label key (e.g. 'severity', 'feature', or just 'smoke' for labels without values)
+ * @param {string} [value] - optional label value (e.g. 'high', 'login')
+ */
+declare function setLabel(key: string, value?: string): void;

package/lib/reporter-functions.js

@@ -44,9 +44,45 @@ function setKeyValue(keyValue, value = null) {
     }
     index_js_1.services.keyValues.put(keyValue);
 }
+/**
+ * Add a single label to the test report
+ * @param {string} key - label key (e.g. 'severity', 'feature', or just 'smoke' for labels without values)
+ * @param {string} [value] - optional label value (e.g. 'high', 'login')
+ */
+function setLabel(key, value = null) {
+    if (!key || typeof key !== 'string') {
+        console.warn('Label key must be a non-empty string');
+        return;
+    }
+    // Limit key length to 255 characters
+    if (key.length > 255) {
+        console.warn('Label key is too long, trimmed to 255 characters:', key);
+        key = key.substring(0, 255);
+    }
+    let labelString = key;
+    if (value !== null && value !== undefined && value !== '') {
+        if (typeof value !== 'string') {
+            console.warn('Label value must be a string, converting:', value);
+            value = String(value);
+        }
+        // Limit value length to 255 characters
+        if (value.length > 255) {
+            console.warn('Label value is too long, trimmed to 255 characters:', value);
+            value = value.substring(0, 255);
+        }
+        labelString = `${key}:${value}`;
+    }
+    // Limit total label length to 255 characters
+    if (labelString.length > 255) {
+        console.warn('Label is too long, trimmed to 255 characters:', labelString);
+        labelString = labelString.substring(0, 255);
+    }
+    index_js_1.services.labels.put([labelString]);
+}
 module.exports = {
     artifact: saveArtifact,
     log: logMessage,
     step: addStep,
     keyValue: setKeyValue,
+    label: setLabel,
 };

package/lib/reporter.d.ts

@@ -81,6 +81,8 @@ export const meta: (keyValue: {
 } | string, value?: string | null) => void;
 export type step = typeof import("./reporter-functions.js");
 export const step: (message: string) => void;
+export type label = typeof import("./reporter-functions.js");
+export const label: (key: string, value?: string) => void;
 declare namespace _default {
     let testomatioLogger: {
         "__#13@#originalUserLogger": {
@@ -228,5 +230,6 @@ declare namespace _default {
         [key: string]: string;
     } | string, value?: string | null) => void;
     let step: (message: string) => void;
+    let label: (key: string, value?: string) => void;
 }
 export default _default;

package/lib/reporter.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.step = exports.meta = exports.logger = exports.log = exports.artifact = void 0;
+exports.label = exports.step = exports.meta = exports.logger = exports.log = exports.artifact = void 0;
 // import TestomatClient from './client.js';
 // import * as TRConstants from './constants.js';
 const index_js_1 = require("./services/index.js");
@@ -13,12 +13,14 @@ exports.log = reporter_functions_js_1.default.log;
 exports.logger = index_js_1.services.logger;
 exports.meta = reporter_functions_js_1.default.keyValue;
 exports.step = reporter_functions_js_1.default.step;
+exports.label = reporter_functions_js_1.default.label;
 /**
  * @typedef {import('./reporter-functions.js')} artifact
  * @typedef {import('./reporter-functions.js')} log
  * @typedef {import('./services/index.js')} logger
  * @typedef {import('./reporter-functions.js')} meta
  * @typedef {import('./reporter-functions.js')} step
+ * @typedef {import('./reporter-functions.js')} label
  */
 module.exports = {
     /**
@@ -30,6 +32,7 @@ module.exports = {
     logger: index_js_1.services.logger,
     meta: reporter_functions_js_1.default.keyValue,
     step: reporter_functions_js_1.default.step,
+    label: reporter_functions_js_1.default.label,
     // TestomatClient,
     // TRConstants,
 };

package/lib/services/index.d.ts

@@ -2,8 +2,10 @@ export namespace services {
     export { logger };
     export { artifactStorage as artifacts };
     export { keyValueStorage as keyValues };
+    export { labelStorage as labels };
     export function setContext(context: any): void;
 }
 import { logger } from './logger.js';
 import { artifactStorage } from './artifacts.js';
 import { keyValueStorage } from './key-values.js';
+import { labelStorage } from './labels.js';

package/lib/services/index.js

@@ -4,11 +4,13 @@ exports.services = void 0;
 const logger_js_1 = require("./logger.js");
 const artifacts_js_1 = require("./artifacts.js");
 const key_values_js_1 = require("./key-values.js");
+const labels_js_1 = require("./labels.js");
 const data_storage_js_1 = require("../data-storage.js");
 exports.services = {
     logger: logger_js_1.logger,
     artifacts: artifacts_js_1.artifactStorage,
     keyValues: key_values_js_1.keyValueStorage,
+    labels: labels_js_1.labelStorage,
     setContext: context => {
         data_storage_js_1.dataStorage.setContext(context);
     },

package/lib/services/labels.d.ts

@@ -0,0 +1,22 @@
+export const labelStorage: LabelStorage;
+declare class LabelStorage {
+    static "__#16@#instance": any;
+    /**
+     *
+     * @returns {LabelStorage}
+     */
+    static getInstance(): LabelStorage;
+    /**
+     * Stores labels array and passes it to reporter
+     * @param {string[]} labels - array of label strings
+     * @param {*} context - full test title
+     */
+    put(labels: string[], context?: any): void;
+    /**
+     * Returns labels array for the test
+     * @param {*} context testId or test context from test runner
+     * @returns {string[]} labels array, e.g. ['smoke', 'severity:high', 'feature:user_account']
+     */
+    get(context?: any): string[];
+}
+export {};

package/lib/services/labels.js

@@ -0,0 +1,62 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.labelStorage = void 0;
+const debug_1 = __importDefault(require("debug"));
+const data_storage_js_1 = require("../data-storage.js");
+const debug = (0, debug_1.default)('@testomatio/reporter:services-labels');
+class LabelStorage {
+    static #instance;
+    /**
+     *
+     * @returns {LabelStorage}
+     */
+    static getInstance() {
+        if (!this.#instance) {
+            this.#instance = new LabelStorage();
+        }
+        return this.#instance;
+    }
+    /**
+     * Stores labels array and passes it to reporter
+     * @param {string[]} labels - array of label strings
+     * @param {*} context - full test title
+     */
+    put(labels, context = null) {
+        if (!labels || !Array.isArray(labels))
+            return;
+        data_storage_js_1.dataStorage.putData('labels', labels, context);
+    }
+    /**
+     * Returns labels array for the test
+     * @param {*} context testId or test context from test runner
+     * @returns {string[]} labels array, e.g. ['smoke', 'severity:high', 'feature:user_account']
+     */
+    get(context = null) {
+        const labelsList = data_storage_js_1.dataStorage.getData('labels', context);
+        if (!labelsList || !labelsList?.length)
+            return [];
+        const allLabels = [];
+        for (const labels of labelsList) {
+            if (Array.isArray(labels)) {
+                allLabels.push(...labels);
+            }
+            else if (typeof labels === 'string') {
+                try {
+                    const parsedLabels = JSON.parse(labels);
+                    if (Array.isArray(parsedLabels)) {
+                        allLabels.push(...parsedLabels);
+                    }
+                }
+                catch (e) {
+                    debug(`Error parsing labels for test ${context}`, labels);
+                }
+            }
+        }
+        // Remove duplicates
+        return [...new Set(allLabels)];
+    }
+}
+exports.labelStorage = LabelStorage.getInstance();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@testomatio/reporter",
-  "version": "2.0.1-beta.6",
+  "version": "2.0.1-beta.7",
   "description": "Testomatio Reporter Client",
   "engines": {
     "node": ">=18"

package/src/adapter/codecept.js

@@ -151,6 +151,7 @@ function CodeceptReporter(config) {
     const logs = getTestLogs(test);
     const manuallyAttachedArtifacts = services.artifacts.get(test.fullTitle());
     const keyValues = services.keyValues.get(test.fullTitle());
+    const labels = services.labels.get(test.fullTitle());
     services.setContext(null);
 
     client.addTestRun(STATUS.PASSED, {
@@ -164,6 +165,7 @@ function CodeceptReporter(config) {
       logs,
       manuallyAttachedArtifacts,
       meta: keyValues,
+      labels: labels,
     });
     // output.stop();
   });
@@ -208,6 +210,7 @@ function CodeceptReporter(config) {
     const logs = getTestLogs(test);
     const manuallyAttachedArtifacts = services.artifacts.get(test.fullTitle());
     const keyValues = services.keyValues.get(test.fullTitle());
+    const labels = services.labels.get(test.fullTitle());
     services.setContext(null);
 
     client.addTestRun(STATUS.FAILED, {
@@ -223,6 +226,7 @@ function CodeceptReporter(config) {
       logs,
       manuallyAttachedArtifacts,
       meta: keyValues,
+      labels: labels,
     });
 
     debug('artifacts', artifacts);

package/src/adapter/playwright.js

@@ -125,20 +125,24 @@ class PlaywrightReporter {
   #getArtifactPath(artifact) {
     if (artifact.path) {
       if (path.isAbsolute(artifact.path)) return artifact.path;
-
       return path.join(this.config.outputDir || this.config.projects[0].outputDir, artifact.path);
     }
-
     if (artifact.body) {
       let filePath = generateTmpFilepath(artifact.name);
-
-      const extension = artifact.contentType?.split('/')[1]?.replace('jpeg', 'jpg');
-      if (extension) filePath += `.${extension}`;
-
+      // Check if file already has an extension
+      const hasExtension = artifact.name && path.extname(artifact.name);
+      if (!hasExtension && artifact.contentType) {
+        const mimeType = artifact.contentType.split('/')[1];
+        const extensionMap = {
+          jpeg: 'jpg',
+          plain: 'txt',
+        };
+        const extension = extensionMap[mimeType] || mimeType;
+        if (extension) filePath += `.${extension}`;
+      }
       fs.writeFileSync(filePath, artifact.body);
       return filePath;
     }
-
     return null;
   }
 

package/src/client.js

@@ -179,6 +179,7 @@ class Client {
       test_id,
       timestamp,
       manuallyAttachedArtifacts,
+      labels,
     } = testData;
     let { message = '', meta = {} } = testData;
 
@@ -219,6 +220,8 @@ class Client {
         return acc;
       }, {});
 
+    // Labels are simple array of strings, no processing needed
+
     let errorFormatted = '';
     if (error) {
       errorFormatted += this.formatError(error) || '';
@@ -273,6 +276,7 @@ class Client {
       timestamp,
       artifacts,
       meta,
+      labels,
       ...(rootSuiteId && { root_suite_id: rootSuiteId }),
     };
 

package/src/data-storage.js

@@ -41,7 +41,7 @@ class DataStorage {
   /**
    * Puts any data to storage (file or global variable).
    * If file: stores data as text, if global variable – stores as array of data.
-   * @param {'log' | 'artifact' | 'keyvalue'} dataType
+   * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
    * @param {*} data anything you want to store (string, object, array, etc)
    * @param {*} context could be testId or any context (test name, suite name, including their IDs etc)
    * suite name + test name is used by default
@@ -70,7 +70,7 @@ class DataStorage {
    * Returns data, stored for specific test/context (or data which was stored without test id specified).
    * This method will get data from global variable and/or from from file (previosly saved with put method).
    *
-   * @param {'log' | 'artifact' | 'keyvalue'} dataType
+   * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
    * @param {string} context
    * @returns {any []} array of data (any type), null (if no data found for context) or string (if data type is log)
    */
@@ -108,7 +108,7 @@ class DataStorage {
   }
 
   /**
-   * @param {'log' | 'artifact' | 'keyvalue'} dataType
+   * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
    * @param {string} context
    * @returns aray of data (any type)
    */
@@ -127,7 +127,7 @@ class DataStorage {
   }
 
   /**
-   * @param {'log' | 'artifact' | 'keyvalue'} dataType
+   * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
    * @param {*} context
    * @returns array of data (any type)
    */
@@ -151,7 +151,7 @@ class DataStorage {
 
   /**
    * Puts data to global variable. Unlike the file storage, stores data in array (file storage just append as string).
-   * @param {'log' | 'artifact' | 'keyvalue'} dataType
+   * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
    * @param {*} data
    * @param {*} context
    */
@@ -166,7 +166,7 @@ class DataStorage {
 
   /**
    * Puts data to file. Unlike the global variable storage, stores data as string
-   * @param {'log' | 'artifact' | 'keyvalue'} dataType
+   * @param {'log' | 'artifact' | 'keyvalue' | 'labels'} dataType
    * @param {*} data
    * @param {string} context
    * @returns

package/src/reporter-functions.js

@@ -47,9 +47,50 @@ function setKeyValue(keyValue, value = null) {
   services.keyValues.put(keyValue);
 }
 
+/**
+ * Add a single label to the test report
+ * @param {string} key - label key (e.g. 'severity', 'feature', or just 'smoke' for labels without values)
+ * @param {string} [value] - optional label value (e.g. 'high', 'login')
+ */
+function setLabel(key, value = null) {
+  if (!key || typeof key !== 'string') {
+    console.warn('Label key must be a non-empty string');
+    return;
+  }
+
+  // Limit key length to 255 characters
+  if (key.length > 255) {
+    console.warn('Label key is too long, trimmed to 255 characters:', key);
+    key = key.substring(0, 255);
+  }
+
+  let labelString = key;
+  if (value !== null && value !== undefined && value !== '') {
+    if (typeof value !== 'string') {
+      console.warn('Label value must be a string, converting:', value);
+      value = String(value);
+    }
+    // Limit value length to 255 characters
+    if (value.length > 255) {
+      console.warn('Label value is too long, trimmed to 255 characters:', value);
+      value = value.substring(0, 255);
+    }
+    labelString = `${key}:${value}`;
+  }
+
+  // Limit total label length to 255 characters
+  if (labelString.length > 255) {
+    console.warn('Label is too long, trimmed to 255 characters:', labelString);
+    labelString = labelString.substring(0, 255);
+  }
+
+  services.labels.put([labelString]);
+}
+
 export default {
   artifact: saveArtifact,
   log: logMessage,
   step: addStep,
   keyValue: setKeyValue,
+  label: setLabel,
 };

package/src/reporter.js

@@ -8,6 +8,7 @@ export const log = reporterFunctions.log;
 export const logger = services.logger;
 export const meta = reporterFunctions.keyValue;
 export const step = reporterFunctions.step;
+export const label = reporterFunctions.label;
 
 /**
  * @typedef {import('./reporter-functions.js')} artifact
@@ -15,6 +16,7 @@ export const step = reporterFunctions.step;
  * @typedef {import('./services/index.js')} logger
  * @typedef {import('./reporter-functions.js')} meta
  * @typedef {import('./reporter-functions.js')} step
+ * @typedef {import('./reporter-functions.js')} label
  */
 export default {
   /**
@@ -27,6 +29,7 @@ export default {
   logger: services.logger,
   meta: reporterFunctions.keyValue,
   step: reporterFunctions.step,
+  label: reporterFunctions.label,
 
   // TestomatClient,
   // TRConstants,

package/src/services/index.js

@@ -1,12 +1,14 @@
 import { logger } from './logger.js';
 import { artifactStorage } from './artifacts.js';
 import { keyValueStorage } from './key-values.js';
+import { labelStorage } from './labels.js';
 import { dataStorage } from '../data-storage.js';
 
 export const services = {
   logger,
   artifacts: artifactStorage,
   keyValues: keyValueStorage,
+  labels: labelStorage,
   setContext: context => {
     dataStorage.setContext(context);
   },

package/src/services/labels.js

@@ -0,0 +1,59 @@
+import createDebugMessages from 'debug';
+import { dataStorage } from '../data-storage.js';
+
+const debug = createDebugMessages('@testomatio/reporter:services-labels');
+class LabelStorage {
+  static #instance;
+
+  /**
+   *
+   * @returns {LabelStorage}
+   */
+  static getInstance() {
+    if (!this.#instance) {
+      this.#instance = new LabelStorage();
+    }
+    return this.#instance;
+  }
+
+  /**
+   * Stores labels array and passes it to reporter
+   * @param {string[]} labels - array of label strings
+   * @param {*} context - full test title
+   */
+  put(labels, context = null) {
+    if (!labels || !Array.isArray(labels)) return;
+    dataStorage.putData('labels', labels, context);
+  }
+
+  /**
+   * Returns labels array for the test
+   * @param {*} context testId or test context from test runner
+   * @returns {string[]} labels array, e.g. ['smoke', 'severity:high', 'feature:user_account']
+   */
+  get(context = null) {
+    const labelsList = dataStorage.getData('labels', context);
+    if (!labelsList || !labelsList?.length) return [];
+
+    const allLabels = [];
+    for (const labels of labelsList) {
+      if (Array.isArray(labels)) {
+        allLabels.push(...labels);
+      } else if (typeof labels === 'string') {
+        try {
+          const parsedLabels = JSON.parse(labels);
+          if (Array.isArray(parsedLabels)) {
+            allLabels.push(...parsedLabels);
+          }
+        } catch (e) {
+          debug(`Error parsing labels for test ${context}`, labels);
+        }
+      }
+    }
+
+    // Remove duplicates
+    return [...new Set(allLabels)];
+  }
+}
+
+export const labelStorage = LabelStorage.getInstance();