Health check for UDP container test

Author: technocidal

packages/testcontainers/src/generic-container/generic-container.test.ts

@@ -5,6 +5,7 @@ import { getContainerRuntimeClient } from "../container-runtime";
 import { PullPolicy } from "../utils/pull-policy";
 import {
   checkContainerIsHealthy,
+  checkContainerIsHealthyUdp,
   getDockerEventStream,
   getRunningContainerNames,
   waitForDockerEvent,
@@ -24,6 +25,7 @@ describe("GenericContainer", { timeout: 180_000 }, () => {
 
   it("should return first mapped port with regardless of protocol", async () => {
     await using container = await new GenericContainer("mendhak/udp-listener").withExposedPorts("5005/udp").start();
+    await checkContainerIsHealthyUdp(container);
     expect(container.getFirstMappedPort()).toBe(container.getMappedPort("5005/udp"));
   });
 
@@ -49,6 +51,7 @@ describe("GenericContainer", { timeout: 180_000 }, () => {
         protocol: "udp",
       })
       .start();
+    await checkContainerIsHealthyUdp(container);
     expect(container.getMappedPort("5005/udp")).toBe(hostPort);
     expect(container.getMappedPort(5005, "udp")).toBe(hostPort);
   });

packages/testcontainers/src/utils/test-helper.ts

@@ -1,5 +1,6 @@
 import { GetEventsOptions, ImageInspectInfo } from "dockerode";
 import { createServer, Server } from "http";
+import { createSocket } from "node:dgram";
 import fs from "node:fs";
 import path from "node:path";
 import { Readable } from "stream";
@@ -30,6 +31,27 @@ export const checkContainerIsHealthyTls = async (container: StartedTestContainer
   expect(response.statusCode).toBe(200);
 };
 
+export const checkContainerIsHealthyUdp = async (container: StartedTestContainer): Promise<void> => {
+  const host = container.getHost();
+  const port = container.getMappedPort("5005/udp");
+  const readyMessage = "Listening on UDP port 5005";
+  const testMessage = "health_check";
+  const client = createSocket("udp4");
+  try {
+    const logs = await container.logs();
+    for await (const log of logs) {
+      if (log.includes(readyMessage)) {
+        client.send(Buffer.from(testMessage), port, host);
+      }
+      if (log.includes(testMessage)) {
+        return;
+      }
+    }
+  } finally {
+    client.close();
+  }
+};
+
 export const checkEnvironmentContainerIsHealthy = async (
   startedEnvironment: StartedDockerComposeEnvironment,
   containerName: string