[nodejs] Enable Test_Blocking_request_body_multipart (#3249)

DataDog · Oct 29, 2024 · ee67d34 · ee67d34
1 parent 9709e9f
commit ee67d34
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 1 deletion.
diff --git a/manifests/nodejs.yml b/manifests/nodejs.yml
@@ -372,7 +372,9 @@ tests/:
       Test_Blocking_request_body:
         '*': *ref_3_19_0
         nextjs: missing_feature
-      Test_Blocking_request_body_multipart: missing_feature
+      Test_Blocking_request_body_multipart:
+        '*': *ref_5_25_0
+        nextjs: missing_feature
       Test_Blocking_request_cookies:
         '*': *ref_4_16_0
         nextjs: missing_feature

diff --git a/utils/build/docker/nodejs/express4-typescript/app.ts b/utils/build/docker/nodejs/express4-typescript/app.ts
@@ -11,6 +11,9 @@ const passport = require('passport')
 const { Kafka } = require("kafkajs")
 const { spawnSync } = require('child_process');
 
+const multer = require('multer')
+const uploadToMemory = multer({ storage: multer.memoryStorage(), limits: { fileSize: 200000 } })
+
 const iast = require('./iast')
 
 iast.initData().catch(() => {})
@@ -40,6 +43,11 @@ app.get('/healthcheck', (req: Request, res: Response) => {
   });
 })
 
+app.post('/waf', uploadToMemory.single('foo'), (req: Request, res: Response) => {
+  res.send('Hello\n')
+})
+
+
 app.all(['/waf', '/waf/*'], (req: Request, res: Response) => {
   res.send('Hello\n');
 });

diff --git a/utils/build/docker/nodejs/express4-typescript/package.json b/utils/build/docker/nodejs/express4-typescript/package.json
@@ -25,6 +25,7 @@
     "kafkajs": "^2.2.4",
     "ldapjs": "2.0.0",
     "mongodb": "^6.1.0",
+    "multer": "^1.4.5-lts.1",
     "passport": "^0.6.0",
     "passport-http": "^0.3.0",
     "passport-local": "^1.0.0",

diff --git a/utils/build/docker/nodejs/express4/app.js b/utils/build/docker/nodejs/express4/app.js
@@ -19,6 +19,9 @@ const pgsql = require('./integrations/db/postgres')
 const mysql = require('./integrations/db/mysql')
 const mssql = require('./integrations/db/mssql')
 
+const multer = require('multer')
+const uploadToMemory = multer({ storage: multer.memoryStorage(), limits: { fileSize: 200000 } })
+
 const { kinesisProduce, kinesisConsume } = require('./integrations/messaging/aws/kinesis')
 const { snsPublish, snsConsume } = require('./integrations/messaging/aws/sns')
 const { sqsProduce, sqsConsume } = require('./integrations/messaging/aws/sqs')
@@ -48,6 +51,10 @@ app.get('/healthcheck', (req, res) => {
   })
 })
 
+app.post('/waf', uploadToMemory.single('foo'), (req, res) => {
+  res.send('Hello\n')
+})
+
 app.all(['/waf', '/waf/*'], (req, res) => {
   res.send('Hello\n')
 })

diff --git a/utils/build/docker/nodejs/express4/package.json b/utils/build/docker/nodejs/express4/package.json
@@ -25,6 +25,7 @@
     "ldapjs": "2.0.0",
     "mongodb": "^6.1.0",
     "mssql": "^9.1.0",
+    "multer": "^1.4.5-lts.1",
     "mysql2": "^3.5.0",
     "passport": "^0.6.0",
     "passport-http": "^0.3.0",