test: add test for memory leak (#3450)

snyamathi · mcollina · commit 52870545df66 · 2024-08-17T10:31:49.000+02:00
* test: add test for memory leak

* lint
diff --git a/test/fetch/response.js b/test/fetch/response.js
@@ -2,6 +2,8 @@
 
 const { test } = require('node:test')
 const assert = require('node:assert')
+const { setImmediate } = require('node:timers/promises')
+const { AsyncLocalStorage } = require('node:async_hooks')
 const { tspl } = require('@matteo.collina/tspl')
 const {
   Response,
@@ -285,3 +287,29 @@ test('fromInnerResponse', () => {
   assert.strictEqual(getHeadersList(response[kHeaders]), innerResponse.headersList)
   assert.strictEqual(getHeadersGuard(response[kHeaders]), 'immutable')
 })
+
+test('clone body garbage collection', async () => {
+  const asyncLocalStorage = new AsyncLocalStorage()
+  let ref
+
+  await new Promise(resolve => {
+    asyncLocalStorage.run(new Map(), async () => {
+      const res = new Response('hello world')
+      const clone = res.clone()
+
+      asyncLocalStorage.getStore().set('key', clone)
+      ref = new WeakRef(clone.body)
+
+      await res.text()
+      await clone.text() // consume body
+
+      resolve()
+    })
+  })
+
+  await setImmediate()
+  global.gc()
+
+  const cloneBody = ref.deref()
+  assert.equal(cloneBody, undefined, 'clone body was not garbage collected')
+})
