feat: forward onRequestSent to handler (#2375)

ronag · web-flow · commit 67e714d0d558 · 2023-10-25T19:43:21.000+02:00
diff --git a/lib/core/request.js b/lib/core/request.js
@@ -222,6 +222,14 @@ class Request {
     if (channels.bodySent.hasSubscribers) {
       channels.bodySent.publish({ request: this })
     }
+
+    if (this[kHandler].onRequestSent) {
+      try {
+        this[kHandler].onRequestSent()
+      } catch (err) {
+        this.onError(err)
+      }
+    }
   }
 
   onConnect (abort) {
diff --git a/test/client-dispatch.js b/test/client-dispatch.js
@@ -671,6 +671,8 @@ test('dispatch onBodySent not a function', (t) => {
 })
 
 test('dispatch onBodySent buffer', (t) => {
+  t.plan(3)
+
   const server = http.createServer((req, res) => {
     res.end('ad')
   })
@@ -688,20 +690,24 @@ test('dispatch onBodySent buffer', (t) => {
       onBodySent (chunk) {
         t.equal(chunk.toString(), body)
       },
+      onRequestSent () {
+        t.pass()
+      },
       onError (err) {
         throw err
       },
       onConnect () {},
       onHeaders () {},
       onData () {},
       onComplete () {
-        t.end()
+        t.pass()
       }
     })
   })
 })
 
 test('dispatch onBodySent stream', (t) => {
+  t.plan(8)
   const server = http.createServer((req, res) => {
     res.end('ad')
   })
@@ -723,6 +729,9 @@ test('dispatch onBodySent stream', (t) => {
         t.equal(chunks[currentChunk++], chunk)
         sentBytes += Buffer.byteLength(chunk)
       },
+      onRequestSent () {
+        t.pass()
+      },
       onError (err) {
         throw err
       },
@@ -732,7 +741,7 @@ test('dispatch onBodySent stream', (t) => {
       onComplete () {
         t.equal(currentChunk, chunks.length)
         t.equal(sentBytes, toSendBytes)
-        t.end()
+        t.pass()
       }
     })
   })

Original file line number	Diff line number	Diff line change
`@@ -222,6 +222,14 @@ class Request {`
`222`	`222`	`if (channels.bodySent.hasSubscribers) {`
`223`	`223`	`channels.bodySent.publish({ request: this })`
`224`	`224`	`}`
	`225`	`+`
	`226`	`+ if (this[kHandler].onRequestSent) {`
	`227`	`+ try {`
	`228`	`+ this[kHandler].onRequestSent()`
	`229`	`+ } catch (err) {`
	`230`	`+ this.onError(err)`
	`231`	`+ }`
	`232`	`+ }`
`225`	`233`	`}`
`226`	`234`
`227`	`235`	`onConnect (abort) {`