allow customization of test containers image name

gateway-ha/src/test/java/io/trino/gateway/CustomImageNameSubstitutor.java

@@ -0,0 +1,25 @@
+package io.trino.gateway;
+
+import org.testcontainers.utility.DockerImageName;
+import org.testcontainers.utility.ImageNameSubstitutor;
+
+public class CustomImageNameSubstitutor
+        extends ImageNameSubstitutor
+{
+    private final String TESTCONTAINERS_TRINO_IMAGE_SUBSTITUTE = "TESTCONTAINERS_TRINO_IMAGE_SUBSTITUTE";
+    @Override
+    public DockerImageName apply(DockerImageName dockerImageName)
+    {
+        if (dockerImageName.asCanonicalNameString().equals("trinodb/trino:latest") && System.getenv(TESTCONTAINERS_TRINO_IMAGE_SUBSTITUTE) != null) {
+            String image_substitute = System.getenv(TESTCONTAINERS_TRINO_IMAGE_SUBSTITUTE);
+            return DockerImageName.parse(image_substitute).asCompatibleSubstituteFor("trinodb/trino");
+        }
+        return dockerImageName;
+    }
+
+    @Override
+    protected String getDescription()
+    {
+        return "custom image name substitutor";
+    }
+}