import { mustCall } from '../common/index.mjs';

import { Readable } from 'node:stream';

import { memoryUsage } from 'node:process';

import assert from 'node:assert';

import { setImmediate } from 'node:timers/promises';

// Based on: https://github.com/nodejs/node/issues/46347#issuecomment-1413886707

// edit: make it cross-platform as /dev/urandom is not available on Windows

const MAX_MEM = 256 * 1024 * 1024; // 256 MiB

function checkMemoryUsage() {

assert(memoryUsage().arrayBuffers < MAX_MEM);

}

const MAX_BUFFERS = 1000;

let buffersCreated = 0;

const randomNodeStream = new Readable({

read(size) {

if (buffersCreated >= MAX_BUFFERS) {

this.push(null);

return;

}

this.push(Buffer.alloc(size));

buffersCreated++;

}

});

randomNodeStream.on('error', (err) => {

assert.fail(err);

});

// Before doing anything, make sure memory usage is okay

checkMemoryUsage();

// Create stream and check memory usage remains okay

const randomWebStream = Readable.toWeb(randomNodeStream);

checkMemoryUsage();

let timeout;

try {

// Wait two seconds before consuming the stream to see if memory usage increases

timeout = setTimeout(mustCall(async () => {

// Did the stream leak memory?

checkMemoryUsage();

// eslint-disable-next-line no-unused-vars

for await (const _ of randomWebStream) {

// Yield event loop to allow garbage collection

await setImmediate();

// consume the stream

// check memory usage remains okay

checkMemoryUsage();

}

}), 2000);

} catch (err) {

if (timeout) {

clearTimeout(timeout);

}

assert.fail(err);

}