fix express test

DataDog · Jun 22, 2024 · 5107046 · 5107046
1 parent f81ccc6
commit 5107046
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/...es/dd-trace/test/appsec/iast/taint-tracking/sources/taint-tracking.express.plugin.spec.js b/...es/dd-trace/test/appsec/iast/taint-tracking/sources/taint-tracking.express.plugin.spec.js
@@ -171,7 +171,9 @@ describe('Path params sourcing with express', () => {
 
       app.use('/:parameterParent', nestedRouter)
 
-      appListener = app.listen(0, 'localhost', port => {
+      appListener = app.listen(0, 'localhost', () => {
+        const port = appListener.address().port
+
         axios
           .get(`http://localhost:${port}/tainted1/tainted2`)
           .then(() => done())
@@ -189,7 +191,9 @@ describe('Path params sourcing with express', () => {
       app.param('parameter1', checkParamIsTaintedAndNext)
       app.param('parameter2', checkParamIsTaintedAndNext)
 
-      appListener = app.listen(0, 'localhost', port => {
+      appListener = app.listen(0, 'localhost', () => {
+        const port = appListener.address().port
+
         axios
           .get(`http://localhost:${port}/tainted1/tainted2`)
           .then(() => done())
@@ -211,7 +215,9 @@ describe('Path params sourcing with express', () => {
       app.param('parameter1')
       app.param('parameter2')
 
-      appListener = app.listen(0, 'localhost', port => {
+      appListener = app.listen(0, 'localhost', () => {
+        const port = appListener.address().port
+
         axios
           .get(`http://localhost:${port}/tainted1/tainted2`)
           .then(() => done())