nightwatch test-observability integration phase-1

nightwatch/globals.js

@@ -1,9 +1,58 @@
 const LocalTunnel = require('../src/local-tunnel');
+const TestObservability = require('../src/testObservability');
+const {CUSTOM_REPORTER_CALLBACK_TIMEOUT} = require('../src/utils/constants');
+const CrashReporter = require('../src/utils/crashReporter');
 const helper = require('../src/utils/helper');
+const Logger = require('../src/utils/logger');
 
 const localTunnel = new LocalTunnel();
+const testObservability = new TestObservability();
+
+const nightwatchRerun = process.env.NIGHTWATCH_RERUN_FAILED;
+const nightwatchRerunFile = process.env.NIGHTWATCH_RERUN_REPORT_FILE;
 
 module.exports = {
+
+  reporter: function(results, done) {
+    if (helper.isTestObservabilitySession()) {
+      const promises = [];
+      try {
+        const modulesWithEnv = results['modulesWithEnv'];
+        for (const testSetting in modulesWithEnv) {
+          for (const testFile in modulesWithEnv[testSetting]) {
+            for (const completedSection in modulesWithEnv[testSetting][testFile].completed) {
+              if (modulesWithEnv[testSetting][testFile].completed[completedSection]) {
+                delete modulesWithEnv[testSetting][testFile].completed[completedSection].steps;
+                delete modulesWithEnv[testSetting][testFile].completed[completedSection].testcases;
+              }
+            }
+            promises.push(testObservability.processTestReportFile(JSON.parse(JSON.stringify(modulesWithEnv[testSetting][testFile]))));
+          }
+        }
+
+        Promise.all(promises).then(() => {
+          done();
+        }).catch((err) =>{
+          Logger.error(`Something went wrong in processing report file for test observability - ${err.message} with stacktrace ${err.stack}`);
+          CrashReporter.uploadCrashReport(err.message, err.stack);
+          done();
+        });
+
+        return;
+      } catch (error) {
+        CrashReporter.uploadCrashReport(error.message, error.stack);
+        Logger.error(`Something went wrong in processing report file for test observability - ${error.message} with stacktrace ${error.stack}`);
+      }
+    } 
+    done(results);
+  },
+
+  onEvent({eventName, hook_type, ...args}) {
+    if (browser && eventName === 'TestRunStarted') {
+      browser.execute(`browserstack_executor: {"action": "annotate", "arguments": {"type":"Annotation","data":"ObservabilitySync:${Date.now()}","level": "debug"}}`);
+    }
+  },
+
   async before(settings) {
     localTunnel.configure(settings);
     await localTunnel.start();
@@ -22,10 +71,42 @@ module.exports = {
         settings.desiredCapabilities['bstack:options'].localIdentifier = localTunnel._localOpts.localIdentifier;
       }
     }
+
+    try {
+      testObservability.configure(settings);
+      if (helper.isTestObservabilitySession()) {
+        settings.globals['customReporterCallbackTimeout'] = CUSTOM_REPORTER_CALLBACK_TIMEOUT;
+        if (testObservability._user && testObservability._key) {
+          await testObservability.launchTestSession();
+        }
+        if (process.env.BROWSERSTACK_RERUN === 'true' && process.env.BROWSERSTACK_RERUN_TESTS && process.env.BROWSERSTACK_RERUN_TESTS!=='null') {
+          const specs = process.env.BROWSERSTACK_RERUN_TESTS.split(',');
+          await helper.handleNightwatchRerun(specs);
+        }
+      } 
+    } catch (error) {
+      Logger.error(`Could not configure or launch test observability - ${error}`);
+    }
+
   },
 
   async after() {
     localTunnel.stop();
+    if (helper.isTestObservabilitySession()) {
+      try {
+        await testObservability.stopBuildUpstream();
+        if (process.env.BS_TESTOPS_BUILD_HASHED_ID) {
+          Logger.info(`\nVisit https://observability.browserstack.com/builds/${process.env.BS_TESTOPS_BUILD_HASHED_ID} to view build report, insights, and many more debugging information all at one place!\n`);
+        }
+      } catch (error) {
+        Logger.error(`Something went wrong in stopping build session for test observability - ${error}`);
+      }
+      process.env.NIGHTWATCH_RERUN_FAILED = nightwatchRerun;
+      process.env.NIGHTWATCH_RERUN_REPORT_FILE = nightwatchRerunFile;
+      if (process.env.BROWSERSTACK_RERUN === 'true' && process.env.BROWSERSTACK_RERUN_TESTS) {
+        await helper.deleteRerunFile();
+      }
+    }
   },
 
   beforeChildProcess(settings) {