Fix StartupStatus in tests

`StartupStatus.startupComplete` was set in production code, but not in tests.
trinodb · May 13, 2024 · aeba8ef · aeba8ef
1 parent 86248b3
commit aeba8ef
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/core/trino-main/src/main/java/io/trino/server/testing/TestingTrinoServer.java b/core/trino-main/src/main/java/io/trino/server/testing/TestingTrinoServer.java
@@ -83,6 +83,7 @@
 import io.trino.server.SessionPropertyDefaults;
 import io.trino.server.SessionSupplier;
 import io.trino.server.ShutdownAction;
+import io.trino.server.StartupStatus;
 import io.trino.server.security.CertificateAuthenticatorManager;
 import io.trino.server.security.ServerSecurityModule;
 import io.trino.spi.ErrorType;
@@ -424,6 +425,7 @@ private TestingTrinoServer(
         eventListeners.forEach(eventListenerManager::addEventListener);
 
         getFutureValue(injector.getInstance(Announcer.class).forceAnnounce());
+        injector.getInstance(StartupStatus.class).startupComplete();
 
         refreshNodes();
     }