From 92f22c9d6613d46759bc76d618a1f0608866fb15 Mon Sep 17 00:00:00 2001
From: Michelangelo Partipilo <michelangelo@weaviate.io>
Date: Wed, 29 Apr 2026 23:50:25 +0200
Subject: [PATCH] test(it): support WV_TEST_HOST/REST_PORT/GRPC_PORT overrides

Allows pointing integration tests at an external Weaviate instance
(e.g., Apache Arrow Flight parity proxy) by skipping TestContainers
boot when WV_TEST_HOST is set, using the env-var endpoint instead.
---
 .../java/io/weaviate/containers/Weaviate.java | 41 ++++++++++++++++---
 1 file changed, 35 insertions(+), 6 deletions(-)

diff --git a/src/it/java/io/weaviate/containers/Weaviate.java b/src/it/java/io/weaviate/containers/Weaviate.java
index d7c7fcd03..a2773ba0a 100644
--- a/src/it/java/io/weaviate/containers/Weaviate.java
+++ b/src/it/java/io/weaviate/containers/Weaviate.java
@@ -30,9 +30,36 @@ public class Weaviate extends WeaviateContainer {
   public static final String VERSION;
   private static final boolean DEBUG;
 
+  private static final String TEST_HOST;
+  private static final int TEST_REST_PORT;
+  private static final int TEST_GRPC_PORT;
+  private static final boolean TEST_HOST_OVERRIDE;
+
   static {
     VERSION = System.getenv().getOrDefault("WEAVIATE_VERSION", LATEST_VERSION);
     DEBUG = System.getenv("DEBUG") != null;
+
+    String hostOverride = System.getenv("WV_TEST_HOST");
+    TEST_HOST_OVERRIDE = hostOverride != null && !hostOverride.isEmpty();
+    TEST_HOST = envStr("WV_TEST_HOST", "localhost");
+    TEST_REST_PORT = envInt("WV_TEST_REST_PORT", 8080);
+    TEST_GRPC_PORT = envInt("WV_TEST_GRPC_PORT", 50051);
+  }
+
+  private static String envStr(String key, String def) {
+    String v = System.getenv(key);
+    return (v != null && !v.isEmpty()) ? v : def;
+  }
+
+  private static int envInt(String key, int def) {
+    String v = System.getenv(key);
+    if (v != null && !v.isEmpty()) {
+      try {
+        return Integer.parseInt(v);
+      } catch (NumberFormatException ignored) {
+      }
+    }
+    return def;
   }
   public static String OIDC_ISSUER = "https://auth.wcs.api.weaviate.io/auth/realms/SeMI";
 
@@ -94,7 +121,7 @@ public String getContainerName() {
    * after the parent Testcontainer is stopped.
    */
   public WeaviateClient getClient() {
-    if (!isRunning()) {
+    if (!TEST_HOST_OVERRIDE && !isRunning()) {
       start();
       if (DEBUG) {
         followOutput(frame -> System.out.println(frame.getUtf8String()));
@@ -122,7 +149,7 @@ public WeaviateClient getClient() {
    * so it can be auto-closed by the try-with-resources statement when it exists.
    */
   public WeaviateClient getBareClient() {
-    if (!isRunning()) {
+    if (!TEST_HOST_OVERRIDE && !isRunning()) {
       start();
     }
     try {
@@ -138,7 +165,7 @@ public WeaviateClient getBareClient() {
    * steps to run, e.g. OIDC authorization grant exchange.
    */
   public WeaviateClient getClient(Function<Config.Custom, ObjectBuilder<Config>> fn) {
-    if (!isRunning()) {
+    if (!TEST_HOST_OVERRIDE && !isRunning()) {
       start();
     }
 
@@ -152,11 +179,13 @@ public WeaviateClient getClient(Function<Config.Custom, ObjectBuilder<Config>> f
   }
 
   private Function<Config.Custom, ObjectBuilder<Config>> defaultConfigFn() {
-    var host = getHost();
+    var host = TEST_HOST_OVERRIDE ? TEST_HOST : getHost();
+    int restPort = TEST_HOST_OVERRIDE ? TEST_REST_PORT : getMappedPort(8080);
+    int grpcPort = TEST_HOST_OVERRIDE ? TEST_GRPC_PORT : getMappedPort(50051);
     return conn -> conn
         .scheme("http")
-        .httpHost(host).httpPort(getMappedPort(8080))
-        .grpcHost(host).grpcPort(getMappedPort(50051));
+        .httpHost(host).httpPort(restPort)
+        .grpcHost(host).grpcPort(grpcPort);
   }
 
   public static Weaviate createDefault() {