test: fix memory test assertion

napi-rs · Dec 28, 2022 · b7ba068 · b7ba068
1 parent 81498ec
commit b7ba068
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/memory-testing/test-util.mjs b/memory-testing/test-util.mjs
@@ -44,14 +44,19 @@ export async function createSuite(testFile, maxMemoryUsage) {
   const stats = await container.stats()
 
   let shouldAssertMemoryUsage = false
-
-  const initialMemoryUsage = await new Promise((resolve, reject) => {
+  let initialMemoryUsage
+  await new Promise((resolve, reject) => {
+    const initialDate = Date.now()
     stats.on('data', (d) => {
       const { memory_stats } = JSON.parse(d.toString('utf8'))
-      resolve(memory_stats.usage)
+      if (Date.now() - initialDate > 10000 && !shouldAssertMemoryUsage) {
+        resolve()
+        initialMemoryUsage = memory_stats.usage
+        shouldAssertMemoryUsage = true
+      }
       if (shouldAssertMemoryUsage && memory_stats?.usage) {
         const memoryGrowth = memory_stats.usage - initialMemoryUsage
-        if (memoryGrowth > maxMemoryUsage ?? initialMemoryUsage) {
+        if (memoryGrowth > (maxMemoryUsage ?? initialMemoryUsage)) {
           console.info(
             chalk.redBright(
               `Potential memory leak, memory growth: ${prettyBytes(
@@ -72,8 +77,6 @@ export async function createSuite(testFile, maxMemoryUsage) {
 
   await sleep(60000)
 
-  shouldAssertMemoryUsage = true
-
   try {
     await container.stop()
     await container.remove()