Merge branch 'master' of github.com:fijijavis/wdio-mochawesome-reporter

webdriverio-community · Apr 29, 2020 · 5d14591 · 5d14591
2 parents 415cd11 + b25837d
commit 5d14591
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 3 deletions.
diff --git a/src/index.js b/src/index.js
@@ -7,12 +7,21 @@ class WdioMochawesomeReporter extends WDIOReporter {
     constructor (options) {
         options = Object.assign(options)
         super(options)
+        this.registerListeners()
     }
 
     onRunnerStart (runner) {
         this.config = runner.config
         this.sanitizedCaps = runner.sanitizedCapabilities
-        this.sessionId = runner.sessionId
+        // Set sessionId
+        if (runner.isMultiremote) {
+            this.sessionId = {}
+            for (const name in runner.capabilities) {
+                this.sessionId[name] = runner.capabilities[name].sessionId
+            }
+        } else {
+            this.sessionId = runner.sessionId
+        }
         // mochawesome requires this root suite for HTML report generation to work properly
         this.results = {
             stats: new Stats(runner.start),
@@ -61,6 +70,19 @@ class WdioMochawesomeReporter extends WDIOReporter {
         this.results.stats.duration = runner.duration
         this.write(JSON.stringify(this.results))
     }
+
+    // addContext functionality
+    registerListeners () {
+        process.on('wdio-mochawesome-reporter:addContext', this.addSomeContext.bind(this))
+    }
+
+    addSomeContext (object) {
+        this.currTest.context.push(object)
+    }
+
+    static addContext (context) {
+        process.emit('wdio-mochawesome-reporter:addContext', context)
+    }
 }
 
 exports.default = WdioMochawesomeReporter
diff --git a/src/test.js b/src/test.js
@@ -42,8 +42,8 @@ module.exports = class {
         /**
          * Jest Matchers returns error strings with ansi color information
          * https://github.com/jest-community/vscode-jest/issues/279
-         * 
-         * stripAnsi will remove those colors.  
+         *
+         * stripAnsi will remove those colors.
          * Tested with Jest Matchers AND Chai (which does not include the ansi characters)
          */
         if (result.error) {