OpenTracingCollector fixes

testing/trino-testing-containers/pom.xml

@@ -36,6 +36,11 @@
             <artifactId>log</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>io.airlift</groupId>
+            <artifactId>units</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>io.minio</groupId>
             <artifactId>minio</artifactId>

testing/trino-testing-containers/src/main/java/io/trino/testing/containers/OpenTracingCollector.java

@@ -13,6 +13,8 @@
  */
 package io.trino.testing.containers;
 
+import io.airlift.units.DataSize;
+
 import java.io.File;
 import java.io.IOException;
 import java.io.UncheckedIOException;
@@ -27,24 +29,16 @@
 import java.util.stream.Stream;
 
 import static com.google.common.net.UrlEscapers.urlFragmentEscaper;
+import static io.airlift.units.DataSize.Unit.GIGABYTE;
 
 public class OpenTracingCollector
         extends BaseTestContainer
 {
-    private static final Path storageDirectory;
-
-    static {
-        try {
-            storageDirectory = Files.createTempDirectory("tracing-collector");
-        }
-        catch (IOException e) {
-            throw new UncheckedIOException(e);
-        }
-    }
-
     private static final int COLLECTOR_PORT = 4317;
     private static final int HTTP_PORT = 16686;
 
+    private final Path storageDirectory;
+
     public OpenTracingCollector()
     {
         super(
@@ -67,8 +61,15 @@ public OpenTracingCollector()
                 "--badger.maintenance-interval=30s"));
 
         withCreateContainerModifier(command -> command.getHostConfig()
-                .withMemory(2 * 1024 * 1024 * 1024L)); // 1 GB limit
-        mountDirectory(storageDirectory.toString(), "/badger");
+                .withMemory(DataSize.of(1, GIGABYTE).toBytes()));
+
+        try {
+            this.storageDirectory = Files.createTempDirectory("tracing-collector");
+            mountDirectory(storageDirectory.toString(), "/badger");
+        }
+        catch (IOException e) {
+            throw new UncheckedIOException(e);
+        }
     }
 
     @Override

testing/trino-testing/src/main/java/io/trino/testing/DistributedQueryRunner.java

@@ -800,7 +800,7 @@ protected SELF self()
         public DistributedQueryRunner build()
                 throws Exception
         {
-            String tracingEnabled = firstNonNull(getenv("TRACING"), "false");
+            String tracingEnabled = firstNonNull(getenv("TESTS_TRACING_ENABLED"), "false");
             if (parseBoolean(tracingEnabled) || tracingEnabled.equals("1")) {
                 withTracing();
             }