@testomatio/reporter 1.0.17 → 1.1.0-beta-2

package/lib/{dataStorage.js → storages/data-storage.js}

@@ -3,16 +3,18 @@ const fs = require('fs');
 const path = require('path');
 const os = require('os');
 const { join } = require('path');
-const JestReporter = require('./adapter/jest');
-const { TESTOMAT_TMP_STORAGE } = require('./constants');
-const { fileSystem } = require('./utils/utils');
-const getTestIdFromTestTitle = require('./utils/utils').parseTest;
+const { TESTOMAT_TMP_STORAGE_DIR } = require('../constants');
+const { fileSystem, jestHelpers, parseTest } = require('../utils/utils');
+
+const getTestIdFromTestTitle = parseTest;
 
 class DataStorage {
   /**
    * Creates data storage instance for specific data type.
    * Stores data to global variable or to file depending on what is applicable for current test runner
    * (running environment).
+   * Recommend to use composition while using this class (instead of inheritance).
+   * ! Also the class which will use data storage should be singleton (to avoid data loss).
    * @param {*} dataType storage type (like 'log', 'artifact'); could be any string, used only to define file path
    */
   constructor(dataType) {
@@ -20,8 +22,35 @@ class DataStorage {
     this.isFileStorage = true;
     this.#refreshStorageType();
 
-    this.dataDirPath = path.join(TESTOMAT_TMP_STORAGE.mainDir, this.dataType);
-    fileSystem.createDir(this.dataDirPath);
+    this.dataDirPath = path.join(TESTOMAT_TMP_STORAGE_DIR, this.dataType);
+
+    // set test title for mocha (could not be moved to adapter)
+    // it does not override existing hook, just add new one
+    try {
+      // @ts-ignore
+      if (!beforeEach) return;
+      // @ts-ignore-next-line
+      beforeEach(function () {
+        if (this.currentTest?.__mocha_id__) {
+          global.testTitle = this.currentTest.fullTitle();
+        }
+      });
+    } catch (e) {
+      // ignore
+    }
+
+    // set test title for playwright
+    try {
+      // eslint-disable-next-line import/no-extraneous-dependencies, import/no-unresolved
+      const { test } = require('@playwright/test');
+      // eslint-disable-next-line no-empty-pattern
+      test.beforeEach(async ({}, testInfo) => {
+        global.testTitle = testInfo.title;
+        global.testomatioRunningEnvironment = 'playwright';
+      });
+    } catch (e) {
+      // ignore
+    }
   }
 
   /**
@@ -44,15 +73,18 @@ class DataStorage {
   }
 
   /**
-   * Puts any data to storage (file or global variable)
-   * @param {*} data anything you want to store
+   * Puts any data to storage (file or global variable).
+   * If file: stores data as text, if global variable – stores as array of data.
+   * @param {*} data anything you want to store (string, object, array, etc)
    * @param {*} context could be testId or any context (test, suite, etc) which is used to define testId
    * @returns
    */
   putData(data, context = null) {
-    this.#refreshStorageType();
+    fileSystem.createDir(this.dataDirPath);
 
-    let testId = this.#tryToRetrieveTestId(context) || null;
+    this.#refreshStorageType();
+    context = context ?? global.testTitle ?? null;
+    let testId = this.#tryToRetrieveTestId(context) || context || null;
 
     if (this.runningEnvironment === 'codeceptjs') {
       this.isFileStorage = false;
@@ -60,7 +92,7 @@ class DataStorage {
     }
 
     if (this.runningEnvironment === 'jest') {
-      testId = testId ?? JestReporter.getIdOfCurrentlyRunningTest();
+      testId = testId ?? jestHelpers.getIdOfCurrentlyRunningTest();
     }
 
     // logs in playwright are gathered by pw framework itself
@@ -74,7 +106,7 @@ class DataStorage {
       testId = this.#tryToRetrieveTestId(global.currentlyRunningTestTitle);
 
     if (!testId) {
-      debug(`No test id provided for ${this.dataType} data: ${data}`);
+      debug(`No test id provided for "${this.dataType}" data:`, data);
       return;
     }
 
@@ -87,12 +119,10 @@ class DataStorage {
 
   /**
    * Returns data, stored for specific testId (or data which was stored without test id specified).
-   * This method will get data from global variable. But if it is not available, it will try to get data from file.
-   * Thus, good approach is to remove file storage folder before each test run (and after, for sure).
+   * This method will get data from global variable and/or from from file (previosly saved with put method).
    *
-   * Defining the execution environment is not guaranteed! Is used only as additional check.
-   * @param {*} context could be testId or any context (test, suite, etc) which is used to define testId
-   * @returns
+   * @param {*} context could be testId or any context (test title, suite, etc) which is used to define testId
+   * @returns array of data (any type), null (if no data found for test) or string (if data type is log)
    */
   getData(context) {
     this.#refreshStorageType();
@@ -100,9 +130,6 @@ class DataStorage {
     let testId = null;
     if (typeof context === 'string') {
       testId = this.#tryToRetrieveTestId(context) || context;
-    } else {
-      // TODO: derive testId from context
-      // testId = context...
     }
 
     if (!testId) {
@@ -110,12 +137,20 @@ class DataStorage {
       return null;
     }
 
-    let testData = '';
+    let testDataFromFile = [];
+    let testDataFromGlobalVar = [];
 
     if (global?.testomatioDataStore) {
-      testData = this.#getDataFromGlobalVar(testId);
+      testDataFromGlobalVar = this.#getDataFromGlobalVar(testId);
       // these frameworks use global variable storage
-      if (testData && ['cucumber', 'codeceptjs'].includes(this.runningEnvironment)) return testData;
+      // if (testDataFromGlobalVar && ['cucumber', 'codeceptjs'].includes(this.runningEnvironment)) {
+      if (testDataFromGlobalVar) {
+        // return as string if its log data
+        if (this.dataType === 'log' && testDataFromGlobalVar.length) return testDataFromGlobalVar.join('\n');
+        // return as array for other data types
+        if (testDataFromGlobalVar.length) return testDataFromGlobalVar;
+      }
+      // don't return nothing if no data in global variable
     }
 
     /* condition is removed for mocha
@@ -126,11 +161,16 @@ class DataStorage {
     //   if (testData) return testData;
     // }
 
-    const testDataFromFile = this.#getDataFromFile(testId);
-    testData += testDataFromFile || '';
+    testDataFromFile = this.#getDataFromFile(testId);
 
-    debug(`No ${this.dataType} data for test id ${testId} in both file and global variable`);
-    return testData || '';
+    if (testDataFromFile.length) {
+      if (this.dataType === 'log') return testDataFromFile.join('\n');
+      return testDataFromFile;
+    }
+    debug(`No "${this.dataType}" data for test id ${testId} in both file and global variable`);
+
+    // in case no data found for test
+    return null;
   }
 
   /**
@@ -166,46 +206,63 @@ class DataStorage {
     this.runningEnvironment = this.getRunningEnviroment();
 
     // some test frameworks do not persist global variables, thus file storage is used for them (by default)
-    if (['playwright', 'codeceptjs'].includes(this.runningEnvironment)) this.isFileStorage = false;
+    if (this.runningEnvironment === 'codeceptjs') this.isFileStorage = false;
+    // playwrigth stores logs by itself; but file storage is used for other data types for playwright
+    if (this.runningEnvironment === 'playwright' && this.dataType === 'log') this.isFileStorage = false;
   }
 
+  /**
+   * @param {*} testId
+   * @returns aray of data (any type)
+   */
   #getDataFromGlobalVar(testId) {
     try {
       if (global?.testomatioDataStore[this.dataType]) {
         const testData = global.testomatioDataStore[this.dataType][testId];
-        debug(`Data for test id ${testId}:\n${testData}`);
-        return testData || '';
+        if (testData) debug(`"${this.dataType}" data for test id ${testId}:`, testData.join(', '));
+        return testData || [];
       }
-      debug(`No ${this.dataType} data for test id ${testId} in <global> storage`);
-      return '';
+      // debug(`No ${this.dataType} data for test id ${testId} in <global> storage`);
+      return [];
     } catch (e) {
       // there could be no data, ignore
     }
   }
 
+  /**
+   *
+   * @param {*} testId
+   * @returns array of data (any type)
+   */
   #getDataFromFile(testId) {
     try {
       const filepath = join(this.dataDirPath, `${this.dataType}_${testId}`);
       if (fs.existsSync(filepath)) {
-        const testData = fs.readFileSync(filepath, 'utf-8');
-        debug(`Data for test id ${testId}:\n${testData}`);
-        return testData || '';
+        const testDataAsText = fs.readFileSync(filepath, 'utf-8');
+        if (testDataAsText) debug(`"${this.dataType}" data for test id ${testId}:`, testDataAsText);
+        const testDataArr = testDataAsText?.split(os.EOL) || [];
+        return testDataArr;
       }
-      debug(`No ${this.dataType} data for test id ${testId} in <file> storage`);
-      return '';
+      // debug(`No ${this.dataType} data for test id ${testId} in <file> storage`);
+      return [];
     } catch (e) {
       // there could be no data, ignore
     }
-    return '';
+    return [];
   }
 
+  /**
+   * Puts data to global variable. Unlike the file storage, stores data in array (file storage just append as string).
+   * @param {*} data
+   * @param {*} testId
+   */
   #putDataToGlobalVar(data, testId) {
-    debug('Saving data to global variable for test', testId, ':\n', data, '\n');
+    debug(`Saving data to global variable for test ${testId}:`, data);
     if (!global.testomatioDataStore) global.testomatioDataStore = {};
     if (!global.testomatioDataStore?.[this.dataType]) global.testomatioDataStore[this.dataType] = {};
-    global.testomatioDataStore?.[this.dataType][testId] // eslint-disable-line no-unused-expressions
-      ? (global.testomatioDataStore[this.dataType][testId] += `\n${data}`)
-      : (global.testomatioDataStore[this.dataType][testId] = data);
+
+    if (!global.testomatioDataStore?.[this.dataType][testId]) global.testomatioDataStore[this.dataType][testId] = [];
+    global.testomatioDataStore[this.dataType][testId].push(data);
   }
 
   #putDataToFile(data, testId) {
@@ -215,18 +272,36 @@ class DataStorage {
     if (!fs.existsSync(this.dataDirPath)) fileSystem.createDir(this.dataDirPath);
     debug('Saving data to file for test', testId, 'to', filepath, ':\n', data, '\n');
 
-    // TODO: handle multiple invocations of JSON.stringify.
-    // UPD: not actual because decided not to use it - it created extra wrapping quotes "" and // (escaped slashes)
-    fs.appendFileSync(filepath, data + os.EOL, 'utf-8');
+    // append new line if file already exists (in this case its definitely includes some data)
+    if (fs.existsSync(filepath)) {
+      fs.appendFileSync(filepath, os.EOL + data, 'utf-8');
+    } else {
+      fs.writeFileSync(filepath, data, 'utf-8');
+    }
   }
 }
 
-module.exports.DataStorage = DataStorage;
+module.exports = DataStorage;
 
 // TODO: consider using fs promises instead of writeSync/appendFileSync to
 // prevent blocking and improve performance (probably queue usage will be required)
 
 // TODO: rewrite client - everything regarding storing artifacts
 // TODO: try to define adapter inside client
-// TODO: use .env
 // TODO: ability to intercept multiple loggers (upd: no need)
+
+/* does not work for Playwright
+    try {
+      const { test } = require('@playwright/test');
+      test.beforeEach(async ({}, testInfo) => {
+        global.testTitle = testInfo.title;
+        global.testomatioRunningEnvironment = 'playwright';
+      });
+    } catch (e) {
+      // ignore
+    }
+// Playwright storage works fine only when running tests from a single file;
+// otherwise it is not possible to define test id/title, it is overwritten
+ TODO: the only way implementation for Pw for now is to pass testInfo inside test
+//  (and process this testInfo inside "put" function)
+*/

package/lib/storages/key-value-storage.js

@@ -0,0 +1,58 @@
+const debug = require('debug')('@testomatio/reporter:key-value-storage');
+const DataStorage = require('./data-storage');
+
+class KeyValueStorage {
+  constructor() {
+    this.dataStorage = new DataStorage('keyvalue');
+
+    // singleton
+    if (!KeyValueStorage.instance) {
+      KeyValueStorage.instance = this;
+    }
+  }
+
+  /**
+   * Stores key-value pair and passes it to reporter
+   * @param {{key: string}} keyValue - key-value pair(s) as object
+   * @param {*} context testId or test title
+   */
+  put(keyValue, context = null) {
+    if (!keyValue) return;
+    this.dataStorage.putData(keyValue, context);
+  }
+
+  #isKeyValueObject(smth) {
+    return smth && typeof smth === 'object' && !Array.isArray(smth) && smth !== null;
+  }
+
+  /**
+   * Returns key-values pairs for the test as object
+   * @param {*} context testId or test context from test runner
+   * @returns {Object} key-values pairs as object
+   */
+  get(context) {
+    if (!context) return null;
+
+    const keyValuesList = this.dataStorage.getData(context);
+    if (!keyValuesList || !keyValuesList?.length) return null;
+
+    const keyValues = {};
+    for (const keyValue of keyValuesList) {
+      if (this.#isKeyValueObject(keyValue)) {
+        Object.assign(keyValues, keyValue);
+      } else if (typeof keyValue === 'string') {
+        try {
+          Object.assign(keyValues, JSON.parse(keyValue));
+        } catch (e) {
+          debug(`Error parsing key-values for test ${context}`, keyValue);
+        }
+      }
+    }
+
+    return Object.keys(keyValues).length ? keyValues : null;
+  }
+}
+
+KeyValueStorage.instance = null;
+
+module.exports = new KeyValueStorage();

package/lib/{logger.js → storages/logger.js}

@@ -1,6 +1,6 @@
 const chalk = require('chalk');
 const debug = require('debug')('@testomatio/reporter:logger');
-const { DataStorage } = require('./dataStorage');
+const DataStorage = require('./data-storage');
 
 const LOG_METHODS = ['assert', 'debug', 'error', 'info', 'log', 'trace', 'warn'];
 const LEVELS = {
@@ -38,20 +38,6 @@ class Logger {
     if (!Logger.instance) {
       Logger.instance = this;
     }
-
-    // add beforeEach hook for mocha. it does not override existing hook, just add new one
-    try {
-      // @ts-ignore
-      if (!beforeEach) return;
-      // @ts-ignore-next-line
-      beforeEach(function () {
-        if (this.currentTest?.__mocha_id__) {
-          global.testTitle = this.currentTest.fullTitle();
-        }
-      });
-    } catch (e) {
-      // ignore
-    }
   }
 
   /**
@@ -61,9 +47,6 @@ class Logger {
    * @param  {...any} values
    */
   step(strings, ...values) {
-    // get testId for mocha
-    const context = global.testTitle ?? null;
-
     let logs = '';
     for (let i = 0; i < strings.length; i++) {
       logs += strings[i];
@@ -72,7 +55,7 @@ class Logger {
       }
     }
     logs = chalk.blue(`> ${logs}`);
-    this.#dataStorage.putData(logs, context);
+    this.#dataStorage.putData(logs);
   }
 
   /**
@@ -117,11 +100,6 @@ class Logger {
   templateLiteralLog(strings, ...args) {
     if (Array.isArray(strings)) strings = strings.filter(item => item !== '').map(item => item.trim());
     if (Array.isArray(args)) args = args.filter(item => item !== '');
-    // entity which is used to define testId
-    let context = null;
-
-    // get testId for mocha
-    context = global.testTitle ?? null;
 
     let logs;
     // this block means tagged template is used (syntax like $`text ${someVar}`)
@@ -146,7 +124,7 @@ class Logger {
       logs = this.#stringifyLogs(strings, ...args);
     }
     this.#originalUserLogger.log(logs);
-    this.#dataStorage.putData(logs, context);
+    this.#dataStorage.putData(logs);
   }
 
   /**
@@ -158,18 +136,12 @@ class Logger {
   #logWrapper(argsArray, level) {
     if (!argsArray.length) return;
 
-    // entity which is used to define testId
-    let context = null;
-
-    // get context for mocha
-    context = global.testTitle ?? null;
-
     const severity = LEVELS[level].severity;
     if (severity < LEVELS[this.logLevel]?.severity) return;
 
     const logs = this.#stringifyLogs(...argsArray);
     const colorizedLogs = chalk[LEVELS[level].color](logs);
-    this.#dataStorage.putData(colorizedLogs, context);
+    this.#dataStorage.putData(colorizedLogs);
     try {
       // level.toLowerCase() represents method name (log, warn, error, etc)
       this.#originalUserLogger[level.toLowerCase()](colorizedLogs);
@@ -269,7 +241,6 @@ class Logger {
 
 Logger.instance = null;
 
-// module.exports.Logger = Logger;
 const logger = new Logger();
 
 module.exports = logger;

package/lib/utils/utils.js

@@ -290,6 +290,19 @@ function removeColorCodes(input) {
   return input.replace(/\x1b\[[0-9;]*m/g, '');
 }
 
+const jestHelpers = {
+  getIdOfCurrentlyRunningTest: () => {
+    if (!process.env.JEST_WORKER_ID) return null;
+    try {
+      // @ts-expect-error "expect" could only be defined inside Jest environement (forbidden to import it outside)
+      // eslint-disable-next-line no-undef
+      if (expect && expect?.getState()?.currentTestName) return parseTest(expect?.getState()?.currentTestName);
+    } catch (e) {
+      return null;
+    }
+  },
+};
+
 module.exports = {
   isSameTest,
   fetchSourceCode,
@@ -299,6 +312,7 @@ module.exports = {
   fetchFilesFromStackTrace,
   fileSystem,
   getCurrentDateTime,
+  jestHelpers,
   specificTestInfo,
   isValidUrl,
   ansiRegExp,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@testomatio/reporter",
-  "version": "1.0.17",
+  "version": "1.1.0-beta-2",
   "description": "Testomatio Reporter Client",
   "main": "./lib/reporter.js",
   "typings": "typings/index.d.ts",