[ci-visibility] Allow using after:run directly (#4090)

ci/cypress/after-run.js

@@ -0,0 +1 @@
+module.exports = require('../../packages/datadog-plugin-cypress/src/after-run')

integration-tests/cypress.config.js

@@ -6,6 +6,13 @@ module.exports = {
         require('cypress-fail-fast/plugin')(on, config)
       }
       require('dd-trace/ci/cypress/plugin')(on, config)
+      if (process.env.CYPRESS_ENABLE_AFTER_RUN_CUSTOM) {
+        on('after:run', (...args) => {
+          // do custom stuff
+          // and call after-run at the end
+          require('dd-trace/ci/cypress/after-run')(...args)
+        })
+      }
     }
   },
   video: false,

integration-tests/cypress/cypress.spec.js

@@ -824,5 +824,44 @@ moduleType.forEach(({
         }).catch(done)
       })
     })
+
+    it('works if after:run is explicitly used', (done) => {
+      const receiverPromise = receiver
+        .gatherPayloadsMaxTimeout(({ url }) => url.endsWith('/api/v2/citestcycle'), payloads => {
+          const events = payloads.flatMap(({ payload }) => payload.events)
+          const testSessionEvent = events.find(event => event.type === 'test_session_end')
+          assert.exists(testSessionEvent)
+          const testModuleEvent = events.find(event => event.type === 'test_module_end')
+          assert.exists(testModuleEvent)
+          const testSuiteEvents = events.filter(event => event.type === 'test_suite_end')
+          assert.equal(testSuiteEvents.length, 4)
+          const testEvents = events.filter(event => event.type === 'test')
+          assert.equal(testEvents.length, 9)
+        })
+
+      const {
+        NODE_OPTIONS, // NODE_OPTIONS dd-trace config does not work with cypress
+        ...restEnvVars
+      } = getCiVisEvpProxyConfig(receiver.port)
+
+      childProcess = exec(
+        testCommand,
+        {
+          cwd,
+          env: {
+            ...restEnvVars,
+            CYPRESS_BASE_URL: `http://localhost:${webAppPort}`,
+            CYPRESS_ENABLE_AFTER_RUN_CUSTOM: '1'
+          },
+          stdio: 'pipe'
+        }
+      )
+
+      childProcess.on('exit', () => {
+        receiverPromise.then(() => {
+          done()
+        }).catch(done)
+      })
+    })
   })
 })

packages/datadog-plugin-cypress/src/after-run.js

@@ -0,0 +1,3 @@
+const cypressPlugin = require('./cypress-plugin')
+
+module.exports = cypressPlugin.afterRun.bind(cypressPlugin)