fix(ext/node): UdpSocket ref and unref (#21777)

1 files changed, 59 insertions, 0 deletions

diff --git a/cli/tests/unit_node/dgram_test.ts b/cli/tests/unit_node/dgram_test.ts
new file mode 100644
index 000000000..d532241f5
--- /dev/null
+++ b/cli/tests/unit_node/dgram_test.ts
@@ -0,0 +1,59 @@
+// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.
+
+import { assertEquals } from "../../../test_util/std/assert/mod.ts";
+import { execCode } from "../unit/test_util.ts";
+import { createSocket } from "node:dgram";
+
+const listenPort = 4503;
+const listenPort2 = 4504;
+
+Deno.test("[node/dgram] udp ref and unref", {
+  permissions: { read: true, run: true, net: true },
+}, async () => {
+  const { promise, resolve } = Promise.withResolvers<void>();
+
+  const udpSocket = createSocket("udp4");
+  udpSocket.bind(listenPort);
+
+  udpSocket.unref();
+  udpSocket.ref();
+
+  let data;
+  udpSocket.on("message", (buffer, _rinfo) => {
+    data = Uint8Array.from(buffer);
+    udpSocket.close();
+  });
+  udpSocket.on("close", () => {
+    resolve();
+  });
+
+  const conn = await Deno.listenDatagram({
+    port: listenPort2,
+    transport: "udp",
+  });
+  await conn.send(new Uint8Array([0, 1, 2, 3]), {
+    transport: "udp",
+    port: listenPort,
+    hostname: "127.0.0.1",
+  });
+
+  await promise;
+  conn.close();
+  assertEquals(data, new Uint8Array([0, 1, 2, 3]));
+});
+
+Deno.test("[node/dgram] udp unref", {
+  permissions: { read: true, run: true, net: true },
+}, async () => {
+  const [statusCode, _output] = await execCode(`
+      import { createSocket } from "node:dgram";
+      const udpSocket = createSocket('udp4');
+      udpSocket.bind(${listenPort2});
+      // This should let the program to exit without waiting for the
+      // udp socket to close.
+      udpSocket.unref();
+      udpSocket.on('message', (buffer, rinfo) => {
+      });
+    `);
+  assertEquals(statusCode, 0);
+});